allow expr as index

rapiz1 · Nov 3, 2021 · 744e1c3 · 744e1c3
1 parent d38686b
commit 744e1c3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/parser.cc b/parser.cc
@@ -439,7 +439,7 @@ Expr* Parser::index() {
   Expr* e = primary();
   if (match(1, LEFT_SQUARE)) {
     advance();
-    auto i = primary();
+    auto i = expression();
     e = new Index(e, i);
     consume(RIGHT_SQUARE, "Expect `]` after indexing");
   }

diff --git a/parser.h b/parser.h
@@ -46,7 +46,7 @@ class Parser {
   Expr* unary();       // (! | - | -- | ++)* POSTFIX
   Expr* postfix();     // CALL (++ | --)*
   Expr* call();        // INDEX ('(' ARGS? ')')*
-  Expr* index();       // PRIM ('[' NUMBER ']')?
+  Expr* index();       // PRIM ('[' EXPR ']')?
   Expr* primary();
 
  public: