1 files changed, 71 insertions, 4 deletions

diff --git a/src/ast.h b/src/ast.h
index b2c565c..42f02ac 100644
--- a/src/ast.h
+++ b/src/ast.h
@@ -16,6 +16,7 @@
  */
 #ifndef AST_H
 #define AST_H
+#include "lexer.h"
 #include "string_view.h"
 #include "vector.h"
 #include <stdint.h>
@@ -29,6 +30,11 @@ typedef enum
 
 typedef struct ast_node_t ast_node_t;
 
+typedef struct ast_namespace_t
+{
+  vector_t *nodes;
+} ast_namespace_t;
+
 typedef struct ast_block_t
 {
   vector_t *body;
@@ -49,10 +55,28 @@ typedef struct ast_variable_t
   ast_identifier_t *identifier;
 } ast_variable_t;
 
-typedef struct ast_function_declaration_t
+typedef struct ast_function_call_t
+{
+  ast_identifier_t *identifier;
+  vector_t *arguments;
+} ast_function_call_t;
+
+typedef struct ast_function_parameter_t
+{
+  ast_identifier_t identifier;
+  type_t type;
+} ast_function_parameter_t;
+
+typedef struct ast_function_prototype_t
 {
   ast_identifier_t identifier;
   type_t return_type;
+  vector_t *parameters;
+} ast_function_prototype_t;
+
+typedef struct ast_function_declaration_t
+{
+  ast_function_prototype_t prototype;
   ast_node_t *body;
 } ast_function_declaration_t;
 
@@ -117,24 +141,36 @@ typedef struct ast_if_stmt_t
   ast_node_t *body;
 } ast_if_stmt_t;
 
+typedef struct ast_unkown_token_t
+{
+  type_t expected_type;
+  token_t reference_token;
+} ast_unkown_token_t;
+
 typedef enum
 {
+  AST_NAMESPACE,
   AST_BINARY_OPERATION,
   AST_BLOCK,
   AST_FUNCTION_DECLARATION,
+  AST_FUNCTION_PARAMETER,
+  AST_FUNCTION_CALL,
   AST_LITERAL,
   AST_RETURN_STMT,
   AST_IF_STMT,
-  AST_UNKOWN_NODE,
   AST_VARIABLE_DECLARATION,
   AST_VARIABLE_ASSIGNMENT,
-  AST_VARIABLE
+  AST_VARIABLE,
+  AST_UNKOWN_NODE
 } ast_node_kind_t;
 
 typedef union
 {
+  ast_namespace_t ns;
   ast_binary_operation_t binary_operation;
   ast_function_declaration_t function;
+  ast_function_parameter_t function_parameter;
+  ast_function_call_t function_call;
   ast_literal_t literal;
   ast_block_t block;
   ast_if_stmt_t if_stmt;
@@ -142,6 +178,7 @@ typedef union
   ast_variable_declaration_t variable_declaration;
   ast_variable_assignment_t variable_assignment;
   ast_variable_t variable;
+  ast_unkown_token_t unknown;
 } ast_node_data_t;
 
 typedef struct ast_node_t
@@ -164,13 +201,25 @@ void
 ast_node_destroy(ast_node_t *node);
 
 ast_node_t *
+ast_node_new_namespace(vector_t *nodes);
+
+ast_node_t *
 ast_node_new_binary_operation(ast_binary_operation_kind_t kind,
                               ast_node_t *left,
                               ast_node_t *right,
                               type_t result_type);
 
 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_declaration(string_view_t function_name,
+                                  type_t return_type,
+                                  vector_t *parameters,
+                                  ast_node_t *body);
+
+ast_node_t *
+ast_node_new_function_parameter(string_view_t name, type_t type);
+
+ast_node_t *
+ast_node_new_function_call(ast_identifier_t *identifier, type_t result_type, vector_t *arguments);
 
 ast_node_t *
 ast_node_new_return_stmt(ast_node_t *argument);
@@ -196,4 +245,22 @@ ast_node_new_variable_assignment(ast_identifier_t *identifier, ast_node_t *expre
 ast_node_t *
 ast_node_new_if_stmt(ast_node_t *condition, ast_node_t *body);
 
+ast_node_t *
+ast_node_ns_get_function_node_by_sv(ast_node_t *ns, string_view_t name);
+
+ast_node_t *
+ast_node_ns_get_function_node_by_name(ast_node_t *ns, char *function_name);
+
+bool
+ast_node_is_variable_declaration(ast_node_t *node);
+
+bool
+ast_node_is_function_declaration(ast_node_t *node);
+
+ast_identifier_t *
+ast_node_function_declaration_identifier(ast_node_t *node);
+
+string_view_t
+ast_node_function_declaration_name(ast_node_t *node);
+
 #endif /* AST_H */