nebu-ctx 0.10.7

use std::path::PathBuf;

use super::{
    ensure_codex_hooks_enabled as shared_ensure_codex_hooks_enabled, generate_rewrite_script,
    install_codex_instruction_docs, make_executable, mcp_server_quiet_mode, resolve_binary_path,
    resolve_binary_path_for_bash, upsert_lean_ctx_codex_hook_entries, write_file,
    REDIRECT_SCRIPT_CLAUDE,
};

pub(super) fn install_claude_hook(global: bool) {
    let home = match crate::config::preferred_os_home_dir() {
        Some(h) => h,
        None => {
            eprintln!("Cannot resolve home directory");
            return;
        }
    };

    install_claude_hook_scripts(&home);
    install_claude_hook_config(&home);

    let scope = crate::core::config::Config::load().rules_scope_effective();
    if global || scope != crate::core::config::RulesScope::Project {
        install_claude_rules_file(&home);
        install_claude_global_claude_md(&home);
        install_claude_skill(&home);
    }
}

const CLAUDE_MD_BLOCK_START: &str = "<!-- nebu-ctx -->";
const CLAUDE_MD_BLOCK_END: &str = "<!-- /nebu-ctx -->";
const CLAUDE_MD_BLOCK_VERSION: &str = crate::public_guidance::CLAUDE_MD_BLOCK_VERSION;

fn claude_hook_payload(
    binary: &str,
    rewrite_cmd: String,
    redirect_cmd: String,
) -> serde_json::Value {
    serde_json::json!({
        "hooks": {
            "PreToolUse": [
                {
                    "matcher": "Bash|bash",
                    "hooks": [{
                        "type": "command",
                        "command": rewrite_cmd
                    }]
                },
                {
                    "matcher": "Read|read|ReadFile|read_file|View|view|Grep|grep|Search|search|ListFiles|list_files|ListDirectory|list_directory",
                    "hooks": [{
                        "type": "command",
                        "command": redirect_cmd
                    }]
                }
            ],
            "PostToolUse": [
                {
                    "matcher": ".*",
                    "hooks": [{
                        "type": "command",
                        "command": format!("{binary} hook post-tool-use"),
                        "timeout": 10
                    }]
                }
            ],
            "SessionStart": [
                {
                    "matcher": "startup|resume|compact",
                    "hooks": [{
                        "type": "command",
                        "command": format!("{binary} hook session-start"),
                        "timeout": 10
                    }]
                }
            ],
            "UserPromptSubmit": [
                {
                    "hooks": [{
                        "type": "command",
                        "command": format!("{binary} hook user-prompt-submit"),
                        "timeout": 5
                    }]
                }
            ],
            "PreCompact": [
                {
                    "hooks": [{
                        "type": "command",
                        "command": format!("{binary} hook pre-compact"),
                        "timeout": 15
                    }]
                }
            ],
            "Stop": [
                {
                    "hooks": [{
                        "type": "command",
                        "command": format!("{binary} hook stop"),
                        "timeout": 30
                    }]
                }
            ]
        }
    })
}

fn claude_content_dirs(home: &std::path::Path) -> Vec<PathBuf> {
    let canonical = home.join(".claude");
    let state_dir = crate::core::editor_registry::claude_state_dir(home);
    if state_dir == canonical {
        vec![canonical]
    } else {
        vec![canonical, state_dir]
    }
}

fn install_claude_global_claude_md(home: &std::path::Path) {
    for claude_dir in claude_content_dirs(home) {
        let _ = std::fs::create_dir_all(&claude_dir);
        let claude_md_path = claude_dir.join("CLAUDE.md");

        let existing = std::fs::read_to_string(&claude_md_path).unwrap_or_default();

        if existing.contains(CLAUDE_MD_BLOCK_START) {
            if existing.contains(CLAUDE_MD_BLOCK_VERSION) {
                continue;
            }
            let cleaned = remove_block(&existing, CLAUDE_MD_BLOCK_START, CLAUDE_MD_BLOCK_END);
            let updated = format!(
                "{}\n\n{}\n",
                cleaned.trim(),
                crate::public_guidance::claude_md_block_content()
            );
            write_file(&claude_md_path, &updated);
            continue;
        }

        if existing.trim().is_empty() {
            write_file(
                &claude_md_path,
                &crate::public_guidance::claude_md_block_content(),
            );
        } else {
            let updated = format!(
                "{}\n\n{}\n",
                existing.trim(),
                crate::public_guidance::claude_md_block_content()
            );
            write_file(&claude_md_path, &updated);
        }
    }
}

