Fill up main symbol table with everything we need

1 files changed, 10 insertions, 3 deletions

diff --git a/src/main.c b/src/main.c
index 45c5213..7e95e2e 100644
--- a/src/main.c
+++ b/src/main.c
@@ -10,6 +10,7 @@
 // TODO: unions
 // TODO: match deconstruct enums
 // TODO: arrays and pointers
+// TODO: if we have errors make sure we mark the analyzer as such
 
 typedef enum ExecMode {
     RUN_NORMAL,
@@ -325,6 +326,9 @@ typecheck_field(Analyzer *a, Node *node, Scope *scope, Str symbol) {
                              .val = node->field_val,
                          },
                          a->storage);
+        symmap_insert(&scope->symbols, field_name,
+                      (Symbol){.name = field_type, .kind = SYM_STRUCT_FIELD},
+                      a->storage);
         node->type = field_type;
     }
 }
@@ -575,6 +579,10 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                 }
                 enummap_insert(&scope->enums, field_name,
                                (Enum){.name = field_name}, a->storage);
+                symmap_insert(
+                    &scope->symbols, field_name,
+                    (Symbol){.name = field_name, .kind = SYM_ENUM_FIELD},
+                    a->storage);
                 field->type = cstr("int");
             }
             symmap_insert(&scope->symbols, symbol,
@@ -629,6 +637,7 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
         } break;
         case NODE_COND:
         case NODE_MATCH: {
+            // TODO: proper typecheck matchcase expressions here...
             if (node->match_expr) {
                 type_inference(a, node->match_expr, scope);
             }
@@ -644,7 +653,6 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                 }
                 previous = next;
             }
-            // TODO: proper typecheck matchcase expressions here...
             node->type = previous;
             return node->type;
         } break;
@@ -911,8 +919,7 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                 args = cstr("nil");
             }
             Str expected = fun->val.param_type;
-            bool err = !str_eq(args, expected);
-            if (err) {
+            if (!str_eq(args, expected) && !str_eq(expected, cstr("..."))) {
                 eprintln(
                     "%s:%d:%d: error: mismatched parameter types: %s expected "
                     "%s",