1 files changed, 52 insertions, 0 deletions
diff --git a/light-client/src/network.rs b/light-client/src/network.rs
new file mode 100644
index 00000000..dc6e894f
--- /dev/null
+++ b/light-client/src/network.rs
@@ -0,0 +1,52 @@
+use hurrycurry_protocol::{PacketC, PacketS, BINCODE_CONFIG};
+use log::warn;
+use std::{collections::VecDeque, net::TcpStream};
+use tungstenite::{stream::MaybeTlsStream, Message, WebSocket};
+
+pub struct Network {
+    sock: WebSocket<MaybeTlsStream<TcpStream>>,
+    queue_in: VecDeque<PacketC>,
+    queue_out: VecDeque<PacketS>,
+}
+
+impl Network {
+    pub fn connect(addr: &str) -> Self {
+        let (sock, _resp) = tungstenite::connect(addr).unwrap();
+        Self {
+            sock,
+            queue_in: VecDeque::new(),
+            queue_out: VecDeque::new(),
+        }
+    }
+    pub fn poll(&mut self) {
+        self.queue_in.extend(match self.sock.read() {
+            Ok(Message::Text(packet)) => match serde_json::from_str(&packet) {
+                Ok(p) => Some(p),
+                Err(e) => {
+                    warn!("invalid json packet: {e:?}");
+                    None
+                }
+            },
+            Ok(Message::Binary(packet)) => {
+                match bincode::decode_from_slice(&packet, BINCODE_CONFIG) {
+                    Ok((p, _)) => Some(p),
+                    Err(e) => {
+                        warn!("invalid bincode packet: {e:?}");
+                        None
+                    }
+                }
+            }
+            Ok(_) => None,
+            Err(e) => {
+                warn!("{e:?}");
+                None
+            }
+        });
+
+        for packet in self.queue_out.drain(..) {
+            self.sock
+                .write(Message::Text(serde_json::to_string(&packet).unwrap()))
+                .unwrap();
+        }
+    }
+}