fn remove_block(content: &str, start: &str, end: &str) -> String {
    let s = content.find(start);
    let e = content.find(end);
    match (s, e) {
        (Some(si), Some(ei)) if ei >= si => {
            let after_end = ei + end.len();
            let before = content[..si].trim_end_matches('\n');
            let after = &content[after_end..];
            let mut out = before.to_string();
            out.push('\n');
            if !after.trim().is_empty() {
                out.push('\n');
                out.push_str(after.trim_start_matches('\n'));
            }
            out
        }
        _ => content.to_string(),
    }
}

fn install_claude_skill(home: &std::path::Path) {
    let skill_md = include_str!("../../assets/skills/project-bootstrap/SKILL.md");
    let install_sh = include_str!("../../assets/skills/project-bootstrap/scripts/install.sh");

    for claude_dir in claude_content_dirs(home) {
        let skill_dir = claude_dir.join(format!(
            "skills/{}",
            crate::core::editor_registry::PROJECT_BOOTSTRAP_SKILL_NAME
        ));
        let _ = std::fs::create_dir_all(skill_dir.join("scripts"));

        let skill_path = skill_dir.join("SKILL.md");
        let script_path = skill_dir.join("scripts/install.sh");

        write_file(&skill_path, skill_md);
        write_file(&script_path, install_sh);

        #[cfg(unix)]
        {
            use std::os::unix::fs::PermissionsExt;
            if let Ok(mut perms) = std::fs::metadata(&script_path).map(|m| m.permissions()) {
                perms.set_mode(0o755);
                let _ = std::fs::set_permissions(&script_path, perms);
            }
        }
    }
}

fn install_claude_rules_file(home: &std::path::Path) {
    let desired = crate::rules_inject::rules_dedicated_markdown();

    for claude_dir in claude_content_dirs(home) {
        let rules_dir = claude_dir.join("rules");
        let _ = std::fs::create_dir_all(&rules_dir);
        let rules_path = rules_dir.join("nebu-ctx.md");

        let existing = std::fs::read_to_string(&rules_path).unwrap_or_default();

        if existing.is_empty() {
            write_file(&rules_path, &desired);
            continue;
        }
        if existing.contains(crate::rules_inject::RULES_VERSION_STR) {
            continue;
        }
        if existing.contains("<!-- nebu-ctx-rules-") {
            write_file(&rules_path, &desired);
        }
    }
}

pub(super) fn install_claude_hook_scripts(home: &std::path::Path) {
    let hooks_dir = crate::core::editor_registry::claude_state_dir(home).join("hooks");
    let _ = std::fs::create_dir_all(&hooks_dir);

    let binary = resolve_binary_path();

    let rewrite_path = hooks_dir.join("nebu-ctx-rewrite.sh");
    let rewrite_script = generate_rewrite_script(&resolve_binary_path_for_bash());
    write_file(&rewrite_path, &rewrite_script);
    make_executable(&rewrite_path);

    let redirect_path = hooks_dir.join("nebu-ctx-redirect.sh");
    write_file(&redirect_path, REDIRECT_SCRIPT_CLAUDE);
    make_executable(&redirect_path);

    let wrapper = |subcommand: &str| -> String {
        if cfg!(windows) {
            format!("{binary} hook {subcommand}")
        } else {
            format!("{} hook {subcommand}", resolve_binary_path_for_bash())
        }
    };

    let rewrite_native = hooks_dir.join("nebu-ctx-rewrite-native");
    write_file(
        &rewrite_native,
        &format!(
            "#!/bin/sh\nexec {} hook rewrite\n",
            resolve_binary_path_for_bash()
        ),
    );
    make_executable(&rewrite_native);

    let redirect_native = hooks_dir.join("nebu-ctx-redirect-native");
    write_file(
        &redirect_native,
        &format!(
            "#!/bin/sh\nexec {} hook redirect\n",
            resolve_binary_path_for_bash()
        ),
    );
    make_executable(&redirect_native);

    let _ = wrapper; // suppress unused warning on unix
}

