added parsing of if statements

src/parser/parser.cpp

@@ -3,9 +3,11 @@
 #include "ast/errors/error.hpp"
 #include "ast/expressions/boolean.hpp"
 #include "ast/expressions/identifier.hpp"
+#include "ast/expressions/if_then_else.hpp"
 #include "ast/expressions/infix.hpp"
 #include "ast/expressions/integer.hpp"
 #include "ast/expressions/prefix.hpp"
+#include "ast/statements/block.hpp"
 #include "token/token.hpp"
 #include "token/type.hpp"
 #include "utils/tracer.hpp"
@@ -51,6 +53,11 @@ namespace parser {
             std::bind(&parser::parse_grouped_expr, this)
         );
 
+        register_prefix(
+            token::type::IF,
+            std::bind(&parser::parse_if_then_else, this)
+        );
+
         using namespace std::placeholders;
         register_infix(
             {token::type::PLUS,
@@ -256,6 +263,59 @@ namespace parser {
         return ret;
     };
 
+    ast::expression* parser::parse_if_then_else() {
+        // TRACE_FUNCTION;
+        ast::if_then_else* ret = new ast::if_then_else(current);
+        if (!expect_next(token::type::LPAREN)) {
+            delete ret;
+            return nullptr;
+        }
+
+        next_token();
+        ret->condition = parse_expression();
+
+        if (!expect_next(token::type::RPAREN)) {
+            delete ret;
+            return nullptr;
+        }
+
+        if (!expect_next(token::type::LBRACE)) {
+            delete ret;
+            return nullptr;
+        }
+
+        ret->consequence = parse_block();
+        if (next.type != token::type::ELSE)
+            return ret;
+        next_token();
+
+        if (!expect_next(token::type::LBRACE)) {
+            delete ret;
+            return nullptr;
+        }
+
+        ret->alternative = parse_block();
+        return ret;
+    };
+
+    ast::block_stmt* parser::parse_block() {
+        // TRACE_FUNCTION;
+        ast::block_stmt* ret = new ast::block_stmt(current);
+        //     ret->statements.push_back(parse_statement());
+        next_token();
+        int i = 0;
+        while (current.type != token::type::RBRACE && i++ < 10) {
+            // for (next_token(); next.type != token::type::RBRACE;
+            // next_token()) {
+            ast::statement* stmt = parse_statement();
+            if (stmt != nullptr)
+                ret->statements.push_back(stmt);
+            next_token();
+        }
+
+        return ret;
+    }
+
     ast::expression* parser::parse_infix_expr(ast::expression* left) {
         // TRACE_FUNCTION;
         ast::infix_expr* ret =

src/parser/parser.hpp

@@ -3,6 +3,7 @@
 #include "ast/ast.hpp"
 #include "ast/errors/error.hpp"
 #include "ast/program.hpp"
+#include "ast/statements/block.hpp"
 #include "ast/statements/expression.hpp"
 #include "ast/statements/let.hpp"
 #include "ast/statements/return.hpp"
@@ -54,6 +55,8 @@ namespace parser {
         ast::expression* parse_boolean();
         ast::expression* parse_prefix_expr();
         ast::expression* parse_grouped_expr();
+        ast::expression* parse_if_then_else();
+        ast::block_stmt* parse_block();
 
         ast::expression* parse_infix_expr(ast::expression*);
     };