use directories_next::ProjectDirs;
use std::{env::*, fs, path::Path, process::Command};
use structopt::clap::Shell;
use toml::Value;
use walkdir::WalkDir;

include!("src/cli.rs");

const BIN_NAME: &str = "zellij";

fn main() {
    // Build Sub-Projects (Temporary?)
    for entry in WalkDir::new(".") {
        let entry = entry.unwrap();
        let ext = entry.path().extension();
        if ext.is_some() && ext.unwrap() == "rs" {
            println!("cargo:rerun-if-changed={}", entry.path().to_string_lossy());
        }
    }

    let manifest_dir = Path::new(env!("CARGO_MANIFEST_DIR"));
    let manifest: Value =
        toml::from_str(&fs::read_to_string(manifest_dir.join("Cargo.toml")).unwrap()).unwrap();
    let members = manifest
        .get("workspace")
        .unwrap()
        .get("members")
        .unwrap()
        .as_array()
        .unwrap();

    let release = if var("PROFILE").unwrap() == "release" {
        "--release"
    } else {
        ""
    };
    let starting_dir = current_dir().unwrap();
    let alt_target = manifest_dir.join("target/tiles");
    for project in members {
        let path = manifest_dir.join(project.as_str().unwrap());
        set_current_dir(&path);

        // FIXME: This is soul-crushingly terrible, but I can't keep the values alive long enough
        if var("PROFILE").unwrap() == "release" {
            Command::new("cargo".to_string())
                .arg("build")
                .arg("--target-dir")
                .arg(&alt_target.as_os_str())
                .arg("--release")
                .status()
                .unwrap();
        } else {
            Command::new("cargo")
                .arg("build")
                .arg("--target-dir")
                .arg(&alt_target.as_os_str())
                .status()
                .unwrap();
        }
    }
    set_current_dir(&starting_dir);

    // FIXME: Deduplicate this with the initial walk all .rs pattern
    for entry in fs::read_dir(alt_target.join("wasm32-wasi/release/")).unwrap() {
        let entry = entry.unwrap().path();
        let ext = entry.extension();
        if ext.is_some() && ext.unwrap() == "wasm" {
            dbg!(&entry);
            Command::new("wasm-opt")
                .arg("-O")
                .arg(entry.as_os_str())
                .arg("-o")
                .arg(format!(
                    "assets/plugins/{}",
                    entry.file_name().unwrap().to_string_lossy()
                ))
                .status()
                .unwrap();
        }
    }

    // Generate Shell Completions
    let mut clap_app = CliArgs::clap();
    println!("cargo:rerun-if-changed=src/cli.rs");
    let mut out_dir = std::env::var_os("CARGO_MANIFEST_DIR").unwrap();
    out_dir.push("/assets/completions");

    println!(
        "Completion files will to added to this location: {:?}",
        out_dir
    );
    fs::create_dir_all(&out_dir).unwrap();
    clap_app.gen_completions(BIN_NAME, Shell::Bash, &out_dir);
    clap_app.gen_completions(BIN_NAME, Shell::Zsh, &out_dir);
    clap_app.gen_completions(BIN_NAME, Shell::Fish, &out_dir);

    // Install Default Plugins and Layouts
    let assets = vec![
        "plugins/status-bar.wasm",
        "plugins/strider.wasm",
        "layouts/default.yaml",
        "layouts/strider.yaml",
    ];
    let project_dirs = ProjectDirs::from("org", "Mosaic Contributors", "Mosaic").unwrap();
    let data_dir = project_dirs.data_dir();
    fs::create_dir_all(data_dir.join("plugins")).unwrap();
    fs::create_dir_all(data_dir.join("layouts")).unwrap();
    for asset in assets {
        fs::copy(Path::new("assets/").join(asset), data_dir.join(asset))
            .expect("Failed to copy asset files");
    }
}