mod doctor;
mod download;
mod install;
mod manifest;
mod state;

use anyhow::{bail, Result};
use clap::{Parser, Subcommand};
use std::path::PathBuf;

#[derive(Parser)]
#[command(name = "bakery", about = "Package manager for the bread ecosystem")]
struct Cli {
    #[command(subcommand)]
    command: Cmd,
    /// Override the directory where binaries are installed
    #[arg(long, env = "BAKERY_BIN_DIR", global = true)]
    bin_dir: Option<PathBuf>,
}

#[derive(Subcommand)]
enum Cmd {
    /// Install a package
    Install {
        package: String,
    },
    /// Remove an installed package (data files are never deleted)
    Remove {
        package: String,
    },
    /// Update one or all installed packages
    Update {
        /// Package to update; omit to update all installed packages
        package: Option<String>,
    },
    /// List packages
    List {
        /// Show only installed packages
        #[arg(long)]
        installed: bool,
    },
    /// Show details for a package
    Info {
        package: String,
    },
    /// Check system dependencies for installed or requested packages
    Doctor {
        /// Package to check; omit to check all installed packages
        package: Option<String>,
    },
}

fn default_bin_dir() -> PathBuf {
    dirs::home_dir()
        .unwrap_or_else(|| PathBuf::from("~"))
        .join(".local/bin")
}

fn main() -> Result<()> {
    let cli = Cli::parse();
    let bin_dir = cli.bin_dir.unwrap_or_else(default_bin_dir);

    match cli.command {
        Cmd::Install { package } => cmd_install(&package, &bin_dir),
        Cmd::Remove { package } => cmd_remove(&package, &bin_dir),
        Cmd::Update { package } => cmd_update(package.as_deref(), &bin_dir),
        Cmd::List { installed } => cmd_list(installed),
        Cmd::Info { package } => cmd_info(&package),
        Cmd::Doctor { package } => cmd_doctor(package.as_deref()),
    }
}

fn cmd_install(name: &str, bin_dir: &std::path::Path) -> Result<()> {
    let index = manifest::load(false)?;
    let pkg = index
        .get(name)
        .ok_or_else(|| anyhow::anyhow!("unknown package: {name}"))?;

    // Doctor runs first — bail if system deps are missing.
    println!("checking system dependencies…");
    let missing = doctor::check_deps(&pkg.system_deps)?;
    if !missing.is_empty() {
        eprintln!("missing system dependencies for {name}: {}", missing.join(", "));
        eprintln!("install with: sudo pacman -S {}", missing.join(" "));
        bail!("system deps not satisfied");
    }

    install::install_package(pkg, bin_dir)
}

fn cmd_remove(name: &str, bin_dir: &std::path::Path) -> Result<()> {
    install::remove_package(name, bin_dir)
}

fn cmd_update(name: Option<&str>, bin_dir: &std::path::Path) -> Result<()> {
    let index = manifest::load(true)?; // force refresh on update
    let state = state::State::load()?;

    let targets: Vec<String> = match name {
        Some(n) => vec![n.to_string()],
        None => state.packages.keys().cloned().collect(),
    };

    for pkg_name in &targets {
        let installed = match state.packages.get(pkg_name.as_str()) {
            Some(p) => p,
            None => {
                eprintln!("{pkg_name} is not installed, skipping");
                continue;
            }
        };
        let latest = match index.get(pkg_name) {
            Some(p) => p,
            None => {
                eprintln!("{pkg_name} not found in index, skipping");
                continue;
            }
        };
        if installed.version == latest.version {
            println!("{pkg_name} is already at {}", installed.version);
        } else {
            println!(
                "updating {pkg_name} {} → {}",
                installed.version, latest.version
            );
            install::install_package(latest, bin_dir)?;
        }
    }
    Ok(())
}

fn cmd_list(installed_only: bool) -> Result<()> {
    let state = state::State::load()?;

    if installed_only {
        if state.packages.is_empty() {
            println!("no packages installed");
        }
        for pkg in state.packages.values() {
            println!("  {} {} (installed {})", pkg.name, pkg.version, pkg.installed_at);
        }
        return Ok(());
    }

    let index = manifest::load(false)?;
    let mut names: Vec<&str> = index.packages.keys().map(|s| s.as_str()).collect();
    names.sort();
    for name in names {
        let pkg = &index.packages[name];
        let tag = if state.is_installed(name) {
            format!(" [installed {}]", state.packages[name].version)
        } else {
            String::new()
        };
        println!("  {} {} — {}{}", pkg.name, pkg.version, pkg.description, tag);
    }
    Ok(())
}

fn cmd_info(name: &str) -> Result<()> {
    let index = manifest::load(false)?;
    let pkg = index
        .get(name)
        .ok_or_else(|| anyhow::anyhow!("unknown package: {name}"))?;

    let state = state::State::load()?;
    let status = if let Some(inst) = state.packages.get(name) {
        format!("installed ({})", inst.version)
    } else {
        "not installed".to_string()
    };

    println!("{} {}", pkg.name, pkg.version);
    println!("  {}", pkg.description);
    println!("  status:      {status}");
    println!("  binaries:    {}", pkg.binaries.iter().map(|b| b.name.as_str()).collect::<Vec<_>>().join(", "));
    if !pkg.system_deps.is_empty() {
        println!("  system deps: {}", pkg.system_deps.join(", "));
    }
    if !pkg.bread_deps.is_empty() {
        println!("  bread deps:  {}", pkg.bread_deps.join(", "));
    }
    if !pkg.services.is_empty() {
        println!("  services:    {}", pkg.services.iter().map(|s| s.unit.as_str()).collect::<Vec<_>>().join(", "));
    }
    Ok(())
}

fn cmd_doctor(name: Option<&str>) -> Result<()> {
    let index = manifest::load(false)?;
    let state = state::State::load()?;

    let targets: Vec<String> = match name {
        Some(n) => vec![n.to_string()],
        None => state.packages.keys().cloned().collect(),
    };

    if targets.is_empty() {
        println!("no packages installed — nothing to check");
        return Ok(());
    }

    let mut all_ok = true;
    for pkg_name in &targets {
        if let Some(pkg) = index.get(pkg_name) {
            if !doctor::report(pkg_name, &pkg.system_deps) {
                all_ok = false;
            }
        }
    }

    if all_ok {
        println!("all checks passed");
    }
    Ok(())
}