Fixed IF+Colon

rhpo · Feb 4, 2024 · 8a09eb0 · 8a09eb0
1 parent 0e2a375
commit 8a09eb0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/lib/parser.ts b/src/lib/parser.ts
@@ -1256,6 +1256,8 @@ export default class Parser {
 
     let object = { kind: "IfStatement" } as IFStatement;
 
+    this.config.skipColonAction = true;
+
     let condition = this.parseExpression();
     object.test = condition;
 
@@ -1345,11 +1347,13 @@ export default class Parser {
     this.expect(TokenType.Semicolon);
     this.eat();
 
+    this.config.skipColonAction = true;
+
     let increaseExpr = this.parseExpression();
 
     object.increaser = increaseExpr;
 
-    if (!this.at() || this.at().type !== TokenType.OpenBrace) {
+    if (!this.at() || (this.at().type !== TokenType.Colon && this.at().type !== TokenType.OpenBrace)) {
       throw Err(
         "SyntaxError",
         `Expected '{ or :' after 'for' statement${this.where()}`
@@ -1399,6 +1403,8 @@ export default class Parser {
     this.expect(TokenType.WHILE);
     this.eat();
 
+    this.config.skipColonAction = true;
+
     let object = { kind: "WhileStatement" } as WhileStatement;
 
     let condition = this.parseExpression();