tuillem-tui 0.1.2

Ratatui TUI layer for tuillem
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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150

use ratatui::{
    Frame,
    layout::Rect,
    style::{Modifier, Style},
    text::{Line, Span},
    widgets::{Block, Borders, Clear, List, ListItem, ListState},
};

use crate::theme::Theme;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ControlAction {
    SwitchModel,
    SwitchProvider,
    NewConversation,
    RegenerateResponse,
    SaveTranscript,
    OpenInEditor,
    ToggleThinking,
}

impl ControlAction {
    pub const ALL: &'static [ControlAction] = &[
        ControlAction::SwitchModel,
        ControlAction::SwitchProvider,
        ControlAction::NewConversation,
        ControlAction::RegenerateResponse,
        ControlAction::SaveTranscript,
        ControlAction::OpenInEditor,
        ControlAction::ToggleThinking,
    ];

    pub fn label(&self) -> &'static str {
        match self {
            ControlAction::SwitchModel => "Switch Model",
            ControlAction::SwitchProvider => "Switch Provider",
            ControlAction::NewConversation => "New Conversation",
            ControlAction::RegenerateResponse => "Regenerate Response",
            ControlAction::SaveTranscript => "Save Transcript",
            ControlAction::OpenInEditor => "Open in Editor",
            ControlAction::ToggleThinking => "Toggle Thinking",
        }
    }

    pub fn hint(&self) -> &'static str {
        match self {
            ControlAction::SwitchModel => "Ctrl+K > m",
            ControlAction::SwitchProvider => "Ctrl+K > p",
            ControlAction::NewConversation => "Ctrl+N",
            ControlAction::RegenerateResponse => "Ctrl+R",
            ControlAction::SaveTranscript => "",
            ControlAction::OpenInEditor => "Ctrl+E",
            ControlAction::ToggleThinking => "",
        }
    }
}

#[derive(Debug, Clone)]
pub struct ControlPanel {
    pub selected: usize,
}

impl Default for ControlPanel {
    fn default() -> Self {
        Self::new()
    }
}

impl ControlPanel {
    pub fn new() -> Self {
        Self { selected: 0 }
    }

    pub fn move_down(&mut self) {
        if self.selected + 1 < ControlAction::ALL.len() {
            self.selected += 1;
        }
    }

    pub fn move_up(&mut self) {
        self.selected = self.selected.saturating_sub(1);
    }

    pub fn selected_action(&self) -> ControlAction {
        ControlAction::ALL[self.selected]
    }

    pub fn render(&self, frame: &mut Frame, area: Rect, theme: &Theme) {
        let popup_width = 45u16.min(area.width.saturating_sub(10));
        let popup_height = (ControlAction::ALL.len() as u16 + 2).min(area.height.saturating_sub(6));
        let x = (area.width.saturating_sub(popup_width)) / 2;
        let y = (area.height.saturating_sub(popup_height)) / 2;
        let popup_area = Rect::new(x, y, popup_width, popup_height);

        frame.render_widget(Clear, popup_area);

        let items: Vec<ListItem> = ControlAction::ALL
            .iter()
            .enumerate()
            .map(|(i, action)| {
                let selected = i == self.selected;
                let style = if selected {
                    Style::default()
                        .fg(theme.accent)
                        .add_modifier(Modifier::BOLD)
                } else {
                    Style::default().fg(theme.fg)
                };
                let marker = if selected { "▸ " } else { "  " };
                let hint = action.hint();
                let label = action.label();
                let inner_width = popup_width.saturating_sub(2) as usize;
                let pad = inner_width.saturating_sub(marker.len() + label.len() + hint.len());
                let line = if hint.is_empty() {
                    Line::from(Span::styled(format!("{}{}", marker, label), style))
                } else {
                    Line::from(vec![
                        Span::styled(format!("{}{}", marker, label), style),
                        Span::styled(
                            format!("{:>width$}", hint, width = pad + hint.len()),
                            Style::default().fg(theme.thinking_fg),
                        ),
                    ])
                };
                ListItem::new(line)
            })
            .collect();

        let block = Block::default()
            .borders(Borders::ALL)
            .border_style(Style::default().fg(theme.accent))
            .title(Line::from(Span::styled(
                " Commands ",
                Style::default()
                    .fg(theme.accent)
                    .add_modifier(Modifier::BOLD),
            )))
            .title_bottom(Line::from(Span::styled(
                " j/k:select  Enter:run  Esc:cancel ",
                Style::default().fg(theme.thinking_fg),
            )))
            .style(Style::default().bg(theme.bg));

        let mut list_state = ListState::default();
        list_state.select(Some(self.selected));

        let list = List::new(items).block(block);
        frame.render_stateful_widget(list, popup_area, &mut list_state);
    }
}