nodes for blocks

1 files changed, 45 insertions, 30 deletions

diff --git a/crates/libsyntax2/src/grammar/items/structs.rs b/crates/libsyntax2/src/grammar/items/structs.rs
index cde9d0ae6..ca027d718 100644
--- a/crates/libsyntax2/src/grammar/items/structs.rs
+++ b/crates/libsyntax2/src/grammar/items/structs.rs
@@ -14,7 +14,7 @@ pub(super) fn struct_def(p: &mut Parser) {
                     p.bump();
                     return;
                 }
-                L_CURLY => named_fields(p),
+                L_CURLY => named_field_def_list(p),
                 _ => {
                     //TODO: special case `(` error message
                     p.error("expected `;` or `{`");
@@ -26,9 +26,9 @@ pub(super) fn struct_def(p: &mut Parser) {
             p.bump();
             return;
         }
-        L_CURLY => named_fields(p),
+        L_CURLY => named_field_def_list(p),
         L_PAREN => {
-            pos_fields(p);
+            pos_field_list(p);
             p.expect(SEMI);
         }
         _ => {
@@ -44,46 +44,58 @@ pub(super) fn enum_def(p: &mut Parser) {
     name(p);
     type_params::opt_type_param_list(p);
     type_params::opt_where_clause(p);
-    if p.expect(L_CURLY) {
-        while !p.at(EOF) && !p.at(R_CURLY) {
-            let var = p.start();
-            attributes::outer_attributes(p);
-            if p.at(IDENT) {
-                name(p);
-                match p.current() {
-                    L_CURLY => named_fields(p),
-                    L_PAREN => pos_fields(p),
-                    EQ => {
-                        p.bump();
-                        expressions::expr(p);
-                    }
-                    _ => (),
+    if p.at(L_CURLY) {
+        enum_variant_list(p);
+    } else {
+        p.error("expected `{`")
+    }
+}
+
+fn enum_variant_list(p: &mut Parser) {
+    assert!(p.at(L_CURLY));
+    let m = p.start();
+    p.bump();
+    while !p.at(EOF) && !p.at(R_CURLY) {
+        let var = p.start();
+        attributes::outer_attributes(p);
+        if p.at(IDENT) {
+            name(p);
+            match p.current() {
+                L_CURLY => named_field_def_list(p),
+                L_PAREN => pos_field_list(p),
+                EQ => {
+                    p.bump();
+                    expressions::expr(p);
                 }
-                var.complete(p, ENUM_VARIANT);
-            } else {
-                var.abandon(p);
-                p.err_and_bump("expected enum variant");
-            }
-            if !p.at(R_CURLY) {
-                p.expect(COMMA);
+                _ => (),
             }
+            var.complete(p, ENUM_VARIANT);
+        } else {
+            var.abandon(p);
+            p.err_and_bump("expected enum variant");
+        }
+        if !p.at(R_CURLY) {
+            p.expect(COMMA);
         }
-        p.expect(R_CURLY);
     }
+    p.expect(R_CURLY);
+    m.complete(p, ENUM_VARIANT_LIST);
 }
 
-fn named_fields(p: &mut Parser) {
+fn named_field_def_list(p: &mut Parser) {
     assert!(p.at(L_CURLY));
+    let m = p.start();
     p.bump();
     while !p.at(R_CURLY) && !p.at(EOF) {
-        named_field(p);
+        named_field_def(p);
         if !p.at(R_CURLY) {
             p.expect(COMMA);
         }
     }
     p.expect(R_CURLY);
+    m.complete(p, NAMED_FIELD_DEF_LIST);
 
-    fn named_field(p: &mut Parser) {
+    fn named_field_def(p: &mut Parser) {
         let m = p.start();
         // test field_attrs
         // struct S {
@@ -96,7 +108,7 @@ fn named_fields(p: &mut Parser) {
             name(p);
             p.expect(COLON);
             types::type_(p);
-            m.complete(p, NAMED_FIELD);
+            m.complete(p, NAMED_FIELD_DEF);
         } else {
             m.abandon(p);
             p.err_and_bump("expected field declaration");
@@ -104,7 +116,9 @@ fn named_fields(p: &mut Parser) {
     }
 }
 
-fn pos_fields(p: &mut Parser) {
+fn pos_field_list(p: &mut Parser) {
+    assert!(p.at(L_PAREN));
+    let m = p.start();
     if !p.expect(L_PAREN) {
         return;
     }
@@ -119,4 +133,5 @@ fn pos_fields(p: &mut Parser) {
         }
     }
     p.expect(R_PAREN);
+    m.complete(p, POS_FIELD_LIST);
 }