vkit 0.1.4

Fast Rust dev CLI: manage git worktrees, Node ports, run scripts, install & sync VS Code / Cursor extensions.
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
112
113
114
115
116
117
118
119
120
121
122
123
124

//! 通过 `glab` 拉取分支对应的 MR 信息（无 glab / 失败则忽略）。

use std::collections::HashMap;
use std::path::Path;
use std::process::Command;

use serde::Deserialize;

/// MR 状态（GitLab：opened / merged / closed）。
#[derive(Debug, Clone, Copy, PartialEq, Eq, Default, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum MrState {
    #[default]
    Opened,
    Merged,
    Closed,
}

impl MrState {
    /// 同分支多条 MR 时，优先保留更「活跃」的状态。
    fn rank(self) -> u8 {
        match self {
            Self::Opened => 3,
            Self::Merged => 2,
            Self::Closed => 1,
        }
    }
}

#[derive(Debug, Clone, PartialEq, Eq, Deserialize)]
pub struct MrRef {
    pub iid: u64,
    #[serde(alias = "source_branch")]
    pub source_branch: String,
    #[serde(default)]
    pub title: String,
    #[serde(default)]
    pub state: MrState,
}

impl MrRef {
    /// 展示：`!12` 打开 · `!12✓` 已合并 · `!12×` 已关闭。
    pub fn label(&self) -> String {
        match self.state {
            MrState::Opened => format!("!{}", self.iid),
            MrState::Merged => format!("!{}✓", self.iid),
            MrState::Closed => format!("!{}×", self.iid),
        }
    }
}

/// 列出当前项目全部 MR（含 merged/closed），按 source_branch 建索引。失败返回空 map。
pub fn index_by_source_branch(cwd: &Path) -> HashMap<String, MrRef> {
    let Ok(output) = Command::new("glab")
        .args(["mr", "list", "--all", "-F", "json", "--per-page", "100"])
        .current_dir(cwd)
        .output()
    else {
        return HashMap::new();
    };
    if !output.status.success() {
        return HashMap::new();
    }
    let raw = String::from_utf8_lossy(&output.stdout);
    parse_mr_list(&raw)
}

/// 解析 `glab mr list -F json` 输出（纯函数）。同分支多条时保留更高优先级状态。
pub fn parse_mr_list(raw: &str) -> HashMap<String, MrRef> {
    let Ok(list) = serde_json::from_str::<Vec<MrRef>>(raw.trim()) else {
        return HashMap::new();
    };
    let mut map: HashMap<String, MrRef> = HashMap::new();
    for mr in list {
        let branch = mr.source_branch.clone();
        let replace = map
            .get(&branch)
            .map(|existing| mr.state.rank() > existing.state.rank())
            .unwrap_or(true);
        if replace {
            map.insert(branch, mr);
        }
    }
    map
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn parses_glab_json() {
        let raw = r#"[
          {"iid": 12, "source_branch": "feat/login", "title": "Add login", "state": "opened"},
          {"iid": 3, "source_branch": "fix-typo", "title": "typo", "state": "merged"}
        ]"#;
        let map = parse_mr_list(raw);
        assert_eq!(map["feat/login"].label(), "!12");
        assert_eq!(map["fix-typo"].label(), "!3✓");
        assert_eq!(map["fix-typo"].state, MrState::Merged);
    }

    #[test]
    fn prefers_opened_over_closed_same_branch() {
        let raw = r#"[
          {"iid": 1, "source_branch": "feat/x", "state": "closed"},
          {"iid": 2, "source_branch": "feat/x", "state": "opened"}
        ]"#;
        let map = parse_mr_list(raw);
        assert_eq!(map["feat/x"].iid, 2);
        assert_eq!(map["feat/x"].label(), "!2");
    }

    #[test]
    fn closed_label() {
        let raw = r#"[{"iid": 9, "source_branch": "old", "state": "closed"}]"#;
        assert_eq!(parse_mr_list(raw)["old"].label(), "!9×");
    }

    #[test]
    fn bad_json_yields_empty() {
        assert!(parse_mr_list("not-json").is_empty());
    }
}