code2prompt 4.2.0

Command-line interface for code2prompt
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

//! Statistics view types and enums.
//!
//! This module contains the StatisticsView enum and related types
//! for managing different statistics views in the TUI.

/// Different views available in the Statistics tab
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum StatisticsView {
    Overview,   // General statistics and summary
    TokenMap,   // Token distribution by directory/file
    Extensions, // Token distribution by file extension
}

impl StatisticsView {
    pub fn next(&self) -> Self {
        match self {
            StatisticsView::Overview => StatisticsView::TokenMap,
            StatisticsView::TokenMap => StatisticsView::Extensions,
            StatisticsView::Extensions => StatisticsView::Overview,
        }
    }

    pub fn prev(&self) -> Self {
        match self {
            StatisticsView::Overview => StatisticsView::Extensions,
            StatisticsView::TokenMap => StatisticsView::Overview,
            StatisticsView::Extensions => StatisticsView::TokenMap,
        }
    }

    pub fn as_str(&self) -> &'static str {
        match self {
            StatisticsView::Overview => "Overview",
            StatisticsView::TokenMap => "Token Map",
            StatisticsView::Extensions => "Extensions",
        }
    }
}