From 88aad97312787c11154afbe16068fc2b1e18b249 Mon Sep 17 00:00:00 2001
From: Oliver Thomas <thomaso@ie-freiburg.mpg.de>
Date: Mon, 5 Aug 2024 21:56:30 +0200
Subject: [PATCH] feat(snakemake): update highlights and indents

---
 queries/snakemake/highlights.scm | 24 ++++++++++++++----------
 queries/snakemake/indents.scm    |  9 +++++++++
 2 files changed, 23 insertions(+), 10 deletions(-)

diff --git a/queries/snakemake/highlights.scm b/queries/snakemake/highlights.scm
index 58e921e88..8781b405c 100644
--- a/queries/snakemake/highlights.scm
+++ b/queries/snakemake/highlights.scm
@@ -13,10 +13,9 @@
     name: _ @keyword))
 
 ; Subordinate directives (eg. input, output)
-(_
-  body: (_
-    (directive
-      name: _ @label)))
+body: (_
+  (directive
+    name: _ @label))
 
 ; rule/module/checkpoint names
 (rule_definition
@@ -30,11 +29,14 @@
 
 ; Rule imports
 (rule_import
-  "use" @keyword.import
-  "rule" @keyword.import
-  "from" @keyword.import
-  "as"? @keyword.import
-  "with"? @keyword.import)
+  [
+    "use"
+    "rule"
+    "from"
+    "exclude"
+    "as"
+    "with"
+  ] @keyword.import)
 
 ; Rule inheritance
 (rule_inheritance
@@ -44,7 +46,9 @@
 
 ; Wildcard names
 (wildcard
-  (identifier) @variable
+  (identifier) @variable)
+
+(wildcard
   (flag) @variable.parameter.builtin)
 
 ; builtin variables
diff --git a/queries/snakemake/indents.scm b/queries/snakemake/indents.scm
index a4648eff4..140b7fb3c 100644
--- a/queries/snakemake/indents.scm
+++ b/queries/snakemake/indents.scm
@@ -9,8 +9,17 @@
 ((rule_inheritance) @indent.begin
   (#set! indent.immediate 1))
 
+((rule_import
+  "with"
+  ":") @indent.begin
+  (#set! indent.immediate 1))
+
 ((module_definition) @indent.begin
   (#set! indent.immediate 1))
 
 ((directive) @indent.begin
   (#set! indent.immediate 1))
+
+; end indentation after last parameter node (no following ',')
+(directive_parameters
+  (_) @indent.end .)