clankerdiff_ratatui/
input.rs

1use crate::{
2    DiffReviewCommand, DiffReviewState, FocusPane, InputOutcome, InteractionPhase, KeyCode,
3    KeyEvent, MouseEvent, MouseEventKind, ReviewCommand, ReviewInput,
4    interaction::{self, ReviewWidget},
5    state::{RepositoryOperationStatus, RepositoryPrompt},
6};
7use clankerdiff_core::{CommentDraft, DiffReviewEvent, RepositoryAction};
8#[cfg(feature = "crossterm-backend")]
9use crossterm::event::Event;
10use ratatui::layout::Position;
11use std::convert::Infallible;
12
13#[cfg(feature = "crossterm-backend")]
14#[must_use]
15pub fn handle_crossterm_event(
16    state: &mut DiffReviewState,
17    event: Event,
18) -> InputOutcome<DiffReviewEvent> {
19    let Ok(outcome) = crate::crossterm_adapter::handle_event(state, event);
20    outcome
21}
22
23impl DiffReviewState {
24    #[must_use]
25    pub fn interaction_phase(&self) -> InteractionPhase {
26        if self.theme_picker.is_some() {
27            InteractionPhase::ThemePicker
28        } else if self.repository_prompt.is_some() {
29            InteractionPhase::RepositoryPrompt
30        } else if self.help {
31            InteractionPhase::Help
32        } else if self.session.draft().is_some() {
33            InteractionPhase::Draft
34        } else {
35            InteractionPhase::Browse
36        }
37    }
38
39    #[must_use]
40    pub fn handle_input(&mut self, input: ReviewInput) -> InputOutcome<DiffReviewEvent> {
41        let Ok(outcome) = interaction::handle_input(self, input);
42        self.install_deferred();
43        outcome
44    }
45}
46
47impl ReviewWidget for DiffReviewState {
48    type Event = DiffReviewEvent;
49    type Error = Infallible;
50    type Draft = CommentDraft;
51
52    fn phase(&self) -> InteractionPhase {
53        self.interaction_phase()
54    }
55    fn handle_review_command(
56        &mut self,
57        command: ReviewCommand,
58    ) -> Result<InputOutcome<DiffReviewEvent>, Infallible> {
59        Ok(self.handle_command(command))
60    }
61    fn contains(&self, position: Position) -> bool {
62        self.hit_layout.drawer.contains(position) || self.hit_layout.patch.contains(position)
63    }
64    fn mark_dirty(&mut self) {
65        Self::mark_dirty(self);
66    }
67    fn draft_mut(&mut self) -> Option<&mut CommentDraft> {
68        self.session.draft_mut()
69    }
70    fn draft_changed(&mut self) {
71        self.request_follow();
72    }
73    fn paste_prompt(&mut self, text: &str) {
74        if let Some(RepositoryPrompt::Commit { message }) = &mut self.repository_prompt {
75            message.push_str(text);
76        }
77    }
78    fn handle_browse_key(
79        &mut self,
80        key: KeyEvent,
81    ) -> Result<InputOutcome<DiffReviewEvent>, Infallible> {
82        Ok(self
83            .command_for_key(key)
84            .map_or(InputOutcome::Ignored, |command| {
85                self.handle_command(command)
86            }))
87    }
88    fn handle_mouse(&mut self, mouse: MouseEvent) -> InputOutcome<DiffReviewEvent> {
89        self.handle_mouse(mouse)
90    }
91    fn handle_prompt_key(&mut self, key: KeyEvent) -> InputOutcome<DiffReviewEvent> {
92        self.handle_repository_prompt_key(key)
93    }
94}
95
96impl DiffReviewState {
97    fn handle_repository_prompt_key(&mut self, key: KeyEvent) -> InputOutcome<DiffReviewEvent> {
98        if key.code == KeyCode::Esc {
99            return self.handle_command(ReviewCommand::Cancel);
100        }
101        if !interaction::is_plain_key(key) {
102            return InputOutcome::Consumed;
103        }
104        let Some(prompt) = self.repository_prompt.as_mut() else {
105            return InputOutcome::Consumed;
106        };
107        let action = match prompt {
108            RepositoryPrompt::Commit { message } => match key.code {
109                KeyCode::Enter => {
110                    let message = message.trim().to_owned();
111                    if message.is_empty() {
112                        self.set_repository_error("Commit message cannot be empty");
113                        None
114                    } else {
115                        Some(RepositoryAction::Commit { message })
116                    }
117                }
118                KeyCode::Backspace => {
119                    message.pop();
120                    None
121                }
122                KeyCode::Char(character) => {
123                    message.push(character);
124                    None
125                }
126                _ => None,
127            },
128            RepositoryPrompt::Discard { path, status } => match key.code {
129                KeyCode::Char('y' | 'Y') => Some(RepositoryAction::Discard {
130                    path: path.clone(),
131                    status: *status,
132                }),
133                KeyCode::Char('n' | 'N') => {
134                    self.repository_prompt = None;
135                    None
136                }
137                _ => None,
138            },
139        };
140        if let Some(action) = action {
141            self.repository_prompt = None;
142            return self.handle_command(DiffReviewCommand::RepositoryAction(action));
143        }
144        InputOutcome::Consumed
145    }
146
147    fn handle_mouse(&mut self, mouse: MouseEvent) -> InputOutcome<DiffReviewEvent> {
148        let position = Position::new(mouse.column, mouse.row);
149        match mouse.kind {
150            MouseEventKind::ScrollUp => self.scroll_at(position, -1),
151            MouseEventKind::ScrollDown => self.scroll_at(position, 1),
152            MouseEventKind::Down(_) => {
153                if self.hit_layout.drawer.contains(position) {
154                    self.focus = FocusPane::Files;
155                    let relative = usize::from(mouse.row.saturating_sub(self.hit_layout.drawer.y));
156                    self.select_drawer_entry(self.drawer_scroll.saturating_add(relative));
157                    if self.hit_layout.drawer_stage_column == Some(mouse.column)
158                        && !matches!(self.repository_status, RepositoryOperationStatus::Pending)
159                    {
160                        return self.handle_command(DiffReviewCommand::ToggleStage);
161                    }
162                } else if self.hit_layout.patch.contains(position) {
163                    self.focus = FocusPane::Diff;
164                    self.select_clicked_row(mouse.row);
165                }
166            }
167            _ => {}
168        }
169        InputOutcome::Consumed
170    }
171
172    fn scroll_at(&mut self, position: Position, direction: isize) {
173        let pane = if self.hit_layout.patch.contains(position) {
174            FocusPane::Diff
175        } else if self.hit_layout.drawer.contains(position) {
176            FocusPane::Files
177        } else {
178            self.focus
179        };
180        match pane {
181            FocusPane::Diff => self.move_row(direction),
182            FocusPane::Files => self.scroll_drawer(direction),
183        }
184    }
185}
clankerdiff_ratatui/input.rs

clankerdiff_ratatui/
input.rs
