From 6ab9c3732793b6fd40e6388cf1b08756194c0ea6 Mon Sep 17 00:00:00 2001
From: Quinn Stephens <quinn@osmora.org>
Date: Thu, 7 Nov 2024 16:59:19 -0500
Subject: [compiler] Parse return statements

Laid groundwork for statements and AST trees.
Currently return values are not supported, expression parsing must be implemented first.
Also stopped dumping parsed type definitions.

Signed-off-by: Quinn Stephens <quinn@osmora.org>
---
 include/parser/ast.h  | 38 +++++++++++++++++++++++++++++++++++++-
 include/parser/proc.h |  3 +++
 include/parser/stmt.h | 16 ++++++++++++++++
 3 files changed, 56 insertions(+), 1 deletion(-)
 create mode 100644 include/parser/stmt.h

(limited to 'include/parser')

diff --git a/include/parser/ast.h b/include/parser/ast.h
index 03be1bd..9463511 100644
--- a/include/parser/ast.h
+++ b/include/parser/ast.h
@@ -7,14 +7,50 @@
 #ifndef _PARSER_AST_H
 #define _PARSER_AST_H
 
+#include <stdlib.h>
+#include "list.h"
+
 enum ast_node_kind {
     NK_UNKNOWN,
 
-    NK_
+    NK_PROCEDURE,
+    NK_RETURN
 };
 
 struct ast_node {
+    struct list_entry list_entry;
+
     enum ast_node_kind kind;
+
+    struct ast_node *parent;
+    struct list children;
 };
 
+static inline void
+ast_append_child(struct ast_node *parent, struct ast_node *child)
+{
+    child->parent = parent;
+    list_append(&parent->children, &child->list_entry);
+}
+
+static inline void
+ast_remove_child(struct ast_node *child)
+{
+    list_remove(&child->list_entry);
+    child->parent->children.length--;
+}
+
+static inline struct ast_node *
+ast_new_node(enum ast_node_kind kind)
+{
+    struct ast_node *node;
+
+    node = malloc(sizeof(struct ast_node));
+    node->kind = kind;
+    node->parent = NULL;
+    list_init(&node->children);
+
+    return node;
+}
+
 #endif /* !_PARSER_AST_H */
diff --git a/include/parser/proc.h b/include/parser/proc.h
index 0cc8471..f74fee6 100644
--- a/include/parser/proc.h
+++ b/include/parser/proc.h
@@ -9,6 +9,7 @@
 
 #include "list.h"
 #include "hashmap.h"
+#include "parser/ast.h"
 #include "parser/var.h"
 
 struct parameter {
@@ -27,6 +28,8 @@ struct procedure {
 
     struct type *ret_typ;
     int ret_n_ptrs;
+
+    struct ast_node *node;
 };
 
 void parse_proc(struct parser *ctx);
diff --git a/include/parser/stmt.h b/include/parser/stmt.h
new file mode 100644
index 0000000..caf022c
--- /dev/null
+++ b/include/parser/stmt.h
@@ -0,0 +1,16 @@
+/*
+ * Statement parser.
+ * Copyright (c) 2023-2024, Quinn Stephens and the OSMORA team.
+ * Provided under the BSD 3-Clause license.
+ */
+
+#ifndef _PARSER_STMT_H
+#define _PARSER_STMT_H
+
+#include "parser/ast.h"
+#include "parser/proc.h"
+#include "parser.h"
+
+void parse_stmt_block(struct parser *ctx, struct ast_node *parent, struct procedure *proc);
+
+#endif /* !_PARSER_STMT_H */
-- 
cgit v1.2.3