pub(super) fn install_claude_hook_config(home: &std::path::Path) {
    let hooks_dir = crate::core::editor_registry::claude_state_dir(home).join("hooks");
    let binary = resolve_binary_path();

    let rewrite_cmd = format!("{binary} hook rewrite");
    let redirect_cmd = format!("{binary} hook redirect");

    let settings_path = crate::core::editor_registry::claude_state_dir(home).join("settings.json");
    let settings_content = if settings_path.exists() {
        std::fs::read_to_string(&settings_path).unwrap_or_default()
    } else {
        String::new()
    };

    let needs_update = !settings_content.contains("hook rewrite")
        || !settings_content.contains("hook redirect")
        || !settings_content.contains("hook stop")
        || !settings_content.contains("hook post-tool-use")
        || !settings_content.contains("hook session-start")
        || !settings_content.contains("hook user-prompt-submit")
        || !settings_content.contains("hook pre-compact");
    let has_old_hooks = settings_content.contains("nebu-ctx-rewrite.sh")
        || settings_content.contains("nebu-ctx-redirect.sh");

    if !needs_update && !has_old_hooks {
        return;
    }

    let hook_entry = claude_hook_payload(&binary, rewrite_cmd, redirect_cmd);

    if settings_content.is_empty() {
        write_file(
            &settings_path,
            &serde_json::to_string_pretty(&hook_entry).unwrap(),
        );
    } else if let Ok(mut existing) = serde_json::from_str::<serde_json::Value>(&settings_content) {
        if let Some(obj) = existing.as_object_mut() {
            obj.insert("hooks".to_string(), hook_entry["hooks"].clone());
            write_file(
                &settings_path,
                &serde_json::to_string_pretty(&existing).unwrap(),
            );
        }
    }
    if !mcp_server_quiet_mode() {
        println!("Installed Claude Code hooks at {}", hooks_dir.display());
    }
}

pub(super) fn install_claude_project_hooks(cwd: &std::path::Path) {
    let binary = resolve_binary_path();
    let rewrite_cmd = format!("{binary} hook rewrite");
    let redirect_cmd = format!("{binary} hook redirect");

    let settings_path = cwd.join(".claude").join("settings.local.json");
    let _ = std::fs::create_dir_all(cwd.join(".claude"));

    let existing = std::fs::read_to_string(&settings_path).unwrap_or_default();
    if existing.contains("hook rewrite")
        && existing.contains("hook redirect")
        && existing.contains("hook stop")
        && existing.contains("hook post-tool-use")
        && existing.contains("hook session-start")
        && existing.contains("hook user-prompt-submit")
        && existing.contains("hook pre-compact")
    {
        return;
    }

    let hook_entry = claude_hook_payload(&binary, rewrite_cmd, redirect_cmd);

    if existing.is_empty() {
        write_file(
            &settings_path,
            &serde_json::to_string_pretty(&hook_entry).unwrap(),
        );
    } else if let Ok(mut json) = serde_json::from_str::<serde_json::Value>(&existing) {
        if let Some(obj) = json.as_object_mut() {
            obj.insert("hooks".to_string(), hook_entry["hooks"].clone());
            write_file(
                &settings_path,
                &serde_json::to_string_pretty(&json).unwrap(),
            );
        }
    }
    println!("Created .claude/settings.local.json (project-local PreToolUse hooks).");
}

pub fn install_codex_hook() {
    let home = match dirs::home_dir() {
        Some(h) => h,
        None => {
            eprintln!("Cannot resolve home directory");
            return;
        }
    };

    let codex_dir = home.join(".codex");
    let _ = std::fs::create_dir_all(&codex_dir);

    let hook_config_changed = install_codex_hook_config(&home);
    let installed_docs = install_codex_instruction_docs(&codex_dir);

    if !mcp_server_quiet_mode() {
        if hook_config_changed {
            eprintln!(
                "Installed Codex-compatible SessionStart/PreToolUse hooks at {}",
                codex_dir.display()
            );
        }
        if installed_docs {
            eprintln!("Installed Codex instructions at {}", codex_dir.display());
        } else {
            eprintln!("Codex AGENTS.md already configured.");
        }
    }
}

