improve highlighting

grammar.js

@@ -6,6 +6,7 @@ module.exports = grammar({
     _unspaced_statement: $ => seq(
       choice(
         $.string,
+        $.number,
         $.function_definition,
         $.type_definition,
         $.with_expr,
@@ -16,7 +17,6 @@ module.exports = grammar({
         $.use,
         $.include,
         $.def,
-        $.number,
         $.call_expr,
         seq(
           repeat('&'),
@@ -41,7 +41,7 @@ module.exports = grammar({
         repeat($._statement), 
       '}',
     ),
-    identifier: $ => /[^ \n\r\t<>:;&{}"'\[\]]+/,
+    identifier: $ => /[^ \n\r\t<>:;&{}"'\[\],]+/,
     call: $ => choice(
       seq(
         optional($.call),
@@ -52,20 +52,20 @@ module.exports = grammar({
       $.operation,
       $.identifier,
     ),
-    number: $ => /\d+(\.\d+)?/,
+    number: $ => token(prec(10, /\d+(\.\d+)?/)),
-    string: $ => seq('"', repeat(choice(/\\./, /./)), '"'),
+    string: $ => choice(
+      token(prec(10, seq('^', /[^ \n\r\t<>:;&{}"'\[\],]+/))),
+      seq('"', repeat(choice(/\\./, /./)), '"'), 
+    ),
     call_expr: $ => seq('<', optional($._spacing), repeat($._statement), $._unspaced_statement, '>'),
     with_expr: $ => seq(
       'with', $._spacing,
-      repeat($.identifier),
+      repeat(seq($.identifier, $._spacing)),
       ';',
     ),
     array: $ => seq('[', $._spacing, optional(repeat($._statement)), ']'),
     operation: $ => choice(/[+\-*\/%&]/, "=>!", "=>?", "=>?!", "=>"),
-    variable: $ => choice(
+    set_variable: $ => seq('=', $.identifier),
-      seq('def', $._spacing, $.identifier),
-      seq('=', $.identifier),
-    ),
     type_definition: $ => seq(
       'construct', $._spacing, 
       $.call, $._spacing,
@@ -115,6 +115,7 @@ module.exports = grammar({
     def: $ => seq(
       'def', $._spacing,
       $.identifier,
+      optional(repeat(seq(',', $._spacing, $.identifier)))
     ),
   },
 });

queries/highlights.scm

@@ -12,4 +12,7 @@
 
 (number) @number
 (string) @string
+(call (identifier) @function)
 (call (call) @function (identifier) @property)
+(def (identifier) @local.definition)
+(with_expr (identifier) @local.definition)

src/grammar.json

@@ -24,6 +24,10 @@
               "type": "SYMBOL",
               "name": "string"
             },
+            {
+              "type": "SYMBOL",
+              "name": "number"
+            },
             {
               "type": "SYMBOL",
               "name": "function_definition"
@@ -64,10 +68,6 @@
               "type": "SYMBOL",
               "name": "def"
             },
-            {
-              "type": "SYMBOL",
-              "name": "number"
-            },
             {
               "type": "SYMBOL",
               "name": "call_expr"
@@ -272,7 +272,7 @@
     },
     "identifier": {
       "type": "PATTERN",
-      "value": "[^ \\n\\r\\t<>:;&{}\"'\\[\\]]+"
+      "value": "[^ \\n\\r\\t<>:;&{}\"'\\[\\],]+"
     },
     "call": {
       "type": "CHOICE",
@@ -328,10 +328,40 @@
       ]
     },
     "number": {
+      "type": "TOKEN",
+      "content": {
+        "type": "PREC",
+        "value": 10,
+        "content": {
           "type": "PATTERN",
           "value": "\\d+(\\.\\d+)?"
+        }
+      }
     },
     "string": {
+      "type": "CHOICE",
+      "members": [
+        {
+          "type": "TOKEN",
+          "content": {
+            "type": "PREC",
+            "value": 10,
+            "content": {
+              "type": "SEQ",
+              "members": [
+                {
+                  "type": "STRING",
+                  "value": "^"
+                },
+                {
+                  "type": "PATTERN",
+                  "value": "[^ \\n\\r\\t<>:;&{}\"'\\[\\],]+"
+                }
+              ]
+            }
+          }
+        },
+        {
           "type": "SEQ",
           "members": [
             {
@@ -359,6 +389,8 @@
               "value": "\""
             }
           ]
+        }
+      ]
     },
     "call_expr": {
       "type": "SEQ",
@@ -410,8 +442,17 @@
         {
           "type": "REPEAT",
           "content": {
+            "type": "SEQ",
+            "members": [
+              {
                 "type": "SYMBOL",
                 "name": "identifier"
+              },
+              {
+                "type": "SYMBOL",
+                "name": "_spacing"
+              }
+            ]
           }
         },
         {
@@ -477,27 +518,7 @@
         }
       ]
     },
-    "variable": {
+    "set_variable": {
-      "type": "CHOICE",
-      "members": [
-        {
-          "type": "SEQ",
-          "members": [
-            {
-              "type": "STRING",
-              "value": "def"
-            },
-            {
-              "type": "SYMBOL",
-              "name": "_spacing"
-            },
-            {
-              "type": "SYMBOL",
-              "name": "identifier"
-            }
-          ]
-        },
-        {
       "type": "SEQ",
       "members": [
         {
@@ -509,8 +530,6 @@
           "name": "identifier"
         }
       ]
-        }
-      ]
     },
     "type_definition": {
       "type": "SEQ",
@@ -845,6 +864,34 @@
         {
           "type": "SYMBOL",
           "name": "identifier"
+        },
+        {
+          "type": "CHOICE",
+          "members": [
+            {
+              "type": "REPEAT",
+              "content": {
+                "type": "SEQ",
+                "members": [
+                  {
+                    "type": "STRING",
+                    "value": ","
+                  },
+                  {
+                    "type": "SYMBOL",
+                    "name": "_spacing"
+                  },
+                  {
+                    "type": "SYMBOL",
+                    "name": "identifier"
+                  }
+                ]
+              }
+            },
+            {
+              "type": "BLANK"
+            }
+          ]
         }
       ]
     }

src/node-types.json

@@ -314,7 +314,7 @@
     "named": true,
     "fields": {},
     "children": {
-      "multiple": false,
+      "multiple": true,
       "required": true,
       "types": [
         {
@@ -658,6 +658,10 @@
     "type": "&",
     "named": false
   },
+  {
+    "type": ",",
+    "named": false
+  },
   {
     "type": ":",
     "named": false