llmgrep 3.11.0

Smart grep over Magellan code maps with schema-aligned JSON output
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

use crate::cli::{resolve_db_path, Cli};
use llmgrep::backend::Backend;
use llmgrep::error::LlmError;
use llmgrep::output::OutputFormat;

pub fn run_find_ast(cli: &Cli, kind: &str) -> Result<(), LlmError> {
    let db_path = resolve_db_path(cli)?;

    if kind.trim().is_empty() {
        return Err(LlmError::InvalidQuery {
            query: "--kind cannot be empty".to_string(),
        });
    }

    let total_start = std::time::Instant::now();

    let detect_start = std::time::Instant::now();
    let backend = Backend::detect_and_open(&db_path)?;
    let backend_detection_ms = detect_start.elapsed().as_millis() as u64;

    let query_start = std::time::Instant::now();
    let json_value = backend.find_ast(kind)?;
    let query_execution_ms = query_start.elapsed().as_millis() as u64;

    let nodes = if json_value["data"]["nodes"].is_array() {
        json_value["data"]["nodes"].as_array()
    } else {
        json_value["nodes"].as_array()
    };

    if let Some(node_array) = nodes {
        if node_array.is_empty() {
            eprintln!("No AST nodes found with kind '{}'", kind);
            eprintln!("Hint: Check available kinds with: magellan label --list");
            return Ok(());
        }
    }

    let format_start = std::time::Instant::now();
    let rendered = if matches!(cli.output, OutputFormat::Pretty) {
        serde_json::to_string_pretty(&json_value)?
    } else {
        serde_json::to_string(&json_value)?
    };
    let output_formatting_ms = format_start.elapsed().as_millis() as u64;
    let total_ms = total_start.elapsed().as_millis() as u64;

    println!("{}", rendered);

    if cli.show_metrics {
        eprintln!("Performance metrics:");
        eprintln!("  Backend detection: {}ms", backend_detection_ms);
        eprintln!("  Query execution: {}ms", query_execution_ms);
        eprintln!("  Output formatting: {}ms", output_formatting_ms);
        eprintln!("  Total: {}ms", total_ms);
    }

    Ok(())
}