1 files changed, 15 insertions, 1 deletions

diff --git a/src/main.c b/src/main.c
index b1dc0ce..cb93649 100644
--- a/src/main.c
+++ b/src/main.c
@@ -867,6 +867,7 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
             node->type = cstr("str");
             return node->type;
         } break;
+        case NODE_ARR_TYPE: 
         case NODE_TYPE: {
             SymbolMap *type = find_type(scope, node->value.str);
             if (!type) {
@@ -1077,9 +1078,15 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
                 Node *param = node->func_params[i];
                 Str symbol = param->param_name->value.str;
                 Str type = param->param_type->value.str;
+                if (param->param_type->is_ptr) {
+                    type = str_concat(cstr("@"), type, a->storage);
+                }
+                if (param->param_type->kind == NODE_ARR_TYPE) {
+                    type = str_concat(cstr("@"), type, a->storage);
+                }
                 param->param_name->type =
                     type_inference(a, param->param_type, scope);
-                param->type = param->param_name->type;
+                param->type = type;
                 symmap_insert(&scope->symbols, symbol,
                               (Symbol){.name = type, .kind = SYM_PARAM},
                               a->storage);
@@ -1097,6 +1104,12 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
             for (sz i = 0; i < array_size(node->func_ret); i++) {
                 Node *expr = node->func_ret[i];
                 Str type = type_inference(a, expr, scope);
+                if (expr->is_ptr) {
+                    type = str_concat(cstr("@"), type, a->storage);
+                }
+                if (expr->kind == NODE_ARR_TYPE) {
+                    type = str_concat(cstr("@"), type, a->storage);
+                }
                 ret_type = str_concat(ret_type, type, a->storage);
                 if (i != array_size(node->func_ret) - 1) {
                     ret_type = str_concat(ret_type, cstr(","), a->storage);
@@ -1163,6 +1176,7 @@ type_inference(Analyzer *a, Node *node, Scope *scope) {
             emit_semantic_error(a, node,
                                 cstr("type inference not implemented for this "
                                      "kind of expression"));
+            println("KIND: %s", node_str[node->kind]);
         } break;
     }
     return cstr("");