vmonitor 0.0.1-alpha.1

A simple and lightweight system monitor
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

use futures_util::SinkExt;
use serde::{Deserialize, Serialize};
use tokio::net::TcpStream;
use tokio_tungstenite::{connect_async, tungstenite::Message, MaybeTlsStream, WebSocketStream};
use tracing::{error, info};

#[derive(Serialize, Deserialize, Debug)]
pub struct ReportMessage {
    pub r#type: String,
    pub data: String,
}

pub async fn connect_websocket(
    ws_url: &str,
    auth_secret: &str,
) -> Option<WebSocketStream<MaybeTlsStream<TcpStream>>> {
    match connect_async(ws_url).await {
        Ok((mut ws_stream, _)) => {
            info!("WebSocket connection established");

            // Send authentication message
            let auth_msg = ReportMessage {
                r#type: "auth".to_string(),
                data: auth_secret.to_string(),
            };

            match ws_stream
                .send(Message::Text(
                    serde_json::to_string(&auth_msg).unwrap().into(),
                ))
                .await
            {
                Ok(_) => {
                    info!("Authentication message sent successfully");
                    Some(ws_stream)
                }
                Err(e) => {
                    error!(error = %e, "Failed to send authentication message");
                    None
                }
            }
        }
        Err(e) => {
            error!(error = %e, url = %ws_url, "WebSocket connection failed");
            None
        }
    }
}