From fe49b1b2a6235e3c5939709b272d8f8ddace87a3 Mon Sep 17 00:00:00 2001
From: mx42 <49202620+Maan2003@users.noreply.github.com>
Date: Fri, 28 Jul 2023 19:15:53 +0000
Subject: [PATCH] treesitter: add more highlight rules (#3036)

follow up to #2902
---
 utils/treesitter/grammar.js             | 8 ++++----
 utils/treesitter/queries/highlights.scm | 7 ++++++-
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/utils/treesitter/grammar.js b/utils/treesitter/grammar.js
index 090a0617153b..da8f2ef7a9a8 100644
--- a/utils/treesitter/grammar.js
+++ b/utils/treesitter/grammar.js
@@ -151,7 +151,7 @@ module.exports = grammar({
         $.struct_type_literal
       ),
 
-    _binding_lhs: ($) => choice($.ident, '_'),
+    binding_lhs: ($) => choice($.ident, '_'),
 
     paren_pattern: ($) =>
       seq(
@@ -163,9 +163,9 @@ module.exports = grammar({
     _pattern_without_expression: ($) =>
       choice(
         'auto',
-        seq($._binding_lhs, ':', $._expression),
-        seq($._binding_lhs, ':!', $._expression),
-        seq('template', $._binding_lhs, ':!', $._expression),
+        seq($.binding_lhs, ':', $._expression),
+        seq($.binding_lhs, ':!', $._expression),
+        seq('template', $.binding_lhs, ':!', $._expression),
         seq('var', $._pattern),
         $.paren_pattern,
         // alternative patterns
diff --git a/utils/treesitter/queries/highlights.scm b/utils/treesitter/queries/highlights.scm
index da90e3917f96..18085e7db84c 100644
--- a/utils/treesitter/queries/highlights.scm
+++ b/utils/treesitter/queries/highlights.scm
@@ -16,7 +16,12 @@
 (function_declaration (declared_name (ident) @function))
 (call_expression (ident) @function)
 
-; TODO: add more specific rules
+(namespace_declaration (declared_name) @namespace)
+(interface_declaration (declared_name) @type)
+(constraint_declaration (declared_name) @type)
+(class_declaration (declared_name) @type)
+(choice_declaration (declared_name) @type)
+(binding_lhs) @variable
 
 ; upper case => type
 ((ident) @type