fn install_codex_hook_config(home: &std::path::Path) -> bool {
    let binary = resolve_binary_path();
    let session_start_cmd = format!("{binary} hook codex-session-start");
    let pre_tool_use_cmd = format!("{binary} hook codex-pretooluse");
    let codex_dir = home.join(".codex");
    let hooks_json_path = codex_dir.join("hooks.json");

    let mut changed = false;
    let mut root = if hooks_json_path.exists() {
        match std::fs::read_to_string(&hooks_json_path)
            .ok()
            .and_then(|content| serde_json::from_str::<serde_json::Value>(&content).ok())
        {
            Some(parsed) => parsed,
            None => {
                changed = true;
                serde_json::json!({ "hooks": {} })
            }
        }
    } else {
        changed = true;
        serde_json::json!({ "hooks": {} })
    };

    if upsert_lean_ctx_codex_hook_entries(&mut root, &session_start_cmd, &pre_tool_use_cmd) {
        changed = true;
    }
    if changed {
        write_file(
            &hooks_json_path,
            &serde_json::to_string_pretty(&root).unwrap(),
        );
    }

    let rewrite_path = codex_dir.join("hooks").join("nebu-ctx-rewrite-codex.sh");
    if rewrite_path.exists() && std::fs::remove_file(&rewrite_path).is_ok() {
        changed = true;
    }

    let config_toml_path = codex_dir.join("config.toml");
    let config_content = std::fs::read_to_string(&config_toml_path).unwrap_or_default();
    if let Some(updated) = ensure_codex_hooks_enabled(&config_content) {
        write_file(&config_toml_path, &updated);
        changed = true;
        if !mcp_server_quiet_mode() {
            eprintln!(
                "Enabled codex_hooks feature in {}",
                config_toml_path.display()
            );
        }
    }

    changed
}

fn ensure_codex_hooks_enabled(config_content: &str) -> Option<String> {
    shared_ensure_codex_hooks_enabled(config_content)
}

pub(super) fn install_copilot_hook(global: bool) {
    let binary = resolve_binary_path();

    if global {
        let mcp_path = crate::core::editor_registry::vscode_mcp_path();
        if mcp_path.as_os_str() == "/nonexistent" {
            println!("  \x1b[2mVS Code not found — skipping global Copilot config\x1b[0m");
            return;
        }
        write_vscode_mcp_file(&mcp_path, &binary, "global VS Code User MCP");
        install_copilot_pretooluse_hook(true);
    } else {
        let vscode_dir = PathBuf::from(".vscode");
        let _ = std::fs::create_dir_all(&vscode_dir);
        let mcp_path = vscode_dir.join("mcp.json");
        write_vscode_mcp_file(&mcp_path, &binary, ".vscode/mcp.json");
        install_copilot_pretooluse_hook(false);
    }
}

fn install_copilot_pretooluse_hook(global: bool) {
    // Use platform-specific command fields that match the working Copilot hooks schema.
    let bash_binary = resolve_binary_path_for_bash();
    let ps_binary = resolve_binary_path();

    let hook_config = copilot_hook_payload(&bash_binary, &ps_binary);

    let hook_path = if global {
        let Some(home) = dirs::home_dir() else { return };
        let dir = home.join(".github").join("hooks");
        let _ = std::fs::create_dir_all(&dir);
        dir.join("hooks.json")
    } else {
        let dir = PathBuf::from(".github").join("hooks");
        let _ = std::fs::create_dir_all(&dir);
        dir.join("hooks.json")
    };

    let needs_write = hook_config_needs_update(&hook_path);

    if !needs_write {
        return;
    }

    if hook_path.exists() {
        if let Ok(mut existing) = serde_json::from_str::<serde_json::Value>(
            &std::fs::read_to_string(&hook_path).unwrap_or_default(),
        ) {
            if let Some(obj) = existing.as_object_mut() {
                obj.remove("version");
                let hooks = obj
                    .entry("hooks".to_string())
                    .or_insert_with(|| serde_json::json!({}));
                if let Some(hooks_obj) = hooks.as_object_mut() {
                    merge_copilot_hooks(hooks_obj, &hook_config["hooks"]);
                } else {
                    obj.insert("hooks".to_string(), hook_config["hooks"].clone());
                }
                write_file(
                    &hook_path,
                    &serde_json::to_string_pretty(&existing).unwrap(),
                );
                if !mcp_server_quiet_mode() {
                    println!("Updated Copilot hooks at {}", hook_path.display());
                }
                return;
            }
        }
    }

    write_file(
        &hook_path,
        &serde_json::to_string_pretty(&hook_config).unwrap(),
    );
    if !mcp_server_quiet_mode() {
        println!("Installed Copilot hooks at {}", hook_path.display());
    }
}

