demo source

source/tbc.cpp

@@ -0,0 +1,105 @@
+#include <stdio.h>
+#include <string.h>
+extern "C" {
+    #include <tree_sitter/api.h>
+    extern const TSLanguage * tree_sitter_c(void);
+}
+
+int tbtraverse(const char * const code) {
+    // init
+    TSParser * parser;
+    TSTree * tree;
+    TSTreeCursor cursor;
+    TSNode current_node;
+    TSNode previous_node;
+
+    parser = ts_parser_new();
+
+    ts_parser_set_language(parser, tree_sitter_c());
+
+    tree = ts_parser_parse_string(parser, NULL, code, strlen(code));
+    cursor = ts_tree_cursor_new(ts_tree_root_node(tree));
+    current_node = ts_tree_root_node(tree);
+
+    // meat
+    do {
+        current_node = ts_tree_cursor_current_node(&cursor);
+
+		const char * previous_node_type = NULL;
+
+		int tblen = ts_node_end_byte(current_node) - ts_node_start_byte(current_node);
+		char * tbtext = (char *)malloc(sizeof(char) * (tblen + 1));
+		memcpy(tbtext, code + ts_node_start_byte(current_node), tblen);
+		tbtext[tblen] = '\0';
+
+		// XXX INJECTION
+		
+        if (!strcmp("function_definition", ts_node_type(current_node))) {
+			
+			puts("ack");
+			puts(tbtext);
+
+			goto end;
+        }
+
+
+        if (!strcmp("number_literal", ts_node_type(current_node))) {
+			
+			puts("++");
+
+			goto end;
+        }
+
+
+
+	  end:
+		free(tbtext);
+    } while ([&] {
+		bool r = false;
+		previous_node = current_node;
+
+        if (ts_tree_cursor_goto_first_child(&cursor)
+        ||  ts_tree_cursor_goto_next_sibling(&cursor)) {
+            r = true;
+			goto eval;
+        }
+		
+		while (ts_tree_cursor_goto_parent(&cursor)) {
+			if (!strcmp(ts_node_type(current_node), "translation_unit")) {
+				r = false;
+				break;
+			}
+
+			if (ts_tree_cursor_goto_next_sibling(&cursor)) {
+				r = true;
+			}
+
+		  eval:
+			if (!strcmp("function_definition", ts_node_type(previous_node))) {
+				puts("^^df");
+
+				goto end;
+			}
+
+		  end:
+			if (r) { break; }
+		}
+
+		return r;
+    }());
+
+    // deinit
+    ts_tree_delete(tree);
+    ts_parser_delete(parser);
+    ts_tree_cursor_delete(&cursor);
+
+    return 0;
+}
+
+
+// @BAKE g++ $@ $(pkg-config --cflags --libs tree-sitter tree-sitter-c) -ggdb
+
+signed main() {
+    tbtraverse("int main() { return 0; }");
+}
+

source/tbc.py

