arkham 0.1.0

A framework for CLI applications
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
54
55
56
57
58
59
60
61
62
63
64

use indicatif::{HumanDuration, MultiProgress, ProgressBar, ProgressStyle};
use std::{
    sync::{Arc, Mutex},
    time::Instant,
};

#[derive(Clone)]
pub struct TaskGroup {
    mp: Arc<MultiProgress>,
}

impl TaskGroup {
    pub fn new() -> Self {
        Self {
            mp: Arc::new(MultiProgress::new()),
        }
    }

    pub fn start_task(&self, desc: &str) -> Task {
        let pb = self.mp.add(ProgressBar::new(1));
        let spinner_style = ProgressStyle::default_spinner()
            .tick_chars("⠁⠂⠄⡀⢀⠠⠐⠈ ")
            .template("{prefix:.bold.dim} {spinner} {wide_msg}");
        pb.set_style(spinner_style);
        Task::new(TaskState {
            pb,
            desc: desc.into(),
            started_at: Instant::now(),
        })
    }

    pub fn join(&self) {
        self.mp.join().unwrap();
    }
}

#[derive(Clone)]
pub struct Task(Arc<Mutex<TaskState>>);

impl Task {
    pub fn new(state: TaskState) -> Self {
        Self(Arc::new(Mutex::new(state)))
    }

    pub fn tick(&self) {
        let state = self.0.lock().unwrap();
        state.pb.set_message(&format!(
            "{} [{}]",
            state.desc,
            style(HumanDuration(state.started_at.elapsed())).yellow()
        ));
        state.pb.tick();
    }

    pub fn complete(&self) {
        self.0.lock().unwrap().pb.finish_with_message("OK");
    }
}

pub struct TaskState {
    pb: ProgressBar,
    desc: String,
    started_at: Instant,
}