srcwalk 0.2.2

Tree-sitter indexed lookups — smart code reading for AI agents
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use std::fs;
use std::process::Command;

fn srcwalk() -> Command {
    Command::new(env!("CARGO_BIN_EXE_srcwalk"))
}

fn temp_dir(name: &str) -> std::path::PathBuf {
    let dir = std::env::temp_dir().join(format!(
        "srcwalk_{name}_{}_{}",
        std::process::id(),
        std::time::SystemTime::now()
            .duration_since(std::time::UNIX_EPOCH)
            .unwrap()
            .as_nanos()
    ));
    fs::create_dir_all(&dir).unwrap();
    dir
}

#[test]
fn flow_filter_slices_ordered_calls_and_resolves_matching_callee() {
    let dir = temp_dir("flow_filter");
    fs::write(
        dir.join("lib.rs"),
        r#"
mod format;

fn entry() {
    let value = helper();
    noisy();
    format();
}

fn helper() -> i32 {
    1
}

fn noisy() {}
"#,
    )
    .unwrap();

    let out = srcwalk()
        .args(["entry", "--flow", "--filter", "callee:helper", "--scope"])
        .arg(&dir)
        .output()
        .unwrap();
    let stdout = String::from_utf8_lossy(&out.stdout);
    let stderr = String::from_utf8_lossy(&out.stderr);

    assert!(
        out.status.success(),
        "flow filter should succeed, stderr:\n{stderr}\nstdout:\n{stdout}"
    );
    assert!(
        stdout.contains("-> calls (ordered, filtered callee:helper)"),
        "expected filtered calls header, got:\n{stdout}"
    );
    assert!(
        stdout.contains("helper()"),
        "expected matching helper call, got:\n{stdout}"
    );
    assert!(
        !stdout.contains("noisy()"),
        "filter should exclude non-matching call, got:\n{stdout}"
    );
    assert!(
        stdout.contains("filter matched 1/3 call sites"),
        "expected filter count footer, got:\n{stdout}"
    );
    assert!(
        stdout.contains("[fn] helper"),
        "expected matching helper resolve, got:\n{stdout}"
    );
}

#[test]
fn flow_resolves_skip_module_like_noise() {
    let dir = temp_dir("flow_resolve_noise");
    fs::write(
        dir.join("lib.rs"),
        r#"
mod format;

fn entry() {
    helper();
    format();
}

fn helper() {}
"#,
    )
    .unwrap();

    let out = srcwalk()
        .args(["entry", "--flow", "--scope"])
        .arg(&dir)
        .output()
        .unwrap();
    let stdout = String::from_utf8_lossy(&out.stdout);
    let stderr = String::from_utf8_lossy(&out.stderr);

    assert!(
        out.status.success(),
        "flow should succeed, stderr:\n{stderr}\nstdout:\n{stdout}"
    );
    assert!(
        stdout.contains("-> resolves (local helpers first)"),
        "expected local-helper resolves heading, got:\n{stdout}"
    );
    assert!(
        stdout.contains("[fn] helper"),
        "expected helper resolve, got:\n{stdout}"
    );
    assert!(
        !stdout.contains("[fn] format"),
        "module-like format resolve should be skipped, got:\n{stdout}"
    );
}