use std::path::Path;
use std::process::exit;

fn main() {
    println!("cargo::rerun-if-env-changed=NO_FIRMWARE_BIN");
    println!("cargo::rerun-if-env-changed=FIRMWARE_PROFILE");
    let profile = std::env::var("FIRMWARE_PROFILE").unwrap_or_else(|_| {
        // Default to firmware-devel for debug builds, firmware for release builds
        if std::env::var("PROFILE").as_deref() == Ok("release") {
            "firmware".to_string()
        } else {
            "firmware-devel".to_string()
        }
    });
    let include_dir = Path::new(env!("CARGO_MANIFEST_DIR"))
        .join("../target/armv7-unknown-linux-musleabihf")
        .join(&profile);
    set_binary_var(&include_dir, "FILE_ROOTSHELL", "rootshell");
    set_binary_var(&include_dir, "FILE_RAYHUNTER_DAEMON", "rayhunter-daemon");
}

fn set_binary_var(include_dir: &Path, var: &str, file: &str) {
    println!("cargo::rerun-if-env-changed={var}");
    if std::env::var_os("NO_FIRMWARE_BIN").is_some() {
        let out_dir = std::env::var("OUT_DIR").unwrap();
        std::fs::create_dir_all(&out_dir).unwrap();
        let blank = Path::new(&out_dir).join("blank");
        std::fs::write(&blank, []).unwrap();
        println!("cargo::rustc-env={var}={}", blank.display());
        return;
    }
    if std::env::var_os(var).is_none() {
        let binary = include_dir.join(file);
        println!("cargo::rerun-if-changed={}", binary.display());
        if !binary.exists() {
            println!(
                "cargo::error=Firmware binary {file} not present at {}",
                binary.display()
            );
            exit(0);
        }
        println!("cargo::rustc-env={var}={}", binary.display());
    }
}