1 files changed, 49 insertions, 0 deletions
diff --git a/src/server/interaction/peer.rs b/src/server/interaction/peer.rs
new file mode 100644
index 0000000..c0f5223
--- /dev/null
+++ b/src/server/interaction/peer.rs
@@ -0,0 +1,49 @@
+// Copyleft 2021-2021 Whirlsplash
+// SPDX-License-Identifier: GPL-3.0-only
+
+use std::sync::Arc;
+
+use tokio::{
+  net::TcpStream,
+  sync::{mpsc, Mutex},
+};
+use tokio_util::codec::{BytesCodec, Framed};
+
+use crate::server::{interaction::shared::Shared, types::Rx};
+
+pub struct Peer {
+  pub bytes: Framed<TcpStream, BytesCodec>,
+  pub rx:    Rx,
+}
+impl Peer {
+  pub async fn new(
+    state: Arc<Mutex<Shared>>,
+    bytes: Framed<TcpStream, BytesCodec>,
+    username: String,
+  ) -> std::io::Result<Peer> {
+    let (tx, rx) = mpsc::unbounded_channel();
+    state.lock().await.peers.insert(username, tx);
+
+    Ok(Peer {
+      bytes,
+      rx,
+    })
+  }
+
+  pub async fn _change_username(
+    self,
+    state: Arc<Mutex<Shared>>,
+    username: &str,
+    new_username: &str,
+  ) {
+    // Remove peer from peers
+    {
+      state.lock().await.peers.remove(username);
+    }
+
+    // Add the peer back with the new username
+    Self::new(state, self.bytes, new_username.to_string())
+      .await
+      .unwrap();
+  }
+}