fix: support await statement

src/nomparser/expression.rs

@@ -102,7 +102,7 @@ fn mul_exp(input: Span) -> IResult<Span, Box<NodeEnum>> {
 #[test_parser("~a")]
 #[test_parser("await a")]
 #[test_parser_error("+a")]
-fn unary_exp(input: Span) -> IResult<Span, Box<NodeEnum>> {
+pub fn unary_exp(input: Span) -> IResult<Span, Box<NodeEnum>> {
     // todo: consider to aligh the implementation with UnaryExp EBNF
     delspace(alt((
         pointer_exp,

src/nomparser/statement.rs

@@ -81,6 +81,7 @@ pub fn statement(input: Span) -> IResult<Span, Box<NodeEnum>> {
         semi_stmt(pointer_exp, pointer_exp),
         empty_statement,
         comment,
+        semi_stmt(await_statement, await_statement),
         terminated(
             except(
                 "\n\r})",
@@ -92,6 +93,24 @@ pub fn statement(input: Span) -> IResult<Span, Box<NodeEnum>> {
     )))(input)
 }
 
+#[test_parser("await a")]
+fn await_statement(input: Span) -> IResult<Span, Box<NodeEnum>> {
+    map(
+        pair(tag_modifier(TokenType::AWAIT), general_exp),
+        |((op, r), e)| {
+            let range = r.start.to(e.range().end);
+            Box::new(
+                UnaryOpNode {
+                    op: (op, r),
+                    exp: e,
+                    range,
+                }
+                .into(),
+            )
+        },
+    )(input)
+}
+
 #[test_parser("let a = 1")]
 #[test_parser_error("leta = 1")]
 #[test_parser("let (a, b) = 1")]