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/utils/spinner.rs | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'src/utils/spinner.rs')

diff --git a/src/utils/spinner.rs b/src/utils/spinner.rs
index c684901..3715eff 100644
--- a/src/utils/spinner.rs
+++ b/src/utils/spinner.rs
@@ -1,6 +1,11 @@
 use anyhow::Result;
 use crossterm::{cursor, queue, style, terminal};
-use std::io::{Stdout, Write};
+use std::{
+    io::{stdout, Stdout, Write},
+    sync::mpsc,
+    thread,
+    time::Duration,
+};
 
 pub struct Spinner {
     index: usize,
@@ -50,3 +55,18 @@ impl Spinner {
         Ok(())
     }
 }
+
+pub fn run_spinner(message: &str, rx: mpsc::Receiver<()>) -> Result<()> {
+    let mut writer = stdout();
+    let mut spinner = Spinner::new(message);
+    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