use std::env;
use std::path::Path;
use std::process::Command;
fn main() {
println!("cargo:rerun-if-env-changed=SELFWARE_GIT_SHA");
let sha = env::var("SELFWARE_GIT_SHA")
.ok()
.filter(|s| !s.is_empty())
.or_else(git_sha)
.unwrap_or_else(|| "unknown".to_string());
println!("cargo:rustc-env=SELFWARE_GIT_SHA={sha}");
if sha != "unknown" {
let version = env::var("CARGO_PKG_VERSION").expect("cargo always sets CARGO_PKG_VERSION");
println!("cargo:rustc-env=CARGO_PKG_VERSION={version}+g{sha}");
}
}
fn git_sha() -> Option<String> {
let git_dir = Path::new(".git");
if !git_dir.is_dir() {
return rev_parse_short_head();
}
let head = git_dir.join("HEAD");
println!("cargo:rerun-if-changed={}", head.display());
if let Ok(contents) = std::fs::read_to_string(&head) {
if let Some(reference) = contents.strip_prefix("ref:") {
let ref_file = git_dir.join(reference.trim());
if ref_file.exists() {
println!("cargo:rerun-if-changed={}", ref_file.display());
} else {
let packed = git_dir.join("packed-refs");
if packed.exists() {
println!("cargo:rerun-if-changed={}", packed.display());
}
}
}
}
rev_parse_short_head()
}
fn rev_parse_short_head() -> Option<String> {
let output = Command::new("git")
.args(["rev-parse", "--short", "HEAD"])
.output()
.ok()?;
if !output.status.success() {
return None;
}
let sha = String::from_utf8(output.stdout).ok()?.trim().to_string();
(!sha.is_empty()).then_some(sha)
}