episteme 0.3.1

//! Install command: install Episteme into AI tools.

use std::path::PathBuf;

use anyhow::Result;

use episteme::adapters::config::EpistemeConfig;

pub fn cmd_install(tools: &[String], all: bool, dry_run: bool, local: bool) -> Result<()> {
    use episteme::adapters::installer;
    use std::io::{self, IsTerminal};

    // --- Data seeding ---
    let seeded = if local {
        // --local: try dist/ archive first, fallback to raw/meta/ source tree
        println!("Seeding data (local)...");
        let cwd = std::env::current_dir().map_err(|e| anyhow::anyhow!(e))?;
        let dist_archive = find_dist_archive(&cwd);

        if let Some(archive_path) = dist_archive {
            println!("  Using dist archive: {}", archive_path.display());
            for msg in installer::seed_data_from_local_archive(&archive_path, dry_run)
                .map_err(|e| anyhow::anyhow!(e))?
            {
                println!("  {msg}");
            }
        } else {
            println!("  No dist archive found, seeding from source tree...");
            for msg in installer::seed_data(dry_run).map_err(|e| anyhow::anyhow!(e))? {
                println!("  {msg}");
            }
        }
        true
    } else if !dry_run {
        // Default: download from GitHub release
        println!("Fetching data from GitHub Releases...");
        let url = resolve_release_url().map_err(|e| anyhow::anyhow!(e))?;
        println!("  Downloading: {url}");
        for msg in
            installer::seed_data_from_release(&url, dry_run).map_err(|e| anyhow::anyhow!(e))?
        {
            println!("  {msg}");
        }
        true
    } else {
        false
    };

    // --- Build RAG index (skip if DB already provided by archive) ---
    if seeded && !dry_run && !episteme::adapters::paths::db_path().exists() {
        println!("\nBuilding RAG index...");
        super::build::cmd_build(None, None, false, false, 64, true, false)?;
    }

    // --- Tool installation ---
    use episteme::adapters::installer::Transport;

    let mut selected: Vec<String> = if all || tools.iter().any(|t| t == "all") {
        vec!["claude", "cursor", "codex", "opencode", "cline"]
            .into_iter()
            .map(|s| s.to_owned())
            .collect()
    } else if tools.is_empty() && io::stdin().is_terminal() {
        let installed: Vec<&str> = detect_installed_tools().into_iter().collect();
        let selected =
            match episteme::adapters::install_wizard::interactive_select_tools(&installed) {
                Ok(s) if !s.is_empty() => s,
                Ok(_) => anyhow::bail!("install cancelled"),
                Err(e) => {
                    eprintln!("Interactive UI failed ({e}); falling back to text prompt.");
                    episteme::adapters::install_wizard::fallback_select_tools()
                }
            };

        let cfg = EpistemeConfig::load().unwrap_or_default();
        if let Some(redis) = episteme::adapters::install_wizard::configure_redis_tui(
            episteme::adapters::install_wizard::RedisConfig {
                enabled: cfg.redis_enabled,
                host: cfg.redis_host.clone(),
                port: cfg.redis_port,
                db: cfg.redis_db,
                ttl: cfg.redis_ttl,
            },
        )
        .map_err(|e| anyhow::anyhow!(e))?
        {
            upsert_config_yaml(redis.enabled, &redis.host, redis.port, redis.db, redis.ttl)?;
        }

        let telemetry_enabled = episteme::adapters::install_wizard::configure_telemetry_tui()
            .map_err(|e| anyhow::anyhow!(e))?;
        episteme::adapters::telemetry::write_consent(telemetry_enabled)
            .map_err(|e| anyhow::anyhow!(e))?;

        selected
    } else {
        tools.to_vec()
    };
    selected.sort();
    selected.dedup();

    let transport = Transport::default();

    for tool in &selected {
        let result = match tool.as_str() {
            "claude" => installer::install_claude(dry_run, &transport),
            "cursor" => installer::install_cursor(dry_run, &transport),
            "codex" => installer::install_codex(dry_run),
            "opencode" => installer::install_opencode(dry_run, &transport),
            "cline" => installer::install_cline(dry_run, &transport),
            _ => Err(format!("Unknown tool: {tool}")),
        };
        match result {
            Ok(msgs) => {
                for msg in msgs {
                    println!("  {msg}");
                }
            }
            Err(e) => eprintln!("  Error ({tool}): {e}"),
        }
    }

    // --- API server configuration ---
    let api_port = if io::stdin().is_terminal() && !dry_run {
        let cfg = EpistemeConfig::load().unwrap_or_default();
        let sc = episteme::adapters::install_wizard::configure_server_tui(
            "API server",
            &cfg.api_host,
            cfg.api_port,
            &cfg.api_keys,
        )
        .map_err(|e| anyhow::anyhow!(e))?;
        upsert_api_config_yaml(&sc.host, sc.port, sc.token.as_deref())?;
        // Ensure enable_service (and install_launchd_agent_for) sees the chosen port
        // even in binary versions that predate yaml api-section reading.
        // SAFETY: single-threaded install path, no concurrent env reads.
        unsafe {
            std::env::set_var("UVICORN_PORT", sc.port.to_string());
            std::env::set_var("UVICORN_HOST", &sc.host);
        }
        sc.port
    } else {
        EpistemeConfig::load().unwrap_or_default().api_port
    };

    // --- Enable API server as login service ---
    if !dry_run {
        use episteme::adapters::service::{ServiceKind, enable_service};
        // Non-TTY path: propagate port so enable_service uses the configured value.
        if !io::stdin().is_terminal() {
            // SAFETY: single-threaded install path, no concurrent env reads.
            unsafe { std::env::set_var("UVICORN_PORT", api_port.to_string()) };
        }
        match enable_service(ServiceKind::Api, true) {
            Ok(msg) => println!("  {msg}"),
            Err(e) => eprintln!(
                "  Warning: could not enable API server: {e}\n  Run 'epis api enable --now' manually."
            ),
        }
    }

    episteme::adapters::telemetry::track_install_completed(selected.len());

    Ok(())
}

