gitcortex_store/
schema.rs

1use gitcortex_core::error::{GitCortexError, Result};
2use kuzu::Connection;
3
4use crate::branch;
5
6// ── Table name helpers ────────────────────────────────────────────────────────
7
8/// Node table name for a branch: `{sanitized_branch}_nodes`
9pub fn node_table(branch: &str) -> String {
10    format!("{}_nodes", branch::sanitize(branch))
11}
12
13/// Relationship table name for a branch: `{sanitized_branch}_edges`
14pub fn edge_table(branch: &str) -> String {
15    format!("{}_edges", branch::sanitize(branch))
16}
17
18// ── DDL ───────────────────────────────────────────────────────────────────────
19
20/// Create the node and edge tables for `branch` if they don't already exist.
21/// Safe to call on every operation — idempotent.
22pub fn ensure_branch(conn: &mut Connection, branch: &str) -> Result<()> {
23    let nt = node_table(branch);
24    let et = edge_table(branch);
25
26    conn.query(&format!(
27        "CREATE NODE TABLE IF NOT EXISTS {nt} (\
28            id             STRING, \
29            kind           STRING, \
30            name           STRING, \
31            qualified_name STRING, \
32            file           STRING, \
33            start_line     INT64,  \
34            end_line       INT64,  \
35            loc            INT64,  \
36            visibility     STRING, \
37            is_async       BOOL,   \
38            is_unsafe      BOOL,   \
39            is_static      BOOL,   \
40            is_abstract    BOOL,   \
41            is_final       BOOL,   \
42            is_property    BOOL,   \
43            is_generator   BOOL,   \
44            is_const       BOOL,   \
45            generic_bounds STRING, \
46            def_signature  STRING, \
47            def_body       STRING, \
48            def_doc        STRING, \
49            def_start_byte INT64,  \
50            def_end_byte   INT64,  \
51            PRIMARY KEY(id)\
52        )"
53    ))
54    .map_err(|e| GitCortexError::Store(format!("create node table: {e}")))?;
55
56    conn.query(&format!(
57        "CREATE REL TABLE IF NOT EXISTS {et} (\
58            FROM {nt} TO {nt},\
59            kind STRING\
60        )"
61    ))
62    .map_err(|e| GitCortexError::Store(format!("create edge table: {e}")))?;
63
64    Ok(())
65}
gitcortex_store/schema.rs

gitcortex_store/
schema.rs
