looprs 0.5.1

Concise coding assistant REPL — core library
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

use std::process::Command;

#[derive(Debug, Default, Clone)]
pub struct GitInfo {
    pub branch: Option<String>,
    pub ahead: u32,
    pub modified: u32,
    pub untracked: u32,
}

pub fn collect() -> GitInfo {
    let branch = branch_name();
    let ahead = commits_ahead();
    let (modified, untracked) = changed_files();
    GitInfo {
        branch,
        ahead,
        modified,
        untracked,
    }
}

fn branch_name() -> Option<String> {
    let out = Command::new("git")
        .args(["rev-parse", "--abbrev-ref", "HEAD"])
        .output()
        .ok()?;
    if out.status.success() {
        let s = String::from_utf8_lossy(&out.stdout).trim().to_string();
        if s.is_empty() || s == "HEAD" {
            None
        } else {
            Some(s)
        }
    } else {
        None
    }
}

fn commits_ahead() -> u32 {
    let out = Command::new("git")
        .args(["rev-list", "--count", "@{u}..HEAD"])
        .output();
    match out {
        Ok(o) if o.status.success() => String::from_utf8_lossy(&o.stdout)
            .trim()
            .parse()
            .unwrap_or(0),
        _ => 0,
    }
}

fn changed_files() -> (u32, u32) {
    let out = Command::new("git").args(["status", "--porcelain"]).output();
    match out {
        Ok(o) if o.status.success() => {
            let mut modified = 0u32;
            let mut untracked = 0u32;
            for line in String::from_utf8_lossy(&o.stdout).lines() {
                if line.starts_with("??") {
                    untracked += 1;
                } else if !line.is_empty() {
                    modified += 1;
                }
            }
            (modified, untracked)
        }
        _ => (0, 0),
    }
}