/// Find the newest `episteme-data-*.tar.gz` in `dist/`.
fn find_dist_archive(cwd: &std::path::Path) -> Option<PathBuf> {
    let dist_dir = cwd.join("dist");
    if !dist_dir.is_dir() {
        return None;
    }
    let mut newest: Option<(PathBuf, std::time::SystemTime)> = None;
    let Ok(entries) = std::fs::read_dir(&dist_dir) else {
        return None;
    };
    for entry in entries.flatten() {
        let name = entry.file_name();
        let name_str = name.to_string_lossy();
        if name_str.starts_with("episteme-data-")
            && name_str.ends_with(".tar.gz")
            && let Ok(meta) = entry.metadata()
            && let Ok(modified) = meta.modified()
            && newest.as_ref().map(|(_, t)| modified > *t).unwrap_or(true)
        {
            newest = Some((entry.path(), modified));
        }
    }
    newest.map(|(p, _)| p)
}

/// Resolve the GitHub release download URL for the current version.
fn resolve_release_url() -> Result<String> {
    let version = get_package_version();
    let repo = "epicsagas/Episteme";
    let prefix = "episteme-data-";

    // Try exact version tag first, then latest
    for endpoint in [
        format!("https://api.github.com/repos/{repo}/releases/tags/v{version}"),
        format!("https://api.github.com/repos/{repo}/releases/latest"),
    ] {
        if let Ok(url) = fetch_release_asset_url(&endpoint, prefix) {
            return Ok(url);
        }
    }

    anyhow::bail!("No data asset found for v{version}.\nCheck: https://github.com/{repo}/releases")
}

fn get_package_version() -> String {
    std::env::var("EPISTEME_VERSION").unwrap_or_else(|_| env!("CARGO_PKG_VERSION").to_owned())
}

