use bytes::Buf;
use flate2::read::GzDecoder;
use reqwest::header::USER_AGENT;
use std::path::PathBuf;
use std::{env, process::Command};
use tar::Archive;

fn find_qnicorn(qnicorn_dir: &PathBuf) -> Option<PathBuf> {
    for entry in std::fs::read_dir(qnicorn_dir).ok()? {
        let entry = entry.unwrap();
        let path = entry.path();

        if path.is_dir() && path.file_name()?.to_str()?.contains("qnicorn") {
            return Some(path);
        }
    }

    None
}

fn download_qnicorn() -> Option<String> {
    // https://docs.github.com/en/rest/reference/repos#download-a-repository-archive-tar
    let pkg_version;
    if let Ok(qnicorn_version) = env::var("QNICORN_VERSION") {
        pkg_version = qnicorn_version;
    } else {
        pkg_version = env::var("CARGO_PKG_VERSION").unwrap();
    }
    let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
    let client = reqwest::blocking::Client::new();
    let resp = client
        .get(format!(
            "https://api.github.com/repos/qilingframework/qnicorn/tarball/{}",
            pkg_version
        ))
        .header(USER_AGENT, "qnicorn-engine-rust-bindings")
        .send()
        .unwrap()
        .bytes()
        .unwrap();
    let tar = GzDecoder::new(resp.reader());

    let mut archive = Archive::new(tar);
    archive.unpack(&out_dir).unwrap();

    match find_qnicorn(&out_dir) {
        Some(dir) => Some(String::from(out_dir.join(dir).to_str()?)),
        None => None,
    }
}

fn main() {
    let profile = env::var("PROFILE").unwrap();

    let qnicorn_dir = download_qnicorn().unwrap();

    println!("cargo:rerun-if-changed={}", &qnicorn_dir);

    // We don't use TARGET since we can't cross-build.
    if env::consts::OS == "windows" {
        // Windows
        let mut cmd = Command::new("cmake");
        cmd.current_dir(&qnicorn_dir)
            .arg("-B")
            .arg("rust_build")
            .arg("-DQNICORN_BUILD_SHARED=off")
            .arg("-G")
            .arg("Visual Studio 16 2019");

        if profile == "debug" {
            cmd.arg("-DCMAKE_BUILD_TYPE=Debug");
        } else {
            cmd.arg("-DCMAKE_BUILD_TYPE=Release");
        }

        cmd.output()
            .expect("Fail to create build directory on Windows.");

        let mut platform = "x64";
        let mut conf = "Release";
        if std::mem::size_of::<usize>() == 4 {
            platform = "Win32";
        }
        if profile == "debug" {
            conf = "Debug";
        }

        Command::new("msbuild")
            .current_dir(format!("{}/rust_build", &qnicorn_dir))
            .arg("qnicorn.sln")
            .arg("-m")
            .arg("-p:Platform=".to_owned() + platform)
            .arg("-p:Configuration=".to_owned() + conf)
            .output()
            .expect("Fail to build qnicorn on Win32.");
        println!(
            "cargo:rustc-link-search={}/rust_build/{}",
            qnicorn_dir, conf
        );
    } else {
        // Most Unix-like systems
        let mut cmd = Command::new("cmake");
        cmd.current_dir(&qnicorn_dir)
            .arg("-B")
            .arg("rust_build")
            .arg("-DQNICORN_BUILD_SHARED=off");

        if profile == "debug" {
            cmd.arg("-DCMAKE_BUILD_TYPE=Debug");
        } else {
            cmd.arg("-DCMAKE_BUILD_TYPE=Release");
        }

        cmd.output()
            .expect("Fail to create build directory on *nix.");

        Command::new("make")
            .current_dir(format!("{}/rust_build", &qnicorn_dir))
            .arg("-j6")
            .output()
            .expect("Fail to build qnicorn on *nix.");

        println!("cargo:rustc-link-search={}/rust_build", qnicorn_dir);
    }

    // This is a workaround for Qnicorn static link since libqnicorn.a is also linked again lib*-softmmu.a.
    // Static libs is just a bundle of objects files. The link relation defined in CMakeLists is only
    // valid within the cmake project scope and cmake would help link again sub static libs automatically.
    //
    // Lazymio(@wtdcode): Why do I stick to static link? See: https://github.com/rust-lang/cargo/issues/5077
    println!("cargo:rustc-link-lib=qnicorn");
    for arch in [
        "x86_64",
        "arm",
        "armeb",
        "aarch64",
        "aarch64eb",
        "riscv32",
        "riscv64",
        "mips",
        "mipsel",
        "mips64",
        "mips64el",
        "sparc",
        "sparc64",
        "m68k",
        "ppc",
        "ppc64",
    ]
    .iter()
    {
        println!("cargo:rustc-link-lib={}-softmmu", arch);
    }
    println!("cargo:rustc-link-lib=qnicorn-common");
}