1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
use std::io::Write;
use std::time::Duration;
use std::sync::Arc;
use anyhow::Result;
use log::info;
use crate::event_bus::{EventBus, Event, EventEmitter};
use crate::impl_event_emitter;
/// Simple terminal UI handler using a spinner progress bar.
pub struct UIHandler {
pub headless: bool,
handle: Option<tokio::task::JoinHandle<()>>,
event_bus: Option<Arc<EventBus>>,
}
impl UIHandler {
pub fn new(headless: bool) -> Self {
Self {
headless,
handle: None,
event_bus: None,
}
}
pub fn start(&mut self) -> Result<()> {
if !self.headless {
info!("Starting UI");
// Clone event bus for the spawned task
let event_bus = self.event_bus.clone();
let handle = tokio::spawn(async move {
let frames = ["|", "/", "-", "\\"];
let mut idx = 0usize;
// If we have event bus, also listen for events
if let Some(bus) = event_bus {
let mut receiver = bus.subscribe();
loop {
// Check for events with timeout
match tokio::time::timeout(
Duration::from_millis(100),
receiver.recv()
).await {
Ok(Ok(event)) => {
// Handle specific events that affect UI
match event {
Event::TaskProgress { message, .. } => {
print!("\r{} {}", frames[idx % frames.len()], message);
}
Event::ShutdownRequested => {
break;
}
_ => {
// Continue with spinner for other events
print!("\r{}", frames[idx % frames.len()]);
}
}
}
_ => {
// No event or timeout, just update spinner
print!("\r{}", frames[idx % frames.len()]);
}
}
let _ = std::io::stdout().flush();
idx = (idx + 1) % frames.len();
}
} else {
// No event bus, just show spinner
loop {
print!("\r{}", frames[idx % frames.len()]);
let _ = std::io::stdout().flush();
idx = (idx + 1) % frames.len();
tokio::time::sleep(Duration::from_millis(100)).await;
}
}
});
self.handle = Some(handle);
}
Ok(())
}
pub fn finish(&mut self) {
if let Some(handle) = &self.handle {
handle.abort();
print!("\r \r");
let _ = std::io::stdout().flush();
}
self.handle = None;
}
/// Display a task description
pub async fn display_task(&mut self, task: &str) -> Result<()> {
println!("Task: {}", task);
Ok(())
}
/// Display an error message
pub async fn display_error(&mut self, error: &str) -> Result<()> {
println!("Error: {}", error);
Ok(())
}
}
// Implement EventEmitter trait for UIHandler
impl_event_emitter!(UIHandler);