axbuild 0.4.2

use std::path::{Path, PathBuf};

use clap::{Args as ClapArgs, Subcommand};

use crate::axvisor::context::AxvisorContext;

pub mod config;
pub mod registry;
pub mod spec;
pub mod storage;

use config::ImageConfig;
use spec::ImageSpecRef;
use storage::Storage;

#[derive(ClapArgs)]
pub struct Args {
    #[command(flatten)]
    pub overrides: ConfigOverrides,

    #[command(subcommand)]
    pub command: Command,
}

#[derive(ClapArgs, Debug, Clone, Default)]
pub struct ConfigOverrides {
    #[arg(short('S'), long, global = true)]
    pub local_storage: Option<PathBuf>,

    #[arg(short('R'), long, global = true)]
    pub registry: Option<String>,

    #[arg(short('N'), long, global = true)]
    pub no_auto_sync: bool,

    #[arg(long, global = true)]
    pub auto_sync_threshold: Option<u64>,
}

impl ConfigOverrides {
    pub fn apply_on(&self, config: &mut ImageConfig) {
        if let Some(local_storage) = self.local_storage.as_ref() {
            config.local_storage = local_storage.clone();
        }
        if let Some(registry) = self.registry.as_ref() {
            config.registry = registry.clone();
        }
        if self.no_auto_sync {
            config.auto_sync = false;
        }
        if let Some(auto_sync_threshold) = self.auto_sync_threshold {
            config.auto_sync_threshold = auto_sync_threshold;
        }
    }
}

#[derive(Subcommand)]
pub enum Command {
    /// List all available images.
    Ls(ArgsLs),
    /// Pull the specified image archive and extract it by default.
    Pull(ArgsPull),
}

#[derive(ClapArgs)]
pub struct ArgsLs {
    #[arg(short, long)]
    pub verbose: bool,

    pub pattern: Option<String>,
}

#[derive(ClapArgs)]
pub struct ArgsPull {
    pub image: String,

    #[arg(short, long)]
    pub output_dir: Option<PathBuf>,

    #[arg(long)]
    pub no_extract: bool,
}

pub(crate) async fn run(args: Args, ctx: &AxvisorContext) -> anyhow::Result<()> {
    match args.command {
        Command::Ls(ls) => list_images(ctx, &args.overrides, ls).await,
        Command::Pull(pull) => pull_image(ctx, &args.overrides, pull).await,
    }
}

async fn list_images(
    ctx: &AxvisorContext,
    overrides: &ConfigOverrides,
    args: ArgsLs,
) -> anyhow::Result<()> {
    let mut config = ImageConfig::read_config(ctx.workspace_root())?;
    overrides.apply_on(&mut config);
    let storage = Storage::new_from_config(&config).await?;
    storage
        .image_registry
        .print(args.verbose, args.pattern.as_deref());
    Ok(())
}

async fn pull_image(
    ctx: &AxvisorContext,
    overrides: &ConfigOverrides,
    args: ArgsPull,
) -> anyhow::Result<()> {
    let mut config = ImageConfig::read_config(ctx.workspace_root())?;
    overrides.apply_on(&mut config);
    let storage = Storage::new_from_config(&config).await?;
    let spec = ImageSpecRef::parse(&args.image);
    let output_dir = args
        .output_dir
        .as_deref()
        .map(to_absolute_path)
        .transpose()?;
    let _ = storage
        .pull_image(spec, output_dir.as_deref(), !args.no_extract)
        .await?;
    Ok(())
}

fn to_absolute_path(path: &Path) -> anyhow::Result<PathBuf> {
    Ok(if path.is_absolute() {
        path.to_path_buf()
    } else {
        std::env::current_dir()?.join(path)
    })
}

#[cfg(test)]
mod tests {
    use clap::Parser;

    use super::*;

    #[test]
    fn overrides_apply_on_config() {
        let mut config = ImageConfig::new_default();
        let overrides = ConfigOverrides {
            local_storage: Some(PathBuf::from("/tmp/custom")),
            registry: Some("https://example.com/registry.toml".to_string()),
            no_auto_sync: true,
            auto_sync_threshold: Some(123),
        };

        overrides.apply_on(&mut config);

        assert_eq!(config.local_storage, PathBuf::from("/tmp/custom"));
        assert_eq!(config.registry, "https://example.com/registry.toml");
        assert!(!config.auto_sync);
        assert_eq!(config.auto_sync_threshold, 123);
    }

    #[test]
    fn parses_pull_command() {
        #[derive(Parser)]
        struct Cli {
            #[command(subcommand)]
            command: Command,
        }

        let cli = Cli::try_parse_from(["axvisor", "pull", "linux"]).unwrap();
        match cli.command {
            Command::Pull(args) => {
                assert_eq!(args.image, "linux");
                assert!(args.output_dir.is_none());
                assert!(!args.no_extract);
            }
            _ => panic!("expected pull command"),
        }
    }

    #[test]
    fn parses_pull_command_with_version_and_output_dir() {
        #[derive(Parser)]
        struct Cli {
            #[command(subcommand)]
            command: Command,
        }

        let cli = Cli::try_parse_from([
            "axvisor",
            "pull",
            "linux:0.0.1",
            "--output-dir",
            "tmp/images",
        ])
        .unwrap();
        match cli.command {
            Command::Pull(args) => {
                assert_eq!(args.image, "linux:0.0.1");
                assert_eq!(args.output_dir, Some(PathBuf::from("tmp/images")));
            }
            _ => panic!("expected pull command"),
        }
    }

    #[test]
    fn parses_pull_command_with_no_extract() {
        #[derive(Parser)]
        struct Cli {
            #[command(subcommand)]
            command: Command,
        }

        let cli = Cli::try_parse_from(["axvisor", "pull", "linux", "--no-extract"]).unwrap();
        match cli.command {
            Command::Pull(args) => assert!(args.no_extract),
            _ => panic!("expected pull command"),
        }
    }
}