cc_token_usage/analysis/
project.rs

1use std::collections::HashMap;
2
3use crate::data::models::SessionData;
4use crate::pricing::calculator::PricingCalculator;
5
6use super::{AggregatedTokens, ProjectResult, ProjectSummary};
7
8pub fn analyze_projects(
9    sessions: &[SessionData],
10    calc: &PricingCalculator,
11    top_n: usize,
12) -> ProjectResult {
13    let mut project_map: HashMap<String, ProjectAccumulator> = HashMap::new();
14
15    for session in sessions {
16        let project_name = session
17            .project
18            .clone()
19            .unwrap_or_else(|| "(unknown)".to_string());
20
21        let acc = project_map
22            .entry(project_name.clone())
23            .or_insert_with(|| ProjectAccumulator {
24                name: project_name,
25                session_count: 0,
26                total_turns: 0,
27                agent_turns: 0,
28                tokens: AggregatedTokens::default(),
29                cost: 0.0,
30                model_counts: HashMap::new(),
31            });
32
33        acc.session_count += 1;
34
35        for turn in session.all_responses() {
36            acc.tokens.add_usage(&turn.usage);
37            acc.total_turns += 1;
38            if turn.is_agent { acc.agent_turns += 1; }
39            *acc.model_counts.entry(turn.model.clone()).or_insert(0) += 1;
40            let cost = calc.calculate_turn_cost(&turn.model, &turn.usage);
41            acc.cost += cost.total;
42        }
43    }
44
45    let mut projects: Vec<ProjectSummary> = project_map
46        .into_values()
47        .map(|acc| {
48            let primary_model = acc.model_counts.into_iter()
49                .max_by_key(|(_, c)| *c)
50                .map(|(m, _)| m)
51                .unwrap_or_default();
52            ProjectSummary {
53                display_name: project_display_name(&acc.name),
54                name: acc.name,
55                session_count: acc.session_count,
56                total_turns: acc.total_turns,
57                agent_turns: acc.agent_turns,
58                tokens: acc.tokens,
59                cost: acc.cost,
60                primary_model,
61            }
62        })
63        .collect();
64
65    // Sort by cost descending
66    projects.sort_by(|a, b| b.cost.partial_cmp(&a.cost).unwrap_or(std::cmp::Ordering::Equal));
67
68    // Take top_n (0 means no limit)
69    if top_n > 0 {
70        projects.truncate(top_n);
71    }
72
73    ProjectResult { projects }
74}
75
76/// Convert internal project path to a human-readable display name.
77///
78/// `-Users-testuser-cc-web3` -> `~/cc/web3`
79pub fn project_display_name(name: &str) -> String {
80    // Pattern: -Users-<username>-<rest>
81    // Try to find the "-Users-" prefix and convert
82    if let Some(rest) = name.strip_prefix("-Users-") {
83        // Skip the username segment (everything up to the next '-')
84        if let Some(after_user) = rest.find('-') {
85            let path_part = &rest[after_user..];
86            // Replace '-' with '/'
87            let display = path_part.replace('-', "/");
88            return format!("~{display}");
89        }
90    }
91
92    name.to_string()
93}
94
95struct ProjectAccumulator {
96    name: String,
97    session_count: usize,
98    total_turns: usize,
99    agent_turns: usize,
100    tokens: AggregatedTokens,
101    cost: f64,
102    model_counts: HashMap<String, usize>,
103}
104
105#[cfg(test)]
106mod tests {
107    use super::*;
108
109    #[test]
110    fn test_project_display_name() {
111        assert_eq!(
112            project_display_name("-Users-testuser-cc-web3"),
113            "~/cc/web3"
114        );
115        assert_eq!(
116            project_display_name("-Users-alice-projects-my-app"),
117            "~/projects/my/app"
118        );
119        assert_eq!(project_display_name("simple-project"), "simple-project");
120        assert_eq!(project_display_name("-Users-bob"), "-Users-bob");
121    }
122}
cc_token_usage/analysis/project.rs

cc_token_usage/analysis/
project.rs
