highlight path separators

This commit is based on the work done by
Jorge Israel Peña (blaenk) in zsh-users#136.

Changes:
* Adjusted to changes on the latest master branch.
* Use regular path highlighter colors by default.
* Break out early if the separator color is the same to
  improve performance.
* Tests.

Author: m0vie

highlighters/main/main-highlighter.zsh

@@ -41,7 +41,9 @@
 : ${ZSH_HIGHLIGHT_STYLES[commandseparator]:=none}
 : ${ZSH_HIGHLIGHT_STYLES[hashed-command]:=fg=green}
 : ${ZSH_HIGHLIGHT_STYLES[path]:=underline}
+: ${ZSH_HIGHLIGHT_STYLES[path_pathseparator]:=${ZSH_HIGHLIGHT_STYLES[path_pathseparator]}}
 : ${ZSH_HIGHLIGHT_STYLES[path_prefix]:=underline}
+: ${ZSH_HIGHLIGHT_STYLES[path_prefix_pathseparator]:=${ZSH_HIGHLIGHT_STYLES[path_prefix_pathseparator]}}
 : ${ZSH_HIGHLIGHT_STYLES[globbing]:=fg=blue}
 : ${ZSH_HIGHLIGHT_STYLES[history-expansion]:=fg=blue}
 : ${ZSH_HIGHLIGHT_STYLES[single-hyphen-option]:=none}
@@ -271,6 +273,7 @@ _zsh_highlight_main_highlighter()
     # which add the entry early so escape sequences within the string override
     # the string's color.
     integer already_added=0
+    integer path_found=0
     local style_override=""
     if [[ $this_word == *':start:'* ]]; then
       in_array_assignment=false
@@ -457,6 +460,7 @@ _zsh_highlight_main_highlighter()
                         else
                           if _zsh_highlight_main_highlighter_check_path; then
                             style=path
+                            path_found=1
                           else
                             style=unknown-token
                           fi
@@ -508,6 +512,7 @@ _zsh_highlight_main_highlighter()
                  else
                    if _zsh_highlight_main_highlighter_check_path; then
                      style=path
+                     path_found=1
                    else
                      style=default
                    fi
@@ -517,7 +522,10 @@ _zsh_highlight_main_highlighter()
     fi
     # if a style_override was set (eg in _zsh_highlight_main_highlighter_check_path), use it
     [[ -n $style_override ]] && style=$style_override
-    (( already_added )) || _zsh_highlight_main_add_region_highlight $start_pos $end_pos $style
+    if ! (( already_added )); then
+      _zsh_highlight_main_add_region_highlight $start_pos $end_pos $style
+     (( path_found )) && _zsh_highlight_main_highlighter_highlight_path_separators
+    fi
     if [[ -n ${(M)ZSH_HIGHLIGHT_TOKENS_COMMANDSEPARATOR:#"$arg"} ]]; then
       next_word=':start:'
       highlight_glob=true
@@ -546,6 +554,18 @@ _zsh_highlight_main_highlighter_check_assign()
     [[ $arg == [[:alpha:]_][[:alnum:]_]#(|\[*\])(|[+])=* ]]
 }
 
+_zsh_highlight_main_highlighter_highlight_path_separators()
+{
+  local pos style_pathsep
+  style_pathsep=${style_override:-path}_pathseparator
+  [[ -z "$style_pathsep" || "$style" == "$style_pathsep" ]] && return 0
+  for (( pos = start_pos; $pos <= end_pos; pos++ )) ; do
+    if [[ $BUFFER[pos+1] == / ]]; then
+      _zsh_highlight_main_add_region_highlight $pos $((pos + 1)) $style_pathsep
+    fi
+  done
+}
+
 # Check if $arg is a path.
 _zsh_highlight_main_highlighter_check_path()
 {

highlighters/main/test-data/path.zsh

@@ -27,11 +27,15 @@
 # vim: ft=zsh sw=2 ts=2 et
 # -------------------------------------------------------------------------------------------------
 
+ZSH_HIGHLIGHT_TEST_ENABLE_PATH_SEPARATORS=1
+
 mkdir A
 touch A/mu
 BUFFER='ls A/mu'
 
 expected_region_highlight=(
-  "1 2 command" # ls
-  "4 7 path"     # A/mu
+  "1 2 command"                # ls
+  "4 4 path"                   # A
+  "5 5 path_pathseparator"     # /
+  "6 7 path"                   # mu
 )

highlighters/main/test-data/path_prefix.zsh

@@ -30,8 +30,13 @@
 # Assumes that '/bin/sh' exists and '/bin/s' does not exist.
 # Related to path_prefix2.zsh
 
+ZSH_HIGHLIGHT_TEST_ENABLE_PATH_SEPARATORS=1
+
 BUFFER='ls /bin/s'
 
 expected_region_highlight=(
-  "4 9 path_prefix"    # /bin/s
+  "4 4 path_prefix_pathseparator"    # /
+  "5 7 path_prefix"                  # bin
+  "8 8 path_prefix_pathseparator"    # /
+  "9 9 path_prefix"                  # s
 )

tests/test-highlighting.zsh

@@ -59,6 +59,13 @@ _zsh_highlight_add_highlight()
 # Activate the highlighter.
 ZSH_HIGHLIGHT_HIGHLIGHTERS=($1)
 
+# Disable path separator highlighting in tests by default.
+# Otherwise all tests using paths would need to be unnecessarily verbose.
+# Tests that actually test the separator handling should set
+# ZSH_HIGHLIGHT_TEST_ENABLE_PATH_SEPARATORS=1
+ZSH_HIGHLIGHT_TEST_ENABLE_PATH_SEPARATORS=0
+functions[_zsh_highlight_main_highlighter_highlight_path_separators]="(( ZSH_HIGHLIGHT_TEST_ENABLE_PATH_SEPARATORS )) || return; $functions[_zsh_highlight_main_highlighter_highlight_path_separators]"
+
 # Runs a highlighting test
 # $1: data file
 run_test_internal() {