driver: Widget binding: Support binding incomplete/nonexistent widgets

Author: m0vie

zsh-syntax-highlighting.zsh

@@ -250,8 +250,10 @@ _zsh_highlight_bind_widgets()
   }
 
   # Override ZLE widgets to make them invoke _zsh_highlight.
+  local -U widgets_to_bind
+  widgets_to_bind=(${${(k)widgets}:#(.*|orig-*|run-help|which-command|beep|set-local-history|yank)})
   local cur_widget
-  for cur_widget in ${${(k)widgets}:#(.*|orig-*|run-help|which-command|beep|set-local-history|yank)}; do
+  for cur_widget in $widgets_to_bind; do
     case $widgets[$cur_widget] in
 
       # Already rebound event: do nothing.
@@ -277,6 +279,10 @@ _zsh_highlight_bind_widgets()
       builtin) eval "_zsh_highlight_widget_${(q)cur_widget}() { _zsh_highlight_call_widget .${(q)cur_widget} -- \"\$@\" }"
                zle -N $cur_widget _zsh_highlight_widget_$cur_widget;;
 
+      # Incomplete or nonexistent widget: Bind to z-sy-h driectly.
+      '') _zsh_highlight_widget_${cur_widget}() { :; _zsh_highlight }
+          zle -N $cur_widget _zsh_highlight_widget_$cur_widget;;
+
       # Default: unhandled case.
       *) print -r -- >&2 "zsh-syntax-highlighting: unhandled ZLE widget '$cur_widget'" ;;
     esac