From c5506fe39377d864f80175821712b33949241583 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 21 Apr 2024 07:57:02 +0800
Subject: refactor: tell user session compression (#426)

---
 src/main.rs | 20 ++------------------
 1 file changed, 2 insertions(+), 18 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 600381b..6d5fe9b 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -25,9 +25,8 @@ use is_terminal::IsTerminal;
 use parking_lot::RwLock;
 use std::io::{stderr, stdin, stdout, Read};
 use std::sync::{mpsc, Arc};
-use std::time::Duration;
 use std::{process, thread};
-use utils::Spinner;
+use utils::run_spinner;
 
 fn main() -> Result<()> {
     let cli = Cli::parse();
@@ -155,7 +154,7 @@ fn execute(config: &GlobalConfig, mut input: Input) -> Result<()> {
     let client = init_client(config)?;
     config.read().maybe_print_send_tokens(&input);
     let (tx, rx) = mpsc::sync_channel::<()>(0);
-    thread::spawn(move || run_spinner(rx));
+    thread::spawn(move || run_spinner(" Generating", rx));
     let ret = client.send_message(input.clone());
     tx.send(())?;
     let mut eval_str = ret?;
@@ -246,18 +245,3 @@ fn create_input(
     };
     Ok(Some(input))
 }
-
-fn run_spinner(rx: mpsc::Receiver<()>) -> Result<()> {
-    let mut writer = stdout();
-    let mut spinner = Spinner::new(" Generating");
-    loop {
-        spinner.step(&mut writer)?;
-        if let Ok(()) = rx.try_recv() {
-            spinner.stop(&mut writer)?;
-            break;
-        }
-        thread::sleep(Duration::from_millis(50))
-    }
-    spinner.stop(&mut writer)?;
-    Ok(())
-}
-- 
cgit v1.2.3