Add compilation for set nodes

2 files changed, 28 insertions, 0 deletions

diff --git a/src/compiler.c b/src/compiler.c
index f46d00d..0f9607e 100644
--- a/src/compiler.c
+++ b/src/compiler.c
@@ -652,6 +652,28 @@ compile_expr(Chunk *chunk, Node *node) {
 
             return (CompResult){.type = COMP_NIL};
         } break;
+        case NODE_SET: {
+            Str name = node->unique_name;
+            StrVarMap *map = varmap_lookup(&chunk->varmap, name);
+            if (!map) {
+                println("you wot mate?: %s", name);
+            }
+            CompResult res = compile_expr(chunk, node->var_val);
+            switch (res.type) {
+                case COMP_CONST: {
+                    EMIT_OP(OP_STGVARI, map->val.idx, res.idx, 0, node->var_val,
+                            chunk);
+                } break;
+                case COMP_REG: {
+                    EMIT_OP(OP_STGVAR, map->val.idx, res.idx, 0, node->var_val,
+                            chunk);
+                } break;
+                default: {
+                    return (CompResult){.type = COMP_ERR};
+                } break;
+            }
+            return (CompResult){.type = COMP_NIL};
+        } break;
         case NODE_SYMBOL: {
             Str name = node->unique_name;
             StrVarMap *map = varmap_lookup(&chunk->varmap, name);
diff --git a/src/semantic.c b/src/semantic.c
index d782cac..9b119df 100644
--- a/src/semantic.c
+++ b/src/semantic.c
@@ -550,6 +550,12 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                 a->err = true;
                 return cstr("");
             }
+            Str symbol = node->var_name->value.str;
+            FindSymbolResult sym = find_symbol(scope, symbol);
+            node->unique_name = str_concat(cstr("."), symbol, a->storage);
+            node->unique_name =
+                str_concat(node->unique_name,
+                           str_from_int(sym.scope->id, a->storage), a->storage);
             node->type = cstr("nil");
             return node->type;
         } break;