Skip to main content

lean_ctx/server/
mod.rs

1mod dispatch;
2mod execute;
3pub mod helpers;
4
5use rmcp::handler::server::ServerHandler;
6use rmcp::model::*;
7use rmcp::service::{RequestContext, RoleServer};
8use rmcp::ErrorData;
9
10use crate::tools::{CrpMode, LeanCtxServer};
11
12impl ServerHandler for LeanCtxServer {
13    fn get_info(&self) -> ServerInfo {
14        let capabilities = ServerCapabilities::builder().enable_tools().build();
15
16        let instructions = crate::instructions::build_instructions(self.crp_mode);
17
18        InitializeResult::new(capabilities)
19            .with_server_info(Implementation::new("lean-ctx", env!("CARGO_PKG_VERSION")))
20            .with_instructions(instructions)
21    }
22
23    async fn initialize(
24        &self,
25        request: InitializeRequestParams,
26        _context: RequestContext<RoleServer>,
27    ) -> Result<InitializeResult, ErrorData> {
28        let name = request.client_info.name.clone();
29        tracing::info!("MCP client connected: {:?}", name);
30        *self.client_name.write().await = name.clone();
31
32        let derived_root = derive_project_root_from_cwd();
33        let cwd_str = std::env::current_dir()
34            .ok()
35            .map(|p| p.to_string_lossy().to_string())
36            .unwrap_or_default();
37        {
38            let mut session = self.session.write().await;
39            if !cwd_str.is_empty() {
40                session.shell_cwd = Some(cwd_str.clone());
41            }
42            if let Some(ref root) = derived_root {
43                session.project_root = Some(root.clone());
44                tracing::info!("Project root set to: {root}");
45            } else if let Some(ref root) = session.project_root {
46                let root_path = std::path::Path::new(root);
47                let root_has_marker = has_project_marker(root_path);
48                let root_str = root_path.to_string_lossy();
49                let root_suspicious = root_str.contains("/.claude")
50                    || root_str.contains("/.codex")
51                    || root_str.contains("/var/folders/")
52                    || root_str.contains("/tmp/")
53                    || root_str.contains("\\.claude")
54                    || root_str.contains("\\.codex")
55                    || root_str.contains("\\AppData\\Local\\Temp")
56                    || root_str.contains("\\Temp\\");
57                if root_suspicious && !root_has_marker {
58                    session.project_root = None;
59                }
60            }
61            let _ = session.save();
62        }
63
64        let agent_name = name.clone();
65        let agent_root = derived_root.clone().unwrap_or_default();
66        let agent_id_handle = self.agent_id.clone();
67        tokio::task::spawn_blocking(move || {
68            if std::env::var("LEAN_CTX_HEADLESS").is_ok() {
69                return;
70            }
71            if let Some(home) = dirs::home_dir() {
72                let _ = crate::rules_inject::inject_all_rules(&home);
73            }
74            crate::hooks::refresh_installed_hooks();
75            crate::core::version_check::check_background();
76
77            if !agent_root.is_empty() {
78                let role = match agent_name.to_lowercase().as_str() {
79                    n if n.contains("cursor") => Some("coder"),
80                    n if n.contains("claude") => Some("coder"),
81                    n if n.contains("codex") => Some("coder"),
82                    n if n.contains("antigravity") || n.contains("gemini") => Some("explorer"),
83                    n if n.contains("review") => Some("reviewer"),
84                    n if n.contains("test") => Some("tester"),
85                    _ => None,
86                };
87                let env_role = std::env::var("LEAN_CTX_AGENT_ROLE").ok();
88                let effective_role = env_role.as_deref().or(role);
89                let mut registry = crate::core::agents::AgentRegistry::load_or_create();
90                registry.cleanup_stale(24);
91                let id = registry.register("mcp", effective_role, &agent_root);
92                let _ = registry.save();
93                if let Ok(mut guard) = agent_id_handle.try_write() {
94                    *guard = Some(id);
95                }
96            }
97        });
98
99        let instructions =
100            crate::instructions::build_instructions_with_client(self.crp_mode, &name);
101        let capabilities = ServerCapabilities::builder().enable_tools().build();
102
103        Ok(InitializeResult::new(capabilities)
104            .with_server_info(Implementation::new("lean-ctx", env!("CARGO_PKG_VERSION")))
105            .with_instructions(instructions))
106    }
107
108    async fn list_tools(
109        &self,
110        _request: Option<PaginatedRequestParams>,
111        _context: RequestContext<RoleServer>,
112    ) -> Result<ListToolsResult, ErrorData> {
113        let all_tools = if crate::tool_defs::is_full_mode() {
114            crate::tool_defs::granular_tool_defs()
115        } else if std::env::var("LEAN_CTX_UNIFIED").is_ok() {
116            crate::tool_defs::unified_tool_defs()
117        } else {
118            crate::tool_defs::lazy_tool_defs()
119        };
120
121        let disabled = crate::core::config::Config::load().disabled_tools_effective();
122        let tools = if disabled.is_empty() {
123            all_tools
124        } else {
125            all_tools
126                .into_iter()
127                .filter(|t| !disabled.iter().any(|d| t.name.as_ref() == d.as_str()))
128                .collect()
129        };
130
131        let tools = {
132            let active = self.workflow.read().await.clone();
133            if let Some(run) = active {
134                if let Some(state) = run.spec.state(&run.current) {
135                    if let Some(allowed) = &state.allowed_tools {
136                        let mut allow: std::collections::HashSet<&str> =
137                            allowed.iter().map(|s| s.as_str()).collect();
138                        allow.insert("ctx");
139                        allow.insert("ctx_workflow");
140                        return Ok(ListToolsResult {
141                            tools: tools
142                                .into_iter()
143                                .filter(|t| allow.contains(t.name.as_ref()))
144                                .collect(),
145                            ..Default::default()
146                        });
147                    }
148                }
149            }
150            tools
151        };
152
153        Ok(ListToolsResult {
154            tools,
155            ..Default::default()
156        })
157    }
158
159    async fn call_tool(
160        &self,
161        request: CallToolRequestParams,
162        _context: RequestContext<RoleServer>,
163    ) -> Result<CallToolResult, ErrorData> {
164        self.check_idle_expiry().await;
165
166        let original_name = request.name.as_ref().to_string();
167        let (resolved_name, resolved_args) = if original_name == "ctx" {
168            let sub = request
169                .arguments
170                .as_ref()
171                .and_then(|a| a.get("tool"))
172                .and_then(|v| v.as_str())
173                .map(|s| s.to_string())
174                .ok_or_else(|| {
175                    ErrorData::invalid_params("'tool' is required for ctx meta-tool", None)
176                })?;
177            let tool_name = if sub.starts_with("ctx_") {
178                sub
179            } else {
180                format!("ctx_{sub}")
181            };
182            let mut args = request.arguments.unwrap_or_default();
183            args.remove("tool");
184            (tool_name, Some(args))
185        } else {
186            (original_name, request.arguments)
187        };
188        let name = resolved_name.as_str();
189        let args = &resolved_args;
190
191        if name != "ctx_workflow" {
192            let active = self.workflow.read().await.clone();
193            if let Some(run) = active {
194                if let Some(state) = run.spec.state(&run.current) {
195                    if let Some(allowed) = &state.allowed_tools {
196                        let allowed_ok = allowed.iter().any(|t| t == name) || name == "ctx";
197                        if !allowed_ok {
198                            let mut shown = allowed.clone();
199                            shown.sort();
200                            shown.truncate(30);
201                            return Ok(CallToolResult::success(vec![Content::text(format!(
202                                "Tool '{name}' blocked by workflow '{}' (state: {}). Allowed ({} shown): {}",
203                                run.spec.name,
204                                run.current,
205                                shown.len(),
206                                shown.join(", ")
207                            ))]));
208                        }
209                    }
210                }
211            }
212        }
213
214        let auto_context = {
215            let task = {
216                let session = self.session.read().await;
217                session.task.as_ref().map(|t| t.description.clone())
218            };
219            let project_root = {
220                let session = self.session.read().await;
221                session.project_root.clone()
222            };
223            let mut cache = self.cache.write().await;
224            crate::tools::autonomy::session_lifecycle_pre_hook(
225                &self.autonomy,
226                name,
227                &mut cache,
228                task.as_deref(),
229                project_root.as_deref(),
230                self.crp_mode,
231            )
232        };
233
234        let throttle_result = {
235            let fp = args
236                .as_ref()
237                .map(|a| {
238                    crate::core::loop_detection::LoopDetector::fingerprint(
239                        &serde_json::Value::Object(a.clone()),
240                    )
241                })
242                .unwrap_or_default();
243            let mut detector = self.loop_detector.write().await;
244
245            let is_search = crate::core::loop_detection::LoopDetector::is_search_tool(name);
246            let is_search_shell = name == "ctx_shell" && {
247                let cmd = args
248                    .as_ref()
249                    .and_then(|a| a.get("command"))
250                    .and_then(|v| v.as_str())
251                    .unwrap_or("");
252                crate::core::loop_detection::LoopDetector::is_search_shell_command(cmd)
253            };
254
255            if is_search || is_search_shell {
256                let search_pattern = args.as_ref().and_then(|a| {
257                    a.get("pattern")
258                        .or_else(|| a.get("query"))
259                        .and_then(|v| v.as_str())
260                });
261                let shell_pattern = if is_search_shell {
262                    args.as_ref()
263                        .and_then(|a| a.get("command"))
264                        .and_then(|v| v.as_str())
265                        .and_then(helpers::extract_search_pattern_from_command)
266                } else {
267                    None
268                };
269                let pat = search_pattern.or(shell_pattern.as_deref());
270                detector.record_search(name, &fp, pat)
271            } else {
272                detector.record_call(name, &fp)
273            }
274        };
275
276        if throttle_result.level == crate::core::loop_detection::ThrottleLevel::Blocked {
277            let msg = throttle_result.message.unwrap_or_default();
278            return Ok(CallToolResult::success(vec![Content::text(msg)]));
279        }
280
281        let throttle_warning =
282            if throttle_result.level == crate::core::loop_detection::ThrottleLevel::Reduced {
283                throttle_result.message.clone()
284            } else {
285                None
286            };
287
288        let config = crate::core::config::Config::load();
289        let minimal = config.minimal_overhead_effective();
290
291        let tool_start = std::time::Instant::now();
292        let result_text = self.dispatch_tool(name, args, minimal).await?;
293
294        let mut result_text = result_text;
295
296        let archive_hint = if !minimal {
297            use crate::core::archive;
298            let archivable = matches!(
299                name,
300                "ctx_shell"
301                    | "ctx_read"
302                    | "ctx_multi_read"
303                    | "ctx_smart_read"
304                    | "ctx_execute"
305                    | "ctx_search"
306                    | "ctx_tree"
307            );
308            if archivable && archive::should_archive(&result_text) {
309                let cmd = helpers::get_str(args, "command")
310                    .or_else(|| helpers::get_str(args, "path"))
311                    .unwrap_or_default();
312                let session_id = self.session.read().await.id.clone();
313                let tokens = crate::core::tokens::count_tokens(&result_text);
314                archive::store(name, &cmd, &result_text, Some(&session_id))
315                    .map(|id| archive::format_hint(&id, result_text.len(), tokens))
316            } else {
317                None
318            }
319        } else {
320            None
321        };
322
323        let density = crate::core::config::OutputDensity::effective(&config.output_density);
324        if density != crate::core::config::OutputDensity::Normal {
325            result_text = crate::core::protocol::compress_output(&result_text, &density);
326        }
327
328        if let Some(hint) = archive_hint {
329            result_text = format!("{result_text}\n{hint}");
330        }
331
332        if let Some(ctx) = auto_context {
333            result_text = format!("{ctx}\n\n{result_text}");
334        }
335
336        if let Some(warning) = throttle_warning {
337            result_text = format!("{result_text}\n\n{warning}");
338        }
339
340        if name == "ctx_read" {
341            if !minimal {
342                let read_path = self
343                    .resolve_path_or_passthrough(
344                        &helpers::get_str(args, "path").unwrap_or_default(),
345                    )
346                    .await;
347                let project_root = {
348                    let session = self.session.read().await;
349                    session.project_root.clone()
350                };
351                let mut cache = self.cache.write().await;
352                let enrich = crate::tools::autonomy::enrich_after_read(
353                    &self.autonomy,
354                    &mut cache,
355                    &read_path,
356                    project_root.as_deref(),
357                );
358                if let Some(hint) = enrich.related_hint {
359                    result_text = format!("{result_text}\n{hint}");
360                }
361                crate::tools::autonomy::maybe_auto_dedup(&self.autonomy, &mut cache);
362            } else {
363                let mut cache = self.cache.write().await;
364                crate::tools::autonomy::maybe_auto_dedup(&self.autonomy, &mut cache);
365            }
366        }
367
368        let output_token_count = crate::core::tokens::count_tokens(&result_text);
369
370        if !minimal && name == "ctx_shell" {
371            let cmd = helpers::get_str(args, "command").unwrap_or_default();
372            let calls = self.tool_calls.read().await;
373            let last_original = calls.last().map(|c| c.original_tokens).unwrap_or(0);
374            drop(calls);
375            if let Some(hint) = crate::tools::autonomy::shell_efficiency_hint(
376                &self.autonomy,
377                &cmd,
378                last_original,
379                output_token_count,
380            ) {
381                result_text = format!("{result_text}\n{hint}");
382            }
383        }
384
385        {
386            let input = helpers::canonical_args_string(args);
387            let input_md5 = helpers::md5_hex_fast(&input);
388            let output_md5 = helpers::md5_hex_fast(&result_text);
389            let action = helpers::get_str(args, "action");
390            let agent_id = self.agent_id.read().await.clone();
391            let client_name = self.client_name.read().await.clone();
392            let mut explicit_intent: Option<(
393                crate::core::intent_protocol::IntentRecord,
394                Option<String>,
395                String,
396            )> = None;
397
398            let pending_session_save = {
399                let empty_args = serde_json::Map::new();
400                let args_map = args.as_ref().unwrap_or(&empty_args);
401                let mut session = self.session.write().await;
402                session.record_tool_receipt(
403                    name,
404                    action.as_deref(),
405                    &input_md5,
406                    &output_md5,
407                    agent_id.as_deref(),
408                    Some(&client_name),
409                );
410
411                if let Some(intent) = crate::core::intent_protocol::infer_from_tool_call(
412                    name,
413                    action.as_deref(),
414                    args_map,
415                    session.project_root.as_deref(),
416                ) {
417                    let is_explicit =
418                        intent.source == crate::core::intent_protocol::IntentSource::Explicit;
419                    let root = session.project_root.clone();
420                    let sid = session.id.clone();
421                    session.record_intent(intent.clone());
422                    if is_explicit {
423                        explicit_intent = Some((intent, root, sid));
424                    }
425                }
426                if session.should_save() {
427                    session.prepare_save().ok()
428                } else {
429                    None
430                }
431            };
432
433            if let Some(prepared) = pending_session_save {
434                tokio::task::spawn_blocking(move || {
435                    let _ = prepared.write_to_disk();
436                });
437            }
438
439            if let Some((intent, root, session_id)) = explicit_intent {
440                crate::core::intent_protocol::apply_side_effects(
441                    &intent,
442                    root.as_deref(),
443                    &session_id,
444                );
445            }
446
447            if self.autonomy.is_enabled() {
448                let (calls, project_root) = {
449                    let session = self.session.read().await;
450                    (session.stats.total_tool_calls, session.project_root.clone())
451                };
452
453                if let Some(root) = project_root {
454                    if crate::tools::autonomy::should_auto_consolidate(&self.autonomy, calls) {
455                        let root_clone = root.clone();
456                        tokio::task::spawn_blocking(move || {
457                            let _ = crate::core::consolidation_engine::consolidate_latest(
458                                &root_clone,
459                                crate::core::consolidation_engine::ConsolidationBudgets::default(),
460                            );
461                        });
462                    }
463                }
464            }
465
466            let agent_key = agent_id.unwrap_or_else(|| "unknown".to_string());
467            let input_token_count = crate::core::tokens::count_tokens(&input) as u64;
468            let output_token_count_u64 = output_token_count as u64;
469            let name_owned = name.to_string();
470            tokio::task::spawn_blocking(move || {
471                let mut store = crate::core::a2a::cost_attribution::CostStore::load();
472                store.record_tool_call(
473                    &agent_key,
474                    &client_name,
475                    &name_owned,
476                    input_token_count,
477                    output_token_count_u64,
478                );
479                let _ = store.save();
480            });
481        }
482
483        let skip_checkpoint = minimal
484            || matches!(
485                name,
486                "ctx_compress"
487                    | "ctx_metrics"
488                    | "ctx_benchmark"
489                    | "ctx_analyze"
490                    | "ctx_cache"
491                    | "ctx_discover"
492                    | "ctx_dedup"
493                    | "ctx_session"
494                    | "ctx_knowledge"
495                    | "ctx_agent"
496                    | "ctx_share"
497                    | "ctx_wrapped"
498                    | "ctx_overview"
499                    | "ctx_preload"
500                    | "ctx_cost"
501                    | "ctx_gain"
502                    | "ctx_heatmap"
503                    | "ctx_task"
504                    | "ctx_impact"
505                    | "ctx_architecture"
506                    | "ctx_workflow"
507            );
508
509        if !skip_checkpoint && self.increment_and_check() {
510            if let Some(checkpoint) = self.auto_checkpoint().await {
511                let combined = format!(
512                    "{result_text}\n\n--- AUTO CHECKPOINT (every {} calls) ---\n{checkpoint}",
513                    self.checkpoint_interval
514                );
515                return Ok(CallToolResult::success(vec![Content::text(combined)]));
516            }
517        }
518
519        let tool_duration_ms = tool_start.elapsed().as_millis() as u64;
520        if tool_duration_ms > 100 {
521            LeanCtxServer::append_tool_call_log(
522                name,
523                tool_duration_ms,
524                0,
525                0,
526                None,
527                &chrono::Local::now().format("%Y-%m-%d %H:%M:%S").to_string(),
528            );
529        }
530
531        let current_count = self.call_count.load(std::sync::atomic::Ordering::Relaxed);
532        if current_count > 0 && current_count.is_multiple_of(100) {
533            std::thread::spawn(crate::cloud_sync::cloud_background_tasks);
534        }
535
536        Ok(CallToolResult::success(vec![Content::text(result_text)]))
537    }
538}
539
540pub fn build_instructions_for_test(crp_mode: CrpMode) -> String {
541    crate::instructions::build_instructions(crp_mode)
542}
543
544pub fn build_claude_code_instructions_for_test() -> String {
545    crate::instructions::claude_code_instructions()
546}
547
548const PROJECT_MARKERS: &[&str] = &[
549    ".git",
550    "Cargo.toml",
551    "package.json",
552    "go.mod",
553    "pyproject.toml",
554    "setup.py",
555    "pom.xml",
556    "build.gradle",
557    "Makefile",
558    ".lean-ctx.toml",
559];
560
561fn has_project_marker(dir: &std::path::Path) -> bool {
562    PROJECT_MARKERS.iter().any(|m| dir.join(m).exists())
563}
564
565fn is_home_or_agent_dir(dir: &std::path::Path) -> bool {
566    if let Some(home) = dirs::home_dir() {
567        if dir == home {
568            return true;
569        }
570    }
571    let dir_str = dir.to_string_lossy();
572    dir_str.ends_with("/.claude")
573        || dir_str.ends_with("/.codex")
574        || dir_str.contains("/.claude/")
575        || dir_str.contains("/.codex/")
576}
577
578fn git_toplevel_from(dir: &std::path::Path) -> Option<String> {
579    std::process::Command::new("git")
580        .args(["rev-parse", "--show-toplevel"])
581        .current_dir(dir)
582        .stdout(std::process::Stdio::piped())
583        .stderr(std::process::Stdio::null())
584        .output()
585        .ok()
586        .and_then(|o| {
587            if o.status.success() {
588                String::from_utf8(o.stdout)
589                    .ok()
590                    .map(|s| s.trim().to_string())
591            } else {
592                None
593            }
594        })
595}
596
597pub fn derive_project_root_from_cwd() -> Option<String> {
598    let cwd = std::env::current_dir().ok()?;
599    let canonical = crate::core::pathutil::safe_canonicalize_or_self(&cwd);
600
601    if is_home_or_agent_dir(&canonical) {
602        return git_toplevel_from(&canonical);
603    }
604
605    if has_project_marker(&canonical) {
606        return Some(canonical.to_string_lossy().to_string());
607    }
608
609    if let Some(git_root) = git_toplevel_from(&canonical) {
610        return Some(git_root);
611    }
612
613    if let Some(root) = detect_multi_root_workspace(&canonical) {
614        return Some(root);
615    }
616
617    None
618}
619
620/// Detect a multi-root workspace: a directory that has no project markers
621/// itself, but contains child directories that do. In this case, use the
622/// parent as jail root and auto-allow all child projects via LEAN_CTX_ALLOW_PATH.
623fn detect_multi_root_workspace(dir: &std::path::Path) -> Option<String> {
624    let entries = std::fs::read_dir(dir).ok()?;
625    let mut child_projects: Vec<String> = Vec::new();
626
627    for entry in entries.flatten() {
628        let path = entry.path();
629        if path.is_dir() && has_project_marker(&path) {
630            let canonical = crate::core::pathutil::safe_canonicalize_or_self(&path);
631            child_projects.push(canonical.to_string_lossy().to_string());
632        }
633    }
634
635    if child_projects.len() >= 2 {
636        let existing = std::env::var("LEAN_CTX_ALLOW_PATH").unwrap_or_default();
637        let sep = if cfg!(windows) { ";" } else { ":" };
638        let merged = if existing.is_empty() {
639            child_projects.join(sep)
640        } else {
641            format!("{existing}{sep}{}", child_projects.join(sep))
642        };
643        std::env::set_var("LEAN_CTX_ALLOW_PATH", &merged);
644        tracing::info!(
645            "Multi-root workspace detected at {}: auto-allowing {} child projects",
646            dir.display(),
647            child_projects.len()
648        );
649        return Some(dir.to_string_lossy().to_string());
650    }
651
652    None
653}
654
655pub fn tool_descriptions_for_test() -> Vec<(&'static str, &'static str)> {
656    crate::tool_defs::list_all_tool_defs()
657        .into_iter()
658        .map(|(name, desc, _)| (name, desc))
659        .collect()
660}
661
662pub fn tool_schemas_json_for_test() -> String {
663    crate::tool_defs::list_all_tool_defs()
664        .iter()
665        .map(|(name, _, schema)| format!("{}: {}", name, schema))
666        .collect::<Vec<_>>()
667        .join("\n")
668}
669
670#[cfg(test)]
671mod tests {
672    use super::*;
673
674    #[test]
675    fn project_markers_detected() {
676        let tmp = tempfile::tempdir().unwrap();
677        let root = tmp.path().join("myproject");
678        std::fs::create_dir_all(&root).unwrap();
679        assert!(!has_project_marker(&root));
680
681        std::fs::create_dir(root.join(".git")).unwrap();
682        assert!(has_project_marker(&root));
683    }
684
685    #[test]
686    fn home_dir_detected_as_agent_dir() {
687        if let Some(home) = dirs::home_dir() {
688            assert!(is_home_or_agent_dir(&home));
689        }
690    }
691
692    #[test]
693    fn agent_dirs_detected() {
694        let claude = std::path::PathBuf::from("/home/user/.claude");
695        assert!(is_home_or_agent_dir(&claude));
696        let codex = std::path::PathBuf::from("/home/user/.codex");
697        assert!(is_home_or_agent_dir(&codex));
698        let project = std::path::PathBuf::from("/home/user/projects/myapp");
699        assert!(!is_home_or_agent_dir(&project));
700    }
701
702    #[test]
703    fn test_unified_tool_count() {
704        let tools = crate::tool_defs::unified_tool_defs();
705        assert_eq!(tools.len(), 5, "Expected 5 unified tools");
706    }
707
708    #[test]
709    fn test_granular_tool_count() {
710        let tools = crate::tool_defs::granular_tool_defs();
711        assert!(tools.len() >= 25, "Expected at least 25 granular tools");
712    }
713
714    #[test]
715    fn disabled_tools_filters_list() {
716        let all = crate::tool_defs::granular_tool_defs();
717        let total = all.len();
718        let disabled = ["ctx_graph".to_string(), "ctx_agent".to_string()];
719        let filtered: Vec<_> = all
720            .into_iter()
721            .filter(|t| !disabled.iter().any(|d| t.name.as_ref() == d.as_str()))
722            .collect();
723        assert_eq!(filtered.len(), total - 2);
724        assert!(!filtered.iter().any(|t| t.name.as_ref() == "ctx_graph"));
725        assert!(!filtered.iter().any(|t| t.name.as_ref() == "ctx_agent"));
726    }
727
728    #[test]
729    fn empty_disabled_tools_returns_all() {
730        let all = crate::tool_defs::granular_tool_defs();
731        let total = all.len();
732        let disabled: Vec<String> = vec![];
733        let filtered: Vec<_> = all
734            .into_iter()
735            .filter(|t| !disabled.iter().any(|d| t.name.as_ref() == d.as_str()))
736            .collect();
737        assert_eq!(filtered.len(), total);
738    }
739
740    #[test]
741    fn misspelled_disabled_tool_is_silently_ignored() {
742        let all = crate::tool_defs::granular_tool_defs();
743        let total = all.len();
744        let disabled = ["ctx_nonexistent_tool".to_string()];
745        let filtered: Vec<_> = all
746            .into_iter()
747            .filter(|t| !disabled.iter().any(|d| t.name.as_ref() == d.as_str()))
748            .collect();
749        assert_eq!(filtered.len(), total);
750    }
751
752    #[test]
753    fn detect_multi_root_workspace_with_child_projects() {
754        let tmp = tempfile::tempdir().unwrap();
755        let workspace = tmp.path().join("workspace");
756        std::fs::create_dir_all(&workspace).unwrap();
757
758        let proj_a = workspace.join("project-a");
759        let proj_b = workspace.join("project-b");
760        std::fs::create_dir_all(proj_a.join(".git")).unwrap();
761        std::fs::create_dir_all(&proj_b).unwrap();
762        std::fs::write(proj_b.join("package.json"), "{}").unwrap();
763
764        let result = detect_multi_root_workspace(&workspace);
765        assert!(
766            result.is_some(),
767            "should detect workspace with 2 child projects"
768        );
769
770        std::env::remove_var("LEAN_CTX_ALLOW_PATH");
771    }
772
773    #[test]
774    fn detect_multi_root_workspace_returns_none_for_single_project() {
775        let tmp = tempfile::tempdir().unwrap();
776        let workspace = tmp.path().join("workspace");
777        std::fs::create_dir_all(&workspace).unwrap();
778
779        let proj_a = workspace.join("project-a");
780        std::fs::create_dir_all(proj_a.join(".git")).unwrap();
781
782        let result = detect_multi_root_workspace(&workspace);
783        assert!(
784            result.is_none(),
785            "should not detect workspace with only 1 child project"
786        );
787    }
788}