Merge pull request #46 from nickel-lang/fix/enum-variant-types

Add support for enum variant in types
nickel-lang · Jun 4, 2024 · 9b15abb · 9b15abb
2 parents da030eb + 170db27
commit 9b15abb
Show file tree

Hide file tree

Showing 7 changed files with 20,814 additions and 19,242 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,3 +4,9 @@ Cargo.lock
 package-lock.json
 target/
 result*
+
+# tree-sitter now generates a bunch of object files in the repository
+*.a
+*.so
+*.o
+*.pc
diff --git a/grammar.js b/grammar.js
@@ -432,6 +432,11 @@ module.exports = grammar({
       field("arg", $.record_operand),
     )),
 
+    enum: $ => choice(
+      $.enum_tag,
+      $.enum_variant,
+    ),
+
     //grammar.lalrpop: 503
     //See NOTE[scanner].
     chunk_literal_single: $ => choice(
@@ -613,7 +618,7 @@ module.exports = grammar({
       "_",
       seq(
         "[|",
-        field("rows", commaSep($.enum_tag)),
+        field("rows", commaSep($.enum)),
         field("tail", optional(seq(";", $.ident))),
         "|]",
       ),

diff --git a/src/grammar.json b/src/grammar.json
@@ -1830,6 +1830,19 @@
         ]
       }
     },
+    "enum": {
+      "type": "CHOICE",
+      "members": [
+        {
+          "type": "SYMBOL",
+          "name": "enum_tag"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "enum_variant"
+        }
+      ]
+    },
     "chunk_literal_single": {
       "type": "CHOICE",
       "members": [
@@ -2558,7 +2571,7 @@
                     "members": [
                       {
                         "type": "SYMBOL",
-                        "name": "enum_tag"
+                        "name": "enum"
                       },
                       {
                         "type": "REPEAT",
@@ -2571,7 +2584,7 @@
                             },
                             {
                               "type": "SYMBOL",
-                              "name": "enum_tag"
+                              "name": "enum"
                             }
                           ]
                         }

diff --git a/src/node-types.json b/src/node-types.json
@@ -391,6 +391,25 @@
       }
     }
   },
+  {
+    "type": "enum",
+    "named": true,
+    "fields": {},
+    "children": {
+      "multiple": false,
+      "required": true,
+      "types": [
+        {
+          "type": "enum_tag",
+          "named": true
+        },
+        {
+          "type": "enum_variant",
+          "named": true
+        }
+      ]
+    }
+  },
   {
     "type": "enum_pattern",
     "named": true,
@@ -1596,7 +1615,7 @@
             "named": false
           },
           {
-            "type": "enum_tag",
+            "type": "enum",
             "named": true
           }
         ]