gitkit-cli 0.1.0

Get insightful metrics on your git repository
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use std::fmt;

use git2::StatusOptions;
use ratatui::{
    style::{
        Color, Stylize,
        palette::material::WHITE,
    },
    text::Line,
};

use crate::git::kit::KitRepo;

#[derive(Debug)]
pub struct KitStatus {
    pub files: Vec<(String, git2::Status)>,
    pub pretty: Vec<(String, String)>,
}

impl KitStatus {
    pub fn new(repo: &KitRepo) -> Self {
        let mut opts = StatusOptions::new();
        opts.include_untracked(true)
            .recurse_untracked_dirs(true)
            .include_ignored(false);

        let empty = Self {
            files: vec![],
            pretty: vec![("".to_owned(), "no files changes".to_owned())],
        };

        if let Ok(statuses) = repo.inner.statuses(Some(&mut opts)) {
            if statuses.is_empty() {
                return empty;
            }

            let files: Vec<(String, git2::Status)> = statuses
                .iter()
                .map(|entry| {
                    let path = entry.path().unwrap_or("(no path)").to_string();
                    let status = entry.status();
                    (path, status)
                })
                .collect();
            let pretty = KitStatus::pretty(&files);
            Self { files, pretty }
        } else {
            // omit the error silently and return empty vecs
            empty
        }
    }

    // (path, status)
    fn pretty(files: &Vec<(String, git2::Status)>) -> Vec<(String, String)> {
        let mut path_status: Vec<(String, String)> = Vec::new();

        for (path, status) in files {
            let code = match status {
                s if s.contains(git2::Status::WT_NEW) => "??",
                s if s.contains(git2::Status::INDEX_NEW) => "A ",
                s if s.contains(git2::Status::WT_MODIFIED) => " M",
                s if s.contains(git2::Status::INDEX_MODIFIED) => "M ",
                s if s.contains(git2::Status::WT_DELETED) => " D",
                s if s.contains(git2::Status::INDEX_DELETED) => "D ",
                _ => " ",
            };
            path_status.push((path.to_string(), code.to_string()));
        }
        path_status
    }

    // todo add pretty print for empty vec
    pub fn tui_print(&self) -> Vec<Line<'_>> {
        let lines: Vec<Line<'_>> = self
            .pretty
            .iter()
            .map(|(path, code)| {
                let line = Line::from(vec![
                    format!("{}", code).fg(Color::Green),
                    " ".into(),
                    format!(" @ {}", path).fg(WHITE),
                ]);
                line
            })
            .collect();
        lines
    }
}

impl fmt::Display for KitStatus {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        if self.files.is_empty() {
            return write!(f, "Your branch is clean.");
        }

        for (path, status) in &self.files {
            // Format the status flags into human-readable shorthand
            let code = match *status {
                s if s.contains(git2::Status::WT_NEW) => "??",
                s if s.contains(git2::Status::INDEX_NEW) => "A ",
                s if s.contains(git2::Status::WT_MODIFIED) => " M",
                s if s.contains(git2::Status::INDEX_MODIFIED) => "M ",
                s if s.contains(git2::Status::WT_DELETED) => " D",
                s if s.contains(git2::Status::INDEX_DELETED) => "D ",
                _ => "  ",
            };
            writeln!(f, "{} {}", code, path)?;
        }
        Ok(())
    }
}