-
-
Notifications
You must be signed in to change notification settings - Fork 169
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add build script to detect Provider support
- Loading branch information
Showing
1 changed file
with
66 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,66 @@ | ||
use std::env; | ||
use std::fs; | ||
use std::path::Path; | ||
use std::process::{Command, ExitStatus, Stdio}; | ||
use std::str; | ||
|
||
// This code exercises the surface area that we expect of the Provider API. If | ||
// the current toolchain is able to compile it, then thiserror is able to use | ||
// providers for backtrace support. | ||
const PROBE: &str = r#" | ||
#![feature(provide_any)] | ||
use std::any::{Demand, Provider}; | ||
fn _f<'a, P: Provider>(p: &'a P, demand: &mut Demand<'a>) { | ||
p.provide(demand); | ||
} | ||
"#; | ||
|
||
fn main() { | ||
match compile_probe() { | ||
Some(status) if status.success() => println!("cargo:rustc-cfg=provide_any"), | ||
_ => {} | ||
} | ||
} | ||
|
||
fn compile_probe() -> Option<ExitStatus> { | ||
let rustc = env::var_os("RUSTC")?; | ||
let out_dir = env::var_os("OUT_DIR")?; | ||
let probefile = Path::new(&out_dir).join("probe.rs"); | ||
fs::write(&probefile, PROBE).ok()?; | ||
|
||
// Make sure to pick up Cargo rustc configuration. | ||
let mut cmd = if let Some(wrapper) = env::var_os("RUSTC_WRAPPER") { | ||
let mut cmd = Command::new(wrapper); | ||
// The wrapper's first argument is supposed to be the path to rustc. | ||
cmd.arg(rustc); | ||
cmd | ||
} else { | ||
Command::new(rustc) | ||
}; | ||
|
||
cmd.stderr(Stdio::null()) | ||
.arg("--edition=2018") | ||
.arg("--crate-name=thiserror_build") | ||
.arg("--crate-type=lib") | ||
.arg("--emit=metadata") | ||
.arg("--out-dir") | ||
.arg(out_dir) | ||
.arg(probefile); | ||
|
||
if let Some(target) = env::var_os("TARGET") { | ||
cmd.arg("--target").arg(target); | ||
} | ||
|
||
// If Cargo wants to set RUSTFLAGS, use that. | ||
if let Ok(rustflags) = env::var("CARGO_ENCODED_RUSTFLAGS") { | ||
if !rustflags.is_empty() { | ||
for arg in rustflags.split('\x1f') { | ||
cmd.arg(arg); | ||
} | ||
} | ||
} | ||
|
||
cmd.status().ok() | ||
} |