Support for mut is clause

vlang · Mar 31, 2024 · 9bdaeb9 · 9bdaeb9
1 parent c8346f4
commit 9bdaeb9
Show file tree

Hide file tree

Showing 6 changed files with 81 additions and 2 deletions.
diff --git a/tree_sitter_v/bindings/node_types.v b/tree_sitter_v/bindings/node_types.v
@@ -108,6 +108,7 @@ pub enum NodeType {
 	match_expression_list
 	module_clause
 	multi_return_type
+	mut_is_clause
 	mutability_modifiers
 	mutable_expression
 	mutable_identifier
@@ -415,6 +416,7 @@ const node_type_name_to_enum = {
 	'match_expression_list':            NodeType.match_expression_list
 	'module_clause':                    NodeType.module_clause
 	'multi_return_type':                NodeType.multi_return_type
+	'mut_is_clause':                    NodeType.mut_is_clause
 	'mutability_modifiers':             NodeType.mutability_modifiers
 	'mutable_expression':               NodeType.mutable_expression
 	'mutable_identifier':               NodeType.mutable_identifier

diff --git a/tree_sitter_v/grammar.js b/tree_sitter_v/grammar.js
@@ -1190,10 +1190,12 @@ module.exports = grammar({
 		for_statement: ($) =>
 			seq(
 				'for',
-				optional(choice($.range_clause, $.for_clause, $._expression)),
+				optional(choice($.range_clause, $.for_clause, $.mut_is_clause, $._expression)),
 				field('body', $.block),
 			),
 
+		mut_is_clause: ($) => prec.left(PREC.primary, seq('mut', $.is_expression)),
+
 		range_clause: ($) =>
 			prec.left(
 				PREC.primary,

diff --git a/tree_sitter_v/package.json b/tree_sitter_v/package.json
@@ -24,7 +24,7 @@
 		"node-gyp-build": "^4.8.0"
 	},
 	"peerDependencies": {
-		"tree-sitter": "^0.21.0"
+		"tree-sitter": "^0.21.1"
 	},
 	"peerDependenciesMeta": {
 		"tree_sitter": {

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
diff --git a/tree_sitter_v/test/corpus/for_statement.txt b/tree_sitter_v/test/corpus/for_statement.txt
@@ -54,6 +54,41 @@ for i in 0 .. 10 {
           (int_literal))))
     (block)))
 
+================================================================================
+Simple for is statement
+================================================================================
+for v is a {
+}
+--------------------------------------------------------------------------------
+
+(source_file
+  (for_statement
+    (is_expression
+      (reference_expression
+        (identifier))
+      (plain_type
+        (type_reference_expression
+          (identifier))))
+    (block)))
+
+================================================================================
+Simple for mut is statement
+================================================================================
+for mut v is a {
+}
+--------------------------------------------------------------------------------
+
+(source_file
+  (for_statement
+    (mut_is_clause
+      (is_expression
+        (reference_expression
+          (identifier))
+        (plain_type
+          (type_reference_expression
+            (identifier)))))
+    (block)))
+
 ================================================================================
 Simple C for statement
 ================================================================================