vlang · spytheman · Mar 16, 2024 · Mar 15, 2024 · Mar 16, 2024
diff --git a/tree_sitter_v/grammar.cjs b/tree_sitter_v/grammar.cjs
@@ -152,6 +152,7 @@ module.exports = grammar({
   rules: {
     source_file: ($) =>
       seq(
+        optional($.shebang),
         optional($.module_clause),
         optional($.import_list),
         repeat(
@@ -162,14 +163,15 @@ module.exports = grammar({
         ),
       ),
 
+    shebang: (_) => token(/\#\!([^\\\r\n]+)+/),
+
     // http://stackoverflow.com/questions/13014947/regex-to-match-a-c-style-multiline-comment/36328890#36328890
     comment: (_) =>
       token(
         choice(
           /\/\/[^\n\r]*/,
           /\/\*(?:[^\/][^\*]+\/\*+[^\/][^\*]+)+(?:[^\*][^\/]+\*+\/[^\*][^\/]+)+\//,
           /\/\*[^\*]*\*\//,
-          /\#\!.*/,
         ),
       ),
 

diff --git a/tree_sitter_v/node_types.v b/tree_sitter_v/node_types.v
@@ -182,6 +182,7 @@ pub enum NodeType {
 	nil_
 	none_
 	rune_literal
+	shebang
 	true_
 }
 
@@ -491,5 +492,6 @@ const node_type_name_to_enum = {
 	'nil':                              NodeType.nil_
 	'none':                             NodeType.none_
 	'rune_literal':                     NodeType.rune_literal
+	'shebang':                          NodeType.shebang
 	'true':                             NodeType.true_
 }
diff --git a/tree_sitter_v/queries/helix.highlights.scm b/tree_sitter_v/queries/helix.highlights.scm
@@ -1,4 +1,7 @@
-(comment) @comment
+[
+ (comment)
+ (shebang)
+] @comment
 
 (module_clause
  (identifier) @namespace)

diff --git a/tree_sitter_v/queries/highlights.scm b/tree_sitter_v/queries/highlights.scm
@@ -1,5 +1,9 @@
 (ERROR) @error
-(comment) @comment
+
+[
+ (comment)
+ (shebang)
+] @comment
 
 (identifier) @variable
 (import_path) @variable

diff --git a/tree_sitter_v/src/grammar.json b/tree_sitter_v/src/grammar.json
diff --git a/tree_sitter_v/src/node-types.json b/tree_sitter_v/src/node-types.json