post-archiver-utils 0.2.1

A set of utilities for Post Archiver
Documentation 
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

use std::{cell::RefCell, collections::HashMap};

use indicatif::{MultiProgress, ProgressBar, ProgressStyle, WeakProgressBar};

pub struct ProgressSet {
    pub multi: MultiProgress,
    pub map: RefCell<HashMap<&'static str, WeakProgressBar>>,
    pub disabled: &'static[&'static str],
}

impl ProgressSet {
    pub fn new(multi: MultiProgress) -> Self {
        Self {
            multi: multi,
            map: Default::default(),
            disabled: &[],
        }
    }

    pub fn disabled(mut self, prefixes: &'static[&'static str]) -> Self {
        self.disabled = prefixes;
        self
    }

    pub fn add(&self, prefix: &'static str) -> ProgressBar {
        if self.disabled.contains(&prefix) {
            return ProgressBar::hidden();
        }

        let mut map = self.map.borrow_mut();

        if let Some(weak) = map.get(prefix)
            && let Some(bar) = weak.upgrade()
        {
            return bar;
        }

        let pb = self.multi.add(
            ProgressBar::new(0)
                .with_style(Self::style())
                .with_prefix(format!("[{prefix}]")),
        );

        map.insert(prefix, pb.downgrade());
        pb
    }

    fn style() -> ProgressStyle {
        ProgressStyle::with_template("{prefix:.bold.dim} {wide_bar:.cyan/blue} {pos:>3}/{len:3}")
            .unwrap()
            .progress_chars("#>-")
    }
}