feat(parsers): add zsh support (#8240)

Co-authored-by: Christian Clason <c.clason@uni-graz.at>
Co-authored-by: Riley Bruins <ribru17@hotmail.com>
Co-authored-by: Tayfun Bocek <tayfunbocek@live.ca>

1 files changed, 358 insertions, 0 deletions

diff --git a/tests/query/highlights/zsh/test.zsh b/tests/query/highlights/zsh/test.zsh
new file mode 100644
index 000000000..b1c4e398f
--- /dev/null
+++ b/tests/query/highlights/zsh/test.zsh
@@ -0,0 +1,358 @@
+#!/bin/zsh
+# ^^^^^^^^ @comment @spell
+
+# Basic comment
+# <- @comment @spell
+
+# Variables and assignments
+name="value"
+#^^^ @variable @constant
+#   ^ @operator
+#    ^^^^^^^ @string
+
+export PATH="/usr/bin:$PATH"
+#^^^^^ @keyword.import
+#      ^^^^ @variable @constant @variable.builtin
+#          ^ @operator
+#           ^^^^^^^^^^^^^^^^ @string
+#                     ^ @punctuation.special
+#                      ^^^^ @variable.builtin
+
+# Function definition
+function myfunction() {
+#^^^^^^^ @keyword.function
+#        ^^^^^^^^^^ @function @number
+    echo "Hello World"
+    #^^^ @function.call @function.builtin @number
+    #    ^^^^^^^^^^^^^ @string
+}
+
+myfunction() {
+#^^^^^^^^^ @function @number
+    local var="test"
+    #^^^^ @keyword
+    #     ^^^ @variable @constant
+    #        ^ @operator
+    #         ^^^^^^ @string
+}
+
+# Built-in commands
+echo "Hello"
+#^^^ @function.call @function.builtin @number
+#    ^^^^^^^ @string
+
+cd /home/user
+#^ @function.call @function.builtin @number
+#  ^^^^^^^^^^ @variable.parameter @number
+
+ls -la
+#^ @function.call @number
+#  ^^^ @variable.parameter @number
+
+# Control structures
+if [[ -f "$file" ]]; then
+#^ @keyword.conditional
+#  ^^ @punctuation.bracket
+#     ^^ @operator @operator
+#        ^^^^^^^ @string
+#         ^ @punctuation.special @none
+#                ^^ @punctuation.bracket
+#                  ^ @punctuation.delimiter
+#                     ^^^^ @keyword.conditional
+    echo "File exists"
+    #^^^ @function.call @function.builtin @number
+    #    ^^^^^^^^^^^^^ @string
+fi
+#^ @keyword.conditional
+
+for i in {1..10}; do
+#^^ @keyword.repeat
+#      ^^ @keyword.conditional
+#        ^ @punctuation.bracket
+#         ^ @number
+#          ^^ @operator
+#            ^^ @number
+#              ^ @punctuation.bracket
+#               ^ @punctuation.delimiter
+#                  ^^ @keyword.repeat
+    echo $i 
+    #^^^ @function.call @function.builtin @number
+    #    ^ @punctuation.special
+done
+#^^^ @keyword.repeat
+
+while read line; do
+#^^^^ @keyword.repeat
+#     ^^^^ @function.call @function.builtin @number
+#          ^^^^ @variable.parameter @number
+#              ^ @punctuation.delimiter
+#                ^^ @keyword.repeat
+    echo "$line"
+    #^^^ @function.call @function.builtin @number
+    #    ^^^^^^^ @string
+    #     ^ @punctuation.special @none
+done
+#^^^ @keyword.repeat
+
+case $input in
+#^^^ @keyword.conditional
+#    ^ @punctuation.special @none
+#           ^^ @keyword.conditional
+    "yes"|"y")
+    #^^^^ @string @variable.parameter
+    #    ^ @operator
+    #     ^^^ @string @variable.parameter
+    #        ^ @punctuation.bracket
+        echo "Yes"
+        #^^^ @function.call @function.builtin @number
+        #    ^^^^^ @string
+        ;; 
+        #^ @punctuation.delimiter
+    "no"|"n")
+    #^^^ @string @variable.parameter
+    #   ^ @operator
+    #    ^^^ @string @variable.parameter
+    #       ^ @punctuation.bracket
+        echo "No"
+        #^^^ @function.call @function.builtin @number
+        #    ^^^^ @string
+        ;;
+        #^ @punctuation.delimiter
+    *)
+   #^ @string.regexp
+   # ^ @punctuation.bracket
+        echo "Unknown"
+        #^^^ @function.call @function.builtin @number
+        #    ^^^^^^^^^ @string
+        ;;
+        #^ @punctuation.delimiter
+esac
+#^^^ @keyword.conditional
+
+# Arrays
+arr=(one two three)
+#^^ @variable @constant
+#  ^ @operator
+#   ^ @punctuation.bracket
+#                 ^ @punctuation.bracket
+
+echo ${arr[1]}
+#^^^ @function.call @function.builtin @number
+#    ^ @punctuation.special
+#     ^ @punctuation.special
+#         ^ @punctuation.bracket
+#          ^ @number
+#           ^ @punctuation.bracket
+#            ^ @punctuation.special
+
+# Parameter expansion
+echo ${name:-default}
+#^^^ @function.call @function.builtin @number
+#    ^^ @punctuation.special
+#                   ^ @punctuation.special
+
+echo ${#name}
+#^^^ @function.call @function.builtin @number
+#    ^^ @punctuation.special
+#           ^ @punctuation.special
+
+# Command substitution
+result=$(date)
+#^^^^^ @variable @constant
+#     ^ @operator
+#      ^ @punctuation.special
+#       ^ @punctuation.special
+#        ^^^^ @function.call @number
+#            ^ @punctuation.special
+
+result=`date`
+#^^^^^ @variable @constant
+#     ^ @operator
+#       ^^^^ @function.call @number
+
+# Pipes and redirection
+ls | grep "test" > output.txt
+#^ @function.call @number
+#  ^ @operator
+#    ^^^^ @function.call @number
+#         ^^^^^^ @string
+#                ^ @operator
+#                  ^^^^^^^^^^ @string.special.path @number
+
+cat < input.txt >> output.txt
+#^^ @function.call @number
+#   ^ @operator
+#     ^^^^^^^^^ @string.special.path @number
+#               ^^ @operator
+#                  ^^^^^^^^^^ @string.special.path @number
+
+# Process substitution
+diff <(ls dir1) <(ls dir2)
+#^^^ @function.call @number
+#    ^ @punctuation.special
+#     ^ @punctuation.special
+#      ^^ @function.call @number
+#         ^^^^ @variable.parameter @number
+#                ^^ @punctuation.special
+#                  ^^ @function.call @number
+#                     ^^^^ @variable.parameter @number
+#                        ^ @punctuation.special @punctuation.bracket
+
+# Test commands
+[[ -f file.txt ]]
+#^ @punctuation.bracket
+#  ^^ @operator @operator
+#              ^^ @punctuation.bracket
+
+ [ -n "$var" ]
+#^ @punctuation.bracket
+#  ^^ @operator @operator
+#     ^^^^^ @string
+#      ^ @punctuation.special
+#            ^ @punctuation.bracket
+
+# Arithmetic expansion
+echo $((2 + 3))
+#^^^ @function.call @function.builtin @number
+#    ^ @punctuation.special
+#     ^^ @punctuation.special
+#       ^ @number
+#         ^ @operator
+#           ^ @number
+#             ^^ @punctuation.special @punctuation.bracket
+
+# Globbing patterns
+ls *.txt
+#^ @function.call @number
+
+# Brace expansion
+echo {a,b,c}
+#^^^ @function.call @function.builtin @number
+
+# Here documents
+cat << EOF
+#^^ @function.call @number
+#   ^^ @operator
+#      ^^^ @label
+This is a heredoc
+#^^^^^^^^^^^^^^^^^ @string
+EOF
+#^^ @label
+
+cat <<< "string"
+#^^ @function.call @number
+#   ^^ @operator
+#     ^ @operator
+#       ^^^^^^^^ @string
+
+# Special variables
+echo $0 $1 $@ $* $# $?
+#^^^ @function.call @function.builtin @number
+#    ^^ @punctuation.special @none @variable.builtin
+#     ^ @variable.builtin
+#       ^^ @punctuation.special @none @variable.builtin
+#        ^ @variable.builtin
+#          ^^ @punctuation.special @none @variable.builtin
+#           ^ @variable.builtin
+#             ^^ @punctuation.special @none @variable.builtin
+#              ^ @variable.builtin
+#                ^^ @punctuation.special @none @variable.builtin
+#                 ^ @variable.builtin
+#                   ^^ @punctuation.special @none @variable.builtin
+#                    ^ @variable.builtin
+# NOTE: $$ is not captured by current query/grammar
+
+# Conditional operators
+[[ $a == $b ]] && echo "equal"
+#^ @punctuation.bracket
+#  ^ @punctuation.special @none
+#     ^^ @operator @operator
+#        ^ @punctuation.special @none
+#           ^^ @punctuation.bracket
+#              ^^ @operator
+#                 ^^^^ @function.builtin @function.call @number
+#                      ^^^^^^^ @string
+
+[[ $a != $b ]] || echo "not equal"
+#^ @punctuation.bracket
+#  ^^ @punctuation.special @none
+#     ^^ @operator
+#        ^^ @punctuation.special @none
+#              ^^ @operator
+#                 ^^^^ @function.call @function.builtin @number
+#                      ^^^^^^^^^^^ @string
+
+# Background jobs
+sleep 10 &
+#^^^^ @function.call @number
+#     ^^ @number
+#        ^ @punctuation.delimiter
+
+# Negation
+ ! command
+#^ @operator
+  #^^^^^^ @function.call @number
+
+# File descriptors
+exec 3> file.txt
+#^^^ @function.call @function.builtin @number
+#    ^ @operator
+#     ^ @operator
+#       ^^^^^^^^ @string.special.path @number
+
+# ZSH-specific features
+setopt AUTO_CD
+#^^^^^ @function.call @number
+#      ^^^^^^^ @variable.parameter @number
+
+autoload -U compinit
+#^^^^^^^ @function.call @number
+#        ^^ @variable.parameter @number
+#           ^^^^^^^^ @variable.parameter @number
+
+# Associative arrays
+typeset -A hash
+#^^^^^^ @keyword
+#       ^^ @variable.parameter @number
+#          ^^^^ @variable.parameter @constant
+
+hash[key]="value"
+#^^^ @variable @constant
+#   ^ punctuation.bracket
+#       ^ punctuation.bracket
+#        ^ @operator
+#         ^^^^^^^ @string
+
+# Regular expressions
+[[ "text" =~ ^[a-z]+$ ]]
+#^ @punctuation.bracket
+#  ^^^^^^ @string
+#         ^^ @operator
+#            ^^^^^^^^^ @string.regexp
+#                     ^^ @punctuation.bracket
+
+# Quotes and escaping
+echo "double quotes with $var"
+#^^^ @function.call @function.builtin @number
+#    ^^^^^^^^^^^^^^^^^^^^^^^^^ @string
+#                        ^ @punctuation.special @none
+
+echo 'single quotes $var'
+#^^^ @function.call @function.builtin @number
+#    ^^^^^^^^^^^^^^^^^^^^ @string
+
+echo $'ansi-c quotes\n'
+#^^^ @function.call @function.builtin @number
+#     ^^^^^^^^^^^^^^^^^ @string
+
+echo "escaped \" quote"
+#^^^ @function.call @function.builtin @number
+#    ^^^^^^^^^^^^^^^^^^ @string
+
+# Command chaining
+cmd1 && cmd2 || cmd3
+#^^^ @function.call @number
+#    ^^ @operator
+#       ^^^^ @function.call @number
+#            ^^ @operator
+#               ^^^^ @function.call @number