d2d01e77e3
This patch updates the CLI to use anyhow and thiserror for error management. The main feature that our custom `Error` type was providing was a _list_ of messages, which would allow us to annotate "lower-level" errors with more contextual information. This is exactly what's provided by anyhow's `Context` trait. (This is setup work for a future PR that will pull the `config` and `loader` modules out into separate crates; by using `anyhow` we wouldn't have to deal with a circular dependency between with the new crates.)
105 lines
3.8 KiB
Rust
105 lines
3.8 KiB
Rust
use crate::query_testing;
|
|
use anyhow::{Context, Result};
|
|
use std::{
|
|
fs,
|
|
io::{self, Write},
|
|
ops::Range,
|
|
path::Path,
|
|
};
|
|
use tree_sitter::{Language, Parser, Query, QueryCursor};
|
|
|
|
pub fn query_files_at_paths(
|
|
language: Language,
|
|
paths: Vec<String>,
|
|
query_path: &Path,
|
|
ordered_captures: bool,
|
|
range: Option<Range<usize>>,
|
|
should_test: bool,
|
|
) -> Result<()> {
|
|
let stdout = io::stdout();
|
|
let mut stdout = stdout.lock();
|
|
|
|
let query_source = fs::read_to_string(query_path)
|
|
.with_context(|| format!("Error reading query file {:?}", query_path))?;
|
|
let query = Query::new(language, &query_source).with_context(|| "Query compilation failed")?;
|
|
|
|
let mut query_cursor = QueryCursor::new();
|
|
if let Some(range) = range {
|
|
query_cursor.set_byte_range(range);
|
|
}
|
|
|
|
let mut parser = Parser::new();
|
|
parser.set_language(language)?;
|
|
|
|
for path in paths {
|
|
let mut results = Vec::new();
|
|
|
|
writeln!(&mut stdout, "{}", path)?;
|
|
|
|
let source_code =
|
|
fs::read(&path).with_context(|| format!("Error reading source file {:?}", path))?;
|
|
let tree = parser.parse(&source_code, None).unwrap();
|
|
|
|
if ordered_captures {
|
|
for (mat, capture_index) in
|
|
query_cursor.captures(&query, tree.root_node(), source_code.as_slice())
|
|
{
|
|
let capture = mat.captures[capture_index];
|
|
let capture_name = &query.capture_names()[capture.index as usize];
|
|
writeln!(
|
|
&mut stdout,
|
|
" pattern: {}, capture: {}, row: {}, text: {:?}",
|
|
mat.pattern_index,
|
|
capture_name,
|
|
capture.node.start_position().row,
|
|
capture.node.utf8_text(&source_code).unwrap_or("")
|
|
)?;
|
|
results.push(query_testing::CaptureInfo {
|
|
name: capture_name.to_string(),
|
|
start: capture.node.start_position(),
|
|
end: capture.node.end_position(),
|
|
});
|
|
}
|
|
} else {
|
|
for m in query_cursor.matches(&query, tree.root_node(), source_code.as_slice()) {
|
|
writeln!(&mut stdout, " pattern: {}", m.pattern_index)?;
|
|
for capture in m.captures {
|
|
let start = capture.node.start_position();
|
|
let end = capture.node.end_position();
|
|
let capture_name = &query.capture_names()[capture.index as usize];
|
|
if end.row == start.row {
|
|
writeln!(
|
|
&mut stdout,
|
|
" capture: {}, start: {}, text: {:?}",
|
|
capture_name,
|
|
start,
|
|
capture.node.utf8_text(&source_code).unwrap_or("")
|
|
)?;
|
|
} else {
|
|
writeln!(
|
|
&mut stdout,
|
|
" capture: {}, start: {}, end: {}",
|
|
capture_name, start, end,
|
|
)?;
|
|
}
|
|
results.push(query_testing::CaptureInfo {
|
|
name: capture_name.to_string(),
|
|
start: capture.node.start_position(),
|
|
end: capture.node.end_position(),
|
|
});
|
|
}
|
|
}
|
|
}
|
|
if query_cursor.did_exceed_match_limit() {
|
|
writeln!(
|
|
&mut stdout,
|
|
" WARNING: Query exceeded maximum number of in-progress captures!"
|
|
)?;
|
|
}
|
|
if should_test {
|
|
query_testing::assert_expected_captures(results, path, &mut parser, language)?
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|