From 8aac9c7a82027ba9d90c0546be7f091df89a6d1f Mon Sep 17 00:00:00 2001
From: c+1
Date: Sun, 29 Oct 2023 17:12:11 -0400
Subject: AST

---
 src/include/tree.h | 76 ++++++++++++++++++++++++++++++++++++------------------
 1 file changed, 51 insertions(+), 25 deletions(-)

(limited to 'src/include')

diff --git a/src/include/tree.h b/src/include/tree.h
index 108f646..d19f1f7 100644
--- a/src/include/tree.h
+++ b/src/include/tree.h
@@ -3,39 +3,65 @@
 
 #include <stdlib.h>
 
-typedef enum {
-      LEAF_TYPE_INT,
-      LEAF_TYPE_STR,
-} leaf_t;
-
-typedef struct ARG_T_STRUCT {
-   leaf_t type;
-   char* id;
-   struct ARG_T_STRUCT* nxt;
-} arg_t;
-
-typedef struct TREE_T_STRUCT {
+typedef struct TREE {
    enum TREE_TYPE {
+      TREE_TYPE_LINT,
+      TREE_TYPE_LSTR,
+      TREE_TYPE_TAG,
+      TREE_TYPE_DARG,
+      TREE_TYPE_CARG,
       TREE_TYPE_DEF,
       TREE_TYPE_CAL,
    } type;
 
-   union {
-      struct {
-         leaf_t type;
-         char* id;
-         arg_t* arg;
-         struct TREE_T_STRUCT* val;
-      } tree_def;
-
-      struct {
-         char* id;
-         arg_t* arg;
-      } tree_cal;
+   union TREE_DATA{
+
+      /* literal integer */
+      struct TREE_DATA_LINT {
+         int val;
+      } lint;
+
+      /* literal string */
+      struct TREE_DATA_LSTR {
+         size_t len;
+         char* val;
+      } lstr;
+
+      /* tags */
+      struct TREE_DATA_TAG {
+         char* val;
+         struct TREE_DATA_TAG* nxt;
+      } tag;
+
+      /* definition arguments */
+      struct TREE_DATA_DARG {
+         struct TREE_DATA_TAG* tag;
+         struct TREE_DATA_DARG* nxt;
+      } darg;
+
+      /* call arguments */
+      struct TREE_DATA_CARG {
+         struct TREE* val;
+         struct TREE_DATA_CARG* nxt;
+      } carg;
+      
+      /* definitions */
+      struct TREE_DATA_DEF {
+         struct TREE_DATA_TAG* tag;
+         struct TREE_DATA_DARG* arg;
+         struct TREE* val;
+      } def;
+
+      /* calls */
+      struct TREE_DATA_CAL {
+         char* target;
+         struct TREE_DATA_CARG* arg;
+      } cal;
+
    } data;
 } tree_t;
 
 tree_t* tree_init(int type);
-void tree_destroy(tree_t*);
+void tree_destroy(tree_t* tree);
 
 #endif
-- 
cgit v1.2.3