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
620fn 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}