Hide TSTree, expose TSNode

src/runtime/node.c

@@ -0,0 +1,95 @@
+#include "runtime/node.h"
+#include "tree_sitter/parser.h"
+#include "runtime/tree.h"
+
+TSNode * ts_node_make(TSTree *tree, TSNode *parent, size_t position, TSParserConfig *config) {
+    if (parent) ts_node_retain(parent);
+    TSNode *result = malloc(sizeof(TSNode));
+    *result = (TSNode) {
+        .ref_count = 1,
+        .parent = parent,
+        .content = tree,
+        .position = position,
+        .config = config
+    };
+    return result;
+}
+
+void ts_node_retain(TSNode *node) {
+    node->ref_count++;
+}
+
+void ts_node_release(TSNode *node) {
+    node->ref_count--;
+    if (node->ref_count == 0) {
+        if (node->parent) ts_node_release(node->parent);
+        free(node);
+    }
+}
+
+size_t ts_node_pos(const TSNode *node) {
+    return node->position;
+}
+
+size_t ts_node_size(const TSNode *node) {
+    return node->content->size;
+}
+
+const char * ts_node_name(const TSNode *node) {
+    return node->config->symbol_names[node->content->symbol];
+}
+
+TSTree * ts_tree_child(const TSTree *tree, size_t goal_index, size_t *offset, size_t *children_seen) {
+    *offset = 0;
+    *children_seen = 0;
+    size_t child_count;
+    TSTree **children = ts_tree_children(tree, &child_count);
+
+    if (!children) {
+        *offset = tree->offset + tree->size;
+    }
+
+    for (size_t i = 0; i < child_count; i++) {
+        size_t delta_index = 0, delta_offset = 0;
+        TSTree *child = children[i];
+        TSTree *result = NULL;
+
+        if (ts_tree_is_visible(child)) {
+            delta_offset = child->offset;
+            if (*children_seen == goal_index) {
+                result = child;
+            } else {
+                delta_index = 1;
+                delta_offset += child->size;
+            }
+        } else {
+            result = ts_tree_child(child, (goal_index - *children_seen), &delta_offset, &delta_index);
+        }
+
+        *offset += delta_offset;
+        *children_seen += delta_index;
+
+        if (result) {
+            return result;
+        }
+    }
+
+    return NULL;
+}
+
+const char * ts_node_string(const TSNode *node) {
+    return ts_tree_string(node->content, node->config->symbol_names);
+}
+
+TSNode * ts_node_child(TSNode *parent, size_t goal_index) {
+    size_t offset, index;
+    TSTree *child = ts_tree_child(parent->content, goal_index, &offset, &index);
+    if (child)
+        return ts_node_make(child, parent, offset, parent->config);
+    else
+        return NULL;
+}
+
+TSNode * ts_node_leaf_at_pos(TSNode *parent, size_t child_index) {
+    return NULL;
+}