Add set typechecking

1 files changed, 23 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
index ffe9e10..bfaca4e 100644
--- a/src/main.c
+++ b/src/main.c
@@ -253,6 +253,12 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
             }
             return node->type;
         } break;
+        case NODE_SET: {
+            type_inference(a, node->var_name, scope);
+            type_inference(a, node->var_val, scope);
+            node->type = cstr("nil");
+            return node->type;
+        } break;
         case NODE_IF: {
             Type cond_type = type_inference(a, node->cond_if, scope);
             if (!str_eq(cond_type, cstr("bool"))) {
@@ -281,6 +287,7 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
                     return cstr("");
                 }
             }
+            return node->type;
         } break;
         case NODE_WHILE: {
             Type cond_type = type_inference(a, node->while_cond, scope);
@@ -402,6 +409,22 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
                 node->type = type->val;
                 return node->type;
             }
+            return node->type;
+        } break;
+        case NODE_FUNCALL: {
+            // TODO: ...
+            // Str symbol = node->value.str;
+            // if (find_symbol(scope, symbol) == NULL) {
+            //     eprintln(
+            //         "%s:%d:%d: error: symbol '%s' doesn't exists in current "
+            //         " scope ",
+            //         a->file_name, node->line, node->col, symbol);
+            //     return;
+            // }
+            // for (sz i = 0; i < array_size(node->elements); i++) {
+            //     Node *expr = node->elements[i];
+            //     analyzer_symbols(a, expr, scope);
+            // }
         } break;
         case NODE_BLOCK: {
             scope = typescope_alloc(a, scope);