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

use std::borrow::Cow;

use indicatif::ProgressStyle;

pub struct ProgressBar {
    bar: indicatif::ProgressBar,
}

impl ProgressBar {
    pub fn new(len: u64) -> Self {
        let bar = indicatif::ProgressBar::new(len);
        bar.set_style(
            ProgressStyle::default_bar()
                .progress_chars("=>-")
                .template("{msg:.bold.cyan/blue} [{bar:20.cyan/blue}][{percent}%] {pos}/{len}"),
        );
        Self { bar }
    }

    pub fn inc(&mut self, delta: u64) {
        self.bar.inc(delta)
    }

    pub fn finish(&self, msg: impl Into<Cow<'static, str>>) {
        self.bar.set_message(msg);
        self.bar
            .set_style(ProgressStyle::default_bar().template("{msg:.green.bold} in {elapsed}"));
        self.bar.finish_at_current_pos();
    }
}