main: Introduce *-quoted-argument-unclosed styles

Closes zsh-users#277.

Author: phy1729

docs/highlighters/main.md

@@ -36,8 +36,11 @@ This highlighter defines the following styles:
 * `double-hyphen-option` - double hyphen options (`--option`)
 * `back-quoted-argument` - backquoted expressions (`` `foo` ``)
 * `single-quoted-argument` - single quoted arguments (`` 'foo' ``)
+* `single-quoted-argument-unclosed` - unclosed single quoted arguments (`` 'foo ``)
 * `double-quoted-argument` - double quoted arguments (`` "foo" ``)
+* `double-quoted-argument-unclosed` - unclosed double quoted arguments (`` "foo ``)
 * `dollar-quoted-argument` - dollar quoted arguments (`` $'foo' ``)
+* `dollar-quoted-argument-unclosed` - unclosed dollar quoted arguments (`` $'foo ``)
 * `rc-quote` - two single quotes inside single quotes when `RC_QUOTES` is set (`` 'foo''bar' ``)
 * `dollar-double-quoted-argument` - parameter expansion inside double quotes (`$foo` inside `""`)
 * `back-double-quoted-argument` -  back double quoted arguments (`\x` inside `""`)

highlighters/main/main-highlighter.zsh

@@ -44,8 +44,11 @@
 : ${ZSH_HIGHLIGHT_STYLES[double-hyphen-option]:=none}
 : ${ZSH_HIGHLIGHT_STYLES[back-quoted-argument]:=none}
 : ${ZSH_HIGHLIGHT_STYLES[single-quoted-argument]:=fg=yellow}
+: ${ZSH_HIGHLIGHT_STYLES[single-quoted-argument-unclosed]:=fg=yellow}
 : ${ZSH_HIGHLIGHT_STYLES[double-quoted-argument]:=fg=yellow}
+: ${ZSH_HIGHLIGHT_STYLES[double-quoted-argument-unclosed]:=fg=yellow}
 : ${ZSH_HIGHLIGHT_STYLES[dollar-quoted-argument]:=fg=yellow}
+: ${ZSH_HIGHLIGHT_STYLES[dollar-quoted-argument-unclosed]:=fg=yellow}
 : ${ZSH_HIGHLIGHT_STYLES[rc-quote]:=fg=cyan}
 : ${ZSH_HIGHLIGHT_STYLES[dollar-double-quoted-argument]:=fg=cyan}
 : ${ZSH_HIGHLIGHT_STYLES[back-double-quoted-argument]:=fg=cyan}
@@ -839,7 +842,7 @@ _zsh_highlight_main_highlighter_highlight_argument()
 # Highlight single-quoted strings
 _zsh_highlight_main_highlighter_highlight_single_quote()
 {
-  local arg1=$1 i q=\'
+  local arg1=$1 i q=\' style
   local -a highlights
   i=$arg[(ib:arg1+1:)$q]
 
@@ -850,7 +853,12 @@ _zsh_highlight_main_highlighter_highlight_single_quote()
     done
   fi
 
-  highlights+=($(( start_pos + $1 - 1 )) $(( start_pos + i )) single-quoted-argument $highlights)
+  if [[ $arg[i] == "'" ]]; then
+    style=single-quoted-argument
+  else
+    style=single-quoted-argument-unclosed
+  fi
+  highlights+=($(( start_pos + $1 - 1 )) $(( start_pos + i )) $style $highlights)
   _zsh_highlight_main_add_region_highlights $highlights
   REPLY=$i
 }
@@ -910,7 +918,12 @@ _zsh_highlight_main_highlighter_highlight_double_quote()
     highlights+=($j $k $style)
   done
 
-  highlights=($(( start_pos + $1 - 1)) $(( start_pos + i )) double-quoted-argument $highlights)
+  if [[ $arg[i] == '"' ]]; then
+    style=double-quoted-argument
+  else
+    style=double-quoted-argument-unclosed
+  fi
+  highlights=($(( start_pos + $1 - 1)) $(( start_pos + i )) $style $highlights)
   _zsh_highlight_main_add_region_highlights $highlights
   REPLY=$i
 }
@@ -957,7 +970,12 @@ _zsh_highlight_main_highlighter_highlight_dollar_quote()
     highlights+=($j $k $style)
   done
 
-  highlights+=($(( start_pos + $1 - 1 )) $(( start_pos + i )) dollar-quoted-argument $highlights)
+  if [[ $arg[i] == "'" ]]; then
+    style=dollar-quoted-argument
+  else
+    style=dollar-quoted-argument-unclosed
+  fi
+  highlights+=($(( start_pos + $1 - 1 )) $(( start_pos + i )) $style $highlights)
   _zsh_highlight_main_add_region_highlights $highlights
   REPLY=$i
 }

highlighters/main/test-data/dollar-quoted3.zsh

@@ -32,6 +32,6 @@
 BUFFER=": \$'\xa1"
 
 expected_region_highlight=(
-  "3 4 dollar-quoted-argument" # $'
+  "3 4 dollar-quoted-argument-unclosed" # $'
   "5 8 back-dollar-quoted-argument" # \xa1
 )

highlighters/main/test-data/double-quoted2.zsh

@@ -32,6 +32,6 @@
 BUFFER=': "foo$bar'
 
 expected_region_highlight=(
-  "3 6 double-quoted-argument" # "foo
+  "3 6 double-quoted-argument-unclosed" # "foo
   "7 10 dollar-double-quoted-argument" # $bar
 )

highlighters/main/test-data/multiline-string2.zsh

@@ -30,5 +30,5 @@
 BUFFER=$'echo "foo1\n'
 
 expected_region_highlight=(
-  "6 10 double-quoted-argument" # 'foo2"'
+  "6 10 double-quoted-argument-unclosed" # 'foo2"'
 )