Re-implement the ws client

snow-scanner/src/worker/modules.rs

@@ -1,6 +1,7 @@
 use std::{net::IpAddr, str::FromStr};
 
 use cidr::IpCidr;
+use rocket_ws::Message as RocketMessage;
 use serde::{Deserialize, Deserializer, Serialize, Serializer};
 
 #[derive(Debug, Clone, PartialEq)]
@@ -66,6 +67,20 @@ impl Into<WorkerMessages> for String {
     }
 }
 
+impl TryInto<WorkerMessages> for RocketMessage {
+    type Error = String;
+
+    fn try_into(self) -> Result<WorkerMessages, Self::Error> {
+        match self {
+            RocketMessage::Text(data) => {
+                let data: WorkerMessages = data.into();
+                Ok(data)
+            }
+            _ => Err("Only text is supported".to_string()),
+        }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use cidr::IpCidr;