merge-ready 0.2.1

Show pull request merge blockers as concise prompt tokens
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

use serde::{Deserialize, Serialize};

const DEFAULT_FORMAT: &str = "$symbol $label";

pub const CURRENT_VERSION: u32 = 1;

#[derive(Deserialize, Serialize, Default)]
pub struct Config {
    #[serde(default)]
    pub version: u32,
    pub merge_ready: Option<TokenConfig>,
    pub conflict: Option<TokenConfig>,
    pub update_branch: Option<TokenConfig>,
    pub sync_unknown: Option<TokenConfig>,
    pub ci_fail: Option<TokenConfig>,
    pub ci_action: Option<TokenConfig>,
    pub review: Option<TokenConfig>,
    pub error: Option<ErrorConfig>,
}

impl Config {
    pub fn fill_defaults(&mut self) {
        let tok = |symbol: &str, label: &str| TokenConfig {
            symbol: Some(symbol.to_owned()),
            label: Some(label.to_owned()),
            format: None,
        };
        self.merge_ready
            .get_or_insert_with(|| tok("✓", "merge-ready"));
        self.conflict.get_or_insert_with(|| tok("✗", "conflict"));
        self.update_branch
            .get_or_insert_with(|| tok("✗", "update-branch"));
        self.sync_unknown
            .get_or_insert_with(|| tok("?", "sync-unknown"));
        self.ci_fail.get_or_insert_with(|| tok("✗", "ci-fail"));
        self.ci_action.get_or_insert_with(|| tok("⚠", "ci-action"));
        self.review.get_or_insert_with(|| tok("⚠", "review"));
        let error = self.error.get_or_insert_with(ErrorConfig::empty);
        error
            .auth_required
            .get_or_insert_with(|| tok("!", "gh auth login"));
        error
            .rate_limited
            .get_or_insert_with(|| tok("✗", "rate-limited"));
        error.api_error.get_or_insert_with(|| tok("✗", "api-error"));
    }
}

#[derive(Deserialize, Serialize)]
pub struct TokenConfig {
    pub symbol: Option<String>,
    pub label: Option<String>,
    pub format: Option<String>,
}

#[must_use]
pub fn render_token(token: &TokenConfig, default_symbol: &str, default_label: &str) -> String {
    let symbol = token.symbol.as_deref().unwrap_or(default_symbol);
    let label = token.label.as_deref().unwrap_or(default_label);
    let fmt = token.format.as_deref().unwrap_or(DEFAULT_FORMAT);
    fmt.replace("$symbol", symbol).replace("$label", label)
}

#[derive(Deserialize, Serialize)]
pub struct ErrorConfig {
    pub auth_required: Option<TokenConfig>,
    pub rate_limited: Option<TokenConfig>,
    pub api_error: Option<TokenConfig>,
}

impl ErrorConfig {
    fn empty() -> Self {
        Self {
            auth_required: None,
            rate_limited: None,
            api_error: None,
        }
    }
}