Add parsing of if statements

1 files changed, 49 insertions, 2 deletions

diff --git a/src/parser.c b/src/parser.c
index 58ad722..b76e32f 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -291,7 +291,46 @@ parse_fun(Parser *parser) {
         return NULL;
     }
     node->fun.body = body;
-    consume_rparen(parser);
+    if (!consume_rparen(parser)) {
+        return NULL;
+    }
+
+    return node;
+}
+
+Node *
+parse_if(Parser *parser) {
+    next_token(parser); // Skip keyword.
+
+    Node *node = alloc_node(NODE_IF);
+    node->ifexpr.cond = NULL;
+    node->ifexpr.expr_true = NULL;
+    node->ifexpr.expr_false = NULL;
+
+    Node *cond = parse_next(parser);
+    if (cond == NULL) {
+        return NULL;
+    }
+    Node *expr_true = parse_next(parser);
+    if (expr_true == NULL) {
+        return NULL;
+    }
+    node->ifexpr.cond = cond;
+    node->ifexpr.expr_true = expr_true;
+    Token tok = peek_token(parser);
+
+    // Optional else statement.
+    if (tok.type != TOKEN_RPAREN) {
+        Node *expr_false = parse_next(parser);
+        if (expr_false == NULL) {
+            return NULL;
+        }
+        node->ifexpr.expr_false = expr_false;
+    }
+
+    if (!consume_rparen(parser)) {
+        return NULL;
+    }
 
     return node;
 }
@@ -299,6 +338,7 @@ parse_fun(Parser *parser) {
 Node *
 parse_paren(Parser *parser) {
     next_token(parser); // Skip paren.
+
     Token tok = peek_token(parser);
 
     switch (tok.type) {
@@ -310,11 +350,17 @@ parse_paren(Parser *parser) {
         case TOKEN_MOD:
         case TOKEN_NOT:
         case TOKEN_AND:
-        case TOKEN_OR: { return parse_builtin(parser); } break;
+        case TOKEN_OR:
+        case TOKEN_EQ:
+        case TOKEN_LT:
+        case TOKEN_GT:
+        case TOKEN_LE:
+        case TOKEN_GE: { return parse_builtin(parser); } break;
         // Special functions.
         case TOKEN_DEF: { return parse_def(parser); } break;
         case TOKEN_SET: { return parse_set(parser); } break;
         case TOKEN_FUN: { return parse_fun(parser); } break;
+        case TOKEN_IF: { return parse_if(parser); } break;
         default: break;
     }
 
@@ -326,6 +372,7 @@ parse_paren(Parser *parser) {
 Node *
 parse_block(Parser *parser) {
     next_token(parser); // Skip paren.
+
     Node *node = alloc_node(NODE_BLOCK);
     array_init(node->block.expr, 0);
     while (true) {