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

rust: In queries, allow set! w/ capture and constant

Browse source
This commit is contained in:
Max Brunsfeld 2020-02-14 20:50:07 -08:00
parent 56e52c76f9
commit b2e79f6438
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
lib/binding_rust/lib.rs
View file

@ -1439,12 +1439,17 @@ impl Query {
let mut value = None;
if i < args.len() {
if args[i].type_ == ffi::TSQueryPredicateStepType_TSQueryPredicateStepTypeCapture {
return Err(QueryError::Predicate(format!(
"Invalid arguments to {} predicate. Expected string, got @{}",
function_name, capture_names[args[i].value_id as usize]
)));
if capture_id.is_some() {
return Err(QueryError::Predicate(format!(
"Invalid arguments to {} predicate. Unexpected second capture name @{}",
function_name, capture_names[args[i].value_id as usize]
)));
} else {
capture_id = Some(args[i].value_id as usize);
}
} else {
value = Some(string_values[args[i].value_id as usize].as_str());
}
value = Some(string_values[args[i].value_id as usize].as_str());
}
Ok(QueryProperty::new(key, value, capture_id))