mirror of https://github.com/rust-lang/cargo
103 lines
3.4 KiB
Rust
103 lines
3.4 KiB
Rust
use flate2::{Compression, GzBuilder};
|
|
use std::ffi::OsStr;
|
|
use std::fs;
|
|
use std::path::Path;
|
|
use std::process::Command;
|
|
|
|
fn main() {
|
|
commit_info();
|
|
compress_man();
|
|
windows_manifest();
|
|
// ALLOWED: Accessing environment during build time shouldn't be prohibited.
|
|
#[allow(clippy::disallowed_methods)]
|
|
let target = std::env::var("TARGET").unwrap();
|
|
println!("cargo:rustc-env=RUST_HOST_TARGET={target}");
|
|
}
|
|
|
|
fn compress_man() {
|
|
// ALLOWED: Accessing environment during build time shouldn't be prohibited.
|
|
#[allow(clippy::disallowed_methods)]
|
|
let out_path = Path::new(&std::env::var("OUT_DIR").unwrap()).join("man.tgz");
|
|
let dst = fs::File::create(out_path).unwrap();
|
|
let encoder = GzBuilder::new()
|
|
.filename("man.tar")
|
|
.write(dst, Compression::best());
|
|
let mut ar = tar::Builder::new(encoder);
|
|
ar.mode(tar::HeaderMode::Deterministic);
|
|
|
|
let mut add_files = |dir, extension| {
|
|
let mut files = fs::read_dir(dir)
|
|
.unwrap()
|
|
.map(|e| e.unwrap().path())
|
|
.collect::<Vec<_>>();
|
|
files.sort();
|
|
for path in files {
|
|
if path.extension() != Some(extension) {
|
|
continue;
|
|
}
|
|
println!("cargo:rerun-if-changed={}", path.display());
|
|
ar.append_path_with_name(&path, path.file_name().unwrap())
|
|
.unwrap();
|
|
}
|
|
};
|
|
|
|
add_files(Path::new("src/etc/man"), OsStr::new("1"));
|
|
add_files(Path::new("src/doc/man/generated_txt"), OsStr::new("txt"));
|
|
let encoder = ar.into_inner().unwrap();
|
|
encoder.finish().unwrap();
|
|
}
|
|
|
|
fn commit_info() {
|
|
if !Path::new(".git").exists() {
|
|
return;
|
|
}
|
|
|
|
// Var set by bootstrap whenever omit-git-hash is enabled in rust-lang/rust's config.toml.
|
|
println!("cargo:rerun-if-env-changed=CFG_OMIT_GIT_HASH");
|
|
#[allow(clippy::disallowed_methods)]
|
|
if std::env::var_os("CFG_OMIT_GIT_HASH").is_some() {
|
|
return;
|
|
}
|
|
|
|
let output = match Command::new("git")
|
|
.arg("log")
|
|
.arg("-1")
|
|
.arg("--date=short")
|
|
.arg("--format=%H %h %cd")
|
|
.arg("--abbrev=9")
|
|
.output()
|
|
{
|
|
Ok(output) if output.status.success() => output,
|
|
_ => return,
|
|
};
|
|
let stdout = String::from_utf8(output.stdout).unwrap();
|
|
let mut parts = stdout.split_whitespace();
|
|
let mut next = || parts.next().unwrap();
|
|
println!("cargo:rustc-env=CARGO_COMMIT_HASH={}", next());
|
|
println!("cargo:rustc-env=CARGO_COMMIT_SHORT_HASH={}", next());
|
|
println!("cargo:rustc-env=CARGO_COMMIT_DATE={}", next())
|
|
}
|
|
|
|
#[allow(clippy::disallowed_methods)]
|
|
fn windows_manifest() {
|
|
use std::env;
|
|
let target_os = env::var("CARGO_CFG_TARGET_OS");
|
|
let target_env = env::var("CARGO_CFG_TARGET_ENV");
|
|
if Ok("windows") == target_os.as_deref() && Ok("msvc") == target_env.as_deref() {
|
|
static WINDOWS_MANIFEST_FILE: &str = "windows.manifest.xml";
|
|
|
|
let mut manifest = env::current_dir().unwrap();
|
|
manifest.push(WINDOWS_MANIFEST_FILE);
|
|
|
|
println!("cargo:rerun-if-changed={WINDOWS_MANIFEST_FILE}");
|
|
// Embed the Windows application manifest file.
|
|
println!("cargo:rustc-link-arg-bin=cargo=/MANIFEST:EMBED");
|
|
println!(
|
|
"cargo:rustc-link-arg-bin=cargo=/MANIFESTINPUT:{}",
|
|
manifest.to_str().unwrap()
|
|
);
|
|
// Turn linker warnings into errors.
|
|
println!("cargo:rustc-link-arg-bin=cargo=/WX");
|
|
}
|
|
}
|