devops-armory 0.7.4

Library to make DevOps work easier, based on russh and actix.
Documentation 
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

use awc::{
    Client, Connector,
    ws::{
        self, 
        Frame
    },
};
use futures::StreamExt;
use futures_util::SinkExt as _;

use openssl::ssl::{
    SslConnector, 
    SslMethod, 
    SslVerifyMode
};

/// Total memory usage
/// Token and gke cluster endpoint, namespace and pod name need to be provided
pub async fn mem_total(
    token: String,
    gke_cluster_endpoint: String,
    gke_cluster_namespace: String,
    gke_pod_name: String
) -> Result<f64, std::io::Error> {

    let mut builder = SslConnector::builder(SslMethod::tls()).unwrap();
    builder.set_verify(SslVerifyMode::NONE);
    let myconnector = builder.build();
    let client = Client::builder()
        .connector(Connector::new().openssl(myconnector))
        .finish();
    loop {
        let Ok((_resp, mut connection2)) = client
                .ws(format!("https://{gke_cluster_endpoint}:443/api/v1/namespaces/{gke_cluster_namespace}/pods/{gke_pod_name}/exec?command=cat&command=/proc/meminfo&stdin=true&stdout=true&stderr=true"))
                .bearer_auth(format!("{token}"))
                .connect()
                .await
                else {
                    continue;
                };

            connection2
                .send(ws::Message::Text("SGVsbG8sIHdvcmxkIQ==".into()))
                .await
                .unwrap();

        loop {
            let response = connection2.next().await;
            match response {
                Some(Ok(Frame::Binary(b))) => {
                    let s = String::from_utf8_lossy(&b);
                    let v = s.trim_matches(|c: char| !c.is_alphanumeric());
                    if v.is_empty() {
                        continue;
                    }

                    let mem: String = v.parse().expect("Incorrect type");

                    let z = mem.split(" ").collect::<Vec<&str>>()[7].trim();

                    let mem_total: f64 = z.parse().expect("Parse number error");

                    return Ok(mem_total);
                }
                Some(Ok(Frame::Close(_c))) => {
                    //println!("Connection closed");
                }
                Some(Ok(_)) => todo!(),
                Some(Err(e)) => {
                    println!("Some error occured {e}");
                    //continue;
                }
                None => {
                    break;
                }
            }
        }
    }
}