infigraph-core 1.0.0

AST-powered code analysis framework — parser, graph, diff, and analysis engine
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

pub const MIGRATIONS: &[&str] = &[
    "ALTER TABLE Symbol ADD parameters STRING DEFAULT ''",
    "ALTER TABLE Symbol ADD return_type STRING DEFAULT ''",
    "CREATE NODE TABLE IF NOT EXISTS Statement(id STRING, kind STRING, condition STRING, start_line INT32, end_line INT32, depth INT32, parent_symbol STRING, PRIMARY KEY(id))",
    "CREATE REL TABLE IF NOT EXISTS HAS_STATEMENT(FROM Symbol TO Statement)",
];

/// Kuzu schema DDL for the infigraph graph.
pub const CREATE_SCHEMA: &[&str] = &[
    // Node tables
    "CREATE NODE TABLE IF NOT EXISTS Symbol(
        id STRING,
        name STRING,
        kind STRING,
        file STRING,
        start_line INT32,
        end_line INT32,
        signature_hash STRING,
        language STRING,
        visibility STRING,
        parent STRING,
        docstring STRING,
        complexity INT32,
        parameters STRING,
        return_type STRING,
        embedding FLOAT[],
        PRIMARY KEY(id)
    )",
    "CREATE NODE TABLE IF NOT EXISTS Module(
        id STRING,
        name STRING,
        file STRING,
        language STRING,
        content_hash STRING,
        summary STRING,
        PRIMARY KEY(id)
    )",
    "CREATE NODE TABLE IF NOT EXISTS Cluster(
        id STRING,
        name STRING,
        description STRING,
        PRIMARY KEY(id)
    )",
    "CREATE NODE TABLE IF NOT EXISTS File(
        id STRING,
        name STRING,
        path STRING,
        language STRING,
        symbol_count INT32,
        PRIMARY KEY(id)
    )",
    "CREATE NODE TABLE IF NOT EXISTS Folder(
        id STRING,
        name STRING,
        path STRING,
        PRIMARY KEY(id)
    )",
    "CREATE NODE TABLE IF NOT EXISTS Dependency(
        id STRING,
        name STRING,
        version STRING,
        ecosystem STRING,
        is_dev BOOLEAN,
        PRIMARY KEY(id)
    )",
    "CREATE NODE TABLE IF NOT EXISTS Statement(
        id STRING,
        kind STRING,
        condition STRING,
        start_line INT32,
        end_line INT32,
        depth INT32,
        parent_symbol STRING,
        PRIMARY KEY(id)
    )",
    // Relationship tables
    "CREATE REL TABLE IF NOT EXISTS CALLS(FROM Symbol TO Symbol)",
    "CREATE REL TABLE IF NOT EXISTS DEPENDS_ON(FROM Module TO Dependency, is_dev BOOLEAN)",
    "CREATE REL TABLE IF NOT EXISTS IMPORTS(FROM Module TO Module)",
    "CREATE REL TABLE IF NOT EXISTS CONTAINS(FROM Module TO Symbol)",
    "CREATE REL TABLE IF NOT EXISTS INHERITS(FROM Symbol TO Symbol)",
    "CREATE REL TABLE IF NOT EXISTS TESTED_BY(FROM Symbol TO Symbol)",
    "CREATE REL TABLE IF NOT EXISTS READS(FROM Symbol TO Symbol)",
    "CREATE REL TABLE IF NOT EXISTS WRITES(FROM Symbol TO Symbol)",
    "CREATE REL TABLE IF NOT EXISTS MEMBER_OF(FROM Symbol TO Cluster)",
    "CREATE REL TABLE IF NOT EXISTS SIMILAR_TO(FROM Symbol TO Symbol, score FLOAT)",
    "CREATE REL TABLE IF NOT EXISTS BRIDGE_TO(FROM Symbol TO Symbol, bridge_kind STRING, detail STRING)",
    "CREATE REL TABLE IF NOT EXISTS CONTAINS_FILE(FROM Folder TO File)",
    "CREATE REL TABLE IF NOT EXISTS CONTAINS_FOLDER(FROM Folder TO Folder)",
    "CREATE REL TABLE IF NOT EXISTS DEFINES(FROM File TO Symbol)",
    "CREATE REL TABLE IF NOT EXISTS CALLS_SERVICE(FROM Symbol TO Symbol, method STRING, path STRING, target_service STRING)",
    "CREATE REL TABLE IF NOT EXISTS HAS_STATEMENT(FROM Symbol TO Statement)",
];

use kuzu::Connection;

pub fn ensure_custom_edge_table(conn: &Connection<'_>, edge_name: &str) -> anyhow::Result<()> {
    let ddl = format!(
        "CREATE REL TABLE IF NOT EXISTS {}(FROM Symbol TO Symbol)",
        edge_name
    );
    match conn.query(&ddl) {
        Ok(_) => Ok(()),
        Err(e) => {
            let msg = format!("{e}");
            if msg.contains("already exists") {
                Ok(())
            } else {
                Err(anyhow::anyhow!(
                    "failed to create custom edge table '{}': {}",
                    edge_name,
                    e
                ))
            }
        }
    }
}