otoroshictl 0.0.8

a CLI to manage your otoroshi clusters with style ;)
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use serde::{Serialize, Deserialize};
use cli_table::{print_stdout, Cell, Style, Table};

use crate::cli::cliopts::{CliOpts, Commands};
use crate::{cli_stderr_printline, cli_stdout_printline};
use crate::utils::otoroshi::Otoroshi;

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct ComponentHealth {
   pub initialized: Option<bool>,
   pub status: String,
}

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct ClusterHealth {
   pub health: String,
}

#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct OtoroshiHealth {
   pub otoroshi: String,
   pub datastore: String,
   pub proxy: ComponentHealth,
   pub storage: ComponentHealth,
   pub eventstore: ComponentHealth,
   pub certificates: ComponentHealth,
   pub scripts: ComponentHealth,
   pub cluster: Option<ClusterHealth>,
}

pub struct HealthCommand {}

impl HealthCommand {

    fn healthy_cell(value: String) -> cli_table::CellStruct {
        match value.as_str() {
            "healthy" => value.cell().foreground_color(Some(cli_table::Color::Green)),
            "unhealthy" => value.cell().foreground_color(Some(cli_table::Color::Magenta)),
            "down" => value.cell().foreground_color(Some(cli_table::Color::Red)),
            "unreachable" => value.cell().foreground_color(Some(cli_table::Color::Red)),
            _ => value.cell().foreground_color(Some(cli_table::Color::White)),
        }
    }

    fn loaded_cell(value: String) -> cli_table::CellStruct {
        match value.as_str() {
            "loaded" => value.cell().foreground_color(Some(cli_table::Color::Green)),
            _ => value.cell().foreground_color(Some(cli_table::Color::White)),
        }
    }

    fn default_display(health: OtoroshiHealth) -> () {
        let table = vec![
            vec![  
                Self::healthy_cell(health.otoroshi),
                Self::healthy_cell(health.datastore),
                Self::healthy_cell(health.storage.status),
                Self::healthy_cell(health.eventstore.status),
                Self::loaded_cell(health.certificates.status),
                Self::loaded_cell(health.scripts.status),
                match health.cluster {
                    None => "".cell(),
                    Some(cluster) => Self::healthy_cell(cluster.health),
                },
            ]
        ]
        .table()
        .title(vec![
            "otoroshi".cell().bold(true),
            "datastore".cell().bold(true),
            "storage".cell().bold(true),
            "eventstore".cell().bold(true),
            "certificates".cell().bold(true),
            "scripts".cell().bold(true),
            "cluster".cell().bold(true),
        ]);
        let _ = print_stdout(table);
    }

    pub async fn display(cli_opts: CliOpts, _command: &Commands) -> () {
        match Otoroshi::get_health(cli_opts.clone()).await {
            None => {
                cli_stderr_printline!("error while fetching health");
                std::process::exit(-1)
            },
            Some(health) => {
                match cli_opts.ouput {
                    Some(str) => {
                        match str.as_str() {
                            "json" => cli_stdout_printline!("{}", serde_json::to_string(&health).unwrap()),
                            "json_pretty" => cli_stdout_printline!("{}", serde_json::to_string_pretty(&health).unwrap()),
                            "yaml" => cli_stdout_printline!("{}", serde_yaml::to_string(&health).unwrap()),
                            _ => Self::default_display(health),
                        }
                    },
                    _ => Self::default_display(health),
                };
            },
        };    
    }
}