@@ -0,0 +1,172 @@
+# @BAKE python tbc.py convert.tbsp > kek.cpp
+import re
+import sys
+
+def header(s : str, lang : str) -> str:
+	r = s.format(language=lang)
+	return r
+
+def definition_section(s : str) -> str:
+	def top(s : str) -> str:
+		regex = re.compile(r"^%top {(.*?^)}", re.MULTILINE | re.DOTALL)
+		matches = regex.findall(s)
+		return matches[0]
+	def lang(s : str) -> str:
+		regex = re.compile(r"%language (\w+)", re.MULTILINE | re.DOTALL)
+		matches = regex.findall(s)
+		return matches[0]
+	return {
+		"code" : top(s),
+		"language" : lang(s),
+	}
+
+def rule_section(s : str) -> str:
+	def rules(s : str, p : r"", shim : str) -> str:
+		r = ""
+		regex = re.compile(p, re.MULTILINE | re.DOTALL)
+		matches = regex.findall(s)
+		l = []
+		for match in matches:
+			l.append({
+				"name": match[0],
+				"code": match[1][1:-1],
+			})
+		for i in l:
+			r += shim.format(**i)
+		return r
+	enter_shims = rules(s, r"^enter (\w+) {(.*?)^}", shim_enter)
+	leave_shims = rules(s, r"^leave (\w+) {(.*?)^}", shim_leave)
+	r = tbtraverse.format(enter_shims=enter_shims, leave_shims=leave_shims)
+	return r
+
+def tbc_main(argv : [str]):
+	if len(argv) < 2:
+		print("tbc <file>")
+		exit(1)
+
+	input_file = argv[1]
+	input_string = open(input_file, "r").read()
+
+	delim = "%%"
+	delims = []
+	delims.append(input_string.find(delim))
+	delims.append(delims[0] + len(delim) + input_string[delims[0]+len(delim):].find(delim))
+
+	definition_data = definition_section(input_string[:delims[0]])
+
+	print(header(header_str, definition_data["language"]))
+	print(definition_data["code"])
+	print(rule_section(input_string[delims[0]+len(delim):delims[1]]))
+	print(input_string[delims[1]+len(delim):])
+
+	return 0
+
+# ------------------------------
+# --- STRING LITERAL SECTION ---
+# ------------------------------
+
+header_str = '''
+#include <stdio.h>
+#include <string.h>
+#include <tree_sitter/api.h>
+
+extern "C" const TSLanguage * tree_sitter_{language}(void);
+
+const TSLanguage * (*tree_sitter_language_function)(void) = tree_sitter_{language};
+'''
+
+tbtraverse = '''
+int tbtraverse(const char * const code) {{
+    // init
+    TSParser * parser;
+    TSTree * tree;
+    TSTreeCursor cursor;
+    TSNode current_node;
+    TSNode previous_node;
+
+    parser = ts_parser_new();
+
+    ts_parser_set_language(parser, tree_sitter_language_function());
+
+    tree = ts_parser_parse_string(parser, NULL, code, strlen(code));
+    cursor = ts_tree_cursor_new(ts_tree_root_node(tree));
+    current_node = ts_tree_root_node(tree);
+
+    // meat
+    do {{
+        current_node = ts_tree_cursor_current_node(&cursor);
+
+        int tblen = ts_node_end_byte(current_node) - ts_node_start_byte(current_node);
+        char * tbtext = (char *)malloc(sizeof(char) * (tblen + 1));
+        memcpy(tbtext, code + ts_node_start_byte(current_node), tblen);
+        tbtext[tblen] = '\\0';
+
+      #if defined(TBDEBUG) && TBDEBUG == 1
+        puts(ts_node_string(current_node));
+      #endif
+
+        // XXX INJECTION
+        {enter_shims}
+
+      end:
+        free(tbtext);
+    }} while ([&](void) -> bool {{
+        bool r = false;
+
+        if (ts_tree_cursor_goto_first_child(&cursor)) {{
+			return true;
+		}}
+
+        if (ts_tree_cursor_goto_next_sibling(&cursor)) {{
+            r = true;
+			previous_node = current_node;
+            goto eval;
+        }}
+        
+        while (ts_tree_cursor_goto_parent(&cursor)) {{
+			previous_node = ts_tree_cursor_current_node(&cursor);
+            if (ts_tree_cursor_goto_next_sibling(&cursor)) {{
+                r = true;
+            }}
+
+          eval:
+          #if defined(TBDEBUG) && TBDEBUG == 1
+            puts(ts_node_string(previous_node));
+          #endif
+
+            // XXX INJECTION
+            {leave_shims}
+
+          end:
+            if (r) {{ break; }}
+        }}
+
+        return r;
+    }}());
+
+    // deinit
+    ts_tree_delete(tree);
+    ts_parser_delete(parser);
+    ts_tree_cursor_delete(&cursor);
+
+    return 0;
+}}
+'''
+
+shim_enter = '''
+        if (!strcmp("{name}", ts_node_type(current_node))) {{
+            {code}
+            goto end;
+        }}
+
+'''
+
+shim_leave = '''
+        if (!strcmp("{name}", ts_node_type(previous_node))) {{
+            {code}
+            goto end;
+        }}
+'''
+
+if __name__ == '__main__':
+	exit(tbc_main(sys.argv))