traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(lib): properly account for aliased root nodes and root nodes with

Browse source 
children in `ts_subtree_string`
This commit is contained in:
Amaan Qureshi 2024-03-18 02:50:01 -04:00
parent ab485da756
commit 24a68697ac
3 changed files with 21 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

16
lib/src/subtree.c
View file

@ -890,9 +890,15 @@ static size_t ts_subtree__write_to_string(
}
}
} else if (is_root) {
TSSymbol symbol = ts_subtree_symbol(self);
TSSymbol symbol = alias_symbol ? alias_symbol : ts_subtree_symbol(self);
const char *symbol_name = ts_language_symbol_name(language, symbol);
cursor += snprintf(*writer, limit, "(\"%s\")", symbol_name);
if (ts_subtree_child_count(self) > 0) {
cursor += snprintf(*writer, limit, "(%s", symbol_name);
} else if (ts_subtree_named(self)) {
cursor += snprintf(*writer, limit, "(%s)", symbol_name);
} else {
cursor += snprintf(*writer, limit, "(\"%s\")", symbol_name);
}
}
if (ts_subtree_child_count(self)) {
@ -947,6 +953,8 @@ static size_t ts_subtree__write_to_string(
char *ts_subtree_string(
Subtree self,
TSSymbol alias_symbol,
bool alias_is_named,
const TSLanguage *language,
bool include_all
) {
@ -954,13 +962,13 @@ char *ts_subtree_string(
size_t size = ts_subtree__write_to_string(
self, scratch_string, 1,
language, include_all,
0, false, ROOT_FIELD
alias_symbol, alias_is_named, ROOT_FIELD
) + 1;
char *result = ts_malloc(size * sizeof(char));
ts_subtree__write_to_string(
self, result, size,
language, include_all,
0, false, ROOT_FIELD
alias_symbol, alias_is_named, ROOT_FIELD
);
return result;
}