1 files changed, 46 insertions, 12 deletions

diff --git a/src/main.c b/src/main.c
index b0207f7..669984b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -57,6 +57,12 @@ typedef struct Symbol {
     SymbolKind kind;
 } Symbol;
 
+typedef struct ArraySymbol {
+    Str name;
+    Str type;
+    sz size;
+} ArraySymbol;
+
 typedef struct Fun {
     Str name;
     Str param_type;
@@ -74,6 +80,7 @@ typedef struct Struct {
     Node *val;
 } Struct;
 
+MAPDEF(ArrayMap, arraymap, Str, ArraySymbol, str_hash, str_eq)
 MAPDEF(SymbolMap, symmap, Str, Symbol, str_hash, str_eq)
 MAPDEF(FunMap, funmap, Str, Fun, str_hash, str_eq)
 MAPDEF(EnumMap, enummap, Str, Enum, str_hash, str_eq)
@@ -87,6 +94,7 @@ typedef struct Scope {
     FunMap *funcs;
     EnumMap *enums;
     StructMap *structs;
+    ArrayMap *arrays;
     struct Scope *parent;
 } Scope;
 
@@ -478,6 +486,14 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                     a->err = true;
                     return cstr("");
                 }
+                if (node->var_type->is_ptr) {
+                    type_name = str_concat(cstr("@"), type_name, a->storage);
+                }
+                if (node->var_type->kind == NODE_ARR_TYPE) {
+                    type_name = str_concat(cstr("@"), type_name, a->storage);
+                    // TODO: typecheck size
+                    // TODO: register array in scope
+                }
                 if (node->var_val) {
                     Str type = type_inference(a, node->var_val, scope);
                     if (!type.size) {
@@ -506,7 +522,12 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                         }
                     }
                 }
-                symmap_insert(&scope->symbols, symbol, type->val, a->storage);
+                symmap_insert(&scope->symbols, symbol,
+                              (Symbol){
+                                  .name = type_name,
+                                  .kind = SYM_VAR,
+                              },
+                              a->storage);
                 return node->type;
             }
 
@@ -856,6 +877,7 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
             node->type = type->val.name;
             return node->type;
         } break;
+        case NODE_SYMBOL_IDX:
         case NODE_SYMBOL: {
             Str symbol = node->value.str;
             SymbolMap *type = find_type(scope, symbol);
@@ -865,9 +887,22 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                 a->err = true;
                 return cstr("");
             }
+            Str type_name = type->val.name;
+            if (node->kind == NODE_SYMBOL_IDX) {
+                Str idx_type = type_inference(a, node->arr_size, scope);
+                if (!strset_lookup(&a->integer_types, idx_type)) {
+                    emit_semantic_error(
+                        a, node, cstr("can't resolve non integer index"));
+                    return cstr("");
+                }
+                type_name = str_remove_prefix(type_name, cstr("@"));
+            }
+            if (node->is_ptr) {
+                type_name = str_concat(cstr("@"), type_name, a->storage);
+            }
 
-            FindEnumResult e = find_enum(scope, type->val.name);
-            if (e.map && str_eq(symbol, type->val.name)) {
+            FindEnumResult e = find_enum(scope, type_name);
+            if (e.map && str_eq(symbol, type_name)) {
                 if (!node->next) {
                     eprintln(
                         "%s:%d:%d: error: unspecified enum field for symbol "
@@ -877,7 +912,7 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                     return cstr("");
                 }
                 // Check if there is a next and it matches the enum field.
-                Str field = str_concat(type->val.name, cstr("."), a->storage);
+                Str field = str_concat(type_name, cstr("."), a->storage);
                 field = str_concat(field, node->next->value.str, a->storage);
                 if (!enummap_lookup(&e.scope->enums, field)) {
                     eprintln(
@@ -888,24 +923,23 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                     a->err = true;
                     return cstr("");
                 }
-                node->next->type = type->val.name;
-                node->type = type->val.name;
+                node->next->type = type_name;
+                node->type = type_name;
                 return node->next->type;
             }
 
-            FindStructResult s = find_struct(scope, type->val.name);
+            FindStructResult s = find_struct(scope, type_name);
             if (s.map) {
-                if (str_eq(symbol, type->val.name)) {
+                if (str_eq(symbol, type_name)) {
                     eprintln(
                         "%s:%d:%d: error: struct incomplete struct literal "
                         "'%s', did you mean to use %s:{}?",
-                        a->file_name, node->line, node->col, symbol,
-                        symbol);
+                        a->file_name, node->line, node->col, symbol, symbol);
                     a->err = true;
                     return cstr("");
                 } else {
                     if (node->next) {
-                        Str chain = type->val.name;
+                        Str chain = type_name;
                         Node *next = node;
                         while (next->next) {
                             next = next->next;
@@ -927,7 +961,7 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                     }
                 }
             }
-            node->type = type->val.name;
+            node->type = type_name;
             return node->type;
         } break;
         case NODE_STRUCT_LIT: {