tree-sitter-batch 0.3.1 → 0.4.1

package/README.md

@@ -48,8 +48,10 @@ Parsed tree:
   (echo_off)
   (comment)
   (setlocal_stmt)
-  (variable_assignment)
-  (variable_assignment)
+  (variable_assignment
+    (set_keyword) (variable_name) (assignment_value))
+  (variable_assignment
+    (set_keyword) (set_option) (variable_name) (assignment_value))
   (if_stmt
     (string)
     (parenthesized

package/grammar.js

@@ -4,6 +4,7 @@ const kw = (word) => token(prec(10, ci(word)));
 export default grammar({
   name: 'batch',
   extras: () => [/[ \t]/],
+  word: ($) => $.command_name,
   rules: {
     program: ($) => repeat(choice(seq($._stmt, /\r?\n/), /\r?\n/)),
     _stmt: ($) => choice(
@@ -19,12 +20,13 @@ export default grammar({
       seq('::', /[^\r\n]*/),
     ))),
     label: () => token(seq(':', /[a-zA-Z_][a-zA-Z0-9_-]*/)),
-    variable_assignment: () => prec(8, seq(
-      optional('@'), kw('set'),
-      optional(seq(/[ \t]+/, /\/[aApP]/)), /[ \t]+/,
+    variable_assignment: ($) => prec(8, seq(
+      optional('@'), alias(kw('set'), $.set_keyword),
+      optional(seq(/[ \t]+/, alias(/\/[aApP]/, $.set_option))),
+      /[ \t]+/,
       choice(
-        seq('"', /[a-zA-Z_][a-zA-Z0-9_]*/, '=', optional(/[^"\r\n]*/), '"'),
-        seq(/[a-zA-Z_][a-zA-Z0-9_]*/, '=', optional(/[^\r\n]*/)),
+        seq('"', alias(/[a-zA-Z_][a-zA-Z0-9_]*/, $.variable_name), '=', optional(alias(/[^"\r\n]+/, $.assignment_value)), '"'),
+        seq(alias(/[a-zA-Z_][a-zA-Z0-9_]*/, $.variable_name), '=', optional(alias(/[^\r\n]+/, $.assignment_value))),
       ),
     )),
     if_stmt: ($) => prec.right(8, seq(
@@ -83,7 +85,7 @@ export default grammar({
     for_options: () => token(prec(10, choice(ci('/d'), seq(ci('/r'), optional(seq(/[ \t]+/, /[^\s]+/))), ci('/l'), seq(ci('/f'), optional(seq(/[ \t]+/, '"', /[^"]*/, '"')))))),
     for_variable: () => token(seq('%%', optional('~'), /[a-zA-Z]/)),
     for_set: () => /[^)\r\n]+/,
-    parenthesized: ($) => seq('(', repeat(choice(seq($._stmt, /\r?\n/), /\r?\n/)), ')'),
+    parenthesized: ($) => seq('(', repeat(choice(seq($._stmt, /\r?\n/), /\r?\n/)), optional($._stmt), ')'),
     redirect_stmt: ($) => prec.right(4, seq(choice($.cmd, $.parenthesized), $.redirection)),
     redirection: ($) => prec.right(seq(
       optional(/[0-2]/), $.redirect_op, $.redirect_target,
@@ -93,8 +95,8 @@ export default grammar({
     redirect_target: () => token(choice(ci('nul'), ci('con'), /[^\s|&><\r\n]+/)),
     pipe_stmt: ($) => prec.left(3, seq(choice($.cmd, $.parenthesized), '|', choice($.cmd, $.parenthesized))),
     cond_exec: ($) => choice(
-      prec.left(2, seq(choice($.cmd, $.parenthesized), '&&', choice($.cmd, $.parenthesized))),
-      prec.left(1, seq(choice($.cmd, $.parenthesized), '||', choice($.cmd, $.parenthesized))),
+      prec.left(1, seq(choice($.cond_exec, $.cmd, $.parenthesized), '&&', choice($.cmd, $.parenthesized))),
+      prec.left(1, seq(choice($.cond_exec, $.cmd, $.parenthesized), '||', choice($.cmd, $.parenthesized))),
     ),
     variable_reference: () => token(choice(
       seq('%', /[a-zA-Z_][a-zA-Z0-9_]*/, '%'),
@@ -108,9 +110,10 @@ export default grammar({
     cmd: ($) => prec.right(5, seq(optional('@'), $.command_name, optional($.argument_list))),
     command_name: () => /[a-zA-Z_][a-zA-Z0-9_.-]*/,
     argument_list: ($) => prec.right(repeat1($._arg)),
-    _arg: ($) => choice($.string, $.variable_reference, $.command_option, $.argument_value),
+    _arg: ($) => choice($.string, $.variable_reference, $.command_option, $.paren_expression, $.argument_value),
     command_option: () => token(seq('/', /[a-zA-Z_?][a-zA-Z0-9_:]*/)),
-    argument_value: () => /[^\s|&><"\r\n%!][^\s|&><"\r\n]*/,
+    paren_expression: ($) => seq('(', repeat($._arg), ')'),
+    argument_value: () => /[^\s|&><"\r\n%!()][^\s|&><"\r\n()]*/,
     integer: () => /[0-9]+/,
   },
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tree-sitter-batch",
-  "version": "0.3.1",
+  "version": "0.4.1",
   "description": "A Windows Batch/CMD grammar for tree-sitter",
   "type": "module",
   "repository": {

package/queries/highlights.scm

@@ -8,7 +8,10 @@
 (label) @label
 
 ; Variable assignment
-(variable_assignment) @variable
+(set_keyword) @keyword
+(variable_name) @variable
+(set_option) @constant
+(assignment_value) @string
 
 ; IF/FOR/GOTO/CALL statements
 (if_stmt) @keyword

package/src/grammar.json

@@ -1,6 +1,7 @@
 {
   "$schema": "https://tree-sitter.github.io/tree-sitter/assets/schemas/grammar.schema.json",
   "name": "batch",
+  "word": "command_name",
   "rules": {
     "program": {
       "type": "REPEAT",
@@ -249,15 +250,20 @@
             ]
           },
           {
-            "type": "TOKEN",
+            "type": "ALIAS",
             "content": {
-              "type": "PREC",
-              "value": 10,
+              "type": "TOKEN",
               "content": {
-                "type": "PATTERN",
-                "value": "[sS][eE][tT]"
+                "type": "PREC",
+                "value": 10,
+                "content": {
+                  "type": "PATTERN",
+                  "value": "[sS][eE][tT]"
+                }
               }
-            }
+            },
+            "named": true,
+            "value": "set_keyword"
           },
           {
             "type": "CHOICE",
@@ -270,8 +276,13 @@
                     "value": "[ \\t]+"
                   },
                   {
-                    "type": "PATTERN",
-                    "value": "\\/[aApP]"
+                    "type": "ALIAS",
+                    "content": {
+                      "type": "PATTERN",
+                      "value": "\\/[aApP]"
+                    },
+                    "named": true,
+                    "value": "set_option"
                   }
                 ]
               },
@@ -295,8 +306,13 @@
                     "value": "\""
                   },
                   {
-                    "type": "PATTERN",
-                    "value": "[a-zA-Z_][a-zA-Z0-9_]*"
+                    "type": "ALIAS",
+                    "content": {
+                      "type": "PATTERN",
+                      "value": "[a-zA-Z_][a-zA-Z0-9_]*"
+                    },
+                    "named": true,
+                    "value": "variable_name"
                   },
                   {
                     "type": "STRING",
@@ -306,8 +322,13 @@
                     "type": "CHOICE",
                     "members": [
                       {
-                        "type": "PATTERN",
-                        "value": "[^\"\\r\\n]*"
+                        "type": "ALIAS",
+                        "content": {
+                          "type": "PATTERN",
+                          "value": "[^\"\\r\\n]+"
+                        },
+                        "named": true,
+                        "value": "assignment_value"
                       },
                       {
                         "type": "BLANK"
@@ -324,8 +345,13 @@
                 "type": "SEQ",
                 "members": [
                   {
-                    "type": "PATTERN",
-                    "value": "[a-zA-Z_][a-zA-Z0-9_]*"
+                    "type": "ALIAS",
+                    "content": {
+                      "type": "PATTERN",
+                      "value": "[a-zA-Z_][a-zA-Z0-9_]*"
+                    },
+                    "named": true,
+                    "value": "variable_name"
                   },
                   {
                     "type": "STRING",
@@ -335,8 +361,13 @@
                     "type": "CHOICE",
                     "members": [
                       {
-                        "type": "PATTERN",
-                        "value": "[^\\r\\n]*"
+                        "type": "ALIAS",
+                        "content": {
+                          "type": "PATTERN",
+                          "value": "[^\\r\\n]+"
+                        },
+                        "named": true,
+                        "value": "assignment_value"
                       },
                       {
                         "type": "BLANK"
@@ -1227,6 +1258,18 @@
             ]
           }
         },
+        {
+          "type": "CHOICE",
+          "members": [
+            {
+              "type": "SYMBOL",
+              "name": "_stmt"
+            },
+            {
+              "type": "BLANK"
+            }
+          ]
+        },
         {
           "type": "STRING",
           "value": ")"
@@ -1421,13 +1464,17 @@
       "members": [
         {
           "type": "PREC_LEFT",
-          "value": 2,
+          "value": 1,
           "content": {
             "type": "SEQ",
             "members": [
               {
                 "type": "CHOICE",
                 "members": [
+                  {
+                    "type": "SYMBOL",
+                    "name": "cond_exec"
+                  },
                   {
                     "type": "SYMBOL",
                     "name": "cmd"
@@ -1467,6 +1514,10 @@
               {
                 "type": "CHOICE",
                 "members": [
+                  {
+                    "type": "SYMBOL",
+                    "name": "cond_exec"
+                  },
                   {
                     "type": "SYMBOL",
                     "name": "cmd"
@@ -1708,6 +1759,10 @@
           "type": "SYMBOL",
           "name": "command_option"
         },
+        {
+          "type": "SYMBOL",
+          "name": "paren_expression"
+        },
         {
           "type": "SYMBOL",
           "name": "argument_value"
@@ -1730,9 +1785,29 @@
         ]
       }
     },
+    "paren_expression": {
+      "type": "SEQ",
+      "members": [
+        {
+          "type": "STRING",
+          "value": "("
+        },
+        {
+          "type": "REPEAT",
+          "content": {
+            "type": "SYMBOL",
+            "name": "_arg"
+          }
+        },
+        {
+          "type": "STRING",
+          "value": ")"
+        }
+      ]
+    },
     "argument_value": {
       "type": "PATTERN",
-      "value": "[^\\s|&><\"\\r\\n%!][^\\s|&><\"\\r\\n]*"
+      "value": "[^\\s|&><\"\\r\\n%!()][^\\s|&><\"\\r\\n()]*"
     },
     "integer": {
       "type": "PATTERN",

package/src/node-types.json

@@ -15,6 +15,10 @@
           "type": "command_option",
           "named": true
         },
+        {
+          "type": "paren_expression",
+          "named": true
+        },
         {
           "type": "string",
           "named": true
@@ -76,6 +80,10 @@
           "type": "cmd",
           "named": true
         },
+        {
+          "type": "cond_exec",
+          "named": true
+        },
         {
           "type": "parenthesized",
           "named": true
@@ -211,6 +219,37 @@
     "named": true,
     "fields": {}
   },
+  {
+    "type": "paren_expression",
+    "named": true,
+    "fields": {},
+    "children": {
+      "multiple": true,
+      "required": false,
+      "types": [
+        {
+          "type": "argument_value",
+          "named": true
+        },
+        {
+          "type": "command_option",
+          "named": true
+        },
+        {
+          "type": "paren_expression",
+          "named": true
+        },
+        {
+          "type": "string",
+          "named": true
+        },
+        {
+          "type": "variable_reference",
+          "named": true
+        }
+      ]
+    }
+  },
   {
     "type": "parenthesized",
     "named": true,
@@ -431,7 +470,29 @@
   {
     "type": "variable_assignment",
     "named": true,
-    "fields": {}
+    "fields": {},
+    "children": {
+      "multiple": true,
+      "required": true,
+      "types": [
+        {
+          "type": "assignment_value",
+          "named": true
+        },
+        {
+          "type": "set_keyword",
+          "named": true
+        },
+        {
+          "type": "set_option",
+          "named": true
+        },
+        {
+          "type": "variable_name",
+          "named": true
+        }
+      ]
+    }
   },
   {
     "type": "\"",
@@ -461,6 +522,10 @@
     "type": "argument_value",
     "named": true
   },
+  {
+    "type": "assignment_value",
+    "named": true
+  },
   {
     "type": "command_name",
     "named": true
@@ -501,10 +566,22 @@
     "type": "redirect_target",
     "named": true
   },
+  {
+    "type": "set_keyword",
+    "named": true
+  },
+  {
+    "type": "set_option",
+    "named": true
+  },
   {
     "type": "string",
     "named": true
   },
+  {
+    "type": "variable_name",
+    "named": true
+  },
   {
     "type": "variable_reference",
     "named": true