opencode-stats 1.3.5

A terminal dashboard for OpenCode usage statistics inspired by the /stats command in Claude Code
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

use ratatui::buffer::Buffer;
use ratatui::layout::Rect;
use ratatui::style::Style;
use ratatui::text::{Line, Span};
use ratatui::widgets::{Paragraph, Widget};

use crate::analytics::heatmap_data::{HeatmapData, weekday_labels};
use crate::ui::theme::Theme;

pub struct HeatmapWidget<'a> {
    pub data: &'a HeatmapData,
    pub theme: &'a Theme,
}

const ROW_LABEL_WIDTH: usize = 4;

impl Widget for HeatmapWidget<'_> {
    fn render(self, area: Rect, buf: &mut Buffer) {
        let mut lines = Vec::new();

        lines.push(Line::styled(
            build_month_header(self.data),
            self.theme.muted_style(),
        ));

        for (row_index, weekday) in weekday_labels().iter().enumerate() {
            let name = weekday_label(row_index, weekday);
            let mut spans = vec![Span::styled(name, self.theme.muted_style())];
            for week in &self.data.weeks {
                let cell = &week[row_index];
                let (symbol, style) = if cell.is_future {
                    (' ', Style::default())
                } else {
                    match cell.intensity {
                        0 => ('·', Style::default().fg(self.theme.heat_0)),
                        1 => ('░', Style::default().fg(self.theme.heat_3)),
                        2 => ('▒', Style::default().fg(self.theme.heat_3)),
                        3 => ('▓', Style::default().fg(self.theme.heat_3)),
                        _ => ('█', Style::default().fg(self.theme.heat_3)),
                    }
                };
                spans.push(Span::styled(symbol.to_string(), style));
            }
            lines.push(Line::from(spans));
        }

        Paragraph::new(lines).render(area, buf);
    }
}

pub fn heatmap_legend_line(theme: &Theme) -> Line<'static> {
    Line::from(vec![
        Span::styled("    Less ", theme.muted_style()),
        Span::styled("·", Style::default().fg(theme.heat_0)),
        Span::raw(" "),
        Span::styled("░", Style::default().fg(theme.heat_3)),
        Span::raw(" "),
        Span::styled("▒", Style::default().fg(theme.heat_3)),
        Span::raw(" "),
        Span::styled("▓", Style::default().fg(theme.heat_3)),
        Span::raw(" "),
        Span::styled("█", Style::default().fg(theme.heat_3)),
        Span::styled(" More", theme.muted_style()),
    ])
}

fn build_month_header(data: &HeatmapData) -> String {
    let mut header = vec![' '; ROW_LABEL_WIDTH + data.weeks.len()];
    let mut next_free_column = ROW_LABEL_WIDTH;

    for (week_index, label) in &data.month_labels {
        let start = ROW_LABEL_WIDTH + *week_index;
        if start < next_free_column || start >= header.len() {
            continue;
        }

        for (offset, ch) in label.chars().enumerate() {
            let column = start + offset;
            if column >= header.len() {
                break;
            }
            header[column] = ch;
        }

        next_free_column = start + label.chars().count() + 1;
    }

    header.into_iter().collect()
}

fn weekday_label(row_index: usize, weekday: &chrono::Weekday) -> String {
    match row_index {
        0 | 2 | 4 | 6 => format!(
            "{:>3} ",
            format!("{:?}", weekday).chars().take(3).collect::<String>()
        ),
        _ => " ".repeat(ROW_LABEL_WIDTH),
    }
}