45 lines
1.1 KiB
Rust
45 lines
1.1 KiB
Rust
use std::{error::Error, future::Future};
|
|
use indicatif::{ProgressBar, ProgressStyle};
|
|
|
|
pub async fn with_spinner<F, Fut, T>(
|
|
start_msg: &str,
|
|
finish_msg: &str,
|
|
f: F,
|
|
) -> Result<T, Box<dyn Error>>
|
|
where
|
|
F: FnOnce() -> Fut,
|
|
Fut: Future<Output = Result<T, Box<dyn Error>>>,
|
|
{
|
|
let pb = ProgressBar::new_spinner();
|
|
pb.set_message(start_msg.to_string());
|
|
pb.enable_steady_tick(std::time::Duration::from_millis(100));
|
|
|
|
let result = f().await;
|
|
|
|
pb.finish_and_clear();
|
|
println!("{} {}", dialoguer::console::style("✔").green(), finish_msg);
|
|
|
|
result
|
|
}
|
|
|
|
pub async fn with_progressbar<F, Fut, T>(
|
|
start_msg: &str,
|
|
finish_msg: &str,
|
|
len: u64,
|
|
f: F,
|
|
) -> Result<T, Box<dyn Error>>
|
|
where
|
|
F: FnOnce(ProgressBar) -> Fut,
|
|
Fut: Future<Output = Result<T, Box<dyn Error>>>
|
|
{
|
|
let pb = ProgressBar::new(len);
|
|
pb.set_style(ProgressStyle::default_bar().template(
|
|
"{msg}\n{bar:40.green/white} {bytes}/{total_bytes} ({bytes_per_sec})",
|
|
)?);
|
|
pb.set_message(start_msg.to_string());
|
|
|
|
let result = f(pb.clone()).await;
|
|
pb.finish_with_message(finish_msg.to_string());
|
|
|
|
result
|
|
}
|