1 files changed, 14 insertions, 6 deletions

diff --git a/src/gas_assembly_generator.c b/src/gas_assembly_generator.c
index 413fbb0..4326b27 100644
--- a/src/gas_assembly_generator.c
+++ b/src/gas_assembly_generator.c
@@ -147,6 +147,9 @@ gas_assembly_generator_compile(gas_assembly_generator_t *gen, ast_node_t *ast)
   gen->latest_evaluation.kind = EVALUATION_RESULT_VOID;
 
   switch (ast->kind) {
+    case AST_NAMESPACE:
+      gas_assembly_generator_compile_ns(gen, &ast->data.ns);
+      break;
     case AST_BINARY_OPERATION:
       gas_assembly_generator_binary_operation(gen, &ast->data.binary_operation);
       break;
@@ -179,6 +182,14 @@ gas_assembly_generator_compile(gas_assembly_generator_t *gen, ast_node_t *ast)
   }
 }
 
+void
+gas_assembly_generator_compile_ns(gas_assembly_generator_t *gen, ast_namespace_t *ns)
+{
+  for (size_t i = 0; i < ns->nodes->size; i++) {
+    gas_assembly_generator_compile(gen, vector_at(ns->nodes, i));
+  }
+}
+
 static void
 gas_assembly_generator_compile_function(gas_assembly_generator_t *gen, ast_function_declaration_t *func)
 {
@@ -201,12 +212,9 @@ gas_assembly_generator_compile_function(gas_assembly_generator_t *gen, ast_funct
 }
 
 void
-gas_assembly_generator_compile_linux_main(gas_assembly_generator_t *gen, ast_node_t *func)
+gas_assembly_generator_compile_linux_main(gas_assembly_generator_t *gen, ast_node_t *ns)
 {
-  assert(func);
-  assert(func->kind == AST_FUNCTION_DECLARATION);
-
-  if (!string_view_eq(func->data.function.identifier.name, string_view_from_str("main"))) {
+  if (ast_node_ns_get_function_node_by_name(ns, "main") == NULL) {
     fprintf(stderr, "[ERROR]: no main function has been defined!\n");
     exit(EXIT_FAILURE);
   }
@@ -214,7 +222,7 @@ gas_assembly_generator_compile_linux_main(gas_assembly_generator_t *gen, ast_nod
   fprintf(gen->stream, ".global _start\n");
   fprintf(gen->stream, ".text\n");
 
-  gas_assembly_generator_compile(gen, func);
+  gas_assembly_generator_compile(gen, ns);
 
   fprintf(gen->stream, "_start:\n");
   fprintf(gen->stream, "  call main\n");