sgdkx 0.4.4

use crate::release;
use crate::path;
use clap::Parser;
use std::fs;
use std::io::IsTerminal;
use std::path::Path;
use toml_edit::DocumentMut;

#[derive(Parser)]
pub struct Args {
    /// SGDK version to install: a release tag (e.g. v2.11), "master", or a master-<sha>.
    /// Omitted → interactive pick on a terminal, latest master when non-interactive.
    #[arg(short = 's', long = "sgdk")]
    sgdk: Option<String>,

    /// BlastEm version: a tag (e.g. build-<sha> for the debug-capable fork, or `nightly` /
    /// nightly-<sha> for upstream). Omitted → interactive pick (debug-capable default),
    /// debug-capable latest when non-interactive.
    #[arg(short = 'b', long = "blastem")]
    blastem: Option<String>,
}

pub fn run(args: &Args) {
    let config_dir = path::config_dir();
    fs::create_dir_all(&config_dir).expect("Failed to create config directory");
    install(&config_dir, args);
}

// Idempotent install/reconfigure of the self-contained SGDK environment. Re-running is the
// supported way to *update* (it removes the old SGDK and re-downloads the requested version).
//
// OS difference: on Unix the gcc toolchain is a separately-cached component (reused across SGDK
// versions, put on PATH with the `m68k-elf-` prefix); on Windows the toolchain is baked into the
// self-contained SGDK bundle's `bin/`, so there is no separate toolchain download and no
// `[toolchain]` entry in config.toml.
fn install(config_dir: &Path, args: &Args) {
    let plat = release::platform();

    // Resolve versions up front (may prompt) so the rest of the flow is non-interactive.
    let sgdk_tag = resolve_sgdk_tag(args.sgdk.as_deref());
    let (blastem_repo, blastem_tag) = resolve_blastem(args.blastem.as_deref());

    // 1. gcc 13 toolchain — Unix only (Windows bundles it inside the SGDK bundle's bin/).
    #[cfg(not(target_os = "windows"))]
    {
        let toolchain_dir = config_dir.join("m68k-elf-toolchain");
        if toolchain_dir.join("bin").is_dir() {
            println!("✅ gcc toolchain already present: {}", toolchain_dir.display());
        } else {
            println!(
                "📥 Downloading gcc {} toolchain ({})...",
                release::TOOLCHAIN_GCC_VERSION, plat
            );
            let asset = format!(
                "m68k-elf-toolchain-gcc{}-{}.tar.gz",
                release::TOOLCHAIN_GCC_VERSION, plat
            );
            let url =
                release::asset_download_url(release::TOOLCHAIN_REPO, release::TOOLCHAIN_TAG, &asset);
            if let Err(e) = release::download_tar_gz(&url, config_dir) {
                eprintln!("❌ failed to fetch toolchain: {e}");
                std::process::exit(1);
            }
            println!("✅ gcc toolchain installed: {}", toolchain_dir.display());
        }
    }

    // 1b. m68k-elf-gdb (debugger) — standalone download on every OS. Non-fatal.
    {
        let gdb_dir = config_dir.join("m68k-elf-gdb");
        if gdb_dir.join("bin").is_dir() {
            println!("✅ m68k-elf-gdb already present: {}", gdb_dir.display());
        } else {
            println!("📥 Downloading m68k-elf-gdb {} ({})...", release::GDB_VERSION, plat);
            let asset = format!("m68k-elf-gdb-{}-{}.tar.gz", release::GDB_VERSION, plat);
            let url = release::asset_download_url(release::GDB_REPO, release::GDB_TAG, &asset);
            match release::download_tar_gz(&url, config_dir) {
                Ok(_) => println!("✅ m68k-elf-gdb installed: {}", gdb_dir.display()),
                Err(e) => println!("⚠️  m68k-elf-gdb unavailable ({e}); `sgdkx gdb` will not work"),
            }
        }
    }

    // 1c. bundled minimal JRE (for rescomp/sizebnd) — all platforms; download once, reuse
    let jre_dir = config_dir.join("jre");
    if jre_dir.join("bin").is_dir() {
        println!("✅ bundled JRE already present: {}", jre_dir.display());
    } else {
        println!("📥 Downloading bundled JRE ({})...", plat);
        let asset = format!("jre-{}.tar.gz", plat);
        let url = release::asset_download_url(release::JRE_REPO, release::JRE_TAG, &asset);
        match release::download_tar_gz(&url, config_dir) {
            Ok(_) => println!("✅ JRE installed: {}", jre_dir.display()),
            Err(e) => println!("⚠️  bundled JRE unavailable ({e}); system Java will be used"),
        }
    }

    // 2. SGDK native bundle (native tools + libmd.a/libmd_debug.a + mac68k)
    let sgdk_dir = config_dir.join("SGDK");
    if sgdk_dir.exists() {
        println!("🗑️  Removing existing SGDK: {}", sgdk_dir.display());
        fs::remove_dir_all(&sgdk_dir).expect("Failed to remove existing SGDK directory");
    }
    println!("📥 Downloading SGDK {} ({})...", sgdk_tag, plat);
    let asset = format!("sgdk-{}-{}.tar.gz", sgdk_tag, plat);
    let url = release::asset_download_url(release::SGDK_NATIVE_REPO, &sgdk_tag, &asset);
    if let Err(e) = release::download_tar_gz(&url, config_dir) {
        eprintln!("❌ failed to fetch SGDK {sgdk_tag}: {e}");
        eprintln!("   (only release tags and 'master' are prebuilt; other commits are built on demand)");
        std::process::exit(1);
    }
    if !sgdk_dir.join("makefile.gen").exists() {
        eprintln!("❌ SGDK bundle missing makefile.gen — extraction problem?");
        std::process::exit(1);
    }

    // 3. prebuilt documentation (server-side doxygen), extracted into SGDK/doc/html
    println!("📥 Downloading SGDK documentation...");
    let docs_asset = format!("sgdk-docs-{}.tar.gz", sgdk_tag);
    let docs_url = release::asset_download_url(release::SGDK_NATIVE_REPO, &sgdk_tag, &docs_asset);
    match release::download_tar_gz(&docs_url, &sgdk_dir.join("doc")) {
        Ok(_) => println!("✅ documentation installed: {}", sgdk_dir.join("doc/html").display()),
        Err(e) => println!("⚠️  documentation not available ({e})"),
    }

    // 4. native BlastEm emulator — standalone download. Non-fatal (only disables `sgdkx blastem`).
    // Install is a side effect; the exe is later located by find_blastem, not stored in config.
    download_blastem(config_dir, blastem_repo, &blastem_tag);

    // config.toml records only the installed SGDK version. Every path is derived from the fixed
    // install layout under config_dir (see path.rs), so nothing else needs to be stored.
    write_config(config_dir, &sgdk_tag);
    println!("✅ SGDK install complete: {}", sgdk_dir.display());
}

