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            });
31
32        acc.session_count += 1;
33
34        for turn in session.all_responses() {
35            acc.tokens.add_usage(&turn.usage);
36            acc.total_turns += 1;
37            if turn.is_agent { acc.agent_turns += 1; }
38            let cost = calc.calculate_turn_cost(&turn.model, &turn.usage);
39            acc.cost += cost.total;
40        }
41    }
42
43    let mut projects: Vec<ProjectSummary> = project_map
44        .into_values()
45        .map(|acc| ProjectSummary {
46            display_name: project_display_name(&acc.name),
47            name: acc.name,
48            session_count: acc.session_count,
49            total_turns: acc.total_turns,
50            agent_turns: acc.agent_turns,
51            tokens: acc.tokens,
52            cost: acc.cost,
53        })
54        .collect();
55
56    // Sort by cost descending
57    projects.sort_by(|a, b| b.cost.partial_cmp(&a.cost).unwrap_or(std::cmp::Ordering::Equal));
58
59    // Take top_n (0 means no limit)
60    if top_n > 0 {
61        projects.truncate(top_n);
62    }
63
64    ProjectResult { projects }
65}
66
67/// Convert internal project path to a human-readable display name.
68///
69/// `-Users-testuser-cc-web3` -> `~/cc/web3`
70pub fn project_display_name(name: &str) -> String {
71    // Pattern: -Users-<username>-<rest>
72    // Try to find the "-Users-" prefix and convert
73    if let Some(rest) = name.strip_prefix("-Users-") {
74        // Skip the username segment (everything up to the next '-')
75        if let Some(after_user) = rest.find('-') {
76            let path_part = &rest[after_user..];
77            // Replace '-' with '/'
78            let display = path_part.replace('-', "/");
79            return format!("~{display}");
80        }
81    }
82
83    name.to_string()
84}
85
86struct ProjectAccumulator {
87    name: String,
88    session_count: usize,
89    total_turns: usize,
90    agent_turns: usize,
91    tokens: AggregatedTokens,
92    cost: f64,
93}
94
95#[cfg(test)]
96mod tests {
97    use super::*;
98
99    #[test]
100    fn test_project_display_name() {
101        assert_eq!(
102            project_display_name("-Users-testuser-cc-web3"),
103            "~/cc/web3"
104        );
105        assert_eq!(
106            project_display_name("-Users-alice-projects-my-app"),
107            "~/projects/my/app"
108        );
109        assert_eq!(project_display_name("simple-project"), "simple-project");
110        assert_eq!(project_display_name("-Users-bob"), "-Users-bob");
111    }
112}
cc_token_usage/analysis/project.rs

cc_token_usage/analysis/
project.rs
