1 files changed, 40 insertions, 0 deletions
diff --git a/client-native-rift/src/signaling.rs b/client-native-rift/src/signaling.rs
new file mode 100644
index 0000000..c61b982
--- /dev/null
+++ b/client-native-rift/src/signaling.rs
@@ -0,0 +1,40 @@
+use crate::protocol::ClientboundPacket;
+use crate::{crypto::hash, protocol::ServerboundPacket};
+use futures_util::{SinkExt, StreamExt};
+use tokio_tungstenite::tungstenite::{self, Message};
+
+pub async fn signaling_connect(
+    host: &str,
+    secret: &str,
+) -> (
+    impl SinkExt<ServerboundPacket>,
+    impl StreamExt<Item = Option<ClientboundPacket>>,
+) {
+    let (conn, _) = tokio_tungstenite::connect_async(
+        url::Url::parse(&format!("wss://{host}/signaling/{}", hash(secret))).unwrap(),
+    )
+    .await
+    .unwrap();
+
+    let (tx, rx) = conn.split();
+    let prx = rx.map(|mesg| {
+        let mesg = mesg.unwrap();
+        match mesg {
+            tungstenite::Message::Text(t) => {
+                let p: ClientboundPacket = serde_json::from_str(t.as_str()).unwrap();
+                Some(p)
+            }
+            tungstenite::Message::Close(_) => {
+                eprintln!("ws closed :(");
+                None
+            }
+            _ => None,
+        }
+    });
+    let ptx = tx.with(async move |p| {
+        Ok::<_, tokio_tungstenite::tungstenite::error::Error>(Message::Text(
+            serde_json::to_string::<ServerboundPacket>(&p).unwrap(),
+        ))
+    });
+    (ptx, prx)
+}