returning functions now work

4 files changed, 20 insertions, 17 deletions

diff --git a/Makefile b/Makefile
index c0327d7..4b79a2c 100644
--- a/Makefile
+++ b/Makefile
@@ -18,9 +18,9 @@ NC := \033[0m
 INC_DIRS := $(SRC_DIR)
 INC_FLAGS := $(addprefix -I,$(INC_DIRS))
 
-CFLAGS := $(INC_FLAGS) -Wall -Wextra -std=gnu23 -O3
+# CFLAGS := $(INC_FLAGS) -Wall -Wextra -std=gnu23 -O3
 # CFLAGS := $(INC_FLAGS) -Wall -Wextra -std=gnu23 -Oz
-# CFLAGS := $(INC_FLAGS) -Wall -Wextra -std=gnu23 -g
+CFLAGS := $(INC_FLAGS) -Wall -Wextra -std=gnu23 -g
 
 EXEC_FILE := $(BUILD_DIR)/$(PROJECT_NAME)
 
diff --git a/code/main.felan b/code/main.felan
index e8e3254..532afaa 100644
--- a/code/main.felan
+++ b/code/main.felan
@@ -6,16 +6,13 @@ float :: f32;
 double :: f64;
 
 main :: () -> void {
-  a :u64= 2;
-  f := (c:u64,b:u64)->void{
-    print_u64 c-b+a;
-  };
-  f = (c:u64,b:u64)->void{
-    print_u64 c*b;
-  };
-  foo(fun = f);
+  f := foo();
+  f(3);
 };
 
-foo :: (fun:(b:u64,u64)->void)->void{
-  fun(b = 3,4);
+foo :: ()->((u64)->void){
+  f :: (a:u64)->void{
+    print_u64 a;
+  };
+  return f;
 };
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;
     }
diff --git a/src/compiler/parser.c b/src/compiler/parser.c
index 12985e1..69a9c8a 100644
--- a/src/compiler/parser.c
+++ b/src/compiler/parser.c
@@ -1156,7 +1156,6 @@ ParserNode *parserFunction(LexerNode *node, LexerNode *begin, LexerNode *end,
                            ParserNode *parent) {
   LexerNode *paramsNode = node - 1;
   LexerNode *retTypeNode = node + 1;
-  LexerNode *bodyNode = node + 2;
   if (paramsNode < begin || paramsNode->parserNode == NULL) {
     printError(node->str_begin, node->str_end, "No params");
     return NULL;
@@ -1166,12 +1165,15 @@ ParserNode *parserFunction(LexerNode *node, LexerNode *begin, LexerNode *end,
   }
   ParserNode *params = getUntilCommonParent(paramsNode->parserNode, parent);
   ParserNode *retType = getUntilCommonParent(retTypeNode->parserNode, parent);
-  ParserNode *body;
 
-  if (bodyNode >= end || bodyNode->parserNode == NULL) {
+  LexerNode *bodyNode =
+      getNextLexerNodeUsingCommonParent(retTypeNode, end, parent);
+
+  ParserNode *body;
+  if (bodyNode == NULL || bodyNode >= end || bodyNode->parserNode == NULL) {
     body = NULL;
   } else {
-    body = getUntilCommonParent(bodyNode->parserNode, parent);
+    body = bodyNode->parserNode;
     if (body == NULL || body->token != PARSER_TOKEN_SYMBOL_CURLY_BRACKET) {
       printError(node->str_begin, node->str_end, "Bad body");
       return NULL;