// Detect when the installed Copilot hooks.json is missing the current nebu-ctx entries.
fn hook_config_needs_update(hook_path: &std::path::Path) -> bool {
    let content = std::fs::read_to_string(hook_path).unwrap_or_default();
    if content.is_empty() {
        return true;
    }

    if !content.contains("hook rewrite")
        || !content.contains("hook stop")
        || !content.contains("hook post-tool-use")
    {
        return true;
    }

    content.contains("\"preToolUse\"")
        || content.contains("\"postToolUse\"")
        || content.contains("\"postSession\"")
        || content.contains("\"bash\"")
        || content.contains("\"powershell\"")
        || content.contains("\"timeoutSec\"")
        || content.contains("\"version\"")
}

// Replace only the nebu-ctx-owned Copilot hook events and preserve unrelated custom hooks.
fn merge_copilot_hooks(
    hooks_obj: &mut serde_json::Map<String, serde_json::Value>,
    desired_hooks: &serde_json::Value,
) {
    for legacy_event_name in ["preToolUse", "postToolUse", "postSession"] {
        hooks_obj.remove(legacy_event_name);
    }

    if let Some(desired_obj) = desired_hooks.as_object() {
        for (event_name, event_value) in desired_obj {
            hooks_obj.insert(event_name.clone(), event_value.clone());
        }
    }
}

// Build the Copilot hooks.json payload using the schema that Copilot currently accepts.
fn copilot_hook_payload(bash_binary: &str, ps_binary: &str) -> serde_json::Value {
    serde_json::json!({
        "hooks": {
            "PreToolUse": [
                {
                    "type": "command",
                    "command": format!("{bash_binary} hook rewrite"),
                    "windows": format!("{ps_binary} hook rewrite"),
                    "linux": format!("{bash_binary} hook rewrite"),
                    "osx": format!("{bash_binary} hook rewrite")
                },
                {
                    "type": "command",
                    "command": format!("{bash_binary} hook redirect"),
                    "windows": format!("{ps_binary} hook redirect"),
                    "linux": format!("{bash_binary} hook redirect"),
                    "osx": format!("{bash_binary} hook redirect")
                }
            ],
            "PostToolUse": [
                {
                    "type": "command",
                    "command": format!("{bash_binary} hook post-tool-use"),
                    "windows": format!("{ps_binary} hook post-tool-use"),
                    "linux": format!("{bash_binary} hook post-tool-use"),
                    "osx": format!("{bash_binary} hook post-tool-use")
                }
            ],
            "PostSession": [
                {
                    "type": "command",
                    "command": format!("{bash_binary} hook stop"),
                    "windows": format!("{ps_binary} hook stop"),
                    "linux": format!("{bash_binary} hook stop"),
                    "osx": format!("{bash_binary} hook stop")
                }
            ]
        }
    })
}

