vct-core 2.3.0

Vibe Coding Tracker core library - parse local AI coding assistant session data into CodeAnalysis results
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
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
109

//! Token-usage aggregation types shared between the usage calculator and the
//! display layer.

use crate::constants::FastHashMap;
use crate::models::Provider;
use serde::Serialize;

/// Token usage data aggregated by model (across all dates)
///
/// Structure: Model Name -> Usage Metrics
/// - Uses FastHashMap (ahash) for better performance than std HashMap
/// - Usage format varies by provider:
///   * Claude/Gemini: `{ input_tokens, output_tokens, cache_read_input_tokens, cache_creation_input_tokens }`
///   * Codex: `{ total_token_usage: { input_tokens, output_tokens } }`
pub type UsageResult = FastHashMap<String, serde_json::Value>;

/// Tracks the number of active days per AI provider
///
/// Used for calculating daily averages when data is aggregated by model only.
/// Day counts are derived from file modification dates during processing.
#[derive(Debug, Clone, Default, Serialize)]
pub struct ProviderActiveDays {
    /// Distinct active days observed for Claude Code.
    pub claude: usize,
    /// Distinct active days observed for Codex.
    pub codex: usize,
    /// Distinct active days observed for Copilot CLI.
    pub copilot: usize,
    /// Distinct active days observed for Gemini CLI.
    pub gemini: usize,
    /// Distinct active days observed for OpenCode.
    pub opencode: usize,
    /// Distinct active days observed for Cursor.
    pub cursor: usize,
    /// Distinct active days observed for Hermes.
    pub hermes: usize,
    /// Distinct active days observed for Grok CLI.
    pub grok: usize,
    /// Distinct active days across all providers combined.
    pub total: usize,
}

/// Per-provider usage data, keyed by source directory rather than by model name.
///
/// The top-level `UsageResult` in `UsageData` intentionally merges same-named
/// models across providers (so the per-model table shows one row for
/// `claude-sonnet-4-6` regardless of whether Claude Code, Copilot CLI, or
/// both invoked it). That merge loses the *source* information though, which
/// matters for the per-provider summary: once Copilot CLI stopped writing
/// the `copilot` sentinel and started recording real model names, the old
/// "classify each row by model-name prefix" logic mis-attributed every
/// Copilot session to Claude Code.
///
/// This struct keeps a separate `UsageResult` per provider so the display
/// layer can sum tokens and cost by source directory directly, with no
/// prefix heuristics involved. It is populated in `usage::aggregator` at
/// the same time the global merged map is built.
#[derive(Debug, Default, Clone, Serialize)]
pub struct PerProviderUsage {
    /// Per-model usage attributed to Claude Code.
    pub claude: UsageResult,
    /// Per-model usage attributed to Codex.
    pub codex: UsageResult,
    /// Per-model usage attributed to Copilot CLI.
    pub copilot: UsageResult,
    /// Per-model usage attributed to Gemini CLI.
    pub gemini: UsageResult,
    /// Per-model usage attributed to OpenCode.
    pub opencode: UsageResult,
    /// Per-model usage attributed to Cursor.
    pub cursor: UsageResult,
    /// Per-model usage attributed to Hermes.
    pub hermes: UsageResult,
    /// Per-model usage attributed to Grok CLI.
    pub grok: UsageResult,
}

impl PerProviderUsage {
    /// Returns the usage map for `provider`, or `None` for [`Provider::Unknown`].
    pub fn get(&self, provider: Provider) -> Option<&UsageResult> {
        match provider {
            Provider::ClaudeCode => Some(&self.claude),
            Provider::Codex => Some(&self.codex),
            Provider::Copilot => Some(&self.copilot),
            Provider::Gemini => Some(&self.gemini),
            Provider::OpenCode => Some(&self.opencode),
            Provider::Cursor => Some(&self.cursor),
            Provider::Hermes => Some(&self.hermes),
            Provider::Grok => Some(&self.grok),
            Provider::Unknown => None,
        }
    }

    /// Returns a mutable handle to the usage map for `provider`, or `None` for
    /// [`Provider::Unknown`].
    pub fn get_mut(&mut self, provider: Provider) -> Option<&mut UsageResult> {
        match provider {
            Provider::ClaudeCode => Some(&mut self.claude),
            Provider::Codex => Some(&mut self.codex),
            Provider::Copilot => Some(&mut self.copilot),
            Provider::Gemini => Some(&mut self.gemini),
            Provider::OpenCode => Some(&mut self.opencode),
            Provider::Cursor => Some(&mut self.cursor),
            Provider::Hermes => Some(&mut self.hermes),
            Provider::Grok => Some(&mut self.grok),
            Provider::Unknown => None,
        }
    }
}