use std::time::Duration;
use std::time::Instant;
use async_signal::{Signal, Signals};
use futures_concurrency::future::Race as _;
use futures_lite::StreamExt as _;
use futures_lite::{Stream, future, stream};
use strides::stream::StreamExt;
use strides::{Theme, bar, spinner, term};
fn throttle<I>(s: impl Stream<Item = I>, interval: Duration) -> impl Stream<Item = I> {
s.zip(async_io::Timer::interval_at(Instant::now(), interval))
.map(|(item, _)| item)
}
fn main() {
let ticks = throttle(stream::iter(0..100), Duration::from_millis(30));
let messages = throttle(
stream::iter(["hello ...", "computing ...", "almost there ...", "done"]),
Duration::from_secs(1),
);
let theme = Theme::default()
.with_bar(bar::styles::PARALLELOGRAM)
.with_spinner(spinner::styles::SAND);
let mut signals = Signals::new([Signal::Int]).expect("signal handler");
future::block_on(async {
let work = async {
let sum = ticks
.progress(theme, |index, _| index as f64 / 100.0)
.with_messages(messages)
.fold(0, |acc, x| acc + x)
.await;
println!("Sum is {sum}, Gauss was right");
};
let on_interrupt = async {
let _ = signals.next().await;
let _ = term::reset();
};
(work, on_interrupt).race().await;
});
}