Add WGSL parser

2 files changed, 111 insertions, 0 deletions

diff --git a/queries/wgsl/folds.scm b/queries/wgsl/folds.scm
new file mode 100644
index 000000000..d8000edee
--- /dev/null
+++ b/queries/wgsl/folds.scm
@@ -0,0 +1,10 @@
+[
+    (struct_declaration)
+    (function_declaration)
+    (if_statement)
+    (switch_statement)
+    (switch_body)
+    (loop_statement)
+    (for_statement)
+    (while_statement)
+] @fold
diff --git a/queries/wgsl/highlights.scm b/queries/wgsl/highlights.scm
new file mode 100644
index 000000000..9f9696e10
--- /dev/null
+++ b/queries/wgsl/highlights.scm
@@ -0,0 +1,101 @@
+(identifier) @variable
+(int_literal) @number
+(float_literal) @float
+(bool_literal) @boolean
+
+(type_declaration) @type
+
+(function_declaration
+    (identifier) @function)
+
+(parameter
+    (variable_identifier_declaration (identifier) @parameter))
+
+(struct_declaration
+    (identifier) @structure)
+	
+(struct_declaration
+	(struct_member (variable_identifier_declaration (identifier) @field)))
+
+(type_constructor_or_function_call_expression
+    (type_declaration) @function)
+
+[
+    "struct"
+    "bitcast"
+    "discard"
+    "enable"
+    "fallthrough"
+    "fn"
+    "let"
+    "private"
+    "read"
+    "read_write"
+    "storage"
+    "type"
+    "uniform"
+    "var"
+    "workgroup"
+    "write"
+    "override"
+    (texel_format)
+] @keyword
+
+"fn" @keyword.function
+
+"return" @keyword.return
+
+[ "," "." ":" ";" ] @punctuation.delimiter
+
+["(" ")" "[" "]" "{" "}"] @punctuation.bracket
+
+[
+    "loop"
+    "for"
+    "while"
+    "break"
+    "continue"
+    "continuing"
+] @repeat
+
+[
+    "if"
+    "else"
+    "switch"
+    "case"
+    "default"
+] @conditional
+
+[
+    "&"
+    "&&"
+    "/"
+    "!"
+    "="
+    "=="
+    "!="
+    ">"
+    ">="
+    ">>"
+    "<"
+    "<="
+    "<<"
+    "%"
+    "-"
+    "+"
+    "|"
+    "||"
+    "*"
+    "~"
+    "^"
+    "@"
+    "++"
+    "--"
+] @operator
+
+(attribute
+    (identifier) @attribute)
+
+(comment) @comment
+
+(ERROR) @error