fn write_vscode_mcp_file(mcp_path: &PathBuf, binary: &str, label: &str) {
    let data_dir = crate::core::data_dir::nebu_ctx_data_dir()
        .map(|d| d.to_string_lossy().to_string())
        .unwrap_or_default();
    let desired = serde_json::json!({ "type": "stdio", "command": binary, "args": [], "env": { "NEBU_CTX_DATA_DIR": data_dir } });
    let preferred_key = crate::core::editor_registry::COPILOT_MCP_SERVER_KEY;
    let legacy_keys = crate::core::editor_registry::COPILOT_LEGACY_MCP_SERVER_KEYS;
    if mcp_path.exists() {
        let content = std::fs::read_to_string(mcp_path).unwrap_or_default();
        match serde_json::from_str::<serde_json::Value>(&content) {
            Ok(mut json) => {
                if let Some(obj) = json.as_object_mut() {
                    let servers = obj
                        .entry("servers")
                        .or_insert_with(|| serde_json::json!({}));
                    if let Some(servers_obj) = servers.as_object_mut() {
                        let existing = servers_obj.get(preferred_key).cloned().or_else(|| {
                            legacy_keys
                                .iter()
                                .find_map(|key| servers_obj.get(*key).cloned())
                        });
                        let has_preferred = servers_obj.contains_key(preferred_key);
                        let had_legacy =
                            legacy_keys.iter().any(|key| servers_obj.contains_key(*key));
                        if existing.as_ref() == Some(&desired) && has_preferred && !had_legacy {
                            if !crate::hooks::mcp_server_quiet_mode() {
                                println!(
                                    "  \x1b[32m✓\x1b[0m Copilot already configured in {label}"
                                );
                            }
                            return;
                        }
                        for key in legacy_keys {
                            let _ = servers_obj.remove(*key);
                        }
                        servers_obj.insert(preferred_key.to_string(), desired);
                    }
                    write_file(
                        mcp_path,
                        &serde_json::to_string_pretty(&json).unwrap_or_default(),
                    );
                    if !crate::hooks::mcp_server_quiet_mode() {
                        println!("  \x1b[32m✓\x1b[0m Configured nebu-ctx MCP server in {label}");
                    }
                    return;
                }
            }
            Err(e) => {
                eprintln!(
                    "Could not parse VS Code MCP config at {}: {e}\nAdd to \"servers\": \"{}\": {{ \"command\": \"{}\", \"args\": [] }}",
                    mcp_path.display(),
                    preferred_key,
                    binary
                );
                return;
            }
        };
    }

    if let Some(parent) = mcp_path.parent() {
        let _ = std::fs::create_dir_all(parent);
    }

    let data_dir = crate::core::data_dir::nebu_ctx_data_dir()
        .map(|d| d.to_string_lossy().to_string())
        .unwrap_or_default();
    let config = serde_json::json!({
        "servers": {
            preferred_key: {
                "type": "stdio",
                "command": binary,
                "args": [],
                "env": { "NEBU_CTX_DATA_DIR": data_dir }
            }
        }
    });

    write_file(
        mcp_path,
        &serde_json::to_string_pretty(&config).unwrap_or_default(),
    );
    if !crate::hooks::mcp_server_quiet_mode() {
        println!("  \x1b[32m✓\x1b[0m Created {label} with nebu-ctx MCP server");
    }
}

pub(super) fn install_opencode_hook() {
    let home = dirs::home_dir().unwrap_or_default();
    let binary = resolve_binary_path();
    let target = crate::core::editor_registry::EditorTarget {
        name: "OpenCode",
        agent_key: "opencode".to_string(),
        config_path: crate::core::editor_registry::opencode_config_path(&home),
        detect_path: crate::core::editor_registry::opencode_config_dir(&home),
        config_type: crate::core::editor_registry::ConfigType::OpenCode,
    };

    match crate::core::editor_registry::write_config_with_options(
        &target,
        &binary,
        crate::core::editor_registry::WriteOptions {
            overwrite_invalid: true,
        },
    ) {
        Ok(result) => {
            let message = match result.action {
                crate::core::editor_registry::WriteAction::Already => {
                    "OpenCode MCP already configured at ~/.config/opencode/opencode.json"
                }
                _ => {
                    "  \x1b[32m✓\x1b[0m OpenCode MCP configured at ~/.config/opencode/opencode.json"
                }
            };
            println!("{message}");
        }
        Err(_) => {
            eprintln!("  \x1b[31m✗\x1b[0m Failed to configure OpenCode");
        }
    }
}

#[cfg(test)]
mod memory_hook_tests {
    use super::{
        claude_hook_payload, copilot_hook_payload, hook_config_needs_update, merge_copilot_hooks,
        write_vscode_mcp_file,
    };

    #[test]
    fn claude_hook_payload_contains_memory_lifecycle_hooks() {
        let payload = claude_hook_payload(
            "nebu-ctx",
            "nebu-ctx hook rewrite".to_string(),
            "nebu-ctx hook redirect".to_string(),
        );

        let hooks = payload["hooks"].as_object().unwrap();
        assert!(hooks.contains_key("SessionStart"));
        assert!(hooks.contains_key("UserPromptSubmit"));
        assert!(hooks.contains_key("PreCompact"));
        assert!(hooks.contains_key("Stop"));
    }

