Fix control flow

bindings/dust

@@ -1 +1 @@
-Subproject commit 41483f6b84b83488ff02512bfad8719cd62481fc
+Subproject commit a42b43ed7853b73e8bb42b0ab23765acc1fc78c1

grammar.js

@@ -8,7 +8,7 @@ module.exports = grammar({
     
     item: $ => choice(
       $.comment,
-      $.statement,
+      field('source', $.statement),
     ),
 
     comment: $ => seq(token('#'), /.*/),
@@ -48,7 +48,7 @@ module.exports = grammar({
 
     identifier: $ => /[a-z|_|.]+[0-9]?/,
 
-    value: $ => prec.left(1, choice(
+    value: $ => choice(
       $.integer,
       $.float,
       $.string,
@@ -57,7 +57,7 @@ module.exports = grammar({
       $.function,
       $.table,
       $.map,
-    )),
+    ),
 
     float: $ => /[-]*[0-9]*[.]{1}[0-9]+/,
 
@@ -95,7 +95,7 @@ module.exports = grammar({
     map: $ => seq(
       'map',
       '{',
-      repeat(seq($.identifier, "=", $.value)),
+      repeat(field('key_value_pair', seq($.identifier, "=", $.value))),
       '}',
     ),
 
@@ -155,10 +155,14 @@ module.exports = grammar({
 
     control_flow: $ => prec.right(seq(
       'if',
-      $.expression,
+      field('if_expression', $.expression),
       'then',
-      $.statement,
-      optional(seq('else', $.statement))
+      field('then_statement', $.statement),
+      optional(
+        seq('else', 
+          field('else_statement', $.statement),
+        ),
+      ),
     )),
 
     function_call: $ => prec.right(seq(

src/grammar.json

@@ -17,9 +17,13 @@
           "name": "comment"
         },
         {
+          "type": "FIELD",
+          "name": "source",
+          "content": {
             "type": "SYMBOL",
             "name": "statement"
           }
+        }
       ]
     },
     "comment": {
@@ -180,9 +184,6 @@
       "value": "[a-z|_|.]+[0-9]?"
     },
     "value": {
-      "type": "PREC_LEFT",
-      "value": 1,
-      "content": {
       "type": "CHOICE",
       "members": [
         {
@@ -218,7 +219,6 @@
           "name": "map"
         }
       ]
-      }
     },
     "float": {
       "type": "PATTERN",
@@ -429,6 +429,9 @@
         },
         {
           "type": "REPEAT",
+          "content": {
+            "type": "FIELD",
+            "name": "key_value_pair",
             "content": {
               "type": "SEQ",
               "members": [
@@ -446,6 +449,7 @@
                 }
               ]
             }
+          }
         },
         {
           "type": "STRING",
@@ -685,16 +689,24 @@
             "value": "if"
           },
           {
+            "type": "FIELD",
+            "name": "if_expression",
+            "content": {
               "type": "SYMBOL",
               "name": "expression"
+            }
           },
           {
             "type": "STRING",
             "value": "then"
           },
           {
+            "type": "FIELD",
+            "name": "then_statement",
+            "content": {
               "type": "SYMBOL",
               "name": "statement"
+            }
           },
           {
             "type": "CHOICE",
@@ -707,9 +719,13 @@
                     "value": "else"
                   },
                   {
+                    "type": "FIELD",
+                    "name": "else_statement",
+                    "content": {
                       "type": "SYMBOL",
                       "name": "statement"
                     }
+                  }
                 ]
               },
               {

src/node-types.json

@@ -46,20 +46,37 @@
   {
     "type": "control_flow",
     "named": true,
-    "fields": {},
-    "children": {
-      "multiple": true,
-      "required": true,
+    "fields": {
+      "else_statement": {
+        "multiple": false,
+        "required": false,
         "types": [
-        {
-          "type": "expression",
-          "named": true
-        },
           {
             "type": "statement",
             "named": true
           }
         ]
+      },
+      "if_expression": {
+        "multiple": false,
+        "required": true,
+        "types": [
+          {
+            "type": "expression",
+            "named": true
+          }
+        ]
+      },
+      "then_statement": {
+        "multiple": false,
+        "required": true,
+        "types": [
+          {
+            "type": "statement",
+            "named": true
+          }
+        ]
+      }
     }
   },
   {
@@ -181,18 +198,25 @@
   {
     "type": "item",
     "named": true,
-    "fields": {},
+    "fields": {
+      "source": {
+        "multiple": false,
+        "required": false,
+        "types": [
+          {
+            "type": "statement",
+            "named": true
+          }
+        ]
+      }
+    },
     "children": {
       "multiple": false,
-      "required": true,
+      "required": false,
       "types": [
         {
           "type": "comment",
           "named": true
-        },
-        {
-          "type": "statement",
-          "named": true
         }
       ]
     }
@@ -259,11 +283,15 @@
   {
     "type": "map",
     "named": true,
-    "fields": {},
-    "children": {
+    "fields": {
+      "key_value_pair": {
         "multiple": true,
         "required": false,
         "types": [
+          {
+            "type": "=",
+            "named": false
+          },
           {
             "type": "identifier",
             "named": true
@@ -274,6 +302,7 @@
           }
         ]
       }
+    }
   },
   {
     "type": "match",