FEAT: bidirectional webtransport

Author: BryanFRD

Cargo.lock

@@ -7,18 +7,19 @@ edition = "2024"
 axum = "0.8.1"
 bollard = "0.18.1"
 tokio = { version = "1.43.0", features = ["full"] }
-serde = { version = "1.0.217", features = ["derive"] }
-serde_json = "1.0.138"
+serde = { version = "1.0.218", features = ["derive"] }
+serde_json = "1.0.140"
 tower-http = { version = "0.6.2", features = ["cors"] }
 tower = "0.5.2"
 tokio-tungstenite = "0.26.2"
 futures = "0.3.31"
 futures-util = "0.3.31"
 quiche = "0.23.2"
 env_logger = "0.11.6"
-log = "0.4.25"
+log = "0.4.26"
 quinn = "0.11.6"
 hyper = "1.6.0"
 hyper-rustls = "0.27.5"
 wtransport = "0.5.0"
 rustls = "0.23.23"
+brotli = "7.0.0"

Cargo.toml

@@ -1,10 +1,10 @@
 use bollard::secret::ContainerSummary;
-use serde_json::json;
+use serde_json::{json, Value};
 
 use super::EventDTO;
 
 impl EventDTO for ContainerSummary {
-  fn to_json(&self) -> String {
+  fn to_json(&self) -> Value {
     json!({
       "id": self.id,
       "names": self.names,
@@ -15,12 +15,15 @@ impl EventDTO for ContainerSummary {
       "status": self.status,
       "ports": self.ports,
       "labels": self.labels,
-    }).to_string()
+      "size_rw": self.size_rw,
+      "size_root_fs": self.size_root_fs,
+      "mounts": self.mounts,
+    })
   }
 }
 
 impl EventDTO for Vec<ContainerSummary> {
-  fn to_json(&self) -> String {
-    json!(self.iter().map(|f| f.to_json()).collect::<Vec<String>>()).to_string()
+  fn to_json(&self) -> Value {
+    json!(self.iter().map(|f| f.to_json()).collect::<Vec<Value>>())
   }
 }

src/datas/docker.rs

@@ -1,5 +1,23 @@
+use serde_json::{json, Value};
+
+use crate::events::Event;
+
 pub mod docker;
 
+pub fn create_event_dto(event: Event, data: Value) -> String {
+  let mut event_json = serde_json::to_value(event).unwrap_or_else(|_| json!({}));
+  if let Value::Object(ref mut map) = event_json {
+    map.insert("data".to_string(), data);
+  }
+  event_json.to_string()
+}
+
 pub trait EventDTO {
-  fn to_json(&self) -> String;
+  fn to_json(&self) -> Value;
+}
+
+impl EventDTO for i8 {
+  fn to_json(&self) -> Value {
+    json!(*self)
+  }
 }

src/datas/mod.rs

@@ -0,0 +1,14 @@
+use crate::{datas::{create_event_dto, EventDTO}, events::{docker::DockerEvent, Event}, services::docker};
+
+pub async fn handle_message(send_stream: &mut wtransport::SendStream, event: &DockerEvent) {
+    match event {
+        DockerEvent::DockerStatus => {
+            if let Err(error) = send_stream.write_all(create_event_dto(Event::Docker(DockerEvent::DockerStatus), docker::ping().await.to_json()).as_bytes()).await {
+                log::error!("Failed to send event: {:?}", error);
+            };
+        },
+        DockerEvent::DockerContainersRestart { data } => {
+            log::info!("Restarting containers: {:?}", data);
+        }
+    }
+}