csusage 0.1.1

Token usage reports for Claude Science from its local metadata database
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

//! csusage — token usage reports for Claude Science.
//!
//! Reads the local Claude Science metadata database (read-only) and prints
//! daily, monthly, or per-session token usage reports.

mod discovery;
mod loader;
mod report;

use clap::{Parser, Subcommand};

#[derive(Parser)]
#[command(
    name = "csusage",
    version,
    about = "Token usage reports for Claude Science",
    long_about = "Reads the local Claude Science metadata database (read-only) and prints daily, monthly, or per-session token usage reports."
)]
struct Cli {
    #[command(subcommand)]
    report: Option<Report>,

    /// Path to the metadata database (overrides CLAUDE_SCIENCE_DB and discovery)
    #[arg(long, global = true)]
    db: Option<String>,

    /// First date to include, YYYY-MM-DD
    #[arg(long, global = true)]
    since: Option<String>,

    /// Last date to include, YYYY-MM-DD
    #[arg(long, global = true)]
    until: Option<String>,

    /// Timezone for date bucketing (IANA name, default UTC)
    #[arg(long, global = true)]
    timezone: Option<String>,

    /// Emit machine-readable JSON
    #[arg(long, global = true)]
    json: bool,
}

#[derive(Subcommand, Clone, Copy)]
enum Report {
    /// Per-day usage
    Daily,
    /// Per-month usage
    Monthly,
    /// Per-session usage
    Session,
}

fn main() {
    let cli = Cli::parse();
    let report = cli.report.unwrap_or(Report::Daily);
    if let Err(error) = run(cli, report) {
        eprintln!("csusage: {error}");
        std::process::exit(1);
    }
}

fn run(cli: Cli, report: Report) -> Result<(), String> {
    let databases = discovery::database_paths(cli.db.as_deref())?;
    if databases.is_empty() {
        return Err(
            "no Claude Science database found; set CLAUDE_SCIENCE_DB or pass --db".to_string(),
        );
    }
    let mut frames = Vec::new();
    for path in &databases {
        match loader::read_frames(path) {
            Ok(mut frames_for_path) => frames.append(&mut frames_for_path),
            Err(error) => eprintln!("csusage: skipping {path:?}: {error}"),
        }
    }
    let timezone = report::parse_timezone(cli.timezone.as_deref())?;
    let rows = report::aggregate(
        &frames,
        report,
        cli.since.as_deref(),
        cli.until.as_deref(),
        &timezone,
    )?;
    if cli.json {
        report::print_json(&rows);
    } else {
        report::print_table(&rows, report);
    }
    Ok(())
}