cmd-usage 0.6.0

Live Command Code (commandcode.ai) usage dashboard: credits, 5-hour and weekly windows, plan limits. Watch mode or one-shot.
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

//! Plan-based model gating, mirroring the opencode plugin's tables
//! (opencode/src/gating.ts + access.ts). Kept compact: category comes from a
//! prefix map; the plugin's exact per-id table drifts, so this is a close
//! approximation and the API enforces the real gate. plan ids absent from
//! RULES allow everything; any purchased/free credits unlock everything.

#[derive(PartialEq)]
pub enum Category {
    OpenSource,
    Premium,
}

/// Longest matching prefix wins. Mirrors MODEL_CATEGORIES for the families
/// that matter; unknown non-claude ids default to OpenSource (the server
/// rejects a truly disallowed model anyway). Matches on the full model path
/// (some premium ids are provider-scoped, e.g. "sakana/fugu-ultra").
fn category(id: &str) -> Category {
    let full = id.to_lowercase();
    const PREMIUM_PREFIXES: &[&str] = &[
        "claude-opus",
        "claude-sonnet",
        "claude-fable",
        "claude-haiku",
        "gpt-5.6-terra",
        "gpt-5.5",
        "gpt-5.4",
        "gpt-5.3-codex",
        "google/gemini-3.5-flash",
        "google/gemini-3.1-flash-lite",
        "sakana/fugu-ultra",
        "meta/muse-spark-1.1",
    ];
    if PREMIUM_PREFIXES.iter().any(|p| full.starts_with(p)) {
        Category::Premium
    } else {
        Category::OpenSource
    }
}

struct Rule {
    open_only: bool,
    /// bare model ids blocked regardless of category (provider qualifier stripped)
    blocked: &'static [&'static str],
}

fn rule(plan_id: &str) -> Option<Rule> {
    match plan_id {
        "individual-go" => Some(Rule {
            open_only: true,
            blocked: &["meta/muse-spark-1.2", "xai/grok-4.6", "google/gemini-3.7-flash", "gpt-5.6-sol"],
        }),
        "individual-goat" => Some(Rule { open_only: true, blocked: &[] }),
        "individual-pro" | "individual-pro-v1" => Some(Rule {
            open_only: false,
            blocked: &["claude-fable-5", "claude-opus-5", "claude-opus-4-8", "claude-opus-4-7", "claude-opus-4-6", "claude-opus-4-5-20251101", "sakana/fugu-ultra"],
        }),
        _ => None, // unknown / max / ultra / provider / team: everything
    }
}

pub fn allowed(model_id: &str, plan_id: &str, unlocked: bool) -> bool {
    if unlocked || plan_id.is_empty() {
        return true;
    }
    let Some(r) = rule(plan_id) else { return true };
    let full = model_id.to_lowercase();
    if r.blocked.iter().any(|b| *b == full) {
        return false;
    }
    if r.open_only && category(model_id) == Category::Premium {
        return false;
    }
    true
}

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

    #[test]
    fn goat_blocks_premium_allows_open() {
        assert!(!allowed("claude-opus-5", "individual-goat", false));
        assert!(!allowed("claude-sonnet-5", "individual-goat", false));
        assert!(!allowed("sakana/fugu-ultra", "individual-goat", false));
        assert!(allowed("deepseek/deepseek-v4-pro", "individual-goat", false));
        assert!(allowed("z-ai/glm-5.3-flash", "individual-goat", false));
    }

    #[test]
    fn credits_unlock_everything() {
        assert!(allowed("claude-opus-5", "individual-goat", true));
        assert!(allowed("claude-opus-5", "individual-go", true));
    }

    #[test]
    fn pro_blocks_opus_fable_not_sonnet() {
        assert!(!allowed("claude-opus-5", "individual-pro", false));
        assert!(!allowed("claude-fable-5", "individual-pro", false));
        assert!(!allowed("sakana/fugu-ultra", "individual-pro", false));
        assert!(allowed("claude-sonnet-5", "individual-pro", false));
    }

    #[test]
    fn unknown_plan_allows_all() {
        assert!(allowed("claude-opus-5", "individual-max", false));
        assert!(allowed("claude-opus-5", "", false));
    }
}