feat: add `config.stream` and `.set stream` repl command (#759)

2 files changed, 20 insertions, 6 deletions

diff --git a/src/utils/mod.rs b/src/utils/mod.rs
index 340359c..4e5428a 100644
--- a/src/utils/mod.rs
+++ b/src/utils/mod.rs
@@ -16,7 +16,7 @@ pub use self::path::*;
 pub use self::prompt_input::*;
 pub use self::render_prompt::render_prompt;
 pub use self::request::*;
-pub use self::spinner::{create_spinner, Spinner};
+pub use self::spinner::*;
 
 use anyhow::{Context, Result};
 use fancy_regex::Regex;
diff --git a/src/utils/spinner.rs b/src/utils/spinner.rs
index 8f386db..53969f4 100644
--- a/src/utils/spinner.rs
+++ b/src/utils/spinner.rs
@@ -1,7 +1,9 @@
+use super::IS_STDOUT_TERMINAL;
+
 use anyhow::Result;
 use crossterm::{cursor, queue, style, terminal};
-use is_terminal::IsTerminal;
 use std::{
+    future::Future,
     io::{stdout, Write},
     time::Duration,
 };
@@ -10,7 +12,6 @@ use tokio::{sync::mpsc, time::interval};
 pub struct SpinnerInner {
     index: usize,
     message: String,
-    is_not_terminal: bool,
 }
 
 impl SpinnerInner {
@@ -20,12 +21,11 @@ impl SpinnerInner {
         SpinnerInner {
             index: 0,
             message: message.to_string(),
-            is_not_terminal: !stdout().is_terminal(),
         }
     }
 
     fn step(&mut self) -> Result<()> {
-        if self.is_not_terminal || self.message.is_empty() {
+        if !*IS_STDOUT_TERMINAL || self.message.is_empty() {
             return Ok(());
         }
         let mut writer = stdout();
@@ -50,7 +50,7 @@ impl SpinnerInner {
     }
 
     fn clear_message(&mut self) -> Result<()> {
-        if self.is_not_terminal || self.message.is_empty() {
+        if !*IS_STDOUT_TERMINAL || self.message.is_empty() {
             return Ok(());
         }
         self.message.clear();
@@ -126,3 +126,17 @@ async fn run_spinner(message: String, mut rx: mpsc::UnboundedReceiver<SpinnerEve
     }
     Ok(())
 }
+
+pub async fn run_with_spinner<F, T>(task: F, spinner_message: &str) -> Result<T>
+where
+    F: Future<Output = Result<T>>,
+{
+    if *IS_STDOUT_TERMINAL {
+        let spinner = create_spinner(spinner_message).await;
+        let ret = task.await;
+        spinner.stop();
+        ret
+    } else {
+        task.await
+    }
+}