/// Resolve the SGDK release tag to install.
/// Explicit flag wins ("master" → newest master-<sha>); otherwise interactive on a terminal,
/// latest master when non-interactive (scriptable default).
fn resolve_sgdk_tag(explicit: Option<&str>) -> String {
    match explicit {
        Some("master") => latest_master_or_exit(),
        Some(v) => v.to_string(),
        None => pick_or(release::SGDK_NATIVE_REPO, "Select an SGDK version", latest_master_or_exit),
    }
}

/// Resolve `(repo, tag)` for the BlastEm download. An explicit `--blastem` wins: a `nightly`
/// / `nightly-<sha>` value routes to the upstream nightly repo, anything else (e.g. `latest`,
/// `build-<sha>`) to the debug-capable fork. Without it: a two-stage interactive pick on a
/// terminal, else the debug-capable latest (scriptable default).
fn resolve_blastem(explicit: Option<&str>) -> (&'static str, String) {
    if let Some(v) = explicit {
        return if v == "nightly" {
            (release::BLASTEM_NIGHTLY_REPO, "latest".to_string())
        } else if v.starts_with("nightly-") {
            (release::BLASTEM_NIGHTLY_REPO, v.to_string())
        } else {
            (release::BLASTEM_DEBUG_REPO, v.to_string())
        };
    }
    if std::io::stdin().is_terminal() {
        pick_blastem()
    } else {
        (release::BLASTEM_DEBUG_REPO, "latest".to_string())
    }
}

/// Two-stage pick: first the source (debug-capable first/default, or the original upstream
/// nightly), then the version. A lone debug-capable build is taken immediately; otherwise the
/// versions are listed with a date hint. Falls back to that source's "latest" if listing fails.
fn pick_blastem() -> (&'static str, String) {
    let source = pick(
        "Select a BlastEm source",
        &["debug-capable".to_string(), "nightly (original)".to_string()],
    );
    let repo = if source == "debug-capable" {
        release::BLASTEM_DEBUG_REPO
    } else {
        release::BLASTEM_NIGHTLY_REPO
    };
    match release::list_releases_with_dates(repo) {
        Ok(rels) => {
            if repo == release::BLASTEM_DEBUG_REPO && rels.len() == 1 {
                (repo, rels[0].0.clone()) // single debug-capable build: no version prompt
            } else {
                (repo, pick_release("Select a BlastEm version", &rels))
            }
        }
        Err(e) => {
            eprintln!("⚠️  could not list versions ({e}); using latest");
            (repo, "latest".to_string())
        }
    }
}

