harness-cli 0.0.9

Precise and reproducible benchmarking
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

use std::process::Command;

use git_info2::types::GitInfo;

pub fn get_git_hash() -> anyhow::Result<String> {
    let git_info = git_info2::get();
    let mut hash = git_info
        .head
        .last_commit_hash
        .ok_or_else(|| anyhow::anyhow!("Failed to get the current commit hash"))?;
    if git_info.dirty.unwrap_or_default() {
        hash += "-dirty";
    }
    Ok(hash)
}

pub fn get_second_last_git_hash() -> anyhow::Result<String> {
    Command::new("git")
        .args(["rev-parse", "@~"])
        .output()
        .map_err::<anyhow::Error, _>(|e| e.into())
        .and_then(|o| String::from_utf8(o.stdout).map_err(|e| e.into()))
        .map(|s| s.trim().to_owned())
}

pub fn get_branch_last_git_hash(branch: &str) -> anyhow::Result<String> {
    Command::new("git")
        .args(["rev-parse", branch])
        .output()
        .map_err::<anyhow::Error, _>(|e| e.into())
        .and_then(|o| String::from_utf8(o.stdout).map_err(|e| e.into()))
        .map(|s| s.trim().to_owned())
}

pub fn restore_git_state(prev: &GitInfo) -> anyhow::Result<()> {
    let curr = git_info2::get();
    if prev.head.last_commit_hash != curr.head.last_commit_hash {
        let checkout_target = if let Some(branch) = prev.current_branch.as_ref() {
            let hash = get_branch_last_git_hash(branch)?;
            if Some(hash) == prev.head.last_commit_hash {
                branch
            } else {
                prev.head.last_commit_hash.as_ref().unwrap()
            }
        } else {
            prev.head.last_commit_hash.as_ref().unwrap()
        };
        checkout_no_guard(checkout_target)?;
    }
    Ok(())
}

fn checkout_no_guard(mut commit: &str) -> anyhow::Result<bool> {
    let current_commit = git_info2::get().head.last_commit_hash.unwrap();
    if commit.ends_with("-dirty") {
        commit = commit.trim_end_matches("-dirty");
    }
    if current_commit == commit {
        return Ok(false);
    }
    let output = Command::new("git").args(["checkout", commit]).output()?;
    if !output.status.success() {
        anyhow::bail!(
            "Failed to checkout git commit: {}: {}",
            commit,
            String::from_utf8_lossy(&output.stderr)
        );
    }
    Ok(true)
}

pub struct TempGitCommitGuard {
    prev: GitInfo,
}

impl Drop for TempGitCommitGuard {
    fn drop(&mut self) {
        restore_git_state(&self.prev).unwrap();
    }
}

pub fn checkout(commit: &str) -> anyhow::Result<TempGitCommitGuard> {
    let prev = git_info2::get();
    checkout_no_guard(commit)?;
    Ok(TempGitCommitGuard { prev })
}