otoroshictl 0.0.12

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
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use cli_table::{Cell, CellStruct, Color, Style, Table, print_stdout};

use std::vec::Vec;

pub struct TableResource {
    pub raw: serde_json::Value,
}

pub struct TableHelper {}

impl TableHelper {
    fn walk_through_object(
        path: Vec<String>,
        value: &serde_json::Map<String, serde_json::Value>,
    ) -> serde_json::Map<String, serde_json::Value> {
        if path.is_empty() {
            value.to_owned()
        } else {
            value
                .get(path.first().unwrap())
                .unwrap()
                .as_object()
                .unwrap()
                .to_owned()
        }
    }

    fn column_as_cell(
        name: &String,
        value: &serde_json::Map<String, serde_json::Value>,
    ) -> CellStruct {
        match value.get(name) {
            Some(serde_json::Value::Array(arr)) => {
                arr.len().cell().justify(cli_table::format::Justify::Center)
            }
            Some(serde_json::Value::Object(obj)) => {
                obj.len().cell().justify(cli_table::format::Justify::Center)
            }
            Some(serde_json::Value::Bool(true)) => {
                "yes".cell().justify(cli_table::format::Justify::Center)
            }
            Some(serde_json::Value::Bool(false)) => {
                "no".cell().justify(cli_table::format::Justify::Center)
            }
            Some(serde_json::Value::Number(v)) => {
                v.cell().justify(cli_table::format::Justify::Center)
            }
            Some(serde_json::Value::String(str)) => str.cell(),
            _ => "".cell(),
        }
    }

    pub fn display_table_of_resources_with_custom_columns(
        vec: Vec<TableResource>,
        columns: Vec<String>,
    ) {
        let table = vec
            .into_iter()
            .map(|item| {
                let value = item.raw.as_object().unwrap();
                let values: Vec<CellStruct> = columns
                    .iter()
                    .cloned()
                    .map(|name| {
                        if name.contains(".") {
                            let path: Vec<String> =
                                name.split(".").map(|i| i.to_string()).collect();
                            let last_name: String = path.last().unwrap().to_string();
                            Self::column_as_cell(
                                &last_name,
                                &Self::walk_through_object(
                                    path.get(0..path.len() - 1).unwrap().to_vec(),
                                    value,
                                ),
                            )
                        } else {
                            Self::column_as_cell(&name, value)
                        }
                    })
                    .collect();
                values
            })
            .table()
            .title(columns.into_iter().map(|name| {
                if name.contains(".") {
                    name.split(".")
                        .last()
                        .unwrap_or(name.as_str())
                        .cell()
                        .bold(true)
                        .foreground_color(Some(Color::White))
                } else {
                    name.cell()
                        .bold(true)
                        .background_color(Some(Color::Rgb(0, 0, 0)))
                        .foreground_color(Some(Color::White))
                }
            }));
        let _ = print_stdout(table);
    }

    pub fn display_table_of_resources_default(vec: Vec<TableResource>) {
        Self::display_table_of_resources_with_custom_columns(
            vec,
            vec![
                "id".to_string(),
                "name".to_string(),
                "description".to_string(),
                "enabled".to_string(),
                "tags".to_string(),
                "metadata".to_string(),
            ],
        )
    }
}