Prepare parse tree for return

2 files changed, 25 insertions, 8 deletions

diff --git a/src/parser.c b/src/parser.c
index 2c1c24f..ba063f1 100644
--- a/src/parser.c
+++ b/src/parser.c
@@ -3,17 +3,17 @@
 
 Token
 next_token(Parser *parser) {
-    return parser->tokens[parser->current++];
+    return parser->tokens[parser->current_token++];
 }
 
 Token
 peek_token(Parser *parser) {
-    return parser->tokens[parser->current];
+    return parser->tokens[parser->current_token];
 }
 
 bool
 has_next(Parser *parser) {
-    return parser->current < array_size(parser->tokens);
+    return parser->current_token < array_size(parser->tokens);
 }
 
 bool
@@ -408,12 +408,20 @@ parse_next(Parser *parser) {
     }
 }
 
-void
+ParseTree *
+alloc_parsetree(void) {
+    ParseTree *parse_tree = malloc(sizeof(ParseTree));
+    array_init(parse_tree->roots, 0);
+    return parse_tree;
+}
+
+ParseTree *
 parse(Token *tokens) {
     Parser parser = {
         .tokens = tokens,
-        .current = 0,
+        .current_token = 0,
     };
+    parser.parse_tree = alloc_parsetree();
 
     // DEBUG: TOKENS
     printf("-- tokens --\n");
@@ -423,9 +431,13 @@ parse(Token *tokens) {
     while (has_next(&parser)) {
         Node *node = parse_next(&parser);
         if (node == NULL) {
-            return;
+            return NULL;
         }
+        // DEBUG:...
         print_node(node);
         printf("\n");
+
+        array_push(parser.parse_tree->roots, node);
     }
+    return parser.parse_tree;
 }
diff --git a/src/parser.h b/src/parser.h
index 47316b9..4d871ad 100644
--- a/src/parser.h
+++ b/src/parser.h
@@ -4,12 +4,17 @@
 #include "lexer.h"
 #include "nodes.h"
 
+typedef struct ParseTree {
+    Node **roots;
+} ParseTree;
+
 typedef struct Parser {
     Token *tokens;
-    size_t current;
+    size_t current_token;
+    ParseTree *parse_tree;
 } Parser;
 
-void parse(Token *tokens);
+ParseTree * parse(Token *tokens);
 Node * parse_next(Parser *parser);
 
 #endif // BDL_PARSER_H