build.rs

extern crate bindgen;
extern crate cbindgen;

use std::env;

fn main() {
    let project_dir = env::var("CARGO_MANIFEST_DIR").unwrap();

    println!("cargo:rustc-link-search={}/deps/root/lib", project_dir); // the "-L" flag
    if env::var("AG2PC").is_ok() {
        println!("cargo:rustc-link-lib=tokenm-utils"); // the "-l" flag
        println!("cargo:rustc-link-lib=test-e2e"); // the "-l" flag
    } else {
        println!("cargo:rustc-link-lib=token-utils"); // the "-l" flag
    }

    // Create bindings
    let header_file: &str;
    if env::var("AG2PC").is_ok() {
        header_file = "deps/root/include/update-state/ag2pc/tokens.h";
    } else {
        header_file = "deps/root/include/update-state/tokens.h";
    }
    let path_to_header_file = format!("{}/{}", project_dir, header_file);
    //println!("token header file path: {}", path_to_header_file);
    let bindings = bindgen::Builder::default()
        // The input header we would like to generate
        // bindings for.
        .header(path_to_header_file)
        .clang_arg("-x")
        .clang_arg("c++")
        .clang_arg("-Ideps/root/include")
        .trust_clang_mangling(false)
        // 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.
    bindings
        .write_to_file("src/bindings.rs")
        .expect("Couldn't write bindings!");

    // TODO: Create build for libtoken-utils here

    let crate_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
    cbindgen::generate(crate_dir)
        .expect("Unable to generate bindings")
        .write_to_file("include/bindings.h");
}