feat(treesitter): upstream get_named_children() as a node method

Util from the nvim-treesitter project.
This commit is contained in:
Quentin Rasmont
2022-04-30 20:54:25 +02:00
committed by bfredl
parent 6b2d42eb03
commit a577fb778a
2 changed files with 44 additions and 0 deletions

View File

@@ -88,6 +88,7 @@ static struct luaL_Reg node_meta[] = {
{ "prev_sibling", node_prev_sibling },
{ "next_named_sibling", node_next_named_sibling },
{ "prev_named_sibling", node_prev_named_sibling },
{ "named_children", node_named_children },
{ NULL, NULL }
};
@@ -1062,6 +1063,31 @@ static int node_prev_named_sibling(lua_State *L)
return 1;
}
static int node_named_children(lua_State *L)
{
TSNode source;
if (!node_check(L, 1, &source)) {
return 0;
}
TSTreeCursor cursor = ts_tree_cursor_new(source);
lua_newtable(L);
int curr_index = 0;
if (ts_tree_cursor_goto_first_child(&cursor)) {
do {
TSNode node = ts_tree_cursor_current_node(&cursor);
if (ts_node_is_named(node)) {
push_node(L, node, 1);
lua_rawseti(L, -2, ++curr_index);
}
} while (ts_tree_cursor_goto_next_sibling(&cursor));
}
ts_tree_cursor_delete(&cursor);
return 1;
}
/// assumes the match table being on top of the stack
static void set_match(lua_State *L, TSQueryMatch *match, int nodeidx)
{