rtx-cli 2023.12.21

Polyglot runtime manager (asdf rust clone)
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

use console::style;
use indicatif::MultiProgress;

use crate::ui::progress_report::ProgressReport;

#[derive(Debug)]
pub struct MultiProgressReport {
    mp: Option<MultiProgress>,
}

impl MultiProgressReport {
    pub fn new(verbose: bool) -> Self {
        match verbose {
            true => Self { mp: None },
            false => Self {
                mp: Some(MultiProgress::new()),
            },
        }
    }
    pub fn add(&self) -> ProgressReport {
        match &self.mp {
            Some(mp) => {
                let mut pr = ProgressReport::new(false);
                pr.pb = Some(mp.add(pr.pb.unwrap()));
                pr
            }
            None => ProgressReport::new(true),
        }
    }
    pub fn suspend<F: FnOnce() -> R, R>(&self, f: F) -> R {
        match &self.mp {
            Some(mp) => mp.suspend(f),
            None => f(),
        }
    }
    pub fn warn(&self, message: String) {
        match &self.mp {
            Some(pb) => {
                let _ = pb.println(format!(
                    "{} {}",
                    style("[WARN]").yellow().for_stderr(),
                    message
                ));
            }
            None => warn!("{}", message),
        }
    }
    // pub fn clear(&self) {
    //     match &self.mp {
    //         Some(mp) => {
    //             let _ = mp.clear();
    //         },
    //         None => ()
    //     }
    // }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_multi_progress_report() {
        let mpr = MultiProgressReport::new(false);
        let pr = mpr.add();
        pr.set_style(indicatif::ProgressStyle::with_template("").unwrap());
        pr.enable_steady_tick();
        pr.finish_with_message("test");
        pr.println("");
        pr.set_message("test");
    }
}