repopilot 0.12.0

Local-first CLI for repository audit, architecture risk detection, baseline tracking, and CI-friendly code review.
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

use std::collections::HashSet;
use tree_sitter::{Node, Parser};

pub(super) fn extract(content: &str) -> HashSet<String> {
    let Some(tree) = parse(content) else {
        return HashSet::new();
    };

    let mut result = HashSet::new();
    visit(tree.root_node(), content, &mut result);
    result
}

fn parse(content: &str) -> Option<tree_sitter::Tree> {
    let mut parser = Parser::new();
    parser
        .set_language(&tree_sitter_python::LANGUAGE.into())
        .ok()?;
    parser.parse(content, None)
}

fn visit(node: Node<'_>, content: &str, result: &mut HashSet<String>) {
    if node.kind() == "import_from_statement"
        && let Some(module) = relative_from_module(node, content)
    {
        result.insert(module);
    }

    let mut cursor = node.walk();
    for child in node.children(&mut cursor) {
        visit(child, content, result);
    }
}

fn relative_from_module(node: Node<'_>, content: &str) -> Option<String> {
    let text = node.utf8_text(content.as_bytes()).ok()?;
    let normalized = text.split_whitespace().collect::<Vec<_>>().join(" ");
    let rest = normalized.strip_prefix("from ")?;
    let import_pos = rest.find(" import ")?;
    let module = rest[..import_pos].trim();
    module.starts_with('.').then(|| module.to_string())
}