1use crate::hooks::to_bash_compatible_path;
2
3pub(crate) fn quiet_enabled() -> bool {
4 matches!(std::env::var("LEAN_CTX_QUIET"), Ok(v) if v.trim() == "1")
5}
6
7macro_rules! qprintln {
8 ($($t:tt)*) => {
9 if !quiet_enabled() {
10 println!($($t)*);
11 }
12 };
13}
14
15pub fn cmd_init(args: &[String]) {
16 let global = args.iter().any(|a| a == "--global" || a == "-g");
17 let project = args.iter().any(|a| a == "--project");
18 let dry_run = args.iter().any(|a| a == "--dry-run");
19 let no_hook = args.iter().any(|a| a == "--no-shell-hook")
20 || crate::core::config::Config::load().shell_hook_disabled_effective();
21
22 let explicit_mode = args
23 .windows(2)
24 .find(|w| w[0] == "--mode")
25 .and_then(|w| crate::hooks::HookMode::from_str_loose(&w[1]));
26
27 if args.windows(2).any(|w| w[0] == "--mode")
28 && !args
29 .windows(2)
30 .any(|w| w[0] == "--mode" && crate::hooks::HookMode::from_str_loose(&w[1]).is_some())
31 {
32 let bad = args
33 .windows(2)
34 .find(|w| w[0] == "--mode")
35 .map_or("?", |w| w[1].as_str());
36 eprintln!("Unknown hook mode: '{bad}'. Valid: mcp, hybrid");
37 std::process::exit(1);
38 }
39
40 let agents: Vec<&str> = args
41 .windows(2)
42 .filter(|w| w[0] == "--agent")
43 .map(|w| w[1].as_str())
44 .collect();
45
46 if !agents.is_empty() {
47 let cwd = std::env::current_dir().unwrap_or_default();
48 for agent_name in &agents {
49 let mode =
50 explicit_mode.unwrap_or_else(|| crate::hooks::recommend_hook_mode(agent_name));
51 crate::hooks::install_agent_hook_with_mode(agent_name, global, mode);
52 if let Err(e) = crate::setup::configure_agent_mcp(agent_name) {
53 eprintln!("MCP config for '{agent_name}' not updated: {e}");
54 }
55 if project {
56 crate::hooks::install_agent_project_hooks(agent_name, &cwd);
57 }
58 }
59 if !global {
60 crate::hooks::install_project_rules_for_agents(&agents);
61 }
62 qprintln!("\nRun 'lean-ctx gain' after using some commands to see your savings.");
63 return;
64 }
65
66 let eval_shell = args
67 .iter()
68 .find(|a| matches!(a.as_str(), "bash" | "zsh" | "fish" | "powershell" | "pwsh"));
69 if let Some(shell) = eval_shell {
70 if !global {
71 super::shell_init::print_hook_stdout(shell);
72 return;
73 }
74 }
75
76 let shell_name = std::env::var("SHELL").unwrap_or_default();
77 let is_zsh = shell_name.contains("zsh");
78 let is_fish = shell_name.contains("fish");
79 let is_powershell = cfg!(windows) && shell_name.is_empty();
80
81 let binary = crate::core::portable_binary::resolve_portable_binary();
82
83 if dry_run {
84 let rc = if is_powershell {
85 "Documents/PowerShell/Microsoft.PowerShell_profile.ps1".to_string()
86 } else if is_fish {
87 "~/.config/fish/config.fish".to_string()
88 } else if is_zsh {
89 "~/.zshrc".to_string()
90 } else {
91 "~/.bashrc".to_string()
92 };
93 qprintln!("\nlean-ctx init --dry-run\n");
94 qprintln!(" Would modify: {rc}");
95 qprintln!(" Would backup: {rc}.lean-ctx.bak");
96 qprintln!(" Would alias: git npm pnpm yarn cargo docker docker-compose kubectl");
97 qprintln!(" gh pip pip3 ruff go golangci-lint eslint prettier tsc");
98 qprintln!(" curl wget php composer (24 commands + k)");
99 let data_dir = crate::core::data_dir::lean_ctx_data_dir().map_or_else(
100 |_| "~/.config/lean-ctx/".to_string(),
101 |p| p.to_string_lossy().to_string(),
102 );
103 qprintln!(" Would create: {data_dir}");
104 qprintln!(" Binary: {binary}");
105 qprintln!("\n Safety: aliases auto-fallback to original command if lean-ctx is removed.");
106 qprintln!("\n Run without --dry-run to apply.");
107 return;
108 }
109
110 if no_hook {
111 qprintln!("Shell hook disabled (--no-shell-hook or shell_hook_disabled config).");
112 qprintln!("MCP tools remain active. Set LEAN_CTX_NO_HOOK=1 to disable at runtime.");
113 } else if is_powershell {
114 super::shell_init::init_powershell(&binary);
115 } else {
116 let bash_binary = to_bash_compatible_path(&binary);
117 if is_fish {
118 super::shell_init::init_fish(&bash_binary);
119 } else {
120 super::shell_init::init_posix(is_zsh, &bash_binary);
121 }
122 }
123
124 if let Ok(lean_dir) = crate::core::data_dir::lean_ctx_data_dir() {
125 if !lean_dir.exists() {
126 let _ = std::fs::create_dir_all(&lean_dir);
127 qprintln!("Created {}", lean_dir.display());
128 }
129 }
130
131 let rc = if is_powershell {
132 "$PROFILE"
133 } else if is_fish {
134 "config.fish"
135 } else if is_zsh {
136 ".zshrc"
137 } else {
138 ".bashrc"
139 };
140
141 qprintln!("\nlean-ctx init complete (24 aliases installed)");
142 qprintln!();
143 qprintln!(" Disable temporarily: lean-ctx-off");
144 qprintln!(" Re-enable: lean-ctx-on");
145 qprintln!(" Check status: lean-ctx-status");
146 qprintln!(" Full uninstall: lean-ctx uninstall");
147 qprintln!(" Diagnose issues: lean-ctx doctor");
148 qprintln!(" Preview changes: lean-ctx init --global --dry-run");
149 qprintln!();
150 if is_powershell {
151 qprintln!(" Restart PowerShell or run: . {rc}");
152 } else {
153 qprintln!(" Restart your shell or run: source ~/{rc}");
154 }
155 qprintln!();
156 qprintln!("For AI tool integration: lean-ctx init --agent <tool> [--mode <mode>]");
157 qprintln!(" Supported: aider, amazonq, amp, antigravity, claude, cline, codex,");
158 qprintln!(" continue, copilot, crush, cursor, emacs, gemini, hermes, jetbrains,");
159 qprintln!(" kiro, neovim, opencode, pi, qoder, qoderwork, qwen, roo, sublime,");
160 qprintln!(" trae, verdent, vscode, windsurf, zed");
161 qprintln!(" Modes: mcp, hybrid (auto-detected per agent, override with --mode)");
162}
163
164pub fn cmd_init_quiet(args: &[String]) {
165 std::env::set_var("LEAN_CTX_QUIET", "1");
166 cmd_init(args);
167 std::env::remove_var("LEAN_CTX_QUIET");
168}