116 lines
3.9 KiB
Rust
116 lines
3.9 KiB
Rust
#![allow(clippy::expect_fun_call)]
|
|
use std::collections::hash_map::DefaultHasher;
|
|
use std::env;
|
|
use std::fs::{metadata, write};
|
|
use std::hash::{Hash, Hasher};
|
|
use std::io::Read;
|
|
use std::path::PathBuf;
|
|
use std::process::Command;
|
|
|
|
fn get_node_version() -> std::io::Result<String> {
|
|
let output = Command::new("node").arg("-v").output()?;
|
|
let stdout_str = String::from_utf8_lossy(&output.stdout);
|
|
|
|
// version should not have a leading "v" or trailing whitespace
|
|
Ok(stdout_str.trim().trim_start_matches('v').to_string())
|
|
}
|
|
|
|
fn download_node_lib(dist_url: &str, version: &str, arch: &str) -> Vec<u8> {
|
|
// Assume windows since we know we are building on windows.
|
|
let url = format!(
|
|
"{dist_url}/v{version}/win-{arch}/node.lib",
|
|
dist_url = dist_url,
|
|
version = version,
|
|
arch = arch
|
|
);
|
|
|
|
match ureq::get(&url).call() {
|
|
Ok(response) => {
|
|
let mut reader = response.into_reader();
|
|
let mut bytes = vec![];
|
|
reader.read_to_end(&mut bytes).unwrap();
|
|
bytes
|
|
}
|
|
Err(error) => {
|
|
panic!("Failed to download node.lib: {:#?}", error);
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn setup() {
|
|
let out_dir = env::var("OUT_DIR").expect("OUT_DIR is not set");
|
|
|
|
// Assume nodejs if not specified.
|
|
let dist_url =
|
|
env::var("NPM_CONFIG_DISTURL").unwrap_or_else(|_| "https://nodejs.org/dist".to_string());
|
|
|
|
// Try to get local nodejs version if not specified.
|
|
let node_version = env::var("NPM_CONFIG_TARGET")
|
|
.or_else(|_| get_node_version())
|
|
.expect("Failed to determine nodejs version");
|
|
|
|
// NPM also gives us an arch var, but let's trust cargo more.
|
|
// We translate from cargo's arch env format into npm/gyps's.
|
|
// See https://doc.rust-lang.org/reference/conditional-compilation.html#target_arch for rust env values.
|
|
// Nodejs appears to follow `process.arch`.
|
|
// See https://nodejs.org/docs/latest/api/process.html#process_process_arch for npm env values.
|
|
// For windows, we only support `['ia32', 'x64', 'arm64']`
|
|
// https://github.com/nodejs/node-gyp/blob/master/lib/install.js#L301
|
|
let arch = env::var("CARGO_CFG_TARGET_ARCH")
|
|
.map(|arch| match arch.as_str() {
|
|
"x86" => "x86",
|
|
"x86_64" => "x64",
|
|
// https://github.com/nodejs/node/issues/25998
|
|
// actually not supported for now
|
|
// because we can not get `node.lib` file for `aarch64` device
|
|
"aarch64" => "arm64",
|
|
arch => panic!("Unsupported CPU Architecture: {}", arch),
|
|
})
|
|
.expect("Failed to determine target arch");
|
|
|
|
println!("cargo:rerun-if-env-changed=NPM_CONFIG_DISTURL");
|
|
println!("cargo:rerun-if-env-changed=NPM_CONFIG_TARGET");
|
|
|
|
let mut node_lib_file_path = PathBuf::from(out_dir);
|
|
let link_search_dir = node_lib_file_path.clone();
|
|
|
|
// Hash the dist_url and store it in the node lib file name.
|
|
let dist_url_hash = {
|
|
let mut hasher = DefaultHasher::new();
|
|
dist_url.hash(&mut hasher);
|
|
hasher.finish()
|
|
};
|
|
|
|
// Encode version, arch, and dist_url to detect and reaquire node.lib when these 3 change.
|
|
let node_lib_file_name = format!(
|
|
"node-{version}-{arch}-{dist_url_hash}.lib",
|
|
version = node_version,
|
|
arch = arch,
|
|
dist_url_hash = dist_url_hash
|
|
);
|
|
node_lib_file_path.push(&node_lib_file_name);
|
|
|
|
// If file does not exist, download it.
|
|
if metadata(&node_lib_file_path).is_err() {
|
|
let node_lib = download_node_lib(&dist_url, &node_version, &arch);
|
|
|
|
write(&node_lib_file_path, &node_lib).expect(&format!(
|
|
"Could not save file to {}",
|
|
node_lib_file_path.to_str().unwrap()
|
|
));
|
|
}
|
|
|
|
println!(
|
|
"cargo:rustc-link-lib={}",
|
|
node_lib_file_path.file_stem().unwrap().to_str().unwrap()
|
|
);
|
|
println!(
|
|
"cargo:rustc-link-search=native={}",
|
|
link_search_dir.display()
|
|
);
|
|
// Link `win_delay_load_hook.obj`
|
|
// Needed for electron, but okay for other environments
|
|
// https://github.com/neon-bindings/neon/pull/627
|
|
println!("cargo:rustc-cdylib-link-arg=delayimp.lib");
|
|
println!("cargo:rustc-cdylib-link-arg=/DELAYLOAD:node.exe");
|
|
}
|