feat: add spinner

Author: pythops

src/app.rs

@@ -3,6 +3,7 @@ use std::collections::HashMap;
 
 use crate::config::Config;
 use crate::notification::Notification;
+use crate::spinner::Spinner;
 use crossterm::event::KeyCode;
 
 use std::sync::Arc;
@@ -39,6 +40,7 @@ pub struct App {
     pub history_thread_index: usize,
     pub config: Arc<Config>,
     pub notifications: Vec<Notification>,
+    pub spinner: Spinner,
 }
 
 impl App {
@@ -58,11 +60,19 @@ impl App {
             history_thread_index: 0,
             config,
             notifications: Vec::new(),
+            spinner: Spinner::default(),
         }
     }
 
     pub fn tick(&mut self) {
         self.notifications.retain(|n| n.ttl > 0);
         self.notifications.iter_mut().for_each(|n| n.ttl -= 1);
+
+        if self.spinner.active {
+            self.chat.pop();
+            self.chat
+                .push(format!("🤖: Waiting {}", self.spinner.draw()));
+            self.spinner.update();
+        }
     }
 }

src/handler.rs

@@ -55,7 +55,8 @@ pub fn handle_key_events(
                         }))
                         .unwrap();
                 });
-                app.chat.push("🤖: Waiting ...".to_string());
+                app.spinner.active = true;
+                app.chat.push("🤖: Waiting ..".to_string());
             }
 
             // scroll down

src/lib.rs

@@ -17,3 +17,5 @@ pub mod ui;
 pub mod notification;
 
 pub mod llm;
+
+pub mod spinner;

src/main.rs

@@ -39,6 +39,7 @@ fn main() -> AppResult<()> {
             Event::Resize(_, _) => {}
             Event::LLMAnswer(answer) => {
                 app.chat.pop();
+                app.spinner.active = false;
                 app.chat.push(format!("🤖: {}\n", answer));
                 app.chat.push("\n".to_string());
                 let mut conv: HashMap<String, String> = HashMap::new();

src/spinner.rs

@@ -0,0 +1,18 @@
+static SPINNER_CHARS: &[char] = &['⣷', '⣯', '⣟', '⡿', '⢿', '⣻', '⣽', '⣾'];
+
+#[derive(Default, Debug)]
+pub struct Spinner {
+    pub active: bool,
+    pub index: usize,
+}
+
+impl Spinner {
+    pub fn draw(&self) -> char {
+        SPINNER_CHARS[self.index]
+    }
+
+    pub fn update(&mut self) {
+        self.index += 1;
+        self.index %= SPINNER_CHARS.len();
+    }
+}