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 = if let Some(p) = ctx.resolved_path("path") {
36            p.to_string()
37        } else if let Some(err) = ctx.path_error("path") {
38            return Err(ErrorData::invalid_params(format!("path: {err}"), None));
39        } else {
40            ".".to_string()
41        };
42        let depth = (get_int(args, "depth").unwrap_or(3) as usize).min(10);
43        let show_hidden = get_bool(args, "show_hidden").unwrap_or(false);
44
45        let (result, original) = crate::tools::ctx_tree::handle(&path, depth, show_hidden);
46        let sent = crate::core::tokens::count_tokens(&result);
47        let saved = original.saturating_sub(sent);
48
49        let final_out = crate::core::protocol::append_savings(&result, original, sent);
50
51        Ok(ToolOutput {
52            text: final_out,
53            original_tokens: original,
54            saved_tokens: saved,
55            mode: None,
56            path: Some(path),
57            changed: false,
58        })
59    }
60}
lean_ctx/tools/registered/ctx_tree.rs

lean_ctx/tools/registered/
ctx_tree.rs
