implements scroll in chat window

Author: dingfeli

Cargo.lock

@@ -28,6 +28,7 @@ eyre.workspace = true
 tokio-util.workspace = true
 futures.workspace = true
 rustyline.workspace = true
+strip-ansi-escapes.workspace = true
 ratatui = "0.29.0"
 
 [target.'cfg(unix)'.dependencies]

crates/chat-cli-ui/Cargo.toml

@@ -16,4 +16,17 @@ pub enum Action {
     Error(String),
     Help,
     Input(InputEvent),
+    Scroll(Scroll),
+}
+
+#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
+pub enum Scroll {
+    Up(ScrollDistance),
+    Down(ScrollDistance),
+}
+
+#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
+pub enum ScrollDistance {
+    Message,
+    Line(u16),
 }

crates/chat-cli-ui/src/ui/action.rs

@@ -156,6 +156,7 @@ impl App {
                         Action::ClearScreen => {},
                         Action::Error(_) => {},
                         Action::Help => {},
+                        Action::Scroll(_) => {},
                         Action::Input(input_event) => {
                             if let Err(e) = self.view_end.sender.send(input_event.clone()).await {
                                 error!("Error sending input event to control end: {:?}", e);

crates/chat-cli-ui/src/ui/components/app.rs

@@ -18,7 +18,11 @@ use crate::protocol::{
     Event as SessionEvent,
     InputEvent,
 };
-use crate::ui::action::Action;
+use crate::ui::action::{
+    Action,
+    Scroll,
+    ScrollDistance,
+};
 
 #[derive(Debug, Clone)]
 struct Message {
@@ -43,6 +47,7 @@ pub struct ChatWindow {
     messages: Vec<Message>,
     current_message: Option<Message>,
     scroll_offset: u16,
+    nearest_message_idx: usize,
     visible: bool,
     // TODO: update on resize
     pub dimension: (u16, u16),
@@ -108,7 +113,7 @@ impl Component for ChatWindow {
                     Style::default().fg(Color::DarkGray),
                 ),
                 Span::styled(format!("{}: ", prefix), style.bold()),
-                Span::raw(&message.content),
+                Span::raw(strip_ansi_escapes::strip_str(&message.content)),
             ]));
             lines.push(Line::from("")); // Empty line between messages
         }
@@ -196,20 +201,63 @@ impl Component for ChatWindow {
     }
 
     fn update(&mut self, action: Action) -> eyre::Result<Option<Action>> {
-        if let Action::Input(input_event) = action {
-            match input_event {
-                InputEvent::Text(text) => {
-                    self.add_message(MessageRole::User, text);
-                    self.scroll_offset = self.messages.last().as_ref().map(|msg| msg.offset).unwrap_or_default();
-                    return Ok(Some(Action::Render));
-                },
-                InputEvent::Interrupt => {
-                    // Handle interrupt - could be used to cancel current streaming message
-                    if self.current_message.is_some() {
-                        self.finalize_current_message();
-                        return Ok(Some(Action::Render));
+        if self.visible {
+            match action {
+                Action::Input(input_event) => {
+                    match input_event {
+                        InputEvent::Text(text) => {
+                            self.add_message(MessageRole::User, text);
+                            self.scroll_offset =
+                                self.messages.last().as_ref().map(|msg| msg.offset).unwrap_or_default();
+                            self.nearest_message_idx = self.messages.len();
+                            return Ok(Some(Action::Render));
+                        },
+                        InputEvent::Interrupt => {
+                            // Handle interrupt - could be used to cancel current streaming message
+                            if self.current_message.is_some() {
+                                self.finalize_current_message();
+                                return Ok(Some(Action::Render));
+                            }
+                        },
                     }
                 },
+                Action::Scroll(scroll) => match scroll {
+                    Scroll::Up(scroll_distance) => match scroll_distance {
+                        ScrollDistance::Message => {
+                            if self.nearest_message_idx == 0 {
+                                return Ok(None);
+                            }
+                            self.nearest_message_idx -= 1;
+                            self.scroll_offset = self
+                                .messages
+                                .get(self.nearest_message_idx)
+                                .as_ref()
+                                .map(|msg| msg.offset)
+                                .unwrap_or_default();
+
+                            return Ok(Some(Action::Render));
+                        },
+                        ScrollDistance::Line(_) => {},
+                    },
+                    Scroll::Down(scroll_distance) => match scroll_distance {
+                        ScrollDistance::Message => {
+                            if self.messages.is_empty() || self.nearest_message_idx == self.messages.len() - 1 {
+                                return Ok(None);
+                            }
+                            self.nearest_message_idx += 1;
+                            self.scroll_offset = self
+                                .messages
+                                .get(self.nearest_message_idx)
+                                .as_ref()
+                                .map(|msg| msg.offset)
+                                .unwrap_or_default();
+
+                            return Ok(Some(Action::Render));
+                        },
+                        ScrollDistance::Line(_) => {},
+                    },
+                },
+                _ => {},
             }
         }
 

crates/chat-cli-ui/src/ui/components/chat_window.rs

@@ -1,13 +1,23 @@
 use std::collections::HashMap;
 
-use crossterm::event::KeyEvent;
+use crossterm::event::{
+    KeyCode,
+    KeyEvent,
+    KeyEventKind,
+    KeyEventState,
+    KeyModifiers,
+};
 use serde::{
     Deserialize,
     Deserializer,
     Serialize,
 };
 
-use super::action::Action;
+use super::action::{
+    Action,
+    Scroll,
+    ScrollDistance,
+};
 
 #[derive(Default, Debug, Copy, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
 pub enum Mode {
@@ -21,9 +31,38 @@ pub struct Config {
     pub keybindings: KeyBindings,
 }
 
-#[derive(Clone, Debug, Default)]
+#[derive(Clone, Debug)]
 pub struct KeyBindings(pub HashMap<Mode, HashMap<Vec<KeyEvent>, Action>>);
 
+impl Default for KeyBindings {
+    fn default() -> Self {
+        let mut mapping = HashMap::<Vec<KeyEvent>, Action>::new();
+        mapping.insert(
+            vec![KeyEvent {
+                code: KeyCode::Char('p'),
+                modifiers: KeyModifiers::CONTROL,
+                kind: KeyEventKind::Press,
+                state: KeyEventState::NONE,
+            }],
+            Action::Scroll(Scroll::Up(ScrollDistance::Message)),
+        );
+        mapping.insert(
+            vec![KeyEvent {
+                code: KeyCode::Char('n'),
+                modifiers: KeyModifiers::CONTROL,
+                kind: KeyEventKind::Press,
+                state: KeyEventState::NONE,
+            }],
+            Action::Scroll(Scroll::Down(ScrollDistance::Message)),
+        );
+
+        let mut inner = HashMap::<Mode, _>::new();
+        inner.insert(Default::default(), mapping);
+
+        KeyBindings(inner)
+    }
+}
+
 impl<'de> Deserialize<'de> for KeyBindings {
     fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
     where