Add parsing for builtin arithmetic ops

1 files changed, 57 insertions, 3 deletions

diff --git a/src/parser.c b/src/parser.c
index a82a97f..2a5e3e3 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -108,7 +108,44 @@ parse_string(Parser *parser) {
 }
 
 Node *
+parse_builtin(Parser *parser) {
+    Token op = next_token(parser);
+    Node *node = alloc_node(NODE_BUILTIN);
+    node->builtin.type = op.type;
+    array_init(node->builtin.args, 0);
+    while (has_next(parser)) {
+        Token next = peek_token(parser);
+        if (next.type == TOKEN_RPAREN) {
+            next_token(parser);
+            return node;
+        }
+        Node *arg = parse_next(parser);
+        if (arg == NULL) {
+            break;
+        }
+        array_push(node->builtin.args, arg);
+    }
+    push_error(ERR_TYPE_PARSER, ERR_UNMATCHED_PAREN, op.line, op.col);
+    return NULL;
+}
+
+Node *
 parse_paren(Parser *parser) {
+    next_token(parser); // Skip paren.
+    Token tok = peek_token(parser);
+    // TODO: is keyword?
+    switch (tok.type) {
+        case TOKEN_ADD:
+        case TOKEN_SUB:
+        case TOKEN_MUL:
+        case TOKEN_DIV:
+        case TOKEN_MOD: {
+            return parse_builtin(parser);
+        } break;
+        default: break;
+    }
+    // print_token(tok);
+    // TODO: is in symbol table and its value is a function?
     return NULL; // TODO: Not implemented
 }
 
@@ -143,14 +180,28 @@ print_node(Node *node) {
                 printf("-");
             }
             if (node->number.fractional != 0) {
-                printf("%zu.%zu\n", node->number.integral, node->number.fractional);
+                printf("%zu.%zu", node->number.integral, node->number.fractional);
             } else {
-                printf("%zu\n", node->number.integral);
+                printf("%zu", node->number.integral);
             }
         } break;
         case NODE_STRING: {
             sv_write(&node->string);
-            printf("\n");
+        } break;
+        case NODE_BUILTIN: {
+            printf("(");
+            printf("{#%s}", token_str[node->builtin.type]);
+            size_t n_args = array_size(node->builtin.args);
+            if (n_args != 0) {
+                printf(" ");
+            }
+            for (size_t i = 0; i < n_args; ++i) {
+                print_node(node->builtin.args[i]);
+                if (i < n_args - 1) {
+                    printf(" ");
+                }
+            }
+            printf(")");
         } break;
         default: { printf("{#unk}"); } break;
     }
@@ -164,9 +215,11 @@ parse(Token *tokens) {
     };
 
     // DEBUG: TOKENS
+    printf("-- tokens --\n");
     for (size_t i = 0; i < array_size(tokens); i++) {
         print_token(tokens[i]);
     }
+    printf("------------\n");
 
     while (has_next(&parser)) {
         Node *node = parse_next(&parser);
@@ -174,5 +227,6 @@ parse(Token *tokens) {
             return;
         }
         print_node(node);
+        printf("\n");
     }
 }