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
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

//! Template Picker sub-widget.
//!
//! This widget provides template selection with separate default and custom lists.

use crate::model::template::{ActiveList, PickerState};
use ratatui::{
    prelude::*,
    widgets::{Block, Borders, List, ListItem},
};

/// Template Picker sub-widget
pub struct TemplatePickerWidget;

impl TemplatePickerWidget {
    pub fn new() -> Self {
        Self
    }

    /// Render the template picker as a single unified list with groups
    pub fn render(&self, area: Rect, buf: &mut Buffer, state: &PickerState, is_focused: bool) {
        let border_style = if is_focused {
            Style::default().fg(Color::Yellow)
        } else {
            Style::default().fg(Color::Gray)
        };

        // Create unified list with section headers
        let mut items = Vec::new();
        let mut item_index = 0;
        let global_cursor = state.get_global_cursor_position();

        // Default Templates Section
        if !state.default_templates.is_empty() {
            // Section header
            items.push(ListItem::new(Line::from(vec![
                Span::styled("📄 ", Style::default().fg(Color::White)),
                Span::styled(
                    "Default Templates",
                    Style::default()
                        .fg(Color::Cyan)
                        .add_modifier(Modifier::BOLD),
                ),
            ])));
            item_index += 1;

            // Default template items
            for template in state.default_templates.iter() {
                let is_selected = global_cursor == item_index;
                let style = if is_selected && is_focused {
                    Style::default()
                        .fg(Color::Yellow)
                        .add_modifier(Modifier::BOLD)
                } else if is_selected {
                    Style::default()
                        .fg(Color::Cyan)
                        .add_modifier(Modifier::BOLD)
                } else {
                    Style::default().fg(Color::White)
                };

                let prefix = if is_selected { "► " } else { "  " };
                items.push(ListItem::new(format!("{}📄 {}", prefix, template.name)).style(style));
                item_index += 1;
            }
        }

        // Custom Templates Section
        if !state.custom_templates.is_empty() {
            // Add separator if we have default templates
            if !state.default_templates.is_empty() {
                items.push(ListItem::new(""));
                item_index += 1;
            }

            // Section header
            items.push(ListItem::new(Line::from(vec![
                Span::styled("📝 ", Style::default().fg(Color::White)),
                Span::styled(
                    "Custom Templates",
                    Style::default()
                        .fg(Color::Green)
                        .add_modifier(Modifier::BOLD),
                ),
            ])));
            item_index += 1;

            // Custom template items
            for template in state.custom_templates.iter() {
                let is_selected = global_cursor == item_index;
                let style = if is_selected && is_focused {
                    Style::default()
                        .fg(Color::Yellow)
                        .add_modifier(Modifier::BOLD)
                } else if is_selected {
                    Style::default()
                        .fg(Color::Green)
                        .add_modifier(Modifier::BOLD)
                } else {
                    Style::default().fg(Color::White)
                };

                let prefix = if is_selected { "► " } else { "  " };
                items.push(ListItem::new(format!("{}📝 {}", prefix, template.name)).style(style));
                item_index += 1;
            }
        }

        // Create title with focus indicators
        let title_spans = vec![
            Span::styled("Template ", Style::default().fg(Color::White)),
            Span::styled(
                "p",
                Style::default().fg(Color::Red).add_modifier(Modifier::BOLD),
            ),
            Span::styled("icker", Style::default().fg(Color::White)),
        ];

        let list = List::new(items).block(
            Block::default()
                .borders(Borders::ALL)
                .title(Line::from(title_spans))
                .border_style(border_style),
        );

        Widget::render(list, area, buf);
    }

    /// Get help text for the picker
    pub fn get_help_text(is_focused: bool, _active_list: ActiveList) -> &'static str {
        if is_focused {
            "↑↓: Navigate | l/Space: Load | r: Refresh"
        } else {
            "Press 'p' to focus picker"
        }
    }
}

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