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 savings_note =
44            if !ctx.minimal && saved > 0 && crate::core::protocol::savings_footer_visible() {
45                format!("\n[saved {saved} tokens vs native ls]")
46            } else {
47                String::new()
48            };
49
50        Ok(ToolOutput {
51            text: format!("{result}{savings_note}"),
52            original_tokens: original,
53            saved_tokens: saved,
54            mode: None,
55            path: Some(path),
56        })
57    }
58}
lean_ctx/tools/registered/ctx_tree.rs

lean_ctx/tools/registered/
ctx_tree.rs
