rust-relations-explorer 0.1.2

Explore relationships in Rust codebases: build a knowledge graph and run queries (connected files, function usage, cycles, paths, hubs, module centrality, trait impls).
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
121
122
123
124
125
126
127
128
129
130

use rust_relations_explorer::graph::{
    FileNode, Item, ItemId, ItemType, KnowledgeGraph, Location, Relationship, RelationshipType,
    Visibility,
};
use rust_relations_explorer::query::{
    CentralityMetric, HubsQuery, ModuleCentralityQuery, Query, ShortestPathQuery,
};
use std::path::{Path, PathBuf};
use std::sync::Arc;

fn make_fn(path: &Path, name: &str) -> Item {
    Item {
        id: ItemId(format!("fn:{}:{}", name, path.display())),
        item_type: ItemType::Function { is_async: false, is_const: false },
        name: Arc::from(name),
        visibility: Visibility::Public,
        location: Location { file: path.to_path_buf(), line_start: 1, line_end: 1 },
        attributes: vec![],
    }
}

fn graph_chain() -> KnowledgeGraph {
    // a -> b, b -> c, c isolated to module m2 with d
    let mut g = KnowledgeGraph::default();
    let a = PathBuf::from("src/a.rs");
    let b = PathBuf::from("src/b.rs");
    let c = PathBuf::from("src/m1/c.rs");
    let d = PathBuf::from("src/m2/d.rs");

    let ia = make_fn(&a, "fa");
    let ib = make_fn(&b, "fb");
    let ic = make_fn(&c, "fc");
    let id = make_fn(&d, "fd");

    g.files.insert(
        a.clone(),
        FileNode {
            path: a.clone(),
            items: vec![ia.clone()],
            imports: vec![],
            metrics: Default::default(),
        },
    );
    g.files.insert(
        b.clone(),
        FileNode {
            path: b.clone(),
            items: vec![ib.clone()],
            imports: vec![],
            metrics: Default::default(),
        },
    );
    g.files.insert(
        c.clone(),
        FileNode {
            path: c.clone(),
            items: vec![ic.clone()],
            imports: vec![],
            metrics: Default::default(),
        },
    );
    g.files.insert(
        d.clone(),
        FileNode {
            path: d.clone(),
            items: vec![id.clone()],
            imports: vec![],
            metrics: Default::default(),
        },
    );

    g.relationships.push(Relationship {
        from_item: ia.id.clone(),
        to_item: ib.id.clone(),
        relationship_type: RelationshipType::Calls { call_type: "test".into() },
        strength: 1.0,
        context: String::new(),
    });
    g.relationships.push(Relationship {
        from_item: ib.id.clone(),
        to_item: ic.id.clone(),
        relationship_type: RelationshipType::Calls { call_type: "test".into() },
        strength: 1.0,
        context: String::new(),
    });
    // d has no edges, different module directory

    g
}

#[test]
fn shortest_path_found_and_absent() {
    let g = graph_chain();
    // Path a -> b -> c exists
    let p = ShortestPathQuery::new("src/a.rs", "src/m1/c.rs").run(&g);
    assert_eq!(p.first().unwrap().display().to_string(), "src/a.rs");
    assert_eq!(p.last().unwrap().display().to_string(), "src/m1/c.rs");

    // No path from c to a
    let none = ShortestPathQuery::new("src/m1/c.rs", "src/a.rs").run(&g);
    assert!(none.is_empty());
}

#[test]
fn hubs_query_metrics_and_sorting() {
    let g = graph_chain();
    // Total metric: b should have total 2 (1 in, 1 out) and be near top
    let total = HubsQuery::new(CentralityMetric::Total, 10).run(&g);
    assert!(total.iter().any(|(p, i, o)| p.ends_with("src/b.rs") && *i == 1 && *o == 1));

    // In metric: c has indegree 1, d has 0
    let ins = HubsQuery::new(CentralityMetric::In, 10).run(&g);
    assert!(ins.iter().any(|(p, i, _)| p.ends_with("src/m1/c.rs") && *i == 1));

    // Out metric: a has outdegree 1
    let outs = HubsQuery::new(CentralityMetric::Out, 10).run(&g);
    assert!(outs.iter().any(|(p, _, o)| p.ends_with("src/a.rs") && *o == 1));
}

#[test]
fn module_centrality_inter_module_edges() {
    let g = graph_chain();
    // Expect edges counted between parent directories of files
    let rows = ModuleCentralityQuery::new(CentralityMetric::Total, 10).run(&g);
    // At least two modules should appear
    assert!(rows.len() >= 2);
    // Ensure modules include src and src/m1, src/m2
    let mods: Vec<String> = rows.iter().map(|(p, _, _)| p.display().to_string()).collect();
    assert!(mods.iter().any(|m| m == "src" || m.ends_with("/src")));
}