use std::fs;

use color_eyre::eyre::Result;
use path_absolutize::Absolutize;

use crate::{
    config::ProjectConfig,
    error::Error,
    terminal_output::{print_error, print_info},
};

use super::BuildArgs;

#[derive(Debug, clap::Args, Clone)]
pub struct PackageArgs {
    #[clap(flatten)]
    build_args: BuildArgs,
}

pub fn package(_verbose: bool, args: &PackageArgs) -> Result<()> {
    let path = args.build_args.path.as_path();

    print_info(&format!(
        "Packaging project at {}",
        path.absolutize()?.display()
    ));

    let toml_path = if !path.exists() {
        print_error("The specified path does not exist.");
        return Err(Error::PathNotFoundError(path.to_path_buf()))?;
    } else if path.is_dir() {
        let toml_path = path.join("pack.toml");
        if !toml_path.exists() {
            print_error("The specified directory does not contain a pack.toml file.");
            Err(Error::InvalidPackPathError(path.to_path_buf()))?;
        }
        toml_path
    } else if path.is_file()
        && path
            .file_name()
            .ok_or(Error::InvalidPackPathError(path.to_path_buf()))?
            == "pack.toml"
    {
        path.to_path_buf()
    } else {
        print_error("The specified path is neither a directory nor a pack.toml file.");
        return Err(Error::InvalidPackPathError(path.to_path_buf()))?;
    };

    let toml_content = fs::read_to_string(&toml_path)?;
    let project_config = toml::from_str::<ProjectConfig>(&toml_content)?;

    let script_paths = super::build::get_script_paths(
        &toml_path
            .parent()
            .ok_or(Error::InvalidPackPathError(path.to_path_buf()))?
            .join("src"),
    )?;

    let compiled = shulkerscript_lang::compile(&script_paths)?;

    let dist_path = toml_path
        .parent()
        .expect("Failed to get parent directory of pack.toml")
        .join("dist")
        .join(project_config.pack.name + ".zip");

    compiled.zip(&dist_path)?;

    print_info(&format!(
        "Finished packaging project to {}",
        dist_path.absolutize_from(path)?.display()
    ));

    Ok(())
}