forgectl 0.1.1

Keep a GitHub account's repositories in line with one manifest
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

//! A live line per repository while forge works through them, like
//! `docker compose` shows its containers: `·` waiting, a spinner while in
//! flight, its mark once done. Results are printed above the live lines in
//! the repositories' order, as soon as every one before them is done, so the
//! live block shrinks from the top and what stays on screen is the same
//! report a pipe gets.
//!
//! Off a terminal there are no live lines, and results are just printed.

use std::io::{IsTerminal, stdout};
use std::time::Duration;

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

pub struct Progress {
    /// `None` when stdout isn't a terminal.
    live: Option<Live>,
}

struct Live {
    multi: MultiProgress,
    lines: Vec<ProgressBar>,
}

impl Progress {
    pub fn new(names: &[&str]) -> Self {
        if !stdout().is_terminal() {
            return Self { live: None };
        }
        let multi = MultiProgress::with_draw_target(ProgressDrawTarget::stdout());
        let waiting = ProgressStyle::with_template("· {msg}").expect("valid template");
        let lines = names
            .iter()
            .map(|name| {
                multi.add(
                    ProgressBar::new_spinner()
                        .with_style(waiting.clone())
                        .with_message((*name).to_owned()),
                )
            })
            .collect();
        Self {
            live: Some(Live { multi, lines }),
        }
    }

    /// Item `i` is in flight.
    pub fn start(&self, i: usize) {
        if let Some(live) = &self.live {
            let line = &live.lines[i];
            line.set_style(
                ProgressStyle::with_template("{spinner} {msg}").expect("valid template"),
            );
            line.enable_steady_tick(Duration::from_millis(80));
        }
    }

    /// Item `i` is done, with `mark` for its result, though its report may
    /// still wait for those before it.
    pub fn finish(&self, i: usize, mark: &str) {
        if let Some(live) = &self.live {
            let line = &live.lines[i];
            line.set_style(
                ProgressStyle::with_template(&format!("{mark} {{msg}}")).expect("valid template"),
            );
            line.finish();
        }
    }

    /// Prints item `i`'s report in place of its live line.
    pub fn report(&self, i: usize, text: &str) {
        match &self.live {
            Some(live) => {
                live.lines[i].finish_and_clear();
                live.multi.remove(&live.lines[i]);
                // Only fails when the draw target is gone; nothing to do then.
                let _ = live.multi.println(text);
            }
            None => println!("{text}"),
        }
    }
}