mog 0.0.0

A commad-line tool to log ML experiments
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

use anyhow::Context;
use git_url_parse::GitUrl;
use std::path::{Path, PathBuf};

#[derive(Debug)]
pub struct GitInfo {
    pub commit: git2::Oid,
    pub origin_url: GitUrl,
    pub is_dirty: bool,
    pub current_dir: PathBuf,
}

impl GitInfo {
    pub fn repository_name(&self) -> &str {
        &self.origin_url.name
    }

    pub fn https_url(&self) -> Option<String> {
        if let Some(host) = &self.origin_url.host {
            Some(format!("https://{}/{}", host, self.origin_url.fullname))
        } else {
            None
        }
    }
}

impl GitInfo {
    pub fn new<P: AsRef<Path>>(dir: P) -> anyhow::Result<Self> {
        let repo = git2::Repository::discover(dir)?;
        let head = repo.head()?;
        let commit = head.peel_to_commit()?.id();
        let origin_url = repo
            .find_remote("origin")?
            .url()
            .ok_or_else(|| anyhow::anyhow!("origin URL is not a valid UTF-8"))
            .and_then(|s| GitUrl::parse(s).with_context(|| format!("malformed URL: {:?}", s)))?;

        let head_tree = head.peel_to_tree()?;
        let mut diff_opt = git2::DiffOptions::new();
        diff_opt.include_untracked(true);
        let diff_files = repo
            .diff_tree_to_workdir_with_index(Some(&head_tree), Some(&mut diff_opt))?
            .deltas()
            .len();
        let is_dirty = diff_files > 0;

        let rootdir = repo
            .workdir()
            .ok_or_else(|| anyhow::anyhow!("this is a bare repository"))?;
        let current_dir = std::env::current_dir()?
            .strip_prefix(rootdir)?
            .to_path_buf();
        Ok(Self {
            commit,
            origin_url,
            is_dirty,
            current_dir,
        })
    }
}