From d098ac074e980d378fa2920893a04f5903cb5bda Mon Sep 17 00:00:00 2001
From: monosans <hsyqixco@protonmail.com>
Date: Fri, 11 Apr 2025 15:18:46 +0300
Subject: [PATCH] fix non-tui uis

---
 src/ui/dummy.rs  | 8 ++++++--
 src/ui/logger.rs | 8 ++++++--
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/src/ui/dummy.rs b/src/ui/dummy.rs
index 05a02a7..5c50134 100644
--- a/src/ui/dummy.rs
+++ b/src/ui/dummy.rs
@@ -1,4 +1,4 @@
-use crate::event::Event;
+use crate::event::{AppEvent, Event};
 
 pub(crate) struct DummyUI {}
 
@@ -14,7 +14,11 @@ impl super::UI for DummyUI {
         _tx: tokio::sync::mpsc::UnboundedSender<Event>,
         mut rx: tokio::sync::mpsc::UnboundedReceiver<Event>,
     ) -> color_eyre::Result<()> {
-        while rx.recv().await.is_some() {}
+        while let Some(event) = rx.recv().await {
+            if let Event::App(AppEvent::Done) = event {
+                break;
+            }
+        }
         Ok(())
     }
 }
diff --git a/src/ui/logger.rs b/src/ui/logger.rs
index 460f44f..ed54800 100644
--- a/src/ui/logger.rs
+++ b/src/ui/logger.rs
@@ -1,4 +1,4 @@
-use crate::event::Event;
+use crate::event::{AppEvent, Event};
 
 pub(crate) struct LoggerUI {}
 
@@ -17,7 +17,11 @@ impl super::UI for LoggerUI {
         _tx: tokio::sync::mpsc::UnboundedSender<Event>,
         mut rx: tokio::sync::mpsc::UnboundedReceiver<Event>,
     ) -> color_eyre::Result<()> {
-        while rx.recv().await.is_some() {}
+        while let Some(event) = rx.recv().await {
+            if let Event::App(AppEvent::Done) = event {
+                break;
+            }
+        }
         Ok(())
     }
 }