From a4c52bedef04cfb927f3d7809680fed0425a5125 Mon Sep 17 00:00:00 2001
From: metamuffin <metamuffin@disroot.org>
Date: Fri, 5 Jul 2024 23:37:39 +0200
Subject: udp transport

---
 src/transport/udp.rs | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 src/transport/udp.rs

(limited to 'src/transport/udp.rs')

diff --git a/src/transport/udp.rs b/src/transport/udp.rs
new file mode 100644
index 0000000..c86ce50
--- /dev/null
+++ b/src/transport/udp.rs
@@ -0,0 +1,30 @@
+use super::Transport;
+use crate::encoding::{request::Request, response::Response};
+use anyhow::Result;
+use log::debug;
+use std::str::FromStr;
+use tokio::net::UdpSocket;
+
+pub struct UdpTransport {
+    sock: UdpSocket,
+}
+
+impl UdpTransport {
+    pub async fn new(sock: UdpSocket) -> Result<Self> {
+        Ok(Self { sock })
+    }
+}
+impl Transport for UdpTransport {
+    async fn recv(&self) -> Result<Response> {
+        let mut buf = [0; 1024];
+        let size = self.sock.recv(&mut buf).await?;
+        let message = String::from_utf8(buf[..size].to_vec())?;
+        debug!("{message}");
+        Response::from_str(message.trim_end())
+    }
+    async fn send(&self, request: Request) -> Result<()> {
+        debug!("-> {request}");
+        self.sock.send(format!("{request}").as_bytes()).await?;
+        Ok(())
+    }
+}
-- 
cgit v1.2.3-70-g09d2