68 lines
2.5 KiB
Rust
68 lines
2.5 KiB
Rust
#![allow(clippy::expect_fun_call)]
|
|
use std::env;
|
|
use std::fs::{metadata, write};
|
|
use std::path::PathBuf;
|
|
|
|
fn copy_node_lib(arch: &str) -> Vec<u8> {
|
|
match arch {
|
|
"x64" => include_bytes!("libs/node-x64.lib").to_vec(),
|
|
"x86" => include_bytes!("libs/node-x86.lib").to_vec(),
|
|
"arm64" => include_bytes!("libs/node-arm64.lib").to_vec(),
|
|
_ => unreachable!(),
|
|
}
|
|
}
|
|
|
|
pub fn setup() {
|
|
let out_dir = env::var("OUT_DIR").expect("OUT_DIR is not set");
|
|
|
|
// 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
|
|
// but we can get it from https://unofficial-builds.nodejs.org/download/release
|
|
// just set the `NPM_CONFIG_DISTURL` to `https://unofficial-builds.nodejs.org/download/release`
|
|
"aarch64" => "arm64",
|
|
arch => panic!("Unsupported CPU Architecture: {}", arch),
|
|
})
|
|
.expect("Failed to determine target arch");
|
|
|
|
let mut node_lib_file_path = PathBuf::from(out_dir);
|
|
let link_search_dir = node_lib_file_path.clone();
|
|
|
|
// Encode arch to detect and require node.lib.
|
|
let node_lib_file_name = format!("node-{arch}.lib", arch = arch,);
|
|
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 = copy_node_lib(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");
|
|
}
|