use std::{
future::Future,
io::{self, IsTerminal},
time::Instant,
};
use spinners::{Spinner, Spinners};
use super::Result;
pub(super) fn with_elapsed<F, R, EndMsg>(f: F, start_msg: &str, end_msg: EndMsg) -> Result<R>
where
F: FnOnce() -> Result<R>,
EndMsg: FnOnce(&R) -> String,
{
let timer = Instant::now();
let mut maybe_sp = start(start_msg);
Ok(end(f()?, timer, maybe_sp.as_mut(), end_msg))
}
pub(super) async fn with_elapsed_async<F, Fut, R, EndMsg>(
f: F,
start_msg: &str,
end_msg: EndMsg,
) -> Result<R>
where
F: FnOnce() -> Fut,
Fut: Future<Output = Result<R>>,
EndMsg: FnOnce(&R) -> String,
{
let timer = Instant::now();
let mut maybe_sp = start(start_msg);
Ok(end(f().await?, timer, maybe_sp.as_mut(), end_msg))
}
fn start(start_msg: &str) -> Option<Spinner> {
let msg = format!("⏳ {start_msg}");
if io::stdout().is_terminal() {
Some(Spinner::new(Spinners::Dots, msg))
} else {
println!("{msg}");
None
}
}
fn end<T, EndMsg>(val: T, timer: Instant, maybe_sp: Option<&mut Spinner>, end_msg: EndMsg) -> T
where
EndMsg: FnOnce(&T) -> String,
{
let msg = format!("✅ {} in {:.2}s", end_msg(&val), timer.elapsed().as_secs_f32());
if let Some(sp) = maybe_sp {
sp.stop_with_message(msg);
} else {
println!("{msg}");
}
val
}