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

Fix failure to match queries with wildcard at root with range restrictions

Browse source
This commit is contained in:
Max Brunsfeld 2022-06-22 15:54:06 -07:00
parent 46a2a5933f
commit 58b719541b
2 changed files with 66 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
lib/src/query.c
View file

@ -2631,8 +2631,8 @@ TSQuery *ts_query_new(
// Determine whether the pattern has a single root node. This affects
// decisions about whether or not to start matching the pattern when
// a query cursor has a range restriction.
bool is_rooted = true;
uint32_t start_depth = step->depth;
bool is_rooted = start_depth == 0;
for (uint32_t step_index = start_step_index + 1; step_index < self->steps.size; step_index++) {
QueryStep *step = &self->steps.contents[step_index];
if (step->depth == start_depth) {
@ -3318,7 +3318,6 @@ static inline bool ts_query_cursor__advance(
point_gt(ts_node_end_point(node), self->start_point) &&
point_lt(ts_node_start_point(node), self->end_point)
);
bool parent_intersects_range = ts_node_is_null(parent_node) || (
ts_node_end_byte(parent_node) > self->start_byte &&
ts_node_start_byte(parent_node) < self->end_byte &&
@ -3326,7 +3325,7 @@ static inline bool ts_query_cursor__advance(
point_lt(ts_node_start_point(parent_node), self->end_point)
);
// Add new states for any patterns whose root node is a wildcard.
// Add new states for any patterns whose root node is a wildcard.
for (unsigned i = 0; i < self->query->wildcard_root_pattern_count; i++) {
PatternEntry *pattern = &self->query->pattern_map.contents[i];
@ -3334,7 +3333,7 @@ static inline bool ts_query_cursor__advance(
// state at the start of this pattern.
QueryStep *step = &self->query->steps.contents[pattern->step_index];
if (
(node_intersects_range || (!pattern->is_rooted && parent_intersects_range)) &&
(pattern->is_rooted ? node_intersects_range : parent_intersects_range) &&
(!step->field || field_id == step->field) &&
(!step->supertype_symbol || supertype_count > 0)
) {
@ -3352,7 +3351,7 @@ static inline bool ts_query_cursor__advance(
// If this node matches the first step of the pattern, then add a new
// state at the start of this pattern.
if (
(node_intersects_range || (!pattern->is_rooted && parent_intersects_range)) &&
(pattern->is_rooted ? node_intersects_range : parent_intersects_range) &&
(!step->field || field_id == step->field)
) {
ts_query_cursor__add_state(self, pattern);