build.rs

use cmake::Config;
use std::env;

fn main() {
    let custom_cc = env::var("CC");
    let custom_cxx = env::var("CXX");
    let conda_build = env::var("CONDA_BUILD");
    let nopie_build = env::var("NOPIE");
    let nobmi2_var = env::var("NO_BMI2");

    let is_conda_build = match conda_build {
        Ok(val) => match val.to_uppercase().as_str() {
            "TRUE" | "1" | "YES" => true,
            "FALSE" | "0" | "NO" => false,
            _ => true,
        },
        Err(_e) => false,
    };

    println!("cargo:rerun-if-changed=cuttlefish/CMakeLists.txt");
    println!("cargo:rerun-if-changed=piscem-cpp/CMakeLists.txt");

    let mut cfg_piscem_cpp = Box::new(Config::new("piscem-cpp"));
    let mut cfg_cf = Box::new(Config::new("cuttlefish"));

    (*cfg_cf).define("INSTANCE_COUNT", "32");
    if let Ok(cc_var) = custom_cc {
        (*cfg_piscem_cpp).define("CMAKE_C_COMPILER", cc_var.clone());
        (*cfg_cf).define("CMAKE_C_COMPILER", cc_var);
    }

    if let Ok(cxx_var) = custom_cxx {
        (*cfg_piscem_cpp).define("CMAKE_CXX_COMPILER", cxx_var.clone());
        (*cfg_cf).define("CMAKE_CXX_COMPILER", cxx_var);
    }

    if is_conda_build {
        (*cfg_cf).define("CONDA_BUILD", "TRUE");
        (*cfg_cf).define("CMAKE_OSX_DEPLOYMENT_TARGET", "10.15");
        (*cfg_cf).define("MACOSX_SDK_VERSION", "10.15");
    }

    if let Ok(nobmi2) = nobmi2_var {
        match nobmi2.as_str() {
            "1" | "TRUE" | "true" | "True" => {
                (*cfg_piscem_cpp).define("NO_BMI2", "TRUE");
            }
            _ => {}
        }
    }

    (*cfg_piscem_cpp).always_configure(false);
    (*cfg_cf).always_configure(false);

    let dst_piscem_cpp = (*cfg_piscem_cpp).build();
    let dst_cf = (*cfg_cf).build();

    if let Ok(nopie) = nopie_build {
        match nopie.as_str() {
            "1" | "TRUE" | "true" | "True" => {
                println!("cargo:rustc-link-arg=-no-pie");
            }
            _ => {}
        }
    }

    println!(
        "cargo:rustc-link-search=native={}",
        dst_cf.join("lib").display()
    );
    println!(
        "cargo:rustc-link-search=native={}",
        dst_piscem_cpp.join("lib").display()
    );

    // For some reason, if we are using
    // *some* linux distros (and on conda) and are
    // building for the linux target;
    // things get put in the lib64 directory
    // rather than lib... So, we add that here
    println!(
        "cargo:rustc-link-search=native={}",
        dst_cf.join("lib64").display()
    );
    println!(
        "cargo:rustc-link-search=native={}",
        dst_piscem_cpp.join("lib64").display()
    );

    let profile = std::env::var("PROFILE").unwrap();
    match profile.as_str() {
        "debug" => {
            println!(
                "cargo:rustc-link-search=native={}",
                dst_piscem_cpp.join("Debug").join("lib64").display()
            );
            println!(
                "cargo:rustc-link-search=native={}",
                dst_piscem_cpp.join("Debug").join("lib").display()
            );
        }
        "release" => {
            println!(
                "cargo:rustc-link-search=native={}",
                dst_piscem_cpp.join("Release").join("lib64").display()
            );
            println!(
                "cargo:rustc-link-search=native={}",
                dst_piscem_cpp.join("Release").join("lib").display()
            );
        }
        _ => (),
    }

    println!("cargo:rustc-link-lib=static=kmc_core");
    //println!("cargo:rustc-link-lib=static=pesc_static");
    //println!("cargo:rustc-link-lib=static=build_static");
    println!("cargo:rustc-link-lib=static=sshash_static");
    println!("cargo:rustc-link-lib=static=zcf");
    println!("cargo:rustc-link-lib=static=bz2");

    #[cfg(target_os = "linux")]
    println!("cargo:rustc-link-lib=dylib=stdc++");

    #[cfg(target_os = "macos")]
    println!("cargo:rustc-link-lib=dylib=c++");
}