From 0c04a61ef15f02eadc24d46b9527e0afb74040a4 Mon Sep 17 00:00:00 2001
From: Kyle Gunger <kgunger12@gmail.com>
Date: Sun, 21 Jul 2024 01:12:19 -0400
Subject: Begin ast funcs

---
 tnslc/parse/ast.tnsl | 127 ++++++++++++++++++++++++++++++++++++++++++++++++---
 tnslc/tnslc.tnsl     |   6 +--
 2 files changed, 123 insertions(+), 10 deletions(-)

(limited to 'tnslc')

diff --git a/tnslc/parse/ast.tnsl b/tnslc/parse/ast.tnsl
index 554aac2..86788e7 100644
--- a/tnslc/parse/ast.tnsl
+++ b/tnslc/parse/ast.tnsl
@@ -1,11 +1,13 @@
 
-uint16 NTYPE_MOD = 0
-uint16 NTYPE_STRUCT = 1
-uint16 NTYPE_ID = 2
-uint16 NTYPE_BINOP = 3
-uint16 NTYPE_PREOP = 4
-uint16 NTYPE_POSTOP = 5
-uint16 NTYPE_FUNCTION = 6
+uint16 NTYPE_MODULE = 0
+uint16 NTYPE_EXPORT = 1
+uint16 NTYPE_STRUCT = 2
+uint16 NTYPE_ID = 3
+uint16 NTYPE_BIN_OP = 4
+uint16 NTYPE_PRE_OP = 5
+uint16 NTYPE_POST_OP = 6
+uint16 NTYPE_FUNCTION = 7
+uint16 NTYPE_METHOD = 8
 
 
 struct Node {
@@ -14,8 +16,119 @@ struct Node {
 	utils.Vector sub
 }
 
+/; method Node
+	/; init (uint16 _type, ~uint8 data)
+		self.data = data
+		self._type = _type
+
+		Node n
+		self.sub.init(len n)
+	;/
+
+	/; end
+		_delete(self.data)
+		
+		/; loop (uint i = 0; i < self.sub.count) [i++]
+			~Node cur = self.sub.get(i)
+			cur`.end()
+		;/
+	;/
+;/
+
+/; _ast_root_loop (~utils.File fin, ~Node mod, Token first)
+;/
+
+/; _ast_module (~utils.File fin, ~Node mod, Token first)
+	Node out
+
+	uint16 t = NTYPE_MODULE
+	/; if (first.eq("export\0") == true)
+		Token tmp = produce_next_token(fin, first)
+		first.end()
+		first = tmp
+		t = NTYPE_EXPORT
+	;/
+
+	Token tmp = produce_next_token(fin, first)
+	first.end()
+	out.init(t, tmp.data)
+
+	tmp = produce_next_token(fin, tmp)
+	_ast_root_loop(fin, ~out, tmp)
+	mod`.sub.push(~out)
+;/
+
+/; _ast_file (~utils.File fin, ~Node mod)
+	fin`.open()
+	
+	Token tok = produce_first_token(fin)
+	_ast_root_loop(fin, mod, tok)
+
+	fin`.close()
+;/
+
 /; generate_ast (~utils.File fin) [Node]
 	Node out
+
+	utils.Vector v
+	v.init(1)
+
+	out.init(NTYPE_MODULE, v.as_cstr())
+
+	_ast_file(fin, ~out)
+
 	return out
 ;/
 
+#
+# Print out the AST from a specific node
+#
+
+/; print_node_type (~Node n)
+	/; if (n`._type == NTYPE_MODULE)
+		_printf("Mod\0")
+	;; else if (n`._type == NTYPE_EXPORT)
+		_printf("Exported Module\0")
+	;; else if (n`._type == NTYPE_STRUCT)
+		_printf("Struct\0")
+	;; else if (n`._type == NTYPE_ID)
+		_printf("ID\0")
+	;; else if (n`._type == NTYPE_BIN_OP)
+		_printf("Binary OP\0")
+	;; else if (n`._type == NTYPE_PRE_OP)
+		_printf("Pre OP\0")
+	;; else if (n`._type == NTYPE_POST_OP)
+		_printf("Post OP\0")
+	;; else if (n`._type == NTYPE_FUNCTION)
+		_printf("Function\0")
+	;/
+;/
+
+/; print_node_head (~Node n)
+	_printf("{ NODE_TYPE: \0")
+	print_node_type(n)
+	_printf(", DATA: \0")
+	_printf(n`.data)
+	_printf("\n\0")
+;/
+
+/; print_ast_rec(~Node n, uint depth)
+	/; loop (int i = 0; i < depth) [i++]
+		_printf("  \0")
+	;/
+	print_node_head(n)
+
+	/; loop (uint i = 0; i < n`.sub.count) [i++]
+		~Node s = n`.sub.get(i)
+		print_ast_rec(s, depth + 1)
+	;/
+	
+	/; loop (int i = 0; i < depth) [i++]
+		_printf("  \0")
+	;/
+	_printf("}\n\0")
+;/
+
+/; print_ast (~Node n)
+	print_ast_rec(n, 0)
+;/
diff --git a/tnslc/tnslc.tnsl b/tnslc/tnslc.tnsl
index bb7992c..2df15a3 100644
--- a/tnslc/tnslc.tnsl
+++ b/tnslc/tnslc.tnsl
@@ -35,9 +35,9 @@ usage:
 		fout.init(DEFAULT_FOUT)
 	;/
 
-	utils.Vector v = parse.gen_token_list(~fin)
-	parse.print_token_list(~v)
-	parse.end_token_list(~v)
+	parse.Node n = parse.generate_ast(~fin)
+	parse.print_ast(~n)
+	n.end()
 
 	fin.end()
 	fout.end()
-- 
cgit v1.2.3