rustcodegraph 1.2.10

Local-first code intelligence graph and CLI 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

use rustcodegraph::extraction::tree_sitter::extract_from_source;
use rustcodegraph::types::{Language, NodeKind};

#[test]
fn rust_chained_router_extraction_does_not_clone_ast_subtrees_exponentially() {
    let mut source = String::from(
        r#"
use axum::routing::post;
use axum::Router;

pub(crate) fn router<S>() -> Router<S>
where
    S: Clone + Send + Sync + 'static,
{
    Router::new()
"#,
    );
    for index in 0..24 {
        source.push_str(&format!(
            "        .route(\"/task-pool/{index}\", post(handle_{index}))\n"
        ));
    }
    source.push_str("}\n\n");
    for index in 0..24 {
        source.push_str(&format!(
            r#"
async fn handle_{index}() -> Result<(), ApiError> {{
    Ok(())
}}
"#
        ));
    }

    let result = extract_from_source(
        "crates/app/src/task_pool.rs",
        &source,
        Some(Language::Rust),
        None,
    );

    assert!(result.errors.is_empty(), "{:?}", result.errors);
    let functions = result
        .nodes
        .iter()
        .filter(|node| node.kind == NodeKind::Function)
        .count();
    assert!(
        functions >= 25,
        "expected router plus handlers, got {functions} functions"
    );
}