1 files changed, 48 insertions, 6 deletions

diff --git a/src/ast.c b/src/ast.c
index e9aa677..f6f8f08 100644
--- a/src/ast.c
+++ b/src/ast.c
@@ -49,6 +49,7 @@ ast_node_destroy(ast_node_t *node)
       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);
@@ -69,6 +70,7 @@ 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:
@@ -93,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();
 
@@ -101,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,
       }
     },
@@ -277,14 +305,14 @@ ast_node_new_variable(ast_identifier_t *identifier, type_t result_type)
 }
 
 ast_node_t *
-ast_node_new_function_call(ast_identifier_t *identifier, type_t result_type)
+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 } },
+    .data = { .function_call = { .identifier = identifier, .arguments = arguments } },
   };
 
   return node;
@@ -298,7 +326,7 @@ ast_node_ns_get_function_node_by_sv(ast_node_t *ns, string_view_t name)
   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.identifier.name, name)) {
+    if (node->kind == AST_FUNCTION_DECLARATION && string_view_eq(node->data.function.prototype.identifier.name, name)) {
       return node;
     }
   }
@@ -323,6 +351,20 @@ 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)
 {