Skip to main content

lean_ctx/tools/registered/
ctx_heatmap.rs

1use rmcp::model::Tool;
2use rmcp::ErrorData;
3use serde_json::{json, Map, Value};
4
5use crate::server::tool_trait::{get_str, McpTool, ToolContext, ToolOutput};
6use crate::tool_defs::tool_def;
7
8pub struct CtxHeatmapTool;
9
10impl McpTool for CtxHeatmapTool {
11    fn name(&self) -> &'static str {
12        "ctx_heatmap"
13    }
14
15    fn tool_def(&self) -> Tool {
16        tool_def(
17            "ctx_heatmap",
18            "File access heatmap — shows most frequently accessed files.",
19            json!({
20                "type": "object",
21                "properties": {
22                    "action": { "type": "string", "description": "status|detail" },
23                    "path": { "type": "string" }
24                }
25            }),
26        )
27    }
28
29    fn handle(
30        &self,
31        args: &Map<String, Value>,
32        _ctx: &ToolContext,
33    ) -> Result<ToolOutput, ErrorData> {
34        let action = get_str(args, "action").unwrap_or_else(|| "status".to_string());
35        let path = get_str(args, "path");
36        let result = crate::tools::ctx_heatmap::handle(&action, path.as_deref());
37        Ok(ToolOutput {
38            text: result,
39            original_tokens: 0,
40            saved_tokens: 0,
41            mode: Some(action),
42            path: None,
43        })
44    }
45}