From d6cacb27319da9be84de7cad45e7f1264b210052 Mon Sep 17 00:00:00 2001
From: Freddie Gilbraith <75043775+FreddieGilbraith@users.noreply.github.com>
Date: Thu, 20 Mar 2025 13:04:52 +0000
Subject: [PATCH] add werk language and highlights (#13136)

---
 book/src/generated/lang-support.md  |  1 +
 languages.toml                      | 10 +++++
 runtime/queries/werk/highlights.scm | 62 +++++++++++++++++++++++++++++
 3 files changed, 73 insertions(+)
 create mode 100644 runtime/queries/werk/highlights.scm

diff --git a/book/src/generated/lang-support.md b/book/src/generated/lang-support.md
index fd621c943..f6aee3fe3 100644
--- a/book/src/generated/lang-support.md
+++ b/book/src/generated/lang-support.md
@@ -247,6 +247,7 @@
 | wast | ✓ |  |  |  |
 | wat | ✓ |  |  | `wat_server` |
 | webc | ✓ |  |  |  |
+| werk | ✓ |  |  |  |
 | wgsl | ✓ |  |  | `wgsl-analyzer` |
 | wit | ✓ |  | ✓ |  |
 | wren | ✓ | ✓ | ✓ |  |
diff --git a/languages.toml b/languages.toml
index a0f59717c..079671b27 100644
--- a/languages.toml
+++ b/languages.toml
@@ -4244,3 +4244,13 @@ formatter = {command = "tlafmt", args = ["--stdin"]}
 name = "tlaplus"
 source = { git = "https://github.com/tlaplus-community/tree-sitter-tlaplus", rev = "4ba91b07b97741a67f61221d0d50e6d962e4987e"}
 
+[[language]]
+name = "werk"
+scope = "source.werk"
+file-types = [ "werk", { glob = "Werkfile" } ]
+comment-token = "#"
+indent = { tab-width = 2, unit = "  " }
+
+[[grammar]]
+name = "werk"
+source = { git = "https://github.com/little-bonsai/tree-sitter-werk", rev = "92b0f7fe98465c4c435794a58e961306193d1c1e" }
diff --git a/runtime/queries/werk/highlights.scm b/runtime/queries/werk/highlights.scm
new file mode 100644
index 000000000..2636be3cd
--- /dev/null
+++ b/runtime/queries/werk/highlights.scm
@@ -0,0 +1,62 @@
+(buildBlock . ("build") @keyword.function)
+(taskBlock . ("task") @keyword.function )
+(run . ("run") @keyword.function )
+(taskBlock name: (identifier) @function )
+
+(comment) @comment
+(string) @string
+(number) @constant.numeric
+(identifier) @identifier
+
+(include) @keyword.control.import
+(let) @keyword.storage
+(default) @keyword.storage
+(config) @keyword.storage
+
+(interpolation ["{" "}" "<" ">" ] @punctuation.special)
+["{" "}" "<" ">" "(" ")" "[" "]"] @punctuation.bracket
+["=>" "|"] @punctuation
+
+; Statements
+(build "build") @function
+(config "config") @function
+(copy "copy") @function
+(default "default") @function
+(delete "delete") @function
+(depfile "depfile") @function
+(envRemove "env-remove") @function
+(from "from") @function
+(info "info") @function
+(let "let") @function
+(setEnv "env") @function
+(shell "shell") @function
+(write "write") @function
+
+; Expressions
+(error "error" @function.builtin)
+(getEnv "env" @function.builtin)
+(glob "glob" @function.builtin)
+(include "include" @function.builtin)
+(info "info" @function.builtin)
+(read "read" @function.builtin)
+(warn "warn" @function.builtin)
+(which "which" @function.builtin)
+
+; Operations
+(op (string)  @operator) 
+("dedup" @operator)
+("first" @operator)
+("flatten" @operator)
+("last" @operator)
+("len" @operator)
+("lines" @operator)
+("tail" @operator)
+(assertEq "assert-eq" @operator)
+(discard "discard" @operator)
+(filter "filter" @operator)
+(filterMatch "filter-match" @operator)
+(join "join" @operator)
+(map "map" @operator)
+(match "match" @operator)
+(split "split" @operator)
+