use std::fs::create_dir_all;

pub mod cli {
    include!("src/command.rs");
}

fn main() -> std::io::Result<()> {
    println!("cargo:rerun-if-changed=build.rs");

    let out_dir = match std::env::var_os("ASSET_OUT_DIR") {
        Some(dir) => std::path::PathBuf::from(dir),
        None => return Ok(()),
    };

    #[allow(unused_mut)]
    let mut cli = cli::make_command();

    let man = clap_mangen::Man::new(cli.clone());
    let mut man_buffer: Vec<u8> = Default::default();
    man.render(&mut man_buffer)?;

    let man_out_dir = out_dir.as_path().join("man");
    create_dir_all(&man_out_dir)?;
    std::fs::write(man_out_dir.join("minijinja-cli.1"), man_buffer)?;

    #[cfg(feature = "completions")]
    {
        use clap::ValueEnum;

        let completions_out_dir = out_dir.as_path().join("completions");
        create_dir_all(&completions_out_dir)?;

        for shell in clap_complete::Shell::value_variants() {
            clap_complete::generate_to(*shell, &mut cli, "minijinja-cli", &completions_out_dir)?;
        }
        clap_complete::generate_to(
            clap_complete_nushell::Nushell,
            &mut cli,
            "minijinja-cli",
            &completions_out_dir,
        )?;
        clap_complete::generate_to(
            clap_complete_fig::Fig,
            &mut cli,
            "minijinja-cli",
            &completions_out_dir,
        )?;
    }

    Ok(())
}