    #[test]
    fn copilot_hook_payload_routes_shared_memory_handlers() {
        let payload = copilot_hook_payload("nebu-ctx", "nebu-ctx");

        let hooks = payload["hooks"].as_object().unwrap();
        assert!(hooks.contains_key("PostToolUse"));
        assert!(hooks.contains_key("PostSession"));
        assert_eq!(
            hooks["PostToolUse"][0]["command"],
            "nebu-ctx hook post-tool-use"
        );
        assert_eq!(
            hooks["PostToolUse"][0]["windows"],
            "nebu-ctx hook post-tool-use"
        );
        assert_eq!(hooks["PostSession"][0]["command"], "nebu-ctx hook stop");
        assert_eq!(hooks["PostSession"][0]["windows"], "nebu-ctx hook stop");
    }

    #[test]
    fn hook_config_needs_update_flags_legacy_copilot_schema() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("hooks.json");
        std::fs::write(
            &path,
            r#"{
    "version": 1,
    "hooks": {
        "preToolUse": [{
            "type": "command",
            "bash": "nebu-ctx hook rewrite",
            "powershell": "nebu-ctx hook rewrite",
            "timeoutSec": 15
        }],
        "postToolUse": [{
            "type": "command",
            "bash": "nebu-ctx hook post-tool-use",
            "powershell": "nebu-ctx hook post-tool-use",
            "timeoutSec": 10
        }],
        "postSession": [{
            "type": "command",
            "bash": "nebu-ctx hook stop",
            "powershell": "nebu-ctx hook stop",
            "timeoutSec": 30
        }]
    }
}"#,
        )
        .unwrap();

        assert!(hook_config_needs_update(&path));
    }

    #[test]
    fn merge_copilot_hooks_preserves_custom_events() {
        let mut hooks = serde_json::json!({
                "CustomEvent": [{
                        "type": "command",
                        "command": "echo custom"
                }]
        })
        .as_object()
        .unwrap()
        .clone();

        let desired = copilot_hook_payload("nebu-ctx", "nebu-ctx");
        merge_copilot_hooks(&mut hooks, &desired["hooks"]);

        assert!(hooks.contains_key("CustomEvent"));
        assert!(hooks.contains_key("PreToolUse"));
        assert_eq!(hooks["CustomEvent"][0]["command"], "echo custom");
    }

    #[test]
    fn merge_copilot_hooks_replaces_legacy_schema_entries() {
        let mut hooks = serde_json::json!({
            "preToolUse": [{
                "type": "command",
                "bash": "nebu-ctx hook rewrite"
            }],
            "postToolUse": [{
                "type": "command",
                "bash": "nebu-ctx hook post-tool-use"
            }],
            "postSession": [{
                "type": "command",
                "bash": "nebu-ctx hook stop"
            }],
            "CustomEvent": [{
                "type": "command",
                "command": "echo custom"
            }]
        })
        .as_object()
        .unwrap()
        .clone();

        let desired = copilot_hook_payload("nebu-ctx", "nebu-ctx");
        merge_copilot_hooks(&mut hooks, &desired["hooks"]);

        assert!(!hooks.contains_key("preToolUse"));
        assert!(!hooks.contains_key("postToolUse"));
        assert!(!hooks.contains_key("postSession"));
        assert!(hooks.contains_key("PreToolUse"));
        assert!(hooks.contains_key("CustomEvent"));
    }

    #[test]
    fn write_vscode_mcp_file_uses_camel_case_server_key() {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("mcp.json");

        write_vscode_mcp_file(&path, "/usr/local/bin/nebu-ctx", "test");

        let json: serde_json::Value =
            serde_json::from_str(&std::fs::read_to_string(&path).unwrap()).unwrap();
        assert!(json["servers"].get("nebu-ctx").is_none());
        assert_eq!(
            json["servers"][crate::core::editor_registry::COPILOT_MCP_SERVER_KEY]["command"],
            "/usr/local/bin/nebu-ctx"
        );
    }
}

#[cfg(test)]
mod tests {
    use super::{ensure_codex_hooks_enabled, upsert_lean_ctx_codex_hook_entries};
    use serde_json::json;

