Add parsing for (set ...) statements

1 files changed, 33 insertions, 0 deletions

diff --git a/src/parser.c b/src/parser.c
index cc94fd2..b36ddc9 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -178,6 +178,33 @@ parse_def(Parser *parser) {
 }
 
 Node *
+parse_set(Parser *parser) {
+    Token op = next_token(parser);
+    Node *symbol = parse_next(parser);
+    if (symbol == NULL || symbol->type != NODE_SYMBOL) {
+        push_error(ERR_TYPE_PARSER, ERR_MALFORMED_EXPR, op.line, op.col);
+        return NULL;
+    }
+
+    Node *value = parse_next(parser);
+    if (value == NULL) {
+        push_error(ERR_TYPE_PARSER, ERR_MALFORMED_EXPR, op.line, op.col);
+        return NULL;
+    }
+
+    Token end = next_token(parser);
+    if (end.type != TOKEN_RPAREN) {
+        push_error(ERR_TYPE_PARSER, ERR_MALFORMED_EXPR, op.line, op.col);
+        return NULL;
+    }
+
+    Node *node = alloc_node(NODE_SET);
+    node->set.symbol = symbol;
+    node->set.value = value;
+    return node;
+}
+
+Node *
 parse_paren(Parser *parser) {
     next_token(parser); // Skip paren.
     Token tok = peek_token(parser);
@@ -194,6 +221,7 @@ parse_paren(Parser *parser) {
         case TOKEN_OR: { return parse_builtin(parser); } break;
         // Special functions.
         case TOKEN_DEF: { return parse_def(parser); } break;
+        case TOKEN_SET: { return parse_set(parser); } break;
         default: break;
     }
 
@@ -202,6 +230,11 @@ parse_paren(Parser *parser) {
     return NULL;
 }
 
+
+// TODO: Parse if: (def ...) shouldn't be allowed for now on if statements. In
+// the future, an if can create a new block and thus a fresh scope which would
+// make the decision of what variables are active trivial.
+
 Node *
 parse_next(Parser *parser) {
     Token tok = peek_token(parser);