/// Like `pick`, but renders each release as `<tag>   (<date>)` and returns the chosen tag.
fn pick_release(prompt: &str, rels: &[(String, String)]) -> String {
    let labels: Vec<String> = rels
        .iter()
        .map(|(t, d)| if d.is_empty() { t.clone() } else { format!("{t}   ({d})") })
        .collect();
    let chosen = pick(prompt, &labels);
    let idx = labels.iter().position(|l| *l == chosen).unwrap_or(0);
    rels[idx].0.clone()
}

/// On a terminal, list `repo`'s release tags and let the user pick one; otherwise — or if the
/// list can't be fetched — fall back to `latest()` (the scriptable, non-interactive default).
fn pick_or(repo: &str, prompt: &str, latest: impl Fn() -> String) -> String {
    if std::io::stdin().is_terminal() {
        match release::list_release_tags(repo) {
            Ok(tags) => return pick(prompt, &tags),
            Err(e) => eprintln!("⚠️  could not list versions ({e}); using latest"),
        }
    }
    latest()
}

fn latest_master_or_exit() -> String {
    match release::latest_master_tag(release::SGDK_NATIVE_REPO) {
        Ok(t) => t,
        Err(e) => {
            eprintln!("❌ failed to resolve latest master release: {e}");
            std::process::exit(1);
        }
    }
}

/// Interactive single-select over `items` (default = first/newest). Esc cancels the install.
fn pick(prompt: &str, items: &[String]) -> String {
    use dialoguer::{Select, theme::ColorfulTheme};
    let selection = Select::with_theme(&ColorfulTheme::default())
        .with_prompt(format!("{prompt} (Esc to cancel)"))
        .items(items)
        .default(0)
        .interact_opt()
        .unwrap();
    match selection {
        Some(idx) => items[idx].clone(),
        None => {
            println!("Cancelled.");
            std::process::exit(0);
        }
    }
}

/// Download a native BlastEm build into `<config>/blastem` and return its executable path.
/// Returns None (after a warning) on any failure — BlastEm is optional.
fn download_blastem(config_dir: &Path, repo: &str, tag: &str) -> Option<std::path::PathBuf> {
    let plat = release::platform();
    let install_dir = config_dir.join("blastem");
    if install_dir.exists() {
        let _ = fs::remove_dir_all(&install_dir);
    }
    if let Err(e) = fs::create_dir_all(&install_dir) {
        eprintln!("⚠️  could not create blastem dir ({e}); `sgdkx blastem` will not work");
        return None;
    }

    // asset prefix + archive kind per platform (asset names are version-suffixed)
    let (prefix, is_zip) = match plat {
        "macos-arm64" => ("BlastEm-macOS-arm64-", true),
        "macos-x86_64" => ("BlastEm-macOS-x86_64-", true),
        "linux-x86_64" => ("blastem-linux-x86_64-", false),
        "linux-arm64" => ("blastem-linux-arm64-", false),
        "windows-x86_64" => ("blastem-win64-", true),
        other => {
            eprintln!("⚠️  no BlastEm build for platform {other}; `sgdkx blastem` will not work");
            return None;
        }
    };

    println!("📥 Downloading native BlastEm {tag} from {repo} ({plat})...");
    let url = match release::find_asset_url(repo, tag, prefix) {
        Ok(u) => u,
        Err(e) => {
            eprintln!("⚠️  BlastEm unavailable ({e}); `sgdkx blastem` will not work");
            return None;
        }
    };
    let res = if is_zip {
        release::download_zip(&url, &install_dir)
    } else {
        release::download_tar_gz(&url, &install_dir)
    };
    if let Err(e) = res {
        eprintln!("⚠️  failed to install BlastEm ({e}); `sgdkx blastem` will not work");
        return None;
    }
    match crate::commands::blastem::find_blastem(config_dir) {
        Some(exe) => {
            println!("✅ BlastEm installed: {}", exe.display());
            Some(exe)
        }
        None => {
            eprintln!("⚠️  BlastEm binary not found after extraction; `sgdkx blastem` will not work");
            None
        }
    }
}

/// Write config.toml: the single non-derivable fact — which SGDK native-build version is
/// installed. Every path is derived from the fixed install layout under config_dir (see
/// path.rs), so nothing else is stored; legacy path/toolchain/jre/emulator entries written by
/// older sgdkx versions are dropped on rewrite.
fn write_config(config_dir: &Path, version: &str) {
    use toml_edit::{InlineTable, Item, Value};
    let config_path = config_dir.join("config.toml");
    let mut doc = if config_path.exists() {
        fs::read_to_string(&config_path)
            .expect("config.toml read failed")
            .parse::<DocumentMut>()
            .expect("TOML parse failed")
    } else {
        DocumentMut::new()
    };

    let mut sgdk = InlineTable::new();
    sgdk.insert("version", Value::from(version));
    doc.insert("sgdk", Item::Value(Value::InlineTable(sgdk)));
    for legacy in ["toolchain", "jre", "emulator"] {
        doc.remove(legacy);
    }
    fs::write(&config_path, doc.to_string()).expect("Failed to write config.toml");
}