lean_ctx/core/
knowledge_bootstrap.rs

1use std::path::Path;
2
3use crate::core::graph_index::ProjectIndex;
4use crate::core::knowledge::ProjectKnowledge;
5use crate::core::memory_policy::MemoryPolicy;
6
7const BOOTSTRAP_SESSION_ID: &str = "auto-bootstrap";
8const BOOTSTRAP_CONFIDENCE: f32 = 0.95;
9
10/// Seed a minimal set of *real*, deterministic facts so the dashboard Knowledge Graph
11/// is never empty on a new project.
12///
13/// This does not use placeholders — it only derives values from the filesystem and/or index.
14pub fn bootstrap_if_empty(
15    knowledge: &mut ProjectKnowledge,
16    project_root: &str,
17    index: Option<&ProjectIndex>,
18    policy: &MemoryPolicy,
19) -> bool {
20    if !knowledge.facts.is_empty() {
21        return false;
22    }
23
24    let mut changed = false;
25
26    // Always safe + real: makes the graph non-empty even for marker-less folders.
27    changed |= remember_fact(knowledge, "workflow", "project_root", project_root, policy);
28
29    if let Some(identity) = crate::core::project_hash::project_identity(project_root) {
30        changed |= remember_fact(
31            knowledge,
32            "architecture",
33            "project_identity",
34            &identity,
35            policy,
36        );
37        if let Some(url) = identity.strip_prefix("git:") {
38            changed |= remember_fact(knowledge, "deployment", "git_remote", url, policy);
39        }
40    }
41
42    let markers = detect_build_markers(project_root);
43    if !markers.is_empty() {
44        changed |= remember_fact(
45            knowledge,
46            "architecture",
47            "build_markers",
48            &markers.join(", "),
49            policy,
50        );
51    }
52
53    if let Some(idx) = index {
54        let file_count = idx.files.len();
55        let symbol_count = idx.symbols.len();
56        let edge_count = idx.edges.len();
57        changed |= remember_fact(
58            knowledge,
59            "workflow",
60            "index_stats",
61            &format!("files={file_count}, symbols={symbol_count}, edges={edge_count}"),
62            policy,
63        );
64
65        if !idx.last_scan.trim().is_empty() {
66            changed |= remember_fact(
67                knowledge,
68                "workflow",
69                "index_last_scan",
70                &idx.last_scan,
71                policy,
72            );
73        }
74
75        let (langs, total_tokens) = summarize_languages_and_tokens(idx);
76        if !langs.is_empty() {
77            changed |= remember_fact(knowledge, "architecture", "languages_top", &langs, policy);
78        }
79        if total_tokens > 0 {
80            changed |= remember_fact(
81                knowledge,
82                "performance",
83                "tokens_indexed",
84                &total_tokens.to_string(),
85                policy,
86            );
87        }
88    }
89
90    changed
91}
92
93fn remember_fact(
94    knowledge: &mut ProjectKnowledge,
95    category: &str,
96    key: &str,
97    value: &str,
98    policy: &MemoryPolicy,
99) -> bool {
100    if value.trim().is_empty() {
101        return false;
102    }
103    knowledge.remember(
104        category,
105        key,
106        value,
107        BOOTSTRAP_SESSION_ID,
108        BOOTSTRAP_CONFIDENCE,
109        policy,
110    );
111    true
112}
113
114fn detect_build_markers(project_root: &str) -> Vec<&'static str> {
115    let root = Path::new(project_root);
116    let mut out: Vec<&'static str> = Vec::new();
117
118    // macOS TCC (#356): a launchd-standalone process must not stat or read_dir
119    // under ~/Documents/Desktop/Downloads. Skip detection (no markers) without
120    // touching the filesystem; editor/CLI-attached runs detect normally.
121    if !crate::core::pathutil::may_probe_path(root) {
122        return out;
123    }
124
125    if root.join(".git").exists() {
126        out.push("git");
127    }
128    if root.join("Cargo.toml").exists() {
129        out.push("cargo");
130    }
131    if root.join("package.json").exists() {
132        out.push("npm");
133    }
134    if root.join("pyproject.toml").exists() {
135        out.push("python");
136    }
137    if root.join("go.mod").exists() {
138        out.push("go");
139    }
140    if root.join("pom.xml").exists() {
141        out.push("maven");
142    }
143    if root.join("build.gradle").exists() || root.join("build.gradle.kts").exists() {
144        out.push("gradle");
145    }
146    if root.join("CMakeLists.txt").exists() {
147        out.push("cmake");
148    }
149
150    if let Ok(entries) = std::fs::read_dir(root) {
151        if entries
152            .flatten()
153            .any(|e| e.path().extension().is_some_and(|ext| ext == "sln"))
154        {
155            out.push("dotnet");
156        }
157    }
158
159    out
160}
161
162fn summarize_languages_and_tokens(index: &ProjectIndex) -> (String, u64) {
163    let mut total_tokens: u64 = 0;
164    let mut counts: std::collections::HashMap<&str, usize> = std::collections::HashMap::new();
165
166    for f in index.files.values() {
167        total_tokens = total_tokens.saturating_add(f.token_count as u64);
168        let lang = if f.language.trim().is_empty() {
169            "unknown"
170        } else {
171            f.language.as_str()
172        };
173        *counts.entry(lang).or_insert(0) += 1;
174    }
175
176    let mut entries: Vec<(&str, usize)> = counts.into_iter().collect();
177    entries.sort_by(|a, b| b.1.cmp(&a.1).then_with(|| a.0.cmp(b.0)));
178
179    let langs = entries
180        .into_iter()
181        .take(6)
182        .map(|(lang, count)| format!("{lang}:{count}"))
183        .collect::<Vec<_>>()
184        .join(", ");
185
186    (langs, total_tokens)
187}
lean_ctx/core/knowledge_bootstrap.rs

lean_ctx/core/
knowledge_bootstrap.rs
