srcwalk 0.2.4

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

use std::path::{Path, PathBuf};

use crate::types::Match;

/// Faceted search results grouped by match type and location.
pub struct FacetedResult {
    pub definitions: Vec<Match>,
    pub implementations: Vec<Match>,
    pub bases: Vec<Match>,
    pub tests: Vec<Match>,
    pub comments: Vec<Match>,
    pub usages_local: Vec<Match>,
    pub usages_cross: Vec<Match>,
}

/// Group matches into facets when there are many results (>5).
/// Partitions by definition type, test status, and package locality.
pub fn facet_matches(matches: Vec<Match>, _scope: &Path) -> FacetedResult {
    // Find primary definition's package root for local/cross determination
    let primary_pkg = matches
        .iter()
        .find(|m| m.is_definition)
        .and_then(|m| m.path.parent())
        .and_then(package_root)
        .map(std::path::Path::to_path_buf);

    let mut definitions = Vec::new();
    let mut implementations = Vec::new();
    let mut bases = Vec::new();
    let mut tests = Vec::new();
    let mut comments = Vec::new();
    let mut usages_local = Vec::new();
    let mut usages_cross = Vec::new();

    for m in matches {
        if m.is_definition && m.impl_target.is_some() {
            implementations.push(m);
        } else if m.is_definition && m.base_target.is_some() {
            bases.push(m);
        } else if m.is_definition {
            definitions.push(m);
        } else if m.in_comment {
            comments.push(m);
        } else if is_test_match(&m) {
            tests.push(m);
        } else if is_same_package(&m.path, primary_pkg.as_ref()) {
            usages_local.push(m);
        } else {
            usages_cross.push(m);
        }
    }

    FacetedResult {
        definitions,
        implementations,
        bases,
        tests,
        comments,
        usages_local,
        usages_cross,
    }
}

/// Check if a match is in a test file or contains test markers.
fn is_test_match(m: &Match) -> bool {
    // Path-based detection
    let path_str = m.path.to_string_lossy();
    if path_str.contains("_test.")
        || path_str.contains("/test/")
        || path_str.contains("/tests/")
        || path_str.contains("_spec.")
        || path_str.contains("/spec/")
    {
        return true;
    }

    // Content-based detection
    let text = &m.text;
    text.contains("#[test]")
        || text.contains("#[cfg(test)]")
        || text.contains("@Test")
        || text.contains("def test_")
        || text.contains("it(\"")
        || text.contains("it('")
        || text.contains("describe(\"")
        || text.contains("describe('")
        || text.contains("func Test")
}

/// Check if path is in the same package as the primary definition.
fn is_same_package(path: &Path, primary_pkg: Option<&PathBuf>) -> bool {
    let Some(pkg_root) = primary_pkg else {
        return false;
    };

    path.parent()
        .and_then(package_root)
        .is_some_and(|p| p == pkg_root.as_path())
}

/// Re-export from lang module.
fn package_root(path: &Path) -> Option<&Path> {
    crate::lang::package_root(path)
}