fn fetch_release_asset_url(api_url: &str, prefix: &str) -> Result<String> {
    let output = std::process::Command::new("curl")
        .args([
            "-LfsS",
            "-H",
            "Accept: application/vnd.github+json",
            api_url,
        ])
        .output()
        .map_err(|e| anyhow::anyhow!("curl failed: {e}"))?;

    if !output.status.success() {
        anyhow::bail!("curl returned non-zero");
    }

    let json_str = String::from_utf8_lossy(&output.stdout);
    let val: serde_json::Value =
        serde_json::from_str(&json_str).map_err(|e| anyhow::anyhow!("JSON parse: {e}"))?;

    if let Some(assets) = val.get("assets").and_then(|a| a.as_array()) {
        for asset in assets {
            let name = asset.get("name").and_then(|n| n.as_str()).unwrap_or("");
            if name.starts_with(prefix)
                && name.ends_with(".tar.gz")
                && let Some(url) = asset.get("browser_download_url").and_then(|u| u.as_str())
            {
                return Ok(url.to_owned());
            }
        }
    }

    anyhow::bail!("no matching asset in release")
}

pub fn detect_installed_tools() -> std::collections::HashSet<&'static str> {
    use serde_json::Value;
    use std::collections::HashSet;
    use std::path::Path;

    fn has_json_path(path: &Path, parent: &str, child: &str) -> bool {
        let Ok(text) = std::fs::read_to_string(path) else {
            return false;
        };
        let Ok(v) = serde_json::from_str::<Value>(&text) else {
            return false;
        };
        v.get(parent).and_then(|m| m.get(child)).is_some()
    }

    let mut installed = HashSet::new();
    let home = std::env::var("HOME").unwrap_or_default();
    let cwd = std::env::current_dir().unwrap_or_else(|_| PathBuf::from("."));

    if has_json_path(
        &PathBuf::from(&home).join(".claude.json"),
        "mcpServers",
        "episteme",
    ) {
        installed.insert("claude");
    }
    if has_json_path(
        &PathBuf::from(&home).join(".cursor").join("mcp.json"),
        "mcpServers",
        "episteme",
    ) {
        installed.insert("cursor");
    }
    if has_json_path(
        &PathBuf::from(&home)
            .join(".config")
            .join("opencode")
            .join("opencode.json"),
        "mcp",
        "episteme",
    ) {
        installed.insert("opencode");
    }
    let cline_path = if cfg!(target_os = "macos") {
        PathBuf::from(&home)
            .join("Library")
            .join("Application Support")
            .join("Code")
            .join("User")
            .join("globalStorage")
            .join("saoudrizwan.claude-dev")
    } else {
        PathBuf::from(&home)
            .join(".config")
            .join("Code")
            .join("User")
            .join("globalStorage")
            .join("saoudrizwan.claude-dev")
    };
    if cline_path.exists() {
        installed.insert("cline");
    }
    if let Ok(content) = std::fs::read_to_string(cwd.join("AGENTS.md"))
        && (content.contains("EPISTEME-BEGIN") || content.contains("epis mcp"))
    {
        installed.insert("codex");
    }
    installed
}

fn upsert_config_yaml(
    redis_enabled: bool,
    redis_host: &str,
    redis_port: u16,
    redis_db: u16,
    redis_ttl: u64,
) -> Result<()> {
    use serde_yaml::{Mapping, Value};
    let path = episteme::adapters::paths::episteme_home().join("config.yaml");
    let mut root = if path.exists() {
        let text = std::fs::read_to_string(&path)?;
        serde_yaml::from_str::<Value>(&text).unwrap_or_else(|_| Value::Mapping(Mapping::new()))
    } else {
        Value::Mapping(Mapping::new())
    };

    if !root.is_mapping() {
        root = Value::Mapping(Mapping::new());
    }
    let root_map = root.as_mapping_mut().expect("mapping checked above");

    let mut redis_map = Mapping::new();
    redis_map.insert(
        Value::String("enabled".to_owned()),
        Value::Bool(redis_enabled),
    );
    redis_map.insert(
        Value::String("host".to_owned()),
        Value::String(redis_host.to_owned()),
    );
    redis_map.insert(
        Value::String("port".to_owned()),
        Value::Number(serde_yaml::Number::from(redis_port)),
    );
    redis_map.insert(
        Value::String("db".to_owned()),
        Value::Number(serde_yaml::Number::from(redis_db)),
    );
    redis_map.insert(
        Value::String("ttl".to_owned()),
        Value::Number(serde_yaml::Number::from(redis_ttl)),
    );
    root_map.insert(Value::String("redis".to_owned()), Value::Mapping(redis_map));

    std::fs::create_dir_all(episteme::adapters::paths::episteme_home())?;
    let yaml = serde_yaml::to_string(&root)?;
    std::fs::write(path, yaml)?;
    Ok(())
}

