3 files changed, 42 insertions, 12 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;
diff --git a/tests/compilation.bad b/tests/compilation.bad
index 4b25a76..afb2be0 100644
--- a/tests/compilation.bad
+++ b/tests/compilation.bad
@@ -7,19 +7,21 @@
 
 ; a + b + c
 
-if true {
-    2
-}
+; if true {
+;     2
+; }
 
-if true {
-    1 + 2
+; if true {
+;     1 + 2
+; } else {
+;     3 + 4
+; }
+
+let a = 0
+if 1 != 1 {
+    set a = 1
 } else {
-    3 + 4
+    set a = 2
 }
 
-; let a = 0
-; if 1 != 1 {
-;     set a = 1
-; } else {
-;     set a = 2
-; }
+a