tokmesh-cli 0.1.1

CLI and TUI for tokmesh — AI token usage analytics
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125

mod agents;
mod bar_chart;
mod daily;
pub mod dialog;
mod footer;
mod header;
mod hourly;
mod hourly_profile;
mod minutely;
mod models;
mod monthly;
mod overview;
mod sessions;
pub mod spinner;
mod stats;
mod usage;
pub(crate) mod widgets;

use ratatui::prelude::*;
use ratatui::widgets::{Block, Borders, Paragraph};

use crate::tui::app::{App, Tab};

pub fn render(frame: &mut Frame, app: &mut App) {
    let area = frame.area();
    if area.width == 0 || area.height == 0 {
        return;
    }

    app.clear_click_areas();
    app.handle_resize(area.width, area.height);

    let chunks = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Length(3),
            Constraint::Min(0),
            Constraint::Length(5),
        ])
        .split(area);

    header::render(frame, app, chunks[0]);

    if app.data.loading && !app.background_loading {
        render_loading(frame, app, chunks[1]);
    } else if let Some(ref error) = app.data.error {
        render_error(frame, app, chunks[1], error);
    } else {
        match app.current_tab {
            Tab::Overview => overview::render(frame, app, chunks[1]),
            Tab::Models => models::render(frame, app, chunks[1]),
            Tab::Agents => agents::render(frame, app, chunks[1]),
            Tab::Daily => daily::render(frame, app, chunks[1]),
            Tab::Hourly => hourly::render(frame, app, chunks[1]),
            Tab::Minutely => minutely::render(frame, app, chunks[1]),
            Tab::Monthly => monthly::render(frame, app, chunks[1]),
            Tab::Sessions => sessions::render(frame, app, chunks[1]),
            Tab::Stats => stats::render(frame, app, chunks[1]),
            Tab::Usage => usage::render(frame, app, chunks[1]),
        }
    }

    footer::render(frame, app, chunks[2]);

    if app.dialog_stack.is_active() {
        app.dialog_stack.render(frame, area);
    }
}

fn render_loading(frame: &mut Frame, app: &App, area: Rect) {
    let block = Block::default()
        .borders(Borders::ALL)
        .border_style(Style::default().fg(app.theme.border))
        .style(Style::default().bg(app.theme.background));

    let inner = block.inner(area);
    frame.render_widget(block, area);

    let center = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Percentage(40),
            Constraint::Length(3),
            Constraint::Percentage(40),
        ])
        .split(inner)[1];

    let mut spans = spinner::get_scanner_spans(app.spinner_frame, &app.theme);
    spans.push(Span::raw("  "));
    spans.push(Span::styled(
        spinner::get_phase_message("parsing-sources"),
        Style::default().fg(app.theme.muted),
    ));

    let line = Line::from(spans);
    let paragraph = Paragraph::new(line).alignment(Alignment::Center);

    frame.render_widget(paragraph, center);
}

fn render_error(frame: &mut Frame, app: &App, area: Rect, error: &str) {
    let block = Block::default()
        .borders(Borders::ALL)
        .border_style(Style::default().fg(app.theme.border))
        .style(Style::default().bg(app.theme.background));

    let inner = block.inner(area);
    frame.render_widget(block, area);

    let center = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Percentage(40),
            Constraint::Length(3),
            Constraint::Percentage(40),
        ])
        .split(inner)[1];

    let text = format!("Error: {}", error);
    let paragraph = Paragraph::new(text)
        .style(Style::default().fg(Color::Red))
        .alignment(Alignment::Center);

    frame.render_widget(paragraph, center);
}