fn upsert_api_config_yaml(host: &str, port: u16, keys: Option<&str>) -> Result<()> {
    use serde_yaml::{Mapping, Value};
    let path = episteme::adapters::paths::episteme_home().join("config.yaml");
    let mut root = if path.exists() {
        let text = std::fs::read_to_string(&path)?;
        serde_yaml::from_str::<Value>(&text).unwrap_or_else(|_| Value::Mapping(Mapping::new()))
    } else {
        Value::Mapping(Mapping::new())
    };
    if !root.is_mapping() {
        root = Value::Mapping(Mapping::new());
    }
    let root_map = root.as_mapping_mut().expect("mapping checked above");

    let mut api_map = Mapping::new();
    api_map.insert(
        Value::String("host".to_owned()),
        Value::String(host.to_owned()),
    );
    api_map.insert(
        Value::String("port".to_owned()),
        Value::Number(serde_yaml::Number::from(port)),
    );
    if let Some(k) = keys.filter(|k| !k.is_empty()) {
        api_map.insert(
            Value::String("keys".to_owned()),
            Value::String(k.to_owned()),
        );
    }
    root_map.insert(Value::String("api".to_owned()), Value::Mapping(api_map));

    std::fs::create_dir_all(episteme::adapters::paths::episteme_home())?;
    let yaml = serde_yaml::to_string(&root)?;
    std::fs::write(path, yaml)?;
    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::collections::HashSet;

    /// Cline이 VS Code globalStorage 경로로 감지되는지 확인.
    /// 임시 디렉토리에 경로를 생성한 뒤 HOME을 덮어써서 격리 테스트.
    #[test]
    fn detect_cline_via_vscode_global_storage() {
        let tmp = tempfile::tempdir().unwrap();
        let home = tmp.path();

        // macOS 경로 생성
        let cline_dir = home
            .join("Library")
            .join("Application Support")
            .join("Code")
            .join("User")
            .join("globalStorage")
            .join("saoudrizwan.claude-dev");
        std::fs::create_dir_all(&cline_dir).unwrap();

        // HOME을 임시 디렉토리로 교체
        // SAFETY: 단일 스레드 테스트 내에서 환경 변수를 격리 목적으로만 변경
        unsafe { std::env::set_var("HOME", home) };

        let installed: HashSet<&str> = detect_installed_tools();
        assert!(
            installed.contains("cline"),
            "cline이 VS Code globalStorage 경로에서 감지되어야 합니다"
        );
    }

    /// TOOLS 배열에 cline 항목이 존재하는지 확인.
    #[test]
    fn tools_array_contains_cline() {
        use episteme::adapters::install_wizard;
        // fallback_select_tools는 TOOLS를 순회하므로 간접적으로 TOOLS 내용을 확인
        // 직접 접근은 pub이 아니므로 install_wizard의 공개 API를 통해 검증
        let selected = install_wizard::interactive_select_tools(&["cline"]);
        // non-TTY 환경에서는 Err가 반환되므로 Ok/Err 여부는 무관, cline이 TOOLS에 있으면 충분
        // 여기서는 단순히 컴파일되고 패닉 없이 실행되면 통과
        let _ = selected;
    }
}