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

//! Output widget for displaying generated prompt with scrolling capability.

use crate::model::Model;
use ratatui::{
    prelude::*,
    widgets::{Block, Borders, Paragraph, Wrap},
};

/// State for the output widget - no longer needed, read directly from Model
pub type OutputState = ();

/// Widget for output display with scrolling
pub struct OutputWidget<'a> {
    pub model: &'a Model,
}

impl<'a> OutputWidget<'a> {
    pub fn new(model: &'a Model) -> Self {
        Self { model }
    }
}

impl<'a> StatefulWidget for OutputWidget<'a> {
    type State = OutputState;

    fn render(self, area: Rect, buf: &mut Buffer, _state: &mut Self::State) {
        let layout = Layout::default()
            .direction(Direction::Vertical)
            .constraints([
                Constraint::Length(3), // Info bar
                Constraint::Min(0),    // Prompt content
                Constraint::Length(3), // Controls
            ])
            .split(area);

        // Simplified status bar - focus only on prompt availability
        let info_text = if self.model.prompt_output.analysis_in_progress {
            "Generating prompt...".to_string()
        } else if let Some(error) = &self.model.prompt_output.analysis_error {
            format!("Generation failed: {}", error)
        } else if self.model.prompt_output.generated_prompt.is_some() {
            "✓ Prompt ready! Copy (C) or Save (S)".to_string()
        } else {
            "Press Enter to generate prompt from selected files".to_string()
        };

        let info_widget = Paragraph::new(info_text)
            .block(
                Block::default()
                    .borders(Borders::ALL)
                    .title("Generated Prompt"),
            )
            .style(if self.model.prompt_output.analysis_error.is_some() {
                Style::default().fg(Color::Red)
            } else if self.model.prompt_output.analysis_in_progress {
                Style::default().fg(Color::Yellow)
            } else {
                Style::default().fg(Color::Green)
            });
        Widget::render(info_widget, layout[0], buf);

        // Prompt content
        let content = if self.model.prompt_output.analysis_in_progress {
            "Generating prompt...".to_string()
        } else if let Some(prompt) = &self.model.prompt_output.generated_prompt {
            prompt.clone()
        } else {
            "Press <Enter> to run analysis and generate prompt.\n\nSelected files will be processed according to your settings.".to_string()
        };

        // Compute viewport-aware scroll
        let content_height = layout[1].height.saturating_sub(2).max(1) as usize; // borders
        let (display_scroll, scroll_info) =
            if let Some(prompt) = &self.model.prompt_output.generated_prompt {
                let total_lines = prompt.lines().count();
                let max_scroll = total_lines.saturating_sub(content_height);
                let ds = self
                    .model
                    .prompt_output
                    .output_scroll
                    .min(max_scroll as u16);
                let current_line = ds as usize + 1;
                (
                    ds,
                    format!("Generated Prompt (Line {}/{})", current_line, total_lines),
                )
            } else {
                (
                    self.model.prompt_output.output_scroll,
                    "Generated Prompt".to_string(),
                )
            };

        let prompt_widget = Paragraph::new(content)
            .block(Block::default().borders(Borders::ALL).title(scroll_info))
            .wrap(Wrap { trim: false })
            .scroll((display_scroll, 0));
        Widget::render(prompt_widget, layout[1], buf);

        // Controls
        let controls_text = if self.model.prompt_output.generated_prompt.is_some() {
            "↑↓/PgUp/PgDn: Scroll | C: Copy | S: Save | Enter: Re-run"
        } else {
            "Enter: Run Analysis"
        };

        let controls_widget = Paragraph::new(controls_text)
            .block(Block::default().borders(Borders::ALL).title("Controls"))
            .style(Style::default().fg(Color::Gray));
        Widget::render(controls_widget, layout[2], buf);
    }
}