build.rs

use std::{borrow::Cow, env, process::Command};

/// Generate the `cargo:` key output
pub fn generate_cargo_keys() {
    let output = Command::new("git")
        .args(["rev-parse", "--short", "HEAD"])
        .output();

    let commit = match output {
        Ok(o) if o.status.success() => {
            let sha = String::from_utf8_lossy(&o.stdout).trim().to_owned();
            Cow::from(sha)
        }
        Ok(o) => {
            println!("cargo:warning=Git command failed with status: {}", o.status);
            Cow::from("unknown")
        }
        Err(err) => {
            println!("cargo:warning=Failed to execute git command: {}", err);
            Cow::from("unknown")
        }
    };

    println!("cargo:rustc-env=APP_VERSION={}", get_version(&commit))
}

fn get_platform() -> String {
    let env_dash = if env::var("CARGO_CFG_TARGET_ENV").unwrap().is_empty() {
        ""
    } else {
        "-"
    };

    format!(
        "{}-{}{}{}",
        env::var("CARGO_CFG_TARGET_ARCH").unwrap(),
        env::var("CARGO_CFG_TARGET_OS").unwrap(),
        env_dash,
        env::var("CARGO_CFG_TARGET_ENV").unwrap_or(String::from("")),
    )
}

fn get_version(impl_commit: &str) -> String {
    let commit_dash = if impl_commit.is_empty() { "" } else { "-" };

    format!(
        "{}{}{}-{}",
        std::env::var("CARGO_PKG_VERSION").unwrap_or_default(),
        commit_dash,
        impl_commit,
        get_platform(),
    )
}

pub fn main() {
    generate_cargo_keys();
}