pitboard 0.1.0

Park and restore your own Claude Code logins on one machine, and see what each one has left.
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

//! The last usage reading pitboard took for each account.
//!
//! A parked login's access token stops working hours after it was issued, and pitboard does
//! not renew it, so from then on the only honest thing to show is the last number actually
//! measured, and when. Nothing here is secret; it is a disposable cache, written
//! last-writer-wins without a lock.

use crate::usage::{Snapshot, Source};
use crate::{atomic, home};
use std::collections::HashMap;
use std::path::PathBuf;

fn path() -> PathBuf {
    home::dir().join("usage.json")
}

pub fn load() -> HashMap<String, Snapshot> {
    std::fs::read_to_string(path())
        .ok()
        .and_then(|raw| serde_json::from_str::<HashMap<String, Snapshot>>(&raw).ok())
        .unwrap_or_default()
        .into_iter()
        .map(|(uuid, mut snapshot)| {
            snapshot.source = Source::Remembered;
            (uuid, snapshot)
        })
        .collect()
}

/// Keep live readings for when their account can no longer be asked.
pub fn remember(readings: &[(String, Snapshot)]) {
    if readings.is_empty() {
        return;
    }
    let mut all = load();
    for (uuid, snapshot) in readings {
        let mut stored = snapshot.clone();
        stored.account_uuid = Some(uuid.clone());
        all.insert(uuid.clone(), stored);
    }
    if home::ensure().is_ok()
        && let Ok(body) = serde_json::to_string(&all)
    {
        let _ = atomic::write(&path(), body.as_bytes(), atomic::Perms::Secret);
    }
}