Make sure nil return values and parameters are considered as such

1 files changed, 16 insertions, 2 deletions

diff --git a/src/main.c b/src/main.c
index 4bfd104..72ea88b 100644
--- a/src/main.c
+++ b/src/main.c
@@ -261,14 +261,16 @@ graph_functions(TypeScope *scope, Arena a) {
             "<TD ALIGN=\"left\" > %s </TD>"
             "<TD ALIGN=\"left\" > %s </TD>"
             "</TR>",
-            func->val.name, func->val.name, func->val.param_type, func->val.return_type);
+            func->val.name, func->val.name, func->val.param_type,
+            func->val.return_type);
         func = funmap_next(&iter, &a);
     }
     println("</TABLE>>];");
     sz this_id = scope->id;
     while (scope->parent) {
         if (scope->parent->types) {
-            println("fun_%d:e->fun_%d:%s:w;", this_id, scope->parent->id, scope->name);
+            println("fun_%d:e->fun_%d:%s:w;", this_id, scope->parent->id,
+                    scope->name);
             break;
         } else {
             scope = scope->parent;
@@ -696,6 +698,9 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
                     ret_type = str_concat(ret_type, cstr(","), a->storage);
                 }
             }
+            if (!ret_type.size) {
+                ret_type = cstr("nil");
+            }
             node->type = ret_type;
             return node->type;
         } break;
@@ -735,6 +740,9 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
                     param_type = str_concat(param_type, cstr(","), a->storage);
                 }
             }
+            if (!param_type.size) {
+                param_type = cstr("nil");
+            }
             node->fun_params = param_type;
 
             Type ret_type = cstr("");
@@ -746,6 +754,9 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
                     ret_type = str_concat(ret_type, cstr(","), a->storage);
                 }
             }
+            if (!ret_type.size) {
+                ret_type = cstr("nil");
+            }
             node->fun_return = ret_type;
 
             Str symbol = node->func_name->value.str;
@@ -763,6 +774,9 @@ type_inference(Analyzer *a, Node *node, TypeScope *scope) {
                     Node *expr = node->func_body->elements[i];
                     type = type_inference(a, expr, scope);
                 }
+                if (!type.size) {
+                    type = cstr("nil");
+                }
                 node->func_body->type = type;
             } else {
                 type_inference(a, node->func_body, scope);