use std::env;
use std::path::PathBuf;

use bindgen::callbacks::ParseCallbacks;

const SERIALIZABLE_TYPES: &[&str] = &[
    "Vec2",
    "Vec3",
    "Vec4",
    "Mat4",
    "Quat",
    "Transform",
    "Bbox_3D",
    "OBB",
    "DirectionalLight",
    "PointLight",
];
const EQ_TYPES: &[&str] = &[
    "BufferHandle",
    "TextureHandle",
    "MeshHandle",
    "MaterialHandle",
    "ModelHandle",
];
const DEFAULT_TYPES: &[&str] = &["ShaderDataLayout"];

#[derive(Debug)]
struct AdditionalDerives;
impl ParseCallbacks for AdditionalDerives {
    fn add_derives(&self, info: &bindgen::callbacks::DeriveInfo<'_>) -> Vec<String> {
        let mut derives = vec![];
        if SERIALIZABLE_TYPES.contains(&info.name) {
            derives.extend_from_slice(&["Serialize".to_string(), "Deserialize".to_string()]);
        }
        if EQ_TYPES.contains(&info.name) {
            derives.extend_from_slice(&["PartialEq".to_string()]);
        }
        if DEFAULT_TYPES.contains(&info.name) {
            derives.push("Default".to_string());
        }
        derives
    }
}

fn main() {
    // TODO: we need to look based on OS

    // Tell cargo to look for shared libraries in the specified directory
    let static_lib_path = "/home/joshua/repos/cel-core3/build".to_string();
    // let static_lib_path = std::env::var("CELERITAS_CORE_LIB")
    //     .unwrap_or("../../../build/macosx/arm64/debug".to_string());

    println!("cargo:rustc-link-search={static_lib_path}");

    // Tell cargo to tell rustc to link the system bzip2
    // shared library.
    println!("cargo:rustc-link-lib=dylib=celeritas");
    println!("cargo:rustc-link-lib=glfw");

    // The bindgen::Builder is the main entry point
    // to bindgen, and lets you build up options for
    // the resulting bindings.
    let bindings = bindgen::Builder::default()
        // The input header we would like to generate
        // bindings for.
        .header("../../../include/celeritas.h")
        // -- dependencies
        .clang_arg("-I../../../deps/cgltf")
        .clang_arg("-I../../../deps/glfw-3.3.8/include/GLFW")
        .clang_arg("-I../../../deps/glad/include")
        .clang_arg("-I../../../deps/stb_image")
        .clang_arg("-I../../../deps/stb_image_write")
        .clang_arg("-I../../../deps/stb_truetype")
        .generate_inline_functions(true)
        // Tell cargo to invalidate the built crate whenever any of the
        // included header files changed.
        .rustified_enum("GPU_TextureType")
        // .rustified_enum("GPU_TextureFormat")
        .parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
        .parse_callbacks(Box::new(AdditionalDerives))
        // Finish the builder and generate the bindings.
        .generate()
        // Unwrap the Result and panic on failure.
        .expect("Unable to generate bindings");

    // Write the bindings to the $OUT_DIR/bindings.rs file.
    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
    bindings
        .write_to_file(out_path.join("bindings.rs"))
        .expect("Couldn't write bindings!");
}