    #[test]
    fn upsert_replaces_legacy_codex_rewrite_but_keeps_custom_hooks() {
        let mut input = json!({
            "hooks": {
                "PreToolUse": [
                    {
                        "matcher": "Bash",
                        "hooks": [{
                            "type": "command",
                            "command": "/opt/homebrew/bin/nebu-ctx hook rewrite",
                            "timeout": 15
                        }]
                    },
                    {
                        "matcher": "Bash",
                        "hooks": [{
                            "type": "command",
                            "command": "echo keep-me",
                            "timeout": 5
                        }]
                    }
                ],
                "SessionStart": [
                    {
                        "matcher": "startup|resume|clear",
                        "hooks": [{
                            "type": "command",
                            "command": "nebu-ctx hook codex-session-start",
                            "timeout": 15
                        }]
                    }
                ],
                "PostToolUse": [
                    {
                        "matcher": "Bash",
                        "hooks": [{
                            "type": "command",
                            "command": "echo keep-post",
                            "timeout": 5
                        }]
                    }
                ]
            }
        });

        let changed = upsert_lean_ctx_codex_hook_entries(
            &mut input,
            "nebu-ctx hook codex-session-start",
            "nebu-ctx hook codex-pretooluse",
        );
        assert!(changed, "legacy hooks should be migrated");

        let pre_tool_use = input["hooks"]["PreToolUse"]
            .as_array()
            .expect("PreToolUse array should remain");
        assert_eq!(pre_tool_use.len(), 2, "custom hook should be preserved");
        assert_eq!(
            pre_tool_use[0]["hooks"][0]["command"].as_str(),
            Some("echo keep-me")
        );
        assert_eq!(
            pre_tool_use[1]["hooks"][0]["command"].as_str(),
            Some("nebu-ctx hook codex-pretooluse")
        );
        assert_eq!(
            input["hooks"]["SessionStart"][0]["hooks"][0]["command"].as_str(),
            Some("nebu-ctx hook codex-session-start")
        );
        assert_eq!(
            input["hooks"]["PostToolUse"][0]["hooks"][0]["command"].as_str(),
            Some("echo keep-post")
        );
    }

    #[test]
    fn ignores_non_lean_ctx_codex_entries() {
        let custom = json!({
            "matcher": "Bash",
            "hooks": [{
                "type": "command",
                "command": "echo keep-me",
                "timeout": 5
            }]
        });
        assert!(
            !super::super::support::is_lean_ctx_codex_managed_entry("PreToolUse", &custom),
            "custom Codex hooks must be preserved"
        );
    }

    #[test]
    fn detects_managed_codex_session_start_entry() {
        let managed = json!({
            "matcher": "startup|resume|clear",
            "hooks": [{
                "type": "command",
                "command": "/opt/homebrew/bin/nebu-ctx hook codex-session-start",
                "timeout": 15
            }]
        });
        assert!(super::super::support::is_lean_ctx_codex_managed_entry(
            "SessionStart",
            &managed
        ));
    }

    #[test]
    fn ensure_codex_hooks_enabled_updates_existing_features_flag() {
        let input = "\
[features]
other = true
codex_hooks = false

[mcp_servers.other]
command = \"other\"
";

        let output =
            ensure_codex_hooks_enabled(input).expect("codex_hooks=false should be migrated");

        assert!(output.contains("[features]\nother = true\ncodex_hooks = true\n"));
        assert!(!output.contains("codex_hooks = false"));
    }

    #[test]
    fn ensure_codex_hooks_enabled_moves_stray_assignment_into_features_section() {
        let input = "\
[features]
other = true

[mcp_servers.lean-ctx]
command = \"lean-ctx\"
codex_hooks = true
";

        let output = ensure_codex_hooks_enabled(input)
            .expect("stray codex_hooks assignment should be normalized");

        assert!(output.contains("[features]\nother = true\ncodex_hooks = true\n"));
        assert_eq!(output.matches("codex_hooks = true").count(), 1);
        assert!(
            !output.contains("[mcp_servers.lean-ctx]\ncommand = \"lean-ctx\"\ncodex_hooks = true")
        );
    }

    #[test]
    fn ensure_codex_hooks_enabled_adds_features_section_when_missing() {
        let input = "\
[mcp_servers.lean-ctx]
command = \"lean-ctx\"
";

        let output =
            ensure_codex_hooks_enabled(input).expect("missing features section should be added");

        assert!(output.ends_with("\n[features]\ncodex_hooks = true\n"));
    }
}