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

use std::io::{self, Write};

use owo_colors::OwoColorize;

pub struct Printer {
    ratio: f64,
    width: i32,
    full_bar: String,
    rest_bar: String,
    json_format: bool,
    /// only used in status bar
    ratio_char: String,
}

impl Printer {
    pub fn new(width: i32, full_bar: &str, rest_bar: &str, json_format: bool) -> Self {
        Self {
            ratio: 0.0,
            width,
            full_bar: full_bar.to_string(),
            rest_bar: rest_bar.to_string(),
            json_format,
            ratio_char: "y".to_string(),
        }
    }
    pub fn ratio(mut self, ratio: f64) -> Self {
        self.ratio = ratio;
        self
    }
    pub fn ratio_char(mut self, ratio_char: &str) -> Self {
        self.ratio_char = ratio_char.to_string();
        self
    }
    /// Show progress-bar
    pub fn print(&self) {
        let ratio_int = (self.ratio * 100.0) as i32;
        let progress_int = (self.ratio * f64::from(self.width)).round() as i32;
        let rest_int = self.width - progress_int;

        let mut progress_fmt = format!(
            "{}{} {}%",
            self.full_bar.repeat(progress_int as usize),
            self.rest_bar.repeat(rest_int as usize),
            ratio_int
        );
        let state = {
            if ratio_int > 90 {
                "Critical"
            } else {
                "Info"
            }
        };
        // color
        if state == "Critical" && !self.json_format {
            progress_fmt = format!("{}", progress_fmt.red());
        }
        // JSON
        if self.json_format {
            progress_fmt = format!(
                r#"{{"icon": "{}", "state": "{}", "text": "{}: {}"}}"#,
                "zman", state, self.ratio_char, progress_fmt
            );
        }
        writeln!(io::stdout(), "{}", progress_fmt).ok();
    }
}