1 files changed, 114 insertions, 3 deletions

diff --git a/src/ast.c b/src/ast.c
index 5790c05..f6f8f08 100644
--- a/src/ast.c
+++ b/src/ast.c
@@ -44,8 +44,12 @@ void
 ast_node_destroy(ast_node_t *node)
 {
   switch (node->kind) {
+    case AST_NAMESPACE:
+      ast_node_destroy_vector(node->data.ns.nodes);
+      break;
     case AST_FUNCTION_DECLARATION:
       ast_node_destroy(node->data.function.body);
+      ast_node_destroy_vector(node->data.function.prototype.parameters);
       break;
     case AST_IF_STMT:
       ast_node_destroy(node->data.if_stmt.condition);
@@ -66,9 +70,11 @@ ast_node_destroy(ast_node_t *node)
       break;
     case AST_VARIABLE_ASSIGNMENT:
       ast_node_destroy(node->data.variable_assignment.expression);
+    case AST_FUNCTION_PARAMETER:
     case AST_LITERAL:
     case AST_UNKOWN_NODE:
     case AST_VARIABLE:
+    case AST_FUNCTION_CALL:
       break;
   }
   free(node);
@@ -89,7 +95,30 @@ ast_node_new_return_stmt(ast_node_t *argument)
 }
 
 ast_node_t *
-ast_node_new_function_declaration(string_view_t function_name, type_t return_type, ast_node_t *body)
+ast_node_new_function_parameter(string_view_t name, type_t type)
+{
+  ast_node_t *node = ast_node_new();
+
+  *node = (ast_node_t){
+    .kind = AST_FUNCTION_PARAMETER,
+    .data = {
+      .function_parameter = {
+        .identifier = {
+          .name = name,
+        },
+        .type = type,
+      },
+    },
+  };
+
+  return node;
+}
+
+ast_node_t *
+ast_node_new_function_declaration(string_view_t function_name,
+                                  type_t return_type,
+                                  vector_t *parameters,
+                                  ast_node_t *body)
 {
   ast_node_t *node = ast_node_new();
 
@@ -97,8 +126,11 @@ ast_node_new_function_declaration(string_view_t function_name, type_t return_typ
     .kind = AST_FUNCTION_DECLARATION,
     .data = {
       .function = {
-        .identifier = { .name = function_name },
-        .return_type = return_type,
+        .prototype = {
+          .identifier = { .name = function_name },
+          .return_type = return_type,
+          .parameters = parameters,
+        },
         .body = body,
       }
     },
@@ -108,6 +140,24 @@ ast_node_new_function_declaration(string_view_t function_name, type_t return_typ
 }
 
 ast_node_t *
+ast_node_new_namespace(vector_t *nodes)
+{
+  ast_node_t *node = ast_node_new();
+
+  *node = (ast_node_t){
+    .kind = AST_NAMESPACE,
+    .result_type = TYPE_VOID,
+    .data = {
+      .ns = {
+        .nodes = nodes,
+      }
+    },
+  };
+
+  return node;
+}
+
+ast_node_t *
 ast_node_new_block(vector_t *body)
 {
   ast_node_t *node = ast_node_new();
@@ -254,6 +304,67 @@ ast_node_new_variable(ast_identifier_t *identifier, type_t result_type)
   return node;
 }
 
+ast_node_t *
+ast_node_new_function_call(ast_identifier_t *identifier, type_t result_type, vector_t *arguments)
+{
+  ast_node_t *node = ast_node_new();
+
+  *node = (ast_node_t){
+    .kind = AST_FUNCTION_CALL,
+    .result_type = result_type,
+    .data = { .function_call = { .identifier = identifier, .arguments = arguments } },
+  };
+
+  return node;
+}
+
+ast_node_t *
+ast_node_ns_get_function_node_by_sv(ast_node_t *ns, string_view_t name)
+{
+  assert(ns->kind == AST_NAMESPACE);
+
+  for (size_t i = 0; i < ns->data.ns.nodes->size; i++) {
+    ast_node_t *node = vector_at(ns->data.ns.nodes, i);
+
+    if (node->kind == AST_FUNCTION_DECLARATION && string_view_eq(node->data.function.prototype.identifier.name, name)) {
+      return node;
+    }
+  }
+  return NULL;
+}
+
+ast_node_t *
+ast_node_ns_get_function_node_by_name(ast_node_t *ns, char *function_name)
+{
+  return ast_node_ns_get_function_node_by_sv(ns, string_view_from_str(function_name));
+}
+
+bool
+ast_node_is_variable_declaration(ast_node_t *node)
+{
+  return node->kind == AST_VARIABLE_DECLARATION;
+}
+
+bool
+ast_node_is_function_declaration(ast_node_t *node)
+{
+  return node->kind == AST_FUNCTION_DECLARATION;
+}
+
+ast_identifier_t *
+ast_node_function_declaration_identifier(ast_node_t *node)
+{
+  assert(node->kind == AST_FUNCTION_DECLARATION);
+
+  return &node->data.function.prototype.identifier;
+}
+
+string_view_t
+ast_node_function_declaration_name(ast_node_t *node)
+{
+  return ast_node_function_declaration_identifier(node)->name;
+}
+
 char *
 ast_type_to_str(type_t type)
 {