Add builtin object type

1 files changed, 31 insertions, 22 deletions

diff --git a/src/ir.h b/src/ir.h
index b42e02f..a42cd48 100644
--- a/src/ir.h
+++ b/src/ir.h
@@ -115,34 +115,43 @@ proc_alloc(ProgramIr *program, StringView name) {
 void compile_object(ProgramIr *program, Procedure *proc, Object *obj);
 
 void
-compile_arithmetic_list(ProgramIr *program, Procedure *proc, Op op, Object *obj) {
-    size_t op_line = obj->head->line;
-    size_t op_col = obj->head->col;
-    obj = obj->tail;
-
-    compile_object(program, proc, obj->head);
-    obj = obj->tail;
-    while (obj != NULL) {
-        compile_object(program, proc, obj->head);
-        obj = obj->tail;
-        Instruction inst = (Instruction){op, NULL, op_line, op_col};
+compile_arithmetic(ProgramIr *program, Procedure *proc, Op op,
+        size_t line, size_t col, Object *args) {
+    compile_object(program, proc, args->head);
+    args = args->tail;
+    while (args != NULL) {
+        compile_object(program, proc, args->head);
+        args = args->tail;
+        Instruction inst = (Instruction){op, NULL, line, col};
         array_push(proc->instructions, inst);
     }
 }
 
 void
 compile_proc_call(ProgramIr *program, Procedure *proc, Object *obj) {
-    // TODO: Handle this on the parser?
-    if (sv_equal(&obj->head->text, &STRING("+"))) {
-        compile_arithmetic_list(program, proc, OP_ADD, obj);
-    } else if (sv_equal(&obj->head->text, &STRING("-"))) {
-        compile_arithmetic_list(program, proc, OP_SUB, obj);
-    } else if (sv_equal(&obj->head->text, &STRING("*"))) {
-        compile_arithmetic_list(program, proc, OP_MUL, obj);
-    } else if (sv_equal(&obj->head->text, &STRING("/"))) {
-        compile_arithmetic_list(program, proc, OP_DIV, obj);
-    } else if (sv_equal(&obj->head->text, &STRING("%"))) {
-        compile_arithmetic_list(program, proc, OP_MOD, obj);
+    size_t line = obj->line;
+    size_t col = obj->col;
+    if (obj->head->type == OBJ_TYPE_BUILTIN) {
+        switch (obj->head->builtin) {
+            case BUILTIN_ADD: {
+                compile_arithmetic(program, proc, OP_ADD, line, col, obj->tail);
+            } break;
+            case BUILTIN_SUB: {
+                compile_arithmetic(program, proc, OP_SUB, line, col, obj->tail);
+            } break;
+            case BUILTIN_MUL: {
+                compile_arithmetic(program, proc, OP_MUL, line, col, obj->tail);
+            } break;
+            case BUILTIN_DIV: {
+                compile_arithmetic(program, proc, OP_DIV, line, col, obj->tail);
+            } break;
+            case BUILTIN_MOD: {
+                compile_arithmetic(program, proc, OP_MOD, line, col, obj->tail);
+            } break;
+            default: {
+                assert(false && "builtin not implemented");
+            } break;
+        }
     } else {
         assert(false && "compile_proc_call: not implemented");
     }