Fix some parser bugs

crates/parser/src/grammar.rs

@@ -34,20 +34,14 @@ pub mod entry {
             p.skip();
         }
 
-        pragma::pragma(p);
         while !p.eof() {
             match p.current() {
-                TemplateKw => {
-                    template::template(p);
-                }
-                IncludeKw => {
-                    include::include(p);
-                }
+                Pragma => pragma::pragma(p),
+                TemplateKw => template::template(p),
+                IncludeKw => include::include(p),
                 ComponentKw => main_component::main_component(p),
                 FunctionKw => function::function_parse(p),
-                _ => {
-                    p.advance_with_error("invalid token");
-                }
+                _ => p.advance_with_error("invalid token"),
             }
         }
         p.close(m, CircomProgram);
@@ -63,20 +57,10 @@ pub mod entry {
     impl Scope {
         pub fn parse(self, p: &mut Parser) {
             match self {
-                Self::Block => {
-                    let m = p.open();
-                    block::block(p);
-                    p.close(m, ROOT);
-                }
+                Self::Block => block::block(p),
                 Self::CircomProgram => circom_program(p),
-                Self::Pragma => {
-                    let m = p.open();
-                    pragma::pragma(p);
-                    p.close(m, ROOT);
-                }
-                Self::Template => {
-                    template::template(p);
-                }
+                Self::Pragma => pragma::pragma(p),
+                Self::Template => template::template(p),
             }
         }
     }

crates/parser/src/grammar/block.rs

@@ -2,6 +2,7 @@ use super::*;
 
 pub fn block(p: &mut Parser) {
     p.inc_rcurly();
+
     if !p.at(LCurly) {
         p.advance_with_error("Miss {");
     } else {

crates/parser/src/grammar/declaration.rs

@@ -3,6 +3,9 @@ use super::{
     *,
 };
 
+// "signal" --> None
+// "signal input" --> Some(true)
+// "signal output" --> Some(false)
 fn signal_header(p: &mut Parser) -> Option<bool> {
     let mut res = None;
     let m = p.open();
@@ -35,7 +38,7 @@ pub(super) fn var_declaration(p: &mut Parser) {
 
     if p.at(LParen) {
         tuple(p);
-        if p.at_any(&[Assign, RAssignSignal, RAssignConstraintSignal]) {
+        if p.at(Assign) {
             tuple_init(p);
         }
     } else {

crates/parser/src/grammar/expression.rs

@@ -1,6 +1,7 @@
 use crate::parser::Marker;
 
 use super::*;
+
 pub(super) fn expression(p: &mut Parser) {
     let m = p.open();
     circom_expression(p);

crates/parser/src/grammar/function.rs

@@ -1,25 +1,22 @@
 use crate::grammar::*;
 
+// fucntion name()
 pub fn function_parse(p: &mut Parser) {
     let m = p.open();
+
     p.expect(FunctionKw);
-    let fn_name_marker = p.open();
 
+    let fn_name_marker = p.open();
     p.expect(Identifier);
     p.close(fn_name_marker, FunctionName);
+
     p.expect(LParen);
     let arg_marker = p.open();
-    while !p.at(RParen) && !p.eof() {
-        p.expect(Identifier);
-        if p.at(Comma) {
-            p.expect(Comma);
-        }
-    }
-
+    list_identity::parse(p);
     p.close(arg_marker, ParameterList);
-
     p.expect(RParen);
 
     block::block(p);
+
     p.close(m, FunctionDef);
 }

crates/parser/src/grammar/list_identity.rs

@@ -1,5 +1,6 @@
 use super::*;
 
+// a, b, c, d
 pub fn parse(p: &mut Parser) {
     while p.at(Identifier) && !p.eof() {
         p.expect(Identifier);

crates/parser/src/grammar/main_component.rs

@@ -1,13 +1,22 @@
 use super::*;
 
+/*
+component main {public [signal_list]} = tempid(v1,...,vn);
+
+{public [signal_list]} is optional
+*/
 pub fn main_component(p: &mut Parser) {
     p.expect(ComponentKw);
     p.expect(MainKw);
-    p.expect(LCurly);
-    p.expect(PublicKw);
-    p.expect(LBracket);
-    list_identity::parse(p);
-    p.expect(RBracket);
+
+    if p.at(LCurly) {
+        p.expect(LCurly);
+        p.expect(PublicKw);
+        p.expect(LBracket);
+        list_identity::parse(p);
+        p.expect(RBracket);
+    }
+
     p.expect(Assign);
     expression::expression(p);
 }

crates/parser/src/grammar/template.rs

@@ -1,28 +1,26 @@
 use crate::grammar::*;
 /**
  * template Identifier() {content}
- *
+ * template Identifier( param_1, ... , param_n ) { content }
  */
 pub fn template(p: &mut Parser) {
     // assert!(p.at(TemplateKw));
     let m = p.open();
+
     p.expect(TemplateKw);
+
     let name_marker = p.open();
     p.expect(Identifier);
     p.close(name_marker, TemplateName);
 
     p.expect(LParen);
     let arg_marker = p.open();
-    while !p.at(RParen) && !p.eof() {
-        p.expect(Identifier);
-        if p.at(Comma) {
-            p.expect(Comma);
-        }
-    }
-
+    list_identity::parse(p);
     p.close(arg_marker, ParameterList);
     p.expect(RParen);
+
     block::block(p);
+
     p.close(m, TemplateDef);
 }