From 2a03e1e42906605be5ea093bcaca212c41acb62f Mon Sep 17 00:00:00 2001
From: A404M <ahmadmahmoudiprogrammer@gmail.com>
Date: Fri, 4 Apr 2025 13:17:48 +0330
Subject: returning functions now work

---
 src/compiler/ast-tree.c | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'src/compiler/ast-tree.c')

diff --git a/src/compiler/ast-tree.c b/src/compiler/ast-tree.c
index 77b93d6..4c6d36d 100644
--- a/src/compiler/ast-tree.c
+++ b/src/compiler/ast-tree.c
@@ -2897,7 +2897,11 @@ bool setTypesReturn(AstTree *tree, AstTreeSetTypesHelper _helper,
     };
     if (!setAllTypes(metadata->value, helper, NULL)) {
       return false;
-    } else if (!typeIsEqual(metadata->value->type, function->returnType)) {
+    }
+    if (helper.lookingType != function->returnType) {
+      astTreeDelete(helper.lookingType); // TODO: change plan
+    }
+    if (!typeIsEqual(metadata->value->type, function->returnType)) {
       printError(tree->str_begin, tree->str_end, "Type mismatch");
       return false;
     }
-- 
cgit v1.2.3