From a5d484c7275a1bf438eedc7dda7c217c78d9d2d7 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 1 Aug 2025 18:17:30 +0800
Subject: feat: switch to raw stream renderer when no_color/no_highlight
 (#1371)

---
 src/render/stream.rs | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

(limited to 'src/render/stream.rs')

diff --git a/src/render/stream.rs b/src/render/stream.rs
index a6233a5..d6cadde 100644
--- a/src/render/stream.rs
+++ b/src/render/stream.rs
@@ -36,11 +36,17 @@ pub async fn raw_stream(
     mut rx: UnboundedReceiver<SseEvent>,
     abort_signal: &AbortSignal,
 ) -> Result<()> {
+    let mut spinner = Some(spawn_spinner("Generating"));
+
     loop {
         if abort_signal.aborted() {
-            return Ok(());
+            break;
         }
         if let Some(evt) = rx.recv().await {
+            if let Some(spinner) = spinner.take() {
+                spinner.stop();
+            }
+
             match evt {
                 SseEvent::Text(text) => {
                     print!("{text}");
@@ -52,6 +58,9 @@ pub async fn raw_stream(
             }
         }
     }
+    if let Some(spinner) = spinner.take() {
+        spinner.stop();
+    }
     Ok(())
 }
 
@@ -70,7 +79,7 @@ async fn markdown_stream_inner(
 
     'outer: loop {
         if abort_signal.aborted() {
-            return Ok(());
+            break;
         }
         for reply_event in gather_events(&mut rx).await {
             if let Some(spinner) = spinner.take() {
-- 
cgit v1.2.3