feat: support enum

victorhqc · Sep 19, 2019 · dbfd7c4 · dbfd7c4
1 parent eaf6eb9
commit dbfd7c4
Show file tree

Hide file tree

Showing 5 changed files with 5,435 additions and 4,881 deletions.
diff --git a/corpus/enum.txt b/corpus/enum.txt
@@ -0,0 +1,20 @@
+=====================
+Enum declaration
+=====================
+
+enum Role {
+  USER
+  ADMIN
+}
+
+---
+
+(program
+  (enum_declaration
+    (identifier)
+    (enum_block
+      (enumeral)
+      (enumeral)
+    )
+  )
+)
diff --git a/grammar.js b/grammar.js
@@ -79,11 +79,18 @@ module.exports = grammar({
       ),
     )),
 
+    enum_declaration: $ => seq(
+      'enum',
+      $.identifier,
+      $.enum_block,
+    ),
+
     _declaration: $ => choice(
       $.datasource_declaration,
       $.model_declaration,
       $.generator_declaration,
       $.type_declaration,
+      $.enum_declaration
     ),
 
     comment: $ => token(
@@ -96,6 +103,12 @@ module.exports = grammar({
       '}'
     )),
 
+    enum_block: $ => prec.right(seq(
+      '{',
+      repeat($.enumeral),
+      '}'
+    )),
+
     _statement: $ => choice(
       $.column_declaration,
       $.block_attribute_declaration,
@@ -252,6 +265,8 @@ module.exports = grammar({
       seq('"', /([^"\n]|\\(.|\n))*/, '"')
     )),
 
+    enumeral: $ => /[a-zA-Z-_][a-zA-Z0-9-_]*/,
+
     number: $ => /\d+/,
 
     array: $ => seq(

diff --git a/src/grammar.json b/src/grammar.json
@@ -79,6 +79,23 @@
         ]
       }
     },
+    "enum_declaration": {
+      "type": "SEQ",
+      "members": [
+        {
+          "type": "STRING",
+          "value": "enum"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "identifier"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "enum_block"
+        }
+      ]
+    },
     "_declaration": {
       "type": "CHOICE",
       "members": [
@@ -97,6 +114,10 @@
         {
           "type": "SYMBOL",
           "name": "type_declaration"
+        },
+        {
+          "type": "SYMBOL",
+          "name": "enum_declaration"
         }
       ]
     },
@@ -140,6 +161,30 @@
         ]
       }
     },
+    "enum_block": {
+      "type": "PREC_RIGHT",
+      "value": 0,
+      "content": {
+        "type": "SEQ",
+        "members": [
+          {
+            "type": "STRING",
+            "value": "{"
+          },
+          {
+            "type": "REPEAT",
+            "content": {
+              "type": "SYMBOL",
+              "name": "enumeral"
+            }
+          },
+          {
+            "type": "STRING",
+            "value": "}"
+          }
+        ]
+      }
+    },
     "_statement": {
       "type": "CHOICE",
       "members": [
@@ -1085,6 +1130,10 @@
         ]
       }
     },
+    "enumeral": {
+      "type": "PATTERN",
+      "value": "[a-zA-Z-_][a-zA-Z0-9-_]*"
+    },
     "number": {
       "type": "PATTERN",
       "value": "\\d+"

diff --git a/src/node-types.json b/src/node-types.json
@@ -540,6 +540,45 @@
       ]
     }
   },
+  {
+    "type": "enum_block",
+    "named": true,
+    "fields": {},
+    "children": {
+      "multiple": true,
+      "required": false,
+      "types": [
+        {
+          "type": "enumeral",
+          "named": true
+        }
+      ]
+    }
+  },
+  {
+    "type": "enum_declaration",
+    "named": true,
+    "fields": {},
+    "children": {
+      "multiple": true,
+      "required": false,
+      "types": [
+        {
+          "type": "enum_block",
+          "named": true
+        },
+        {
+          "type": "identifier",
+          "named": true
+        }
+      ]
+    }
+  },
+  {
+    "type": "enumeral",
+    "named": true,
+    "fields": {}
+  },
   {
     "type": "formal_parameters",
     "named": true,
@@ -586,6 +625,11 @@
       ]
     }
   },
+  {
+    "type": "identifier",
+    "named": true,
+    "fields": {}
+  },
   {
     "type": "member_expression",
     "named": true,
@@ -712,6 +756,10 @@
           "type": "datasource_declaration",
           "named": true
         },
+        {
+          "type": "enum_declaration",
+          "named": true
+        },
         {
           "type": "generator_declaration",
           "named": true
@@ -900,6 +948,10 @@
     "type": "type",
     "named": false
   },
+  {
+    "type": "enum",
+    "named": false
+  },
   {
     "type": "comment",
     "named": true
@@ -1032,10 +1084,6 @@
     "type": ")",
     "named": false
   },
-  {
-    "type": "identifier",
-    "named": true
-  },
   {
     "type": "string",
     "named": true