Fix some syntax issues with match cases

1 files changed, 7 insertions, 5 deletions

diff --git a/src/main.c b/src/main.c
index 1bb5ff6..ccd4877 100644
--- a/src/main.c
+++ b/src/main.c
@@ -579,20 +579,22 @@ parse_keyword(Parser *parser) {
             if (!node) return;
             parse_expr(parser, PREC_LOW);
             node->match_expr = array_pop(parser->nodes);
+            parse_consume(parser, TOK_ASSIGN,
+                          cstr("malformed match statement"));
             parse_consume(parser, TOK_LCURLY,
                           cstr("expected block of match cases"));
             while (!parse_match(parser, TOK_RCURLY) && !parser->panic) {
-                parse_consume(parser, TOK_CASE,
-                              cstr("expected case statement"));
                 Node *tmp = node_alloc(parser, NODE_CASE, parser->previous);
                 if (!tmp) return;
                 // Are we on the default case.
-                if (!parse_match(parser, TOK_ASSIGN)) {
+                if (!parse_match(parser, TOK_ELSE)) {
+                    parse_consume(parser, TOK_CASE,
+                                cstr("expected case statement"));
                     parse_expr(parser, PREC_LOW);
                     tmp->case_value = array_pop(parser->nodes);
-                    parse_consume(parser, TOK_ASSIGN,
-                                  cstr("malformed case statement"));
                 }
+                parse_consume(parser, TOK_ASSIGN,
+                                cstr("malformed case statement"));
                 parse_expr(parser, PREC_LOW);
                 tmp->case_expr = array_pop(parser->nodes);
                 array_push(node->match_cases, tmp, parser->storage);