From f22e0aa2ae4169851b43d2225e6a43ef36442659 Mon Sep 17 00:00:00 2001
From: voroskoi <60064214+voroskoi@users.noreply.github.com>
Date: Sat, 28 Aug 2021 06:32:01 +0200
Subject: [PATCH] Add zig tree-sitter support (#631)

* Add initial zig tree-sitter support

* zig/highlights.scm: remove unnecessary queries

* Add zig/indents.toml
---
 .gitmodules                            |   4 +
 helix-syntax/languages/tree-sitter-zig |   1 +
 languages.toml                         |  12 ++
 runtime/queries/zig/highlights.scm     | 198 +++++++++++++++++++++++++
 runtime/queries/zig/indents.toml       |  12 ++
 5 files changed, 227 insertions(+)
 create mode 160000 helix-syntax/languages/tree-sitter-zig
 create mode 100644 runtime/queries/zig/highlights.scm
 create mode 100644 runtime/queries/zig/indents.toml

diff --git a/.gitmodules b/.gitmodules
index e750198ae..0e0156589 100644
--- a/.gitmodules
+++ b/.gitmodules
@@ -102,3 +102,7 @@
 	path = helix-syntax/languages/tree-sitter-protobuf
 	url = https://github.com/yusdacra/tree-sitter-protobuf.git
 	shallow = true
+[submodule "helix-syntax/languages/tree-sitter-zig"]
+	path = helix-syntax/languages/tree-sitter-zig
+	url = https://github.com/maxxnino/tree-sitter-zig
+	shallow = true
diff --git a/helix-syntax/languages/tree-sitter-zig b/helix-syntax/languages/tree-sitter-zig
new file mode 160000
index 000000000..049162bea
--- /dev/null
+++ b/helix-syntax/languages/tree-sitter-zig
@@ -0,0 +1 @@
+Subproject commit 049162bea8a44e1a4acd01b06e1c8672d9231a86
diff --git a/languages.toml b/languages.toml
index 47155523b..1ca40377d 100644
--- a/languages.toml
+++ b/languages.toml
@@ -233,3 +233,15 @@ indent = { tab-width = 4, unit = "    " }
 # comment-token = "--"
 #
 # indent = { tab-width = 2, unit = "  " }
+
+[[language]]
+name = "zig"
+scope = "source.zig"
+injection-regex = "zig"
+file-types = ["zig"]
+roots = ["build.zig"]
+auto-format = true
+comment-token = "//"
+
+language-server = { command = "zls" }
+indent = { tab-width = 4, unit = "    " }
diff --git a/runtime/queries/zig/highlights.scm b/runtime/queries/zig/highlights.scm
new file mode 100644
index 000000000..5a3d62dc8
--- /dev/null
+++ b/runtime/queries/zig/highlights.scm
@@ -0,0 +1,198 @@
+[
+  (container_doc_comment)
+  (doc_comment)
+  (line_comment)
+] @comment
+
+; field in top level decl, and in struct, union...
+(ContainerField
+  (IDENTIFIER) @property
+  (SuffixExpr (IDENTIFIER) @type)?
+)
+
+; error.OutOfMemory;
+(SuffixExpr
+  "error"
+  "."
+  (IDENTIFIER) @constant
+)
+
+; var x: IDENTIFIER
+type: (SuffixExpr (IDENTIFIER) @type)
+
+; IDENTIFIER{}
+constructor: (SuffixExpr (IDENTIFIER) @constructor)
+
+; fields
+(FieldInit (IDENTIFIER) @property)
+
+; foo.bar.baz.function() calls
+(
+  (SuffixOp
+    (IDENTIFIER) @function
+  )
+  .
+  (FnCallArguments)
+)
+
+; function() calls
+(
+  (
+    (IDENTIFIER) @function
+  )
+  .
+  (FnCallArguments)
+)
+
+; functionn decl
+(FnProto
+  (IDENTIFIER) @function
+  (SuffixExpr (IDENTIFIER) @type)?
+  ("!")? @function.macro
+)
+
+; function parameters and types
+(ParamDecl
+  (IDENTIFIER) @variable.parameter
+  ":"
+  [
+    (ParamType (SuffixExpr (IDENTIFIER) @type))
+    (ParamType)
+  ]
+)
+
+; switch
+(SwitchItem
+  (SuffixExpr
+    "."
+    .
+    (IDENTIFIER) @constant
+  )
+)
+
+(INTEGER) @number
+
+(FLOAT) @number
+
+[
+  (STRINGLITERAL)
+  (STRINGLITERALSINGLE)
+] @string
+
+(CHAR_LITERAL) @string
+
+[
+  "allowzero"
+  "volatile"
+  "anytype"
+  "anyframe"
+  (BuildinTypeExpr)
+] @type.builtin
+
+(BreakLabel (IDENTIFIER) @label)
+(BlockLabel (IDENTIFIER) @label)
+
+[
+  "true"
+  "false"
+  "undefined"
+  "unreachable"
+  "null"
+] @constant.builtin
+
+[
+  "else"
+  "if"
+  "switch"
+  "for"
+  "while"
+  "return"
+  "break"
+  "continue"
+  "defer"
+  "errdefer"
+  "async"
+  "nosuspend"
+  "await"
+  "suspend"
+  "resume"
+  "try"
+  "catch"
+] @keyword.control
+
+[
+  "struct"
+  "enum"
+  "union"
+  "error"
+  "packed"
+  "opaque"
+  "test"
+  "usingnamespace"
+  "export"
+  "extern"
+  "const"
+  "var"
+  "comptime"
+  "threadlocal"
+] @keyword
+
+[
+  "pub"
+  "fn"
+] @keyword.function
+
+; PrecProc
+[
+  "inline"
+  "noinline"
+  "asm"
+  "callconv"
+  "noalias"
+] @attribute
+
+[
+  (BUILTINIDENTIFIER)
+  "linksection"
+  "align"
+] @function.builtin
+
+[
+  (CompareOp)
+  (BitwiseOp)
+  (BitShiftOp)
+  (AdditionOp)
+  (MultiplyOp)
+  (PrefixOp)
+  "or"
+  "and"
+  "orelse"
+  "*"
+  "**"
+  "->"
+  "=>"
+  ".?"
+  ".*"
+  "="
+] @operator
+
+[
+  ";"
+  "."
+  ","
+  ":"
+] @punctuation.delimiter
+
+[
+  ".."
+  "..."
+  "["
+  "]"
+  "("
+  ")"
+  "{"
+  "}"
+  (Payload "|")
+  (PtrPayload "|")
+  (PtrIndexPayload "|")
+] @punctuation
diff --git a/runtime/queries/zig/indents.toml b/runtime/queries/zig/indents.toml
new file mode 100644
index 000000000..e119078b3
--- /dev/null
+++ b/runtime/queries/zig/indents.toml
@@ -0,0 +1,12 @@
+indent = [
+ "block",
+ "match_block",
+ "arguments",
+ "parameters"
+]
+
+outdent = [
+ "}",
+ "]",
+ ")"
+]