gitcortex 0.1.0

GitCortex CLI and MCP server — branch-aware code knowledge graph for AI assistants
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

use std::path::PathBuf;
use std::time::Instant;

use anyhow::{Context, Result};
use gitcortex_core::store::GraphStore;
use gitcortex_indexer::IncrementalIndexer;
use gitcortex_store::kuzu::KuzuGraphStore;

use crate::cmd::export;

pub fn run(branch_switch: bool) -> Result<()> {
    let t0 = Instant::now();
    let repo_root = repo_root()?;

    if branch_switch {
        // post-checkout: no re-index needed, the target branch has its own graph.
        tracing::debug!(
            elapsed_ms = t0.elapsed().as_millis(),
            "branch-switch: no-op"
        );
        return Ok(());
    }

    let branch = current_branch(&repo_root)?;
    let mut store = KuzuGraphStore::open(&repo_root).context("failed to open graph store")?;

    let last_sha = store.last_indexed_sha(&branch)?;

    let indexer = IncrementalIndexer::new(&repo_root).context("failed to create indexer")?;

    let (diff, head_sha) = indexer
        .run(last_sha.as_deref())
        .context("incremental index failed")?;

    if diff.is_empty() {
        tracing::debug!(elapsed_ms = t0.elapsed().as_millis(), "no changes");
        return Ok(());
    }

    store
        .apply_diff(&branch, &diff)
        .context("failed to apply diff")?;
    store
        .set_last_indexed_sha(&branch, &head_sha)
        .context("failed to persist sha")?;

    let elapsed = t0.elapsed().as_millis();
    let added_n = diff.added_nodes.len();
    let added_e = diff.added_edges.len();
    let removed = diff.removed_files.len();

    // Brief summary visible to the user after every commit.
    eprintln!(
        "gcx  [{branch}]  +{added_n} nodes  +{added_e} edges  -{removed} files  ({elapsed}ms)"
    );

    tracing::info!(
        branch,
        added_nodes = added_n,
        added_edges = added_e,
        removed_files = removed,
        elapsed_ms = elapsed,
        "indexed"
    );
    if elapsed > 500 {
        tracing::warn!(elapsed_ms = elapsed, "hook exceeded 500ms budget");
    }

    // Regenerate context.md if it was previously exported (opt-in).
    export::refresh_if_exists(&repo_root, &store, &branch);

    Ok(())
}

fn repo_root() -> Result<PathBuf> {
    let output = std::process::Command::new("git")
        .args(["rev-parse", "--show-toplevel"])
        .output()
        .context("git rev-parse failed")?;
    Ok(PathBuf::from(
        String::from_utf8(output.stdout)?.trim().to_owned(),
    ))
}

fn current_branch(repo_root: &std::path::Path) -> Result<String> {
    let output = std::process::Command::new("git")
        .args(["symbolic-ref", "--short", "HEAD"])
        .current_dir(repo_root)
        .output()
        .context("git symbolic-ref failed")?;
    if output.status.success() {
        Ok(String::from_utf8(output.stdout)?.trim().to_owned())
    } else {
        let sha = std::process::Command::new("git")
            .args(["rev-parse", "--short", "HEAD"])
            .current_dir(repo_root)
            .output()?;
        Ok(String::from_utf8(sha.stdout)?.trim().to_owned())
    }
}