1use std::path::PathBuf;
2
3pub mod agents;
4mod support;
5
6#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
11#[serde(rename_all = "snake_case")]
12pub enum HookMode {
13 #[default]
14 Mcp,
15 Hybrid,
16}
17
18impl std::fmt::Display for HookMode {
19 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
20 match self {
21 Self::Mcp => write!(f, "MCP"),
22 Self::Hybrid => write!(f, "Hybrid"),
23 }
24 }
25}
26
27impl HookMode {
28 pub fn from_str_loose(s: &str) -> Option<Self> {
29 match s.to_lowercase().replace('-', "").as_str() {
30 "mcp" => Some(Self::Mcp),
31 "hybrid" => Some(Self::Hybrid),
32 _ => None,
33 }
34 }
35
36 pub fn description(&self) -> &'static str {
37 match self {
38 Self::Mcp => "MCP server only (extension/plugin-based agents without reliable shell)",
39 Self::Hybrid => "MCP server + shell hooks for command compression (best of both)",
40 }
41 }
42}
43
44pub const HYBRID_AGENTS: &[&str] = &[
55 "cursor",
56 "gemini",
57 "codex",
58 "claude",
59 "claude-code",
60 "crush",
61 "hermes",
62 "opencode",
63 "openclaw",
64 "pi",
65 "qoder",
66 "windsurf",
67 "amp",
68 "cline",
69 "roo",
70 "copilot",
71 "kiro",
72 "qwen",
73 "trae",
74 "antigravity",
75 "antigravity-cli",
76 "amazonq",
77 "verdent",
78];
79
80pub fn recommend_hook_mode(agent_key: &str) -> HookMode {
81 if HYBRID_AGENTS.contains(&agent_key) {
82 HookMode::Hybrid
83 } else {
84 HookMode::Mcp
86 }
87}
88use agents::{
89 install_amp_hook, install_antigravity_cli_hook, install_antigravity_hook,
90 install_claude_hook_config, install_claude_hook_scripts, install_claude_hook_with_mode,
91 install_claude_project_hooks, install_cline_rules, install_codebuddy_hook_config,
92 install_codebuddy_hook_scripts, install_codebuddy_hook_with_mode,
93 install_codebuddy_project_hooks, install_codex_hook, install_copilot_hook,
94 install_crush_hook_with_mode, install_cursor_hook_config, install_cursor_hook_scripts,
95 install_cursor_hook_with_mode, install_gemini_hook, install_gemini_hook_config,
96 install_gemini_hook_scripts, install_hermes_hook_with_mode, install_jetbrains_hook,
97 install_kiro_hook, install_openclaw_hook, install_opencode_hook_with_mode,
98 install_pi_hook_with_mode, install_qoder_hook, install_qoder_hook_with_mode,
99 install_windsurf_hooks, install_windsurf_rules,
100};
101use support::{
102 ensure_codex_hooks_enabled, install_codex_instruction_docs, install_named_json_server,
103 upsert_lean_ctx_codex_hook_entries,
104};
105
106fn mcp_server_quiet_mode() -> bool {
107 std::env::var_os("LEAN_CTX_MCP_SERVER").is_some()
108 || matches!(std::env::var("LEAN_CTX_QUIET"), Ok(value) if value.trim() == "1")
109}
110
111const REFRESHABLE_HOOK_AGENTS: &[&str] = &[
119 "claude", "cursor", "gemini", "codex", "windsurf", "copilot", "qoder",
120];
121
122#[cfg(test)]
127const REFRESH_EXEMPT_HYBRID_AGENTS: &[&str] = &[
128 "claude-code",
130 "pi",
132 "cline",
135 "roo",
136 "kiro",
137 "antigravity",
140 "antigravity-cli",
141 "amp",
142 "crush",
143 "hermes",
144 "opencode",
145 "openclaw",
146 "qwen",
147 "trae",
148 "amazonq",
149 "verdent",
150];
151
152pub fn refresh_installed_hooks() {
157 let Some(home) = crate::core::home::resolve_home_dir() else {
158 return;
159 };
160 for agent in REFRESHABLE_HOOK_AGENTS {
161 if hooks_installed_for(agent, &home) {
162 refresh_agent_hooks(agent, &home);
163 }
164 }
165}
166
167fn hooks_installed_for(agent: &str, home: &std::path::Path) -> bool {
169 match agent {
170 "claude" => {
171 let dir = crate::setup::claude_config_dir(home);
172 dir.join("hooks/lean-ctx-rewrite.sh").exists()
173 || file_contains_lean_ctx(&dir.join("settings.json"))
174 }
175 "codebuddy" => {
176 let dir = crate::core::editor_registry::codebuddy_state_dir(home);
177 dir.join("hooks/lean-ctx-rewrite.sh").exists()
178 || file_contains_lean_ctx(&dir.join("settings.json"))
179 }
180 "cursor" => {
181 home.join(".cursor/hooks/lean-ctx-rewrite.sh").exists()
182 || file_contains_lean_ctx(&home.join(".cursor/hooks.json"))
183 }
184 "gemini" => {
185 home.join(".gemini/hooks/lean-ctx-rewrite-gemini.sh")
186 .exists()
187 || home.join(".gemini/hooks/lean-ctx-hook-gemini.sh").exists()
188 }
189 "codex" => {
190 let dir = crate::core::home::resolve_codex_dir().unwrap_or_else(|| home.join(".codex"));
191 dir.join("hooks/lean-ctx-rewrite-codex.sh").exists()
192 || file_contains_lean_ctx(&dir.join("hooks.json"))
193 }
194 "windsurf" => file_contains_lean_ctx(&home.join(".codeium/windsurf/hooks.json")),
195 "copilot" => {
196 file_contains_lean_ctx(&home.join(".copilot/hooks/hooks.json"))
199 || file_contains_lean_ctx(&home.join(".github/hooks/hooks.json"))
200 }
201 "qoder" => file_contains_lean_ctx(&home.join(".qoder/settings.json")),
202 _ => false,
203 }
204}
205
206fn refresh_agent_hooks(agent: &str, home: &std::path::Path) {
209 match agent {
210 "claude" => {
211 install_claude_hook_scripts(home);
212 install_claude_hook_config(home);
213 }
214 "codebuddy" => {
215 install_codebuddy_hook_scripts(home);
216 install_codebuddy_hook_config(home);
217 }
218 "cursor" => {
219 install_cursor_hook_scripts(home);
220 install_cursor_hook_config(home);
221 }
222 "gemini" => {
223 install_gemini_hook_scripts(home);
224 install_gemini_hook_config(home);
225 }
226 "codex" => install_codex_hook(),
227 "windsurf" => install_windsurf_hooks(home),
228 "copilot" => install_copilot_hook(true),
229 "qoder" => install_qoder_hook(),
230 _ => {}
231 }
232}
233
234fn file_contains_lean_ctx(path: &std::path::Path) -> bool {
235 std::fs::read_to_string(path).is_ok_and(|c| c.contains("lean-ctx"))
236}
237
238fn resolve_binary_path() -> String {
249 crate::core::portable_binary::resolve_portable_binary()
250}
251
252fn resolve_binary_path_for_bash() -> String {
253 let path = resolve_binary_path();
254 to_bash_compatible_path(&path)
255}
256
257pub fn to_bash_compatible_path(path: &str) -> String {
258 let path = match crate::core::pathutil::strip_verbatim_str(path) {
259 Some(stripped) => stripped,
260 None => path.replace('\\', "/"),
261 };
262 if path.len() >= 2 && path.as_bytes()[1] == b':' {
263 let drive = (path.as_bytes()[0] as char).to_ascii_lowercase();
264 format!("/{drive}{}", &path[2..])
265 } else {
266 path
267 }
268}
269
270pub fn from_bash_to_native_path(path: &str) -> String {
273 crate::core::pathutil::normalize_tool_path(path)
274}
275
276pub fn normalize_tool_path(path: &str) -> String {
279 crate::core::pathutil::normalize_tool_path(path)
280}
281
282pub fn generate_rewrite_script(binary: &str) -> String {
283 let case_pattern = crate::rewrite_registry::bash_case_pattern();
284 format!(
285 r#"#!/usr/bin/env bash
286# lean-ctx PreToolUse hook — rewrites bash commands to lean-ctx equivalents
287set -euo pipefail
288
289LEAN_CTX_BIN="{binary}"
290
291INPUT=$(cat)
292TOOL=$(echo "$INPUT" | grep -oE '"tool_name":"([^"\\]|\\.)*"' | head -1 | sed 's/^"tool_name":"//;s/"$//' | sed 's/\\"/"/g;s/\\\\/\\/g')
293
294case "$TOOL" in
295 Bash|bash|PowerShell|powershell) ;;
296 *) exit 0 ;;
297esac
298
299CMD=$(echo "$INPUT" | grep -oE '"command":"([^"\\]|\\.)*"' | head -1 | sed 's/^"command":"//;s/"$//' | sed 's/\\"/"/g;s/\\\\/\\/g')
300
301if [ -z "$CMD" ] || echo "$CMD" | grep -qE "^(lean-ctx |$LEAN_CTX_BIN )"; then
302 exit 0
303fi
304
305case "$CMD" in
306 {case_pattern})
307 # Shell-escape then JSON-escape (two passes)
308 SHELL_ESC=$(printf '%s' "$CMD" | sed 's/\\/\\\\/g;s/"/\\"/g')
309 REWRITE="$LEAN_CTX_BIN -c \"$SHELL_ESC\""
310 JSON_CMD=$(printf '%s' "$REWRITE" | sed 's/\\/\\\\/g;s/"/\\"/g')
311 printf '{{"hookSpecificOutput":{{"hookEventName":"PreToolUse","permissionDecision":"allow","updatedInput":{{"command":"%s"}}}}}}' "$JSON_CMD"
312 ;;
313 *) exit 0 ;;
314esac
315"#
316 )
317}
318
319pub fn generate_compact_rewrite_script(binary: &str) -> String {
320 let case_pattern = crate::rewrite_registry::bash_case_pattern();
321 format!(
322 r#"#!/usr/bin/env bash
323# lean-ctx hook — rewrites shell commands
324set -euo pipefail
325LEAN_CTX_BIN="{binary}"
326INPUT=$(cat)
327CMD=$(echo "$INPUT" | grep -oE '"command":"([^"\\]|\\.)*"' | head -1 | sed 's/^"command":"//;s/"$//' | sed 's/\\"/"/g;s/\\\\/\\/g' 2>/dev/null || echo "")
328if [ -z "$CMD" ] || echo "$CMD" | grep -qE "^(lean-ctx |$LEAN_CTX_BIN )"; then exit 0; fi
329case "$CMD" in
330 {case_pattern})
331 SHELL_ESC=$(printf '%s' "$CMD" | sed 's/\\/\\\\/g;s/"/\\"/g')
332 REWRITE="$LEAN_CTX_BIN -c \"$SHELL_ESC\""
333 JSON_CMD=$(printf '%s' "$REWRITE" | sed 's/\\/\\\\/g;s/"/\\"/g')
334 printf '{{"hookSpecificOutput":{{"hookEventName":"PreToolUse","permissionDecision":"allow","updatedInput":{{"command":"%s"}}}}}}' "$JSON_CMD" ;;
335 *) exit 0 ;;
336esac
337"#
338 )
339}
340
341const REDIRECT_SCRIPT_CLAUDE: &str = r"#!/usr/bin/env bash
342# lean-ctx PreToolUse hook — all native tools pass through
343# Read/Grep/ListFiles are allowed so Edit (which requires native Read) works.
344# The MCP instructions guide the AI to prefer ctx_read/ctx_search/ctx_tree.
345exit 0
346";
347
348const REDIRECT_SCRIPT_GENERIC: &str = r"#!/usr/bin/env bash
349# lean-ctx hook — all native tools pass through
350exit 0
351";
352
353pub(crate) const HYBRID_RULES: &str = "\
354# lean-ctx — Hybrid Mode (MCP reads + CLI commands)
355
356Use MCP tools for reads (cache benefit), CLI commands for everything else (no schema overhead):
357
358## MCP tools (keep using):
359| Tool | Why MCP |
360|------|---------|
361| `ctx_read(path, mode)` | In-process cache, re-reads ~13 tokens |
362
363## CLI commands (via Shell/Bash):
364| USE (via Shell/Bash) | INSTEAD OF (MCP) | Why |
365|---------------------|-------------------|-----|
366| `lean-ctx -c \"<cmd>\"` | `ctx_shell` | No MCP schema overhead |
367| `lean-ctx grep <pattern> [path]` | `ctx_search` | No MCP schema overhead |
368| `lean-ctx ls [path]` | `ctx_tree` | No MCP schema overhead |
369
370## File editing:
371Use native Edit/StrReplace — lean-ctx only handles READ operations.
372Write, Delete, Glob → use normally.
373";
374
375pub fn install_project_rules() {
376 install_project_rules_for_agents(&[]);
377}
378
379pub fn install_project_rules_for_agents(agents: &[&str]) {
382 if crate::core::config::Config::load().rules_scope_effective()
383 == crate::core::config::RulesScope::Global
384 {
385 return;
386 }
387
388 let cwd = std::env::current_dir().unwrap_or_default();
389
390 if !is_inside_git_repo(&cwd) {
391 eprintln!(
392 " Skipping project files: not inside a git repository.\n \
393 Run this command from your project root to create CLAUDE.md / AGENTS.md."
394 );
395 return;
396 }
397
398 let home = crate::core::home::resolve_home_dir().unwrap_or_default();
399 if cwd == home {
400 eprintln!(
401 " Skipping project files: current directory is your home folder.\n \
402 Run this command from a project directory instead."
403 );
404 return;
405 }
406
407 let all = agents.is_empty();
408 let wants = |name: &str| all || agents.iter().any(|a| a.eq_ignore_ascii_case(name));
409
410 ensure_project_agents_integration(&cwd);
411
412 if wants("cursor") || wants("windsurf") {
413 let cursorrules = cwd.join(".cursorrules");
414 if !cursorrules.exists()
415 || !std::fs::read_to_string(&cursorrules)
416 .unwrap_or_default()
417 .contains("lean-ctx")
418 {
419 let content = CURSORRULES_TEMPLATE;
420 if cursorrules.exists() {
421 let mut existing = std::fs::read_to_string(&cursorrules).unwrap_or_default();
422 if !existing.ends_with('\n') {
423 existing.push('\n');
424 }
425 existing.push('\n');
426 existing.push_str(content);
427 write_file(&cursorrules, &existing);
428 } else {
429 write_file(&cursorrules, content);
430 }
431 if !mcp_server_quiet_mode() {
432 eprintln!("Created/updated .cursorrules in project root.");
433 }
434 }
435 }
436
437 if wants("claude") {
438 let claude_rules_file = cwd.join(".claude").join("rules").join("lean-ctx.md");
444 if let Ok(existing) = std::fs::read_to_string(&claude_rules_file) {
445 if existing.contains("<!-- lean-ctx-rules-")
446 && std::fs::remove_file(&claude_rules_file).is_ok()
447 && !mcp_server_quiet_mode()
448 {
449 eprintln!(
450 "Removed .claude/rules/lean-ctx.md (always-loaded duplicate; AGENTS.md block + skill replace it)."
451 );
452 }
453 }
454
455 install_claude_project_hooks(&cwd);
456 }
457
458 if wants("codebuddy") {
459 let codebuddy_rules_file = cwd.join(".codebuddy").join("rules").join("lean-ctx.md");
460 if let Ok(existing) = std::fs::read_to_string(&codebuddy_rules_file) {
461 if existing.contains("<!-- lean-ctx-rules-")
462 && std::fs::remove_file(&codebuddy_rules_file).is_ok()
463 && !mcp_server_quiet_mode()
464 {
465 eprintln!(
466 "Removed .codebuddy/rules/lean-ctx.md (always-loaded duplicate; CODEBUDDY.md block + skill replace it)."
467 );
468 }
469 }
470
471 install_codebuddy_project_hooks(&cwd);
472 }
473
474 if wants("kiro") {
475 let kiro_dir = cwd.join(".kiro");
476 if kiro_dir.exists() {
477 let steering_dir = kiro_dir.join("steering");
478 let steering_file = steering_dir.join("lean-ctx.md");
479 if !steering_file.exists()
480 || !std::fs::read_to_string(&steering_file)
481 .unwrap_or_default()
482 .contains("lean-ctx")
483 {
484 let _ = std::fs::create_dir_all(&steering_dir);
485 write_file(&steering_file, KIRO_STEERING_TEMPLATE);
486 if !mcp_server_quiet_mode() {
487 eprintln!("Created .kiro/steering/lean-ctx.md (Kiro steering).");
488 }
489 }
490 }
491 }
492}
493
494const PROJECT_LEAN_CTX_MD_MARKER: &str = "<!-- lean-ctx-owned: PROJECT-LEAN-CTX.md v1 -->";
495const PROJECT_LEAN_CTX_MD: &str = "LEAN-CTX.md";
496const PROJECT_AGENTS_MD: &str = "AGENTS.md";
497const AGENTS_BLOCK_START: &str = "<!-- lean-ctx -->";
498const AGENTS_BLOCK_END: &str = "<!-- /lean-ctx -->";
499
500fn ensure_project_agents_integration(cwd: &std::path::Path) {
501 let lean_ctx_md = cwd.join(PROJECT_LEAN_CTX_MD);
502 let desired = format!(
503 "{PROJECT_LEAN_CTX_MD_MARKER}\n{}\n",
504 crate::rules_inject::rules_dedicated_markdown()
505 );
506
507 if !lean_ctx_md.exists() {
508 write_file(&lean_ctx_md, &desired);
509 } else if std::fs::read_to_string(&lean_ctx_md)
510 .unwrap_or_default()
511 .contains(PROJECT_LEAN_CTX_MD_MARKER)
512 {
513 let current = std::fs::read_to_string(&lean_ctx_md).unwrap_or_default();
514 if !current.contains(crate::rules_inject::RULES_VERSION_STR) {
515 write_file(&lean_ctx_md, &desired);
516 }
517 }
518
519 let block = format!(
524 "{AGENTS_BLOCK_START}\n\
525## lean-ctx\n\n\
526Prefer lean-ctx MCP tools over native equivalents for token savings:\n\
527`ctx_read` > Read/cat, `ctx_search` > Grep/rg, `ctx_shell` > bash, `ctx_tree` > ls/find.\n\
528Native Edit/Write/Glob stay as-is; use `ctx_edit` only when Edit needs an unavailable Read.\n\
529Full rules: {PROJECT_LEAN_CTX_MD} (open on demand — do not auto-load).\n\
530{AGENTS_BLOCK_END}\n"
531 );
532
533 let agents_md = cwd.join(PROJECT_AGENTS_MD);
534 if !agents_md.exists() {
535 let content = format!("# Agent Instructions\n\n{block}");
536 write_file(&agents_md, &content);
537 if !mcp_server_quiet_mode() {
538 eprintln!("Created AGENTS.md in project root (lean-ctx reference only).");
539 }
540 return;
541 }
542
543 let existing = std::fs::read_to_string(&agents_md).unwrap_or_default();
544
545 if existing.contains("CLI-first Token Optimization for Pi")
546 && !existing.contains(AGENTS_BLOCK_START)
547 {
548 let content = format!("# Agent Instructions\n\n{block}");
549 write_file(&agents_md, &content);
550 return;
551 }
552
553 if existing.contains(AGENTS_BLOCK_START) {
554 let updated = crate::marked_block::replace_marked_block(
555 &existing,
556 AGENTS_BLOCK_START,
557 AGENTS_BLOCK_END,
558 &block,
559 );
560 if updated != existing {
561 write_file(&agents_md, &updated);
562 }
563 return;
564 }
565
566 if existing.contains("lean-ctx") && existing.contains(PROJECT_LEAN_CTX_MD) {
567 return;
568 }
569
570 let mut out = existing;
571 if !out.ends_with('\n') {
572 out.push('\n');
573 }
574 out.push('\n');
575 out.push_str(&block);
576 write_file(&agents_md, &out);
577 if !mcp_server_quiet_mode() {
578 eprintln!("Updated AGENTS.md (added lean-ctx reference block).");
579 }
580}
581
582const CURSORRULES_TEMPLATE: &str = "\
586<!-- lean-ctx -->
587# lean-ctx
588
589Prefer lean-ctx MCP tools: ctx_read > Read/cat, ctx_search > Grep/rg, ctx_shell > bash, ctx_tree > ls/find.
590Edit/Write/Glob stay native; ctx_edit only when Edit needs an unavailable Read.
591Full rules: ~/.cursor/rules/lean-ctx.mdc (auto-loaded) — do not duplicate here.
592<!-- /lean-ctx -->
593";
594
595pub const KIRO_STEERING_TEMPLATE: &str = "\
596---
597inclusion: always
598---
599
600# lean-ctx — Context Engineering Layer
601
602The workspace has the `lean-ctx` MCP server installed. You MUST prefer lean-ctx tools over native equivalents for token efficiency and caching.
603
604## Mandatory Tool Preferences
605
606| Use this | Instead of | Why |
607|----------|-----------|-----|
608| `mcp_lean_ctx_ctx_read` | `readFile`, `readCode` | Cached reads, 10 compression modes, re-reads cost ~13 tokens |
609| `mcp_lean_ctx_ctx_multi_read` | `readMultipleFiles` | Batch cached reads in one call |
610| `mcp_lean_ctx_ctx_shell` | `executeBash` | Pattern compression for git/npm/test output |
611| `mcp_lean_ctx_ctx_search` | `grepSearch` | Compact, .gitignore-aware results |
612| `mcp_lean_ctx_ctx_tree` | `listDirectory` | Compact directory maps with file counts |
613
614## When to use native Kiro tools instead
615
616- `fsWrite` / `fsAppend` — always use native (lean-ctx doesn't write files)
617- `strReplace` — always use native (precise string replacement)
618- `semanticRename` / `smartRelocate` — always use native (IDE integration)
619- `getDiagnostics` — always use native (language server diagnostics)
620- `deleteFile` — always use native
621
622## Session management
623
624- At the start of a long task, call `mcp_lean_ctx_ctx_preload` with a task description to warm the cache
625- Use `mcp_lean_ctx_ctx_compress` periodically in long conversations to checkpoint context
626- Use `mcp_lean_ctx_ctx_knowledge` to persist important discoveries across sessions
627
628## Rules
629
630- NEVER loop on edit failures — switch to `mcp_lean_ctx_ctx_edit` immediately
631- For large files, use `mcp_lean_ctx_ctx_read` with `mode: \"signatures\"` or `mode: \"map\"` first
632- For re-reading a file you already read, just call `mcp_lean_ctx_ctx_read` again (cache hit = ~13 tokens)
633- When running tests or build commands, use `mcp_lean_ctx_ctx_shell` for compressed output
634";
635
636pub(crate) fn should_register_mcp() -> bool {
643 crate::core::config::Config::load()
644 .setup
645 .should_update_mcp()
646}
647
648pub fn install_agent_hook(agent: &str, global: bool) {
649 install_agent_hook_with_mode(agent, global, HookMode::Mcp);
650}
651
652pub fn install_agent_hook_with_mode(agent: &str, global: bool, mode: HookMode) {
653 let home = crate::core::home::resolve_home_dir().unwrap_or_default();
654 match agent {
655 "claude" | "claude-code" => install_claude_hook_with_mode(global, mode),
656 "codebuddy" => install_codebuddy_hook_with_mode(global, mode),
657 "cursor" => install_cursor_hook_with_mode(global, mode),
658 "gemini" => {
659 install_gemini_hook();
660 install_antigravity_cli_hook();
666 }
667 "antigravity" => install_antigravity_hook(),
668 "antigravity-cli" => install_antigravity_cli_hook(),
669 "augment" => install_mcp_json_agent(
670 "Augment CLI",
671 "~/.augment/settings.json",
672 &crate::core::editor_registry::augment_cli_settings_path(&home),
673 ),
674 "codex" => install_codex_hook(),
675 "windsurf" => install_windsurf_rules(global),
676 "cline" | "roo" => install_cline_rules(global),
677 "copilot" | "vscode" => install_copilot_hook(global),
678 "pi" => install_pi_hook_with_mode(global, mode),
679 "qoder" => install_qoder_hook_with_mode(mode),
680 "qoderwork" => install_mcp_json_agent(
681 "QoderWork",
682 "~/.qoderwork/mcp.json",
683 &home.join(".qoderwork/mcp.json"),
684 ),
685 "qwen" => install_mcp_json_agent(
686 "Qwen Code",
687 "~/.qwen/settings.json",
688 &home.join(".qwen/settings.json"),
689 ),
690 "trae" => install_mcp_json_agent("Trae", "~/.trae/mcp.json", &home.join(".trae/mcp.json")),
691 "amazonq" => install_mcp_json_agent(
692 "Amazon Q Developer",
693 "~/.aws/amazonq/default.json",
694 &home.join(".aws/amazonq/default.json"),
695 ),
696 "jetbrains" => install_jetbrains_hook(),
697 "kiro" => install_kiro_hook(),
698 "verdent" => install_mcp_json_agent(
699 "Verdent",
700 "~/.verdent/mcp.json",
701 &home.join(".verdent/mcp.json"),
702 ),
703 "opencode" => install_opencode_hook_with_mode(mode),
704 "amp" => install_amp_hook(),
705 "crush" => install_crush_hook_with_mode(mode),
706 "openclaw" => install_openclaw_hook(),
707 "hermes" => install_hermes_hook_with_mode(global, mode),
708 "zed" => {
709 let zed_path = crate::core::editor_registry::zed_settings_path(&home);
710 let binary = resolve_binary_path();
711 let entry = full_server_entry(&binary);
712 install_named_json_server("Zed", "settings.json", &zed_path, "context_servers", entry);
713 }
714 "aider" => {
715 install_mcp_json_agent("Aider", "~/.aider/mcp.json", &home.join(".aider/mcp.json"));
716 }
717 "continue" => install_mcp_json_agent(
718 "Continue",
719 "~/.continue/mcp.json",
720 &home.join(".continue/mcp.json"),
721 ),
722 "neovim" => install_mcp_json_agent(
723 "Neovim (mcphub.nvim)",
724 "~/.config/mcphub/servers.json",
725 &home.join(".config/mcphub/servers.json"),
726 ),
727 "emacs" => install_mcp_json_agent(
728 "Emacs (mcp.el)",
729 "~/.emacs.d/mcp.json",
730 &home.join(".emacs.d/mcp.json"),
731 ),
732 "sublime" => install_mcp_json_agent(
733 "Sublime Text",
734 "~/.config/sublime-text/mcp.json",
735 &home.join(".config/sublime-text/mcp.json"),
736 ),
737 _ => {
738 eprintln!("Unknown agent: {agent}");
739 eprintln!(" Supported: aider, amazonq, amp, antigravity, antigravity-cli, augment,");
740 eprintln!(" claude, cline, codebuddy, codex, continue, copilot, crush, cursor, emacs, gemini,");
741 eprintln!(" hermes, jetbrains, kiro, neovim, openclaw, opencode, pi, qoder,");
742 eprintln!(" qoderwork, qwen, roo, sublime, trae, verdent, vscode, windsurf, zed");
743 std::process::exit(1);
744 }
745 }
746}
747
748pub fn install_agent_project_hooks(agent: &str, cwd: &std::path::Path) {
749 match agent {
750 "claude" | "claude-code" => agents::install_claude_project_hooks(cwd),
751 "codebuddy" => agents::install_codebuddy_project_hooks(cwd),
752 _ => {}
753 }
754}
755
756fn write_file(path: &std::path::Path, content: &str) {
757 if std::fs::read_to_string(path).is_ok_and(|existing| existing == content) {
760 return;
761 }
762 if let Err(e) = crate::config_io::write_atomic_with_backup(path, content) {
763 tracing::error!("Error writing {}: {e}", path.display());
764 }
765}
766
767fn is_inside_git_repo(path: &std::path::Path) -> bool {
768 let mut p = path;
769 loop {
770 if p.join(".git").exists() {
771 return true;
772 }
773 match p.parent() {
774 Some(parent) => p = parent,
775 None => return false,
776 }
777 }
778}
779
780#[cfg(unix)]
781fn make_executable(path: &PathBuf) {
782 use std::os::unix::fs::PermissionsExt;
783 let _ = std::fs::set_permissions(path, std::fs::Permissions::from_mode(0o755));
784}
785
786#[cfg(not(unix))]
787fn make_executable(_path: &PathBuf) {}
788
789pub(crate) fn mcp_server_env_pairs() -> Vec<(String, String)> {
799 let mut pairs = Vec::new();
800
801 let data_dir = crate::core::data_dir::lean_ctx_data_dir()
802 .map(|d| d.to_string_lossy().to_string())
803 .unwrap_or_default();
804 pairs.push(("LEAN_CTX_DATA_DIR".to_string(), data_dir));
805
806 let cfg = crate::core::config::Config::load();
807
808 let project_root = std::env::var("LEAN_CTX_PROJECT_ROOT")
809 .ok()
810 .filter(|v| !v.trim().is_empty())
811 .or_else(|| cfg.project_root.clone().filter(|v| !v.trim().is_empty()));
812 if let Some(root) = project_root {
813 pairs.push(("LEAN_CTX_PROJECT_ROOT".to_string(), root));
814 }
815
816 let extra_roots = std::env::var("LEAN_CTX_EXTRA_ROOTS")
819 .ok()
820 .filter(|v| !v.trim().is_empty())
821 .or_else(|| {
822 let roots: Vec<&str> = cfg
823 .extra_roots
824 .iter()
825 .map(String::as_str)
826 .filter(|s| !s.trim().is_empty())
827 .collect();
828 if roots.is_empty() {
829 return None;
830 }
831 std::env::join_paths(roots)
832 .ok()
833 .map(|s| s.to_string_lossy().to_string())
834 });
835 if let Some(extra) = extra_roots {
836 pairs.push(("LEAN_CTX_EXTRA_ROOTS".to_string(), extra));
837 }
838
839 pairs
840}
841
842pub(crate) fn mcp_server_env_json() -> serde_json::Value {
844 let map: serde_json::Map<String, serde_json::Value> = mcp_server_env_pairs()
845 .into_iter()
846 .map(|(k, v)| (k, serde_json::Value::String(v)))
847 .collect();
848 serde_json::Value::Object(map)
849}
850
851fn full_server_entry(binary: &str) -> serde_json::Value {
852 serde_json::json!({
858 "command": binary,
859 "env": mcp_server_env_json()
860 })
861}
862
863pub(crate) fn install_mcp_json_agent(
864 name: &str,
865 display_path: &str,
866 config_path: &std::path::Path,
867) {
868 let binary = resolve_binary_path();
869 let entry = full_server_entry(&binary);
870 install_named_json_server(name, display_path, config_path, "mcpServers", entry);
871}
872
873#[cfg(test)]
874mod tests {
875 use super::*;
876
877 #[test]
878 fn refresh_covers_every_hybrid_agent() {
879 for agent in HYBRID_AGENTS {
882 let refreshed = REFRESHABLE_HOOK_AGENTS.contains(agent);
883 let exempt = REFRESH_EXEMPT_HYBRID_AGENTS.contains(agent);
884 assert!(
885 refreshed ^ exempt,
886 "hybrid agent `{agent}` must be either refreshed or explicitly exempt (exactly one)"
887 );
888 }
889 }
890
891 #[test]
892 fn refresh_sets_reference_only_hybrid_agents() {
893 for agent in REFRESHABLE_HOOK_AGENTS {
894 assert!(
895 HYBRID_AGENTS.contains(agent),
896 "refreshable agent `{agent}` is not a Hybrid agent"
897 );
898 }
899 for agent in REFRESH_EXEMPT_HYBRID_AGENTS {
900 assert!(
901 HYBRID_AGENTS.contains(agent),
902 "exempt agent `{agent}` is not a Hybrid agent (stale exemption?)"
903 );
904 }
905 }
906
907 #[test]
908 fn mcp_env_pairs_propagate_project_and_extra_roots_from_env() {
909 let _iso = crate::core::data_dir::isolated_data_dir();
913 std::env::set_var("LEAN_CTX_PROJECT_ROOT", "/work/main");
914 std::env::set_var("LEAN_CTX_EXTRA_ROOTS", "/work/wt-a:/work/wt-b");
915
916 let pairs = mcp_server_env_pairs();
917 let get = |k: &str| pairs.iter().find(|(p, _)| p == k).map(|(_, v)| v.as_str());
918 assert!(
919 get("LEAN_CTX_DATA_DIR").is_some(),
920 "data dir always emitted"
921 );
922 assert_eq!(get("LEAN_CTX_PROJECT_ROOT"), Some("/work/main"));
923 assert_eq!(get("LEAN_CTX_EXTRA_ROOTS"), Some("/work/wt-a:/work/wt-b"));
924
925 let json = mcp_server_env_json();
927 assert_eq!(json["LEAN_CTX_PROJECT_ROOT"].as_str(), Some("/work/main"));
928
929 std::env::remove_var("LEAN_CTX_PROJECT_ROOT");
930 std::env::remove_var("LEAN_CTX_EXTRA_ROOTS");
931 }
932
933 #[test]
934 fn mcp_env_pairs_omit_roots_when_unset() {
935 let _iso = crate::core::data_dir::isolated_data_dir();
938 std::env::remove_var("LEAN_CTX_PROJECT_ROOT");
939 std::env::remove_var("LEAN_CTX_EXTRA_ROOTS");
940
941 let pairs = mcp_server_env_pairs();
942 let keys: Vec<&str> = pairs.iter().map(|(k, _)| k.as_str()).collect();
943 assert!(keys.contains(&"LEAN_CTX_DATA_DIR"));
944 assert!(!keys.contains(&"LEAN_CTX_PROJECT_ROOT"));
945 assert!(!keys.contains(&"LEAN_CTX_EXTRA_ROOTS"));
946 }
947
948 #[test]
949 fn hooks_installed_for_is_false_without_artifacts() {
950 let tmp = unique_tmp_dir("leanctx_refresh_empty");
951 for agent in REFRESHABLE_HOOK_AGENTS {
952 if *agent == "codex" {
956 continue;
957 }
958 assert!(
959 !hooks_installed_for(agent, &tmp),
960 "`{agent}` should not be detected as installed in an empty home"
961 );
962 }
963 let _ = std::fs::remove_dir_all(&tmp);
964 }
965
966 #[test]
967 fn hooks_installed_for_detects_marker_content() {
968 let tmp = unique_tmp_dir("leanctx_refresh_marker");
969 let hooks = tmp.join(".codeium/windsurf/hooks.json");
970 std::fs::create_dir_all(hooks.parent().unwrap()).unwrap();
971
972 std::fs::write(&hooks, "{\"hooks\":{}}").unwrap();
974 assert!(!hooks_installed_for("windsurf", &tmp));
975
976 std::fs::write(&hooks, "{\"hooks\":{\"cmd\":\"lean-ctx hook rewrite\"}}").unwrap();
978 assert!(hooks_installed_for("windsurf", &tmp));
979
980 let _ = std::fs::remove_dir_all(&tmp);
981 }
982
983 fn unique_tmp_dir(prefix: &str) -> std::path::PathBuf {
984 let nanos = std::time::SystemTime::now()
985 .duration_since(std::time::UNIX_EPOCH)
986 .map_or(0, |d| d.as_nanos());
987 let dir = std::env::temp_dir().join(format!("{prefix}_{}_{nanos}", std::process::id()));
988 std::fs::create_dir_all(&dir).unwrap();
989 dir
990 }
991
992 #[test]
993 fn bash_path_unix_unchanged() {
994 assert_eq!(
995 to_bash_compatible_path("/usr/local/bin/lean-ctx"),
996 "/usr/local/bin/lean-ctx"
997 );
998 }
999
1000 #[test]
1001 fn bash_path_home_unchanged() {
1002 assert_eq!(
1003 to_bash_compatible_path("/home/user/.cargo/bin/lean-ctx"),
1004 "/home/user/.cargo/bin/lean-ctx"
1005 );
1006 }
1007
1008 #[test]
1009 fn bash_path_windows_drive_converted() {
1010 assert_eq!(
1011 to_bash_compatible_path("C:\\Users\\Fraser\\bin\\lean-ctx.exe"),
1012 "/c/Users/Fraser/bin/lean-ctx.exe"
1013 );
1014 }
1015
1016 #[test]
1017 fn bash_path_windows_lowercase_drive() {
1018 assert_eq!(
1019 to_bash_compatible_path("D:\\tools\\lean-ctx.exe"),
1020 "/d/tools/lean-ctx.exe"
1021 );
1022 }
1023
1024 #[test]
1025 fn bash_path_windows_forward_slashes() {
1026 assert_eq!(
1027 to_bash_compatible_path("C:/Users/Fraser/bin/lean-ctx.exe"),
1028 "/c/Users/Fraser/bin/lean-ctx.exe"
1029 );
1030 }
1031
1032 #[test]
1033 fn bash_path_bare_name_unchanged() {
1034 assert_eq!(to_bash_compatible_path("lean-ctx"), "lean-ctx");
1035 }
1036
1037 #[cfg(windows)]
1040 #[test]
1041 fn normalize_msys2_path() {
1042 assert_eq!(
1043 normalize_tool_path("/c/Users/game/Downloads/project"),
1044 "C:/Users/game/Downloads/project"
1045 );
1046 assert_eq!(
1047 normalize_tool_path("/d/Projects/app/src"),
1048 "D:/Projects/app/src"
1049 );
1050 }
1051
1052 #[cfg(not(windows))]
1053 #[test]
1054 fn normalize_msys2_path_untouched_on_unix() {
1055 assert_eq!(
1056 crate::core::pathutil::normalize_tool_path_lexical("/c/Users/game/Downloads/project"),
1057 "/c/Users/game/Downloads/project"
1058 );
1059 }
1060
1061 #[test]
1062 fn normalize_backslashes() {
1063 assert_eq!(
1064 normalize_tool_path("C:\\Users\\game\\project\\src"),
1065 "C:/Users/game/project/src"
1066 );
1067 }
1068
1069 #[test]
1070 fn normalize_mixed_separators() {
1071 assert_eq!(
1072 normalize_tool_path("C:\\Users/game\\project/src"),
1073 "C:/Users/game/project/src"
1074 );
1075 }
1076
1077 #[test]
1078 fn normalize_double_slashes() {
1079 assert_eq!(
1080 normalize_tool_path("/home/user//project///src"),
1081 "/home/user/project/src"
1082 );
1083 }
1084
1085 #[test]
1086 fn normalize_trailing_slash() {
1087 assert_eq!(
1088 normalize_tool_path("/home/user/project/"),
1089 "/home/user/project"
1090 );
1091 }
1092
1093 #[test]
1094 fn normalize_root_preserved() {
1095 assert_eq!(normalize_tool_path("/"), "/");
1096 }
1097
1098 #[test]
1099 fn normalize_windows_root_preserved() {
1100 assert_eq!(normalize_tool_path("C:/"), "C:/");
1101 }
1102
1103 #[test]
1104 fn normalize_unix_path_unchanged() {
1105 assert_eq!(
1106 normalize_tool_path("/home/user/project/src/main.rs"),
1107 "/home/user/project/src/main.rs"
1108 );
1109 }
1110
1111 #[test]
1112 fn normalize_relative_path_unchanged() {
1113 assert_eq!(normalize_tool_path("src/main.rs"), "src/main.rs");
1114 }
1115
1116 #[test]
1117 fn normalize_dot_unchanged() {
1118 assert_eq!(normalize_tool_path("."), ".");
1119 }
1120
1121 #[test]
1122 fn normalize_unc_path_preserved() {
1123 assert_eq!(
1124 normalize_tool_path("//server/share/file"),
1125 "//server/share/file"
1126 );
1127 }
1128
1129 #[test]
1130 fn cursor_hook_config_has_version_and_object_hooks() {
1131 let config = serde_json::json!({
1132 "version": 1,
1133 "hooks": {
1134 "preToolUse": [
1135 {
1136 "matcher": "terminal_command",
1137 "command": "lean-ctx hook rewrite"
1138 },
1139 {
1140 "matcher": "read_file|grep|search|list_files|list_directory",
1141 "command": "lean-ctx hook redirect"
1142 }
1143 ]
1144 }
1145 });
1146
1147 let json_str = serde_json::to_string_pretty(&config).unwrap();
1148 let parsed: serde_json::Value = serde_json::from_str(&json_str).unwrap();
1149
1150 assert_eq!(parsed["version"], 1);
1151 assert!(parsed["hooks"].is_object());
1152 assert!(parsed["hooks"]["preToolUse"].is_array());
1153 assert_eq!(parsed["hooks"]["preToolUse"].as_array().unwrap().len(), 2);
1154 assert_eq!(
1155 parsed["hooks"]["preToolUse"][0]["matcher"],
1156 "terminal_command"
1157 );
1158 }
1159
1160 #[test]
1161 fn cursor_hook_detects_old_format_needs_migration() {
1162 let old_format = r#"{"hooks":[{"event":"preToolUse","command":"lean-ctx hook rewrite"}]}"#;
1163 let has_correct =
1164 old_format.contains("\"version\"") && old_format.contains("\"preToolUse\"");
1165 assert!(
1166 !has_correct,
1167 "Old format should be detected as needing migration"
1168 );
1169 }
1170
1171 #[test]
1172 fn gemini_hook_config_has_type_command() {
1173 let binary = "lean-ctx";
1174 let rewrite_cmd = format!("{binary} hook rewrite");
1175 let redirect_cmd = format!("{binary} hook redirect");
1176
1177 let hook_config = serde_json::json!({
1178 "hooks": {
1179 "BeforeTool": [
1180 {
1181 "hooks": [{
1182 "type": "command",
1183 "command": rewrite_cmd
1184 }]
1185 },
1186 {
1187 "hooks": [{
1188 "type": "command",
1189 "command": redirect_cmd
1190 }]
1191 }
1192 ]
1193 }
1194 });
1195
1196 let parsed = hook_config;
1197 let before_tool = parsed["hooks"]["BeforeTool"].as_array().unwrap();
1198 assert_eq!(before_tool.len(), 2);
1199
1200 let first_hook = &before_tool[0]["hooks"][0];
1201 assert_eq!(first_hook["type"], "command");
1202 assert_eq!(first_hook["command"], "lean-ctx hook rewrite");
1203
1204 let second_hook = &before_tool[1]["hooks"][0];
1205 assert_eq!(second_hook["type"], "command");
1206 assert_eq!(second_hook["command"], "lean-ctx hook redirect");
1207 }
1208
1209 #[test]
1210 fn gemini_hook_old_format_detected() {
1211 let old_format = r#"{"hooks":{"BeforeTool":[{"command":"lean-ctx hook rewrite"}]}}"#;
1212 let has_new = old_format.contains("hook rewrite")
1213 && old_format.contains("hook redirect")
1214 && old_format.contains("\"type\"");
1215 assert!(!has_new, "Missing 'type' field should trigger migration");
1216 }
1217
1218 #[test]
1219 fn rewrite_script_uses_registry_pattern() {
1220 let script = generate_rewrite_script("/usr/bin/lean-ctx");
1221 assert!(script.contains(r"git\ *"), "script missing git pattern");
1222 assert!(script.contains(r"cargo\ *"), "script missing cargo pattern");
1223 assert!(script.contains(r"npm\ *"), "script missing npm pattern");
1224 assert!(script.contains(r"rg\ *"), "script missing rg pattern");
1225 assert!(script.contains(r"ls\ *"), "script missing ls pattern");
1226 assert!(
1227 script.contains("LEAN_CTX_BIN=\"/usr/bin/lean-ctx\""),
1228 "script missing binary path"
1229 );
1230 assert!(
1231 script.contains("PowerShell|powershell"),
1232 "rewrite script must accept PowerShell tool names for Windows compatibility"
1233 );
1234 }
1235
1236 #[test]
1237 fn compact_rewrite_script_uses_registry_pattern() {
1238 let script = generate_compact_rewrite_script("/usr/bin/lean-ctx");
1239 assert!(script.contains(r"git\ *"), "compact script missing git");
1240 assert!(script.contains(r"cargo\ *"), "compact script missing cargo");
1241 assert!(script.contains(r"rg\ *"), "compact script missing rg");
1242 }
1243
1244 #[test]
1245 fn rewrite_scripts_contain_all_registry_commands() {
1246 let script = generate_rewrite_script("lean-ctx");
1247 let compact = generate_compact_rewrite_script("lean-ctx");
1248 for entry in crate::rewrite_registry::REWRITE_COMMANDS {
1249 if matches!(entry.category, crate::rewrite_registry::Category::FileRead) {
1250 continue;
1251 }
1252 let pattern = if entry.command.contains('-') {
1253 format!("{}*", entry.command.replace('-', r"\-"))
1254 } else {
1255 format!(r"{}\ *", entry.command)
1256 };
1257 assert!(
1258 script.contains(&pattern),
1259 "rewrite_script missing '{}' (pattern: {})",
1260 entry.command,
1261 pattern
1262 );
1263 assert!(
1264 compact.contains(&pattern),
1265 "compact_rewrite_script missing '{}' (pattern: {})",
1266 entry.command,
1267 pattern
1268 );
1269 }
1270 }
1271
1272 #[test]
1273 fn codex_is_hybrid() {
1274 assert_eq!(recommend_hook_mode("codex"), HookMode::Hybrid);
1275 }
1276
1277 #[test]
1278 fn cursor_is_hybrid() {
1279 assert_eq!(recommend_hook_mode("cursor"), HookMode::Hybrid);
1280 }
1281
1282 #[test]
1283 fn gemini_is_hybrid() {
1284 assert_eq!(recommend_hook_mode("gemini"), HookMode::Hybrid);
1285 }
1286
1287 #[test]
1288 fn claude_is_hybrid() {
1289 assert_eq!(recommend_hook_mode("claude"), HookMode::Hybrid);
1290 }
1291
1292 #[test]
1293 fn unknown_agent_falls_back_to_mcp() {
1294 assert_eq!(recommend_hook_mode("unknown-agent"), HookMode::Mcp);
1295 }
1296
1297 #[cfg(windows)]
1299 #[test]
1300 fn from_bash_to_native_converts_msys_drive() {
1301 assert_eq!(
1302 from_bash_to_native_path("/c/Users/ABC/lean-ctx"),
1303 "C:/Users/ABC/lean-ctx"
1304 );
1305 assert_eq!(
1306 from_bash_to_native_path("/d/Program Files/lean-ctx.exe"),
1307 "D:/Program Files/lean-ctx.exe"
1308 );
1309 }
1310
1311 #[test]
1312 fn from_bash_to_native_unix_path_unchanged() {
1313 assert_eq!(
1314 from_bash_to_native_path("/usr/local/bin/lean-ctx"),
1315 "/usr/local/bin/lean-ctx"
1316 );
1317 }
1318
1319 #[test]
1320 fn from_bash_to_native_bare_name() {
1321 assert_eq!(from_bash_to_native_path("lean-ctx"), "lean-ctx");
1322 }
1323
1324 #[test]
1325 fn windows_path_to_bash_form() {
1326 let native = r"C:\Users\ABC\AppData\Local\lean-ctx\lean-ctx.exe";
1327 let bash = to_bash_compatible_path(native);
1328 assert_eq!(bash, "/c/Users/ABC/AppData/Local/lean-ctx/lean-ctx.exe");
1329 }
1330
1331 #[cfg(windows)]
1333 #[test]
1334 fn roundtrip_windows_path() {
1335 let native = r"C:\Users\ABC\AppData\Local\lean-ctx\lean-ctx.exe";
1336 let bash = to_bash_compatible_path(native);
1337 let back = from_bash_to_native_path(&bash);
1338 assert_eq!(back, "C:/Users/ABC/AppData/Local/lean-ctx/lean-ctx.exe");
1339 }
1340
1341 #[test]
1342 fn roundtrip_unix_path() {
1343 let native = "/usr/local/bin/lean-ctx";
1344 let bash = to_bash_compatible_path(native);
1345 assert_eq!(bash, native);
1346 let back = from_bash_to_native_path(&bash);
1347 assert_eq!(back, native);
1348 }
1349}