use std::env::var;
use std::io::Write;
fn main() {
if has_io_safety() {
use_feature("io_safety_is_in_std")
}
use_feature_or_nothing("wasi_ext");
if has_panic_in_const_fn() {
use_feature("panic_in_const_fn")
}
println!("cargo:rerun-if-changed=build.rs");
}
fn use_feature_or_nothing(feature: &str) {
if has_feature(feature) {
use_feature(feature);
}
}
fn use_feature(feature: &str) {
println!("cargo:rustc-cfg={}", feature);
}
fn has_feature(feature: &str) -> bool {
can_compile(format!(
"#![allow(stable_features)]\n#![feature({})]",
feature
))
}
fn can_compile<T: AsRef<str>>(test: T) -> bool {
use std::process::Stdio;
let out_dir = var("OUT_DIR").unwrap();
let rustc = var("RUSTC").unwrap();
let target = var("TARGET").unwrap();
let mut cmd = if let Ok(wrapper) = var("RUSTC_WRAPPER") {
let mut cmd = std::process::Command::new(wrapper);
cmd.arg(rustc);
cmd
} else {
std::process::Command::new(rustc)
};
cmd.arg("--crate-type=rlib") .arg("--emit=metadata") .arg("--target")
.arg(target)
.arg("--out-dir")
.arg(out_dir);
if let Ok(rustflags) = var("CARGO_ENCODED_RUSTFLAGS") {
if !rustflags.is_empty() {
for arg in rustflags.split('\x1f') {
cmd.arg(arg);
}
}
}
let mut child = cmd
.arg("-") .stdin(Stdio::piped()) .stderr(Stdio::null()) .spawn()
.unwrap();
writeln!(child.stdin.take().unwrap(), "{}", test.as_ref()).unwrap();
child.wait().unwrap().success()
}
fn has_panic_in_const_fn() -> bool {
can_compile("const fn foo() {{ panic!() }}")
}
fn has_io_safety() -> bool {
can_compile(
"\
#[cfg(unix)]\n\
use std::os::unix::io::OwnedFd as Owned;\n\
#[cfg(target_os = \"wasi\")]\n\
use std::os::wasi::io::OwnedFd as Owned;\n\
#[cfg(windows)]\n\
use std::os::windows::io::OwnedHandle as Owned;\n\
\n\
pub type Success = Owned;\n\
",
)
}