// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

extern crate glob;
extern crate num_cpus;

use std::env;
use std::path;
use std::process::{Command, Stdio};

fn main() {
    let out_dir = env::var("OUT_DIR").expect("Should have env var OUT_DIR");
    let target = env::var("TARGET").expect("Should have env var TARGET");

    let js_src = env::var("CARGO_MANIFEST_DIR").expect("Should have env var CARGO_MANIFEST_DIR");

    env::set_var("MAKEFLAGS", format!("-j{}", num_cpus::get()));

    // The Rust package used in Spidermonkey inherits the default profile from the toplevel
    // Cargo directory. In particular, both profiles (dev or release) set panic=abort,
    // which defines an exception handling personality that will trigger calls to abort().
    // The Rust bindings package wants to be able to unwind at runtime, so override the
    // profile's value here.
    env::set_var("RUSTFLAGS", "-C panic=unwind");

    env::set_current_dir(&js_src).unwrap();

    let variant = if cfg!(feature = "debugmozjs") {
        "plaindebug"
    } else {
        "plain"
    };

    let python = env::var("PYTHON3").unwrap_or("python3".into());
    let mut cmd = Command::new(&python);
    cmd.args(&[
        "./devtools/automation/autospider.py",
        // Only build SpiderMonkey, don't run all the tests.
        "--build-only",
        // Disable Mozilla's jemalloc; Rust has its own jemalloc that we
        // can swap in instead and everything using a single malloc is
        // good.
        "--no-jemalloc",
        // Don't try to clobber the output directory. Without
        // this option, the build will fail because the directory
        // already exists but wasn't created by autospider.
        "--dep",
        "--objdir",
        &out_dir,
        &variant,
    ])
    .env("NO_RUST_PANIC_HOOK", "1")
    .env("SOURCE", &js_src)
    .env("PWD", &js_src)
    .stdout(Stdio::inherit())
    .stderr(Stdio::inherit());
    println!("Running command: {:?}", cmd);
    let result = cmd.status().expect("Should spawn autospider OK");
    assert!(result.success(), "autospider should exit OK");

    println!("cargo:rustc-link-search=native={}/js/src/build", out_dir);
    println!("cargo:rustc-link-search=native={}/js/src", out_dir);
    println!("cargo:rustc-link-lib=static=js_static");

    maybe_add_spidermonkey_rust_lib();

    println!("cargo:rustc-link-search=native={}/dist/bin", out_dir);
    println!("cargo:rustc-link-lib=nspr4");

    if target.contains("windows") {
        println!("cargo:rustc-link-lib=winmm");
        if target.contains("gnu") {
            println!("cargo:rustc-link-lib=stdc++");
        }
    } else {
        println!("cargo:rustc-link-lib=stdc++");
    }

    println!("cargo:outdir={}", out_dir);
}

/// Find if Spidermonkey built the Spidermonkey Rust library, and add it to the
/// link if it was the case.
fn maybe_add_spidermonkey_rust_lib() {
    let out_dir = env::var("OUT_DIR").expect("cargo should invoke us with the OUT_DIR env var set");

    let mut target_build_dir = path::PathBuf::from(out_dir);
    target_build_dir.push("../../");

    let mut build_dir = target_build_dir.display().to_string();
    build_dir.push_str("mozjs_sys-*/out/*/debug");

    let entries = match glob::glob(&build_dir) {
        Err(_) => {
            return;
        }
        Ok(entries) => entries,
    };

    for entry in entries {
        if let Ok(path) = entry {
            let path = path
                .canonicalize()
                .expect("Should canonicalize debug build path");
            let path = path.to_str().expect("Should be utf8");
            println!("cargo:rustc-link-search=native={}", path);
            println!("cargo:rustc-link-lib=static=jsrust");
            return;
        }
    }
}