lean_ctx/tools/
ctx_heatmap.rs

1use crate::core::gain::GainEngine;
2use crate::core::heatmap;
3
4pub fn handle(action: &str, _path: Option<&str>) -> String {
5    let engine = GainEngine::load();
6
7    match action {
8        "directory" | "dirs" => heatmap::format_directory_summary(&engine.heatmap),
9        "cold" => {
10            let all = collect_project_files(_path);
11            let cold = engine.heatmap.cold_files(&all, 20);
12            if cold.is_empty() {
13                "No cold files found (all files have been accessed).".to_string()
14            } else {
15                let mut lines = vec![format!(
16                    "Cold files (never accessed, {} total):",
17                    cold.len()
18                )];
19                for f in &cold {
20                    lines.push(format!("  {f}"));
21                }
22                lines.join("\n")
23            }
24        }
25        "json" => {
26            serde_json::to_string_pretty(&engine.heatmap).unwrap_or_else(|_| "{}".to_string())
27        }
28        _ => heatmap::format_heatmap_status(&engine.heatmap, 20),
29    }
30}
31
32fn collect_project_files(path: Option<&str>) -> Vec<String> {
33    let root = path.unwrap_or(".");
34    let mut files = Vec::new();
35    let walker = walkdir::WalkDir::new(root)
36        .max_depth(5)
37        .into_iter()
38        .filter_entry(|e| {
39            let name = e.file_name().to_string_lossy();
40            !name.starts_with('.')
41                && name != "node_modules"
42                && name != "target"
43                && name != "dist"
44                && name != "__pycache__"
45                && name != ".git"
46        });
47    for entry in walker.flatten() {
48        if entry.file_type().is_file() {
49            if let Some(ext) = entry.path().extension().and_then(|e| e.to_str()) {
50                if is_source_ext(ext) {
51                    files.push(entry.path().to_string_lossy().to_string());
52                }
53            }
54        }
55    }
56    files
57}
58
59fn is_source_ext(ext: &str) -> bool {
60    matches!(
61        ext,
62        "rs" | "ts"
63            | "tsx"
64            | "js"
65            | "jsx"
66            | "py"
67            | "go"
68            | "java"
69            | "c"
70            | "cpp"
71            | "h"
72            | "rb"
73            | "cs"
74            | "kt"
75            | "swift"
76            | "php"
77            | "svelte"
78            | "vue"
79    )
80}
lean_ctx/tools/ctx_heatmap.rs

lean_ctx/tools/
ctx_heatmap.rs
