stacks-puppet-node/build.rs

use std::process::Command;

fn current_git_hash() -> Option<String> {
    let commit = Command::new("git")
        .arg("log")
        .arg("-1")
        .arg("--pretty=format:%h") // Abbreviated commit hash
        .current_dir(env!("CARGO_MANIFEST_DIR"))
        .output();

    if let Ok(commit) = commit {
        if let Ok(commit) = String::from_utf8(commit.stdout) {
            return Some(commit)
        }
    }
    None
}

fn current_git_branch() -> Option<String> {
    let commit = Command::new("git")
                         .arg("rev-parse")
                         .arg("--abbrev-ref")
                         .arg("HEAD")
                         .output();
    if let Ok(commit) = commit {
        if let Ok(commit) = String::from_utf8(commit.stdout) {
            return Some(commit)
        }
    }
    None
}

fn is_working_tree_clean() -> bool {
    let status = Command::new("git")
        .arg("diff")
        .arg("--quiet")
        .arg("--exit-code")
        .current_dir(env!("CARGO_MANIFEST_DIR"))
        .status();

    if let Ok(status) = status {
        status.code() == Some(0)
    } else {
        true
    }
}

fn main() {
    if let Some(git) = current_git_hash() {
        println!("cargo:rustc-env=GIT_COMMIT={}", git);
    }
    if let Some(git) = current_git_branch() {
        println!("cargo:rustc-env=GIT_BRANCH={}", git);
    }
    if !is_working_tree_clean() {
        println!("cargo:rustc-env=GIT_TREE_CLEAN=+");
    }
}