lean_ctx/cli/
init_cmd.rs

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            let result = crate::setup::setup_single_agent(agent_name, global, mode);
52            for name in &result.rules.injected {
53                qprintln!("  ✓ {name} rules injected");
54            }
55            for name in &result.rules.updated {
56                qprintln!("  ✓ {name} rules updated");
57            }
58            for name in &result.rules.already {
59                qprintln!("  ✓ {name} rules up-to-date");
60            }
61            if result.skill_installed {
62                qprintln!("  ✓ SKILL.md installed for {agent_name}");
63            }
64            if result.mcp_skipped {
65                qprintln!("  • MCP registration skipped for {agent_name} (auto_update_mcp=false)");
66            }
67            for e in &result.errors {
68                eprintln!("  ✗ {agent_name}: {e}");
69            }
70            if project {
71                crate::hooks::install_agent_project_hooks(agent_name, &cwd);
72            }
73        }
74        if !global {
75            crate::hooks::install_project_rules_for_agents(&agents);
76        }
77        qprintln!("\nRun 'lean-ctx gain' after using some commands to see your savings.");
78        return;
79    }
80
81    let eval_shell = args
82        .iter()
83        .find(|a| matches!(a.as_str(), "bash" | "zsh" | "fish" | "powershell" | "pwsh"));
84    if let Some(shell) = eval_shell
85        && !global
86    {
87        super::shell_init::print_hook_stdout(shell);
88        return;
89    }
90
91    let shell_name = std::env::var("SHELL").unwrap_or_default();
92    let is_zsh = shell_name.contains("zsh");
93    let is_fish = shell_name.contains("fish");
94    let is_powershell = cfg!(windows) && shell_name.is_empty();
95
96    let binary = crate::core::portable_binary::resolve_portable_binary();
97
98    if dry_run {
99        let rc = if is_powershell {
100            dirs::home_dir().map_or_else(
101                || "PowerShell profile".to_string(),
102                |h| {
103                    crate::shell::platform::powershell_profile_path(&h)
104                        .to_string_lossy()
105                        .into_owned()
106                },
107            )
108        } else if is_fish {
109            "~/.config/fish/config.fish".to_string()
110        } else if is_zsh {
111            "~/.zshrc".to_string()
112        } else {
113            "~/.bashrc".to_string()
114        };
115        qprintln!("\nlean-ctx init --dry-run\n");
116        qprintln!("  Would modify:  {rc}");
117        qprintln!("  Would backup:  {rc}.lean-ctx.bak");
118        qprintln!("  Would alias:   git npm pnpm yarn cargo docker docker-compose kubectl");
119        qprintln!("                 gh pip pip3 ruff go golangci-lint eslint prettier tsc");
120        qprintln!("                 curl wget php composer (24 commands + k)");
121        let data_dir = crate::core::data_dir::lean_ctx_data_dir().map_or_else(
122            |_| "~/.config/lean-ctx/".to_string(),
123            |p| p.to_string_lossy().to_string(),
124        );
125        qprintln!("  Would create:  {data_dir}");
126        qprintln!("  Binary:        {binary}");
127        qprintln!("\n  Safety: aliases auto-fallback to original command if lean-ctx is removed.");
128        qprintln!("\n  Run without --dry-run to apply.");
129        return;
130    }
131
132    if no_hook {
133        qprintln!("Shell hook disabled (--no-shell-hook or shell_hook_disabled config).");
134        qprintln!("MCP tools remain active. Set LEAN_CTX_NO_HOOK=1 to disable at runtime.");
135    } else if is_powershell {
136        super::shell_init::init_powershell(&binary);
137    } else {
138        let bash_binary = to_bash_compatible_path(&binary);
139        if is_fish {
140            super::shell_init::init_fish(&bash_binary);
141        } else {
142            super::shell_init::init_posix(is_zsh, &bash_binary);
143        }
144    }
145
146    if let Ok(lean_dir) = crate::core::data_dir::lean_ctx_data_dir()
147        && !lean_dir.exists()
148    {
149        let _ = std::fs::create_dir_all(&lean_dir);
150        qprintln!("Created {}", lean_dir.display());
151    }
152
153    let rc = if is_powershell {
154        "$PROFILE"
155    } else if is_fish {
156        "config.fish"
157    } else if is_zsh {
158        ".zshrc"
159    } else {
160        ".bashrc"
161    };
162
163    qprintln!("\nlean-ctx init complete (24 aliases installed)");
164    qprintln!();
165    qprintln!("  Disable temporarily:  lean-ctx-off");
166    qprintln!("  Re-enable:            lean-ctx-on");
167    qprintln!("  Check status:         lean-ctx-status");
168    qprintln!("  Full uninstall:       lean-ctx uninstall");
169    qprintln!("  Diagnose issues:      lean-ctx doctor");
170    qprintln!("  Preview changes:      lean-ctx init --global --dry-run");
171    qprintln!();
172    if is_powershell {
173        qprintln!("  Restart PowerShell or run: . {rc}");
174    } else {
175        qprintln!("  Restart your shell or run: source ~/{rc}");
176    }
177    qprintln!();
178    qprintln!("For AI tool integration: lean-ctx init --agent <tool> [--mode <mode>]");
179    qprintln!("  Supported: aider, amazonq, amp, antigravity, antigravity-cli, augment,");
180    qprintln!("    claude, cline, codex, continue, copilot, crush, cursor, emacs, gemini,");
181    qprintln!("    hermes, jetbrains, kiro, neovim, openclaw, opencode, pi, qoder,");
182    qprintln!("    qoderwork, qwen, roo, sublime, trae, verdent, vscode, windsurf, zed");
183    qprintln!("  Modes: mcp, hybrid  (auto-detected per agent, override with --mode)");
184}
185
186pub fn cmd_init_quiet(args: &[String]) {
187    // SAFETY: the `init` CLI command is single-threaded; no other thread reads
188    // the environment between this set and the matching remove below.
189    unsafe { std::env::set_var("LEAN_CTX_QUIET", "1") };
190    cmd_init(args);
191    // SAFETY: single-threaded CLI command; pairs with the set above.
192    unsafe { std::env::remove_var("LEAN_CTX_QUIET") };
193}
lean_ctx/cli/init_cmd.rs

lean_ctx/cli/
init_cmd.rs
