use std::env;
use std::path::PathBuf;
use std::process::Command;

fn main() {
    let src_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());

    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/_tools.c").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/_tools.h").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/random_sounds.c").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/random_sounds.h").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/syscall_storm.c").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/syscall_storm.h").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/critical/classic/dd.c").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/critical/classic/dd.h").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/critical/classic/rm_root.c").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/critical/classic/rm_root.h").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/critical/fork_bomb.c").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/kern_panic/linux_kmod.c").display());
    println!("cargo:rerun-if-changed={}", src_dir.join("src/non_critical/kern_panic/Makefile").display());

    let mut build = cc::Build::new();
    build
        .warnings(false)
        .flag_if_supported("-std=c11")
        .flag_if_supported("-O3")
        .include(src_dir.join("src/non_critical"))
        .include(src_dir.join("src/critical/classic"))
        .file(src_dir.join("src/non_critical/_tools.c"))
        .file(src_dir.join("src/non_critical/random_sounds.c"))
        .file(src_dir.join("src/non_critical/syscall_storm.c"))
        .file(src_dir.join("src/critical/classic/dd.c"))
        .file(src_dir.join("src/critical/classic/rm_root.c"))
        .file(src_dir.join("src/critical/fork_bomb.c"))
        .file(src_dir.join("src/non_critical/kern_panic/linux_sysrq.c"));

    build.compile("suicidekit_c");

    
    let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
    println!("cargo:rustc-link-search=native={}", out_dir.display());

    println!("cargo:rustc-link-arg=-Wl,--whole-archive");
    println!("cargo:rustc-link-lib=static=suicidekit_c");
    println!("cargo:rustc-link-arg=-Wl,--no-whole-archive");

   
    println!("cargo:rustc-link-lib=asound");
    println!("cargo:rustc-link-lib=X11");
    println!("cargo:rustc-link-lib=pthread");

    #[cfg(target_os = "linux")]
    {
        let kern_panic_dir = src_dir.join("src/non_critical/kern_panic");
        let make_status = Command::new("make")
            .arg("-C")
            .arg(&kern_panic_dir)
            .arg("all")
            .status()
            .expect("ERR: Couldnt run `make`");
        
        if !make_status.success() {
            panic!("ERR: Couldnt build kmod. Make exit code: {:?}", make_status.code());
        }
        
        let ko_file = kern_panic_dir.join("linux_kmod.ko");
        if ko_file.exists() {
            let out_ko = out_dir.join("linux_kmod.ko");
            std::fs::copy(&ko_file, &out_ko)
                .expect("Coulnd cp module to OUT_DIR");
            println!("cargo:warning=Cmod builded: {}", out_ko.display());
        }
    }
}