forked from abdul124/rust-core_io
89 lines
2.6 KiB
Rust
89 lines
2.6 KiB
Rust
extern crate rustc_version;
|
|
|
|
use std::env;
|
|
use std::fs::File;
|
|
use std::io::Write;
|
|
use std::path::PathBuf;
|
|
|
|
struct Mapping(&'static str,&'static str);
|
|
|
|
fn parse_mappings(mut mappings: &'static str) -> Vec<Mapping> {
|
|
// FIXME: The format used here used to be parsed directly by rustc, which
|
|
// is why it's kind of weird. It should be changed to a saner format.
|
|
|
|
const P1: &'static str = r#"-Mapping(""#;
|
|
const P2: &'static str = r#"",""#; ;
|
|
const P3: &'static str = "\")\n";
|
|
|
|
trait TakePrefix: Sized {
|
|
fn take_prefix(&mut self, mid: usize) -> Self;
|
|
}
|
|
|
|
impl<'a> TakePrefix for &'a str {
|
|
fn take_prefix(&mut self, mid: usize) -> Self {
|
|
let prefix = &self[..mid];
|
|
*self = &self[mid..];
|
|
prefix
|
|
}
|
|
}
|
|
|
|
let mut result = Vec::with_capacity( mappings.len() / (P1.len()+40+P2.len()+40+P3.len()) );
|
|
|
|
while mappings.len() != 0 {
|
|
match (
|
|
mappings.take_prefix(P1.len()),
|
|
mappings.take_prefix(40),
|
|
mappings.take_prefix(P2.len()),
|
|
mappings.take_prefix(40),
|
|
mappings.take_prefix(P3.len()),
|
|
) {
|
|
(P1, hash1, P2, hash2, P3) => result.push(Mapping(hash1, hash2)),
|
|
_ => panic!("Invalid input in mappings"),
|
|
}
|
|
}
|
|
|
|
result
|
|
}
|
|
|
|
fn main() {
|
|
let ver=rustc_version::version_meta();
|
|
|
|
let io_commit=match env::var("CORE_IO_COMMIT") {
|
|
Ok(c) => c,
|
|
Err(env::VarError::NotUnicode(_)) => panic!("Invalid commit specified in CORE_IO_COMMIT"),
|
|
Err(env::VarError::NotPresent) => {
|
|
let mappings=parse_mappings(include_str!("mapping.rs"));
|
|
|
|
let compiler=ver.commit_hash.expect("Couldn't determine compiler version");
|
|
mappings.iter().find(|&&Mapping(elem,_)|elem==compiler).expect("Unknown compiler version, upgrade core_io?").1.to_owned()
|
|
}
|
|
};
|
|
|
|
if ver.commit_date.as_ref().map_or(false,|d| &**d>="2018-01-01") {
|
|
println!("cargo:rustc-cfg=core_memchr");
|
|
}
|
|
|
|
if ver.commit_date.as_ref().map_or(false,|d| &**d>="2017-06-15") {
|
|
println!("cargo:rustc-cfg=no_collections");
|
|
}
|
|
|
|
if ver.commit_date.as_ref().map_or(false,|d| &**d<"2016-12-15") {
|
|
println!("cargo:rustc-cfg=rustc_unicode");
|
|
} else if ver.commit_date.as_ref().map_or(false,|d| &**d<"2017-03-03") {
|
|
println!("cargo:rustc-cfg=std_unicode");
|
|
}
|
|
|
|
let mut dest_path=PathBuf::from(env::var_os("OUT_DIR").unwrap());
|
|
dest_path.push("io.rs");
|
|
let mut f=File::create(&dest_path).unwrap();
|
|
|
|
let mut target_path=PathBuf::from(env::var_os("CARGO_MANIFEST_DIR").unwrap());
|
|
target_path.push("src");
|
|
target_path.push(io_commit);
|
|
target_path.push("mod.rs");
|
|
|
|
f.write_all(br#"#[path=""#).unwrap();
|
|
f.write_all(target_path.into_os_string().into_string().unwrap().replace("\\", "\\\\").as_bytes()).unwrap();
|
|
f.write_all(br#""] mod io;"#).unwrap();
|
|
}
|