lean_ctx/tools/registered/
ctx_tree.rs

1use rmcp::model::Tool;
2use rmcp::ErrorData;
3use serde_json::{json, Map, Value};
4
5use crate::server::tool_trait::{get_bool, get_int, McpTool, ToolContext, ToolOutput};
6use crate::tool_defs::tool_def;
7
8pub struct CtxTreeTool;
9
10impl McpTool for CtxTreeTool {
11    fn name(&self) -> &'static str {
12        "ctx_tree"
13    }
14
15    fn tool_def(&self) -> Tool {
16        tool_def(
17            "ctx_tree",
18            "Directory listing with file counts.",
19            json!({
20                "type": "object",
21                "properties": {
22                    "path": { "type": "string", "description": "Directory path (default: .)" },
23                    "depth": { "type": "integer", "description": "Max depth (default: 3)" },
24                    "show_hidden": { "type": "boolean", "description": "Show hidden files" }
25                }
26            }),
27        )
28    }
29
30    fn handle(
31        &self,
32        args: &Map<String, Value>,
33        ctx: &ToolContext,
34    ) -> Result<ToolOutput, ErrorData> {
35        let path = ctx.resolved_path("path").unwrap_or(".").to_string();
36        let depth = (get_int(args, "depth").unwrap_or(3) as usize).min(10);
37        let show_hidden = get_bool(args, "show_hidden").unwrap_or(false);
38
39        let (result, original) = crate::tools::ctx_tree::handle(&path, depth, show_hidden);
40        let sent = crate::core::tokens::count_tokens(&result);
41        let saved = original.saturating_sub(sent);
42
43        let final_out = crate::core::protocol::append_savings(&result, original, sent);
44
45        Ok(ToolOutput {
46            text: final_out,
47            original_tokens: original,
48            saved_tokens: saved,
49            mode: None,
50            path: Some(path),
51            changed: false,
52        })
53    }
54}
lean_ctx/tools/registered/ctx_tree.rs

lean_ctx/tools/registered/
ctx_tree.rs
