From 09871cce7654f3035b2ab75d2840aabf3c120984 Mon Sep 17 00:00:00 2001
From: servostar <sven.vogel123@web.de>
Date: Tue, 11 Jun 2024 17:59:25 +0200
Subject: [PATCH] added: import statement

---
 src/ast/ast.c  |  9 ++++++++-
 src/ast/ast.h  |  3 +++
 src/compiler.c | 33 ++++++++++++++++++++++++++++-----
 3 files changed, 39 insertions(+), 6 deletions(-)

diff --git a/src/ast/ast.c b/src/ast/ast.c
index 20132dc..2d6bef6 100644
--- a/src/ast/ast.c
+++ b/src/ast/ast.c
@@ -315,4 +315,11 @@ AST_NODE_PTR AST_get_node_by_kind(AST_NODE_PTR owner, enum AST_SyntaxElement_t k
   }
 
   return NULL;
-}
\ No newline at end of file
+}
+
+void AST_merge_modules(AST_NODE_PTR dst, AST_NODE_PTR src) {
+    for (size_t i = 0; i < src->child_count; i++) {
+        AST_push_node(dst, AST_remove_child(src, i));
+    }
+    AST_delete_node(src);
+}
diff --git a/src/ast/ast.h b/src/ast/ast.h
index c55acb9..1f17d8c 100644
--- a/src/ast/ast.h
+++ b/src/ast/ast.h
@@ -221,4 +221,7 @@ void AST_fprint_graphviz(FILE* stream, const struct AST_Node_t* node);
 
 AST_NODE_PTR AST_get_node_by_kind(AST_NODE_PTR owner, enum AST_SyntaxElement_t kind);
 
+[[gnu::nonnull(1), gnu::nonnull(2)]]
+void AST_merge_modules(AST_NODE_PTR dst, AST_NODE_PTR src);
+
 #endif
diff --git a/src/compiler.c b/src/compiler.c
index 9dcad1b..291ad6d 100644
--- a/src/compiler.c
+++ b/src/compiler.c
@@ -164,6 +164,29 @@ static void run_backend_codegen(const Module* module, const TargetConfig* target
     err = deinit_backend();
 }
 
+static AST_NODE_PTR compile_module_with_dependencies(ModuleFileStack *unit, ModuleFile* file) {
+    AST_NODE_PTR root_module = AST_new_node(empty_location(), AST_Module, NULL);
+
+    if (compile_file_to_ast(root_module, file) == EXIT_SUCCESS) {
+
+        for (size_t i = 0; i < root_module->child_count; i++) {
+            AST_NODE_PTR child = AST_get_node(root_module, i);
+
+            if (child->kind == AST_Import) {
+                AST_NODE_PTR imported_module = AST_new_node(empty_location(), AST_Module, NULL);
+
+                ModuleFile *imported_file = push_file(unit, child->value);
+
+                if (compile_file_to_ast(imported_module, imported_file) == EXIT_SUCCESS) {
+                    AST_merge_modules(root_module, imported_module);
+                }
+            }
+        }
+    }
+
+    return root_module;
+}
+
 /**
  * @brief Build the given target
  * @param unit
@@ -172,22 +195,22 @@ static void run_backend_codegen(const Module* module, const TargetConfig* target
 static void build_target(ModuleFileStack *unit, const TargetConfig *target) {
     print_message(Info, "Building target: %s", target->name);
 
-    AST_NODE_PTR ast = AST_new_node(empty_location(), AST_Module, NULL);
     ModuleFile *file = push_file(unit, target->root_module);
+    AST_NODE_PTR  ast = compile_module_with_dependencies(unit, file);
 
-    if (compile_file_to_ast(ast, file) == EXIT_SUCCESS) {
+    if (ast != NULL) {
         if (setup_target_environment(target) == 0) {
 
             print_ast_to_file(ast, target);
-            Module* module = create_set(ast);
+            Module *module = create_set(ast);
 
             if (module != NULL) {
                 run_backend_codegen(module, target);
             }
         }
-    }
 
-    AST_delete_node(ast);
+        AST_delete_node(ast);
+    }
 
     mem_purge_namespace(MemoryNamespaceLex);
     mem_purge_namespace(MemoryNamespaceAst);