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
47        if result.starts_with("ERROR:") {
48            return Err(ErrorData::invalid_params(result, None));
49        }
50
51        let sent = crate::core::tokens::count_tokens(&result);
52        let saved = original.saturating_sub(sent);
53
54        let final_out = crate::core::protocol::append_savings(&result, original, sent);
55
56        Ok(ToolOutput {
57            text: final_out,
58            original_tokens: original,
59            saved_tokens: saved,
60            mode: None,
61            path: Some(path),
62            changed: false,
63        })
64    }
65}
lean_ctx/tools/registered/ctx_tree.rs

lean_ctx/tools/registered/
ctx_tree.rs
