fix: Fixed parser not parsing power expression (x^y)

alinalihassan · May 1, 2023 · ea0bc42 · ea0bc42
1 parent 2652777
commit ea0bc42
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/liblesma/Frontend/Parser.cpp b/src/liblesma/Frontend/Parser.cpp
@@ -206,8 +206,18 @@ Expression *Parser::ParseCast() {
 }
 
 Expression *Parser::ParseMult() {
-    auto left = ParseCast();
+    auto left = ParsePower();
     while (AdvanceIfMatchAny<TokenType::STAR, TokenType::SLASH, TokenType::MOD>()) {
+        auto op = Previous()->type;
+        auto right = ParsePower();
+        left = new BinaryOp({left->getStart(), right->getEnd()}, left, op, right);
+    }
+    return left;
+}
+
+Expression *Parser::ParsePower() {
+    auto left = ParseCast();
+    while (AdvanceIfMatchAny<TokenType::POWER>()) {
         auto op = Previous()->type;
         auto right = ParseCast();
         left = new BinaryOp({left->getStart(), right->getEnd()}, left, op, right);

diff --git a/src/liblesma/Frontend/Parser.h b/src/liblesma/Frontend/Parser.h
@@ -96,6 +96,7 @@ namespace lesma {
         Expression *ParseCompare();
         Expression *ParseAdd();
         Expression *ParseMult();
+        Expression *ParsePower();
         Expression *ParseCast();
         Expression *ParseUnary();
         Expression *ParseTerm();