clauth 0.5.3

Simple Claude Code account switcher and usage monitor
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

//! DeepSeek provider — balance stats from `GET /user/balance`.
//!
//! Wire shape per <https://api-docs.deepseek.com/api/get-user-balance>.

use serde::Deserialize;

use super::{StatRow, StatRowKind, ThirdPartyError, ThirdPartyStats, url_matches_host};

pub(super) const DISPLAY_NAME: &str = "DeepSeek";

const BALANCE_URL: &str = "https://api.deepseek.com/user/balance";

pub(super) fn matches_base_url(url: &str) -> bool {
    url_matches_host(url, "https://api.deepseek.com")
}

pub(super) fn fetch(api_key: &str) -> Result<ThirdPartyStats, ThirdPartyError> {
    let text = super::get_json(BALANCE_URL, api_key)?;
    let raw: DeepSeekResponse = serde_json::from_str(&text).map_err(|_| ThirdPartyError::Parse)?;
    Ok(stats(&raw))
}

/// Pure response → display-rows mapping, separated from HTTP for testability.
fn stats(raw: &DeepSeekResponse) -> ThirdPartyStats {
    if !raw.is_available {
        return ThirdPartyStats::unavailable("balance unavailable");
    }

    let mut rows: Vec<StatRow> = Vec::new();
    for info in &raw.balance_infos {
        rows.push(StatRow {
            label: format!("{} balance", info.currency),
            value: String::new(),
            kind: StatRowKind::Heading,
        });
        rows.push(StatRow {
            label: "total".to_string(),
            value: format!("{} {}", info.total_balance, info.currency),
            kind: StatRowKind::Body,
        });
        rows.push(StatRow {
            label: "granted".to_string(),
            value: format!("{} {}", info.granted_balance, info.currency),
            kind: StatRowKind::Body,
        });
        rows.push(StatRow {
            label: "topped up".to_string(),
            value: format!("{} {}", info.topped_up_balance, info.currency),
            kind: StatRowKind::Body,
        });
    }
    ThirdPartyStats::from_rows(rows)
}

// ── Wire types ──────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Deserialize)]
struct DeepSeekResponse {
    is_available: bool,
    #[serde(default)]
    balance_infos: Vec<DeepSeekBalance>,
}

#[derive(Debug, Clone, Deserialize)]
struct DeepSeekBalance {
    currency: String,
    total_balance: String,
    granted_balance: String,
    topped_up_balance: String,
}

#[cfg(test)]
#[path = "../../tests/inline/providers_deepseek.rs"]
mod tests;