Daemon wol

Levminer · Jul 24, 2024 · f042c4d · f042c4d
1 parent 4a5f2dd
commit f042c4d
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/platforms/unix/daemon/Cargo.toml b/platforms/unix/daemon/Cargo.toml
@@ -24,3 +24,4 @@ log = "0.4"
 ezrtc = "0.4.0"
 webrtc = "0.10"
 async-channel = "2.3.1"
+wol-rs = "1"
diff --git a/platforms/unix/daemon/src/main.rs b/platforms/unix/daemon/src/main.rs
@@ -18,6 +18,7 @@ use simplelog::{ColorChoice, CombinedLogger, Config, TermLogger, TerminalMode};
 use std::borrow::Cow;
 use std::net::SocketAddr;
 use std::ops::ControlFlow;
+use std::str::FromStr;
 use std::sync::Arc;
 use tower_http::{
     cors::{Any, CorsLayer},
@@ -26,11 +27,12 @@ use tower_http::{
 use webrtc::data_channel::data_channel_state::RTCDataChannelState;
 use webrtc::data_channel::RTCDataChannel;
 use webrtc::ice_transport::ice_server::RTCIceServer;
+use wol::MacAddr;
 
 #[derive(Serialize, Deserialize)]
-struct NetworkData {
+struct GenericMessage<T> {
     pub r#type: String,
-    pub data: HardwareInfo,
+    pub data: T,
 }
 
 struct AppState {
@@ -136,7 +138,7 @@ async fn main() {
                     loop {
                         if dc.ready_state() == RTCDataChannelState::Open {
                             let hw_message = receiver.recv().await.unwrap();
-                            let network_data = NetworkData {
+                            let network_data = GenericMessage::<HardwareInfo> {
                                 r#type: "data".to_string(),
                                 data: hw_message.clone(),
                             };
@@ -153,6 +155,35 @@ async fn main() {
 
             fn handle_data_channel_message(&self, message: String) {
                 warn!("Data channel message received: {:?}", message);
+
+                let network_message: Result<GenericMessage<String>, _> =
+                    serde_json::from_str(&message);
+
+                match network_message {
+                    Ok(message) => match message.r#type.as_str() {
+                        "shutdown" => {
+                            info!("Received shutdown message");
+                        }
+                        "sleep" => {
+                            info!("Received sleep message");
+                        }
+                        "restart" => {
+                            info!("Received restart message");
+                        }
+                        "wol" => {
+                            let formatted_mac = format_mac_address(&message.data);
+                            let mac_addr = MacAddr::from_str(&formatted_mac).unwrap();
+
+                            info!("Sending WOL packet to: {:?}", mac_addr);
+                        }
+                        _ => {
+                            warn!("Unknown message type: {:?}", message.r#type);
+                        }
+                    },
+                    Err(e) => {
+                        warn!("Failed to parse message: {:?}", e);
+                    }
+                }
             }
         }
 
@@ -225,7 +256,7 @@ async fn handle_socket(mut socket: WebSocket, addr: SocketAddr, state: Arc<AppSt
 
         loop {
             let hw_message = receiver.recv().await.unwrap();
-            let network_data = NetworkData {
+            let network_data = GenericMessage::<HardwareInfo> {
                 r#type: "data".to_string(),
                 data: hw_message.clone(),
             };
@@ -312,3 +343,12 @@ fn process_message(msg: Message, who: SocketAddr) -> ControlFlow<(), ()> {
     }
     ControlFlow::Continue(())
 }
+
+fn format_mac_address(mac: &str) -> String {
+    mac.chars()
+        .collect::<Vec<char>>()
+        .chunks(2)
+        .map(|chunk| chunk.iter().collect::<String>())
+        .collect::<Vec<String>>()
+        .join(":")
+}