servers/src/tcp/client.rs

#![allow(clippy::unused_io_amount)]

/// Max size of a TCP packet
pub const MAX_PACKET_LEN: usize = 65536;

use std::{
    io::{self, Read, Write},
    net::TcpStream,
};

/// TCP Client
pub struct Client {
    /// TCP stream of this client
    pub stream: TcpStream,
}

impl Client {
    /// Create new Client
    pub fn new(stream: TcpStream) -> Self {
        Self { stream }
    }

    /// Read message/buffer from client
    pub fn read(&mut self) -> anyhow::Result<String> {
        // allocate an empty buffer
        let mut buf = [0; MAX_PACKET_LEN];

        // read buffer from stream
        self.stream.read(&mut buf)?;

        // encode &[u8] to a String and delete null bytes (empty `\0` bytes)
        let decoded = String::from_utf8(buf.to_vec())?.replace('\0', "");

        Ok(decoded)
    }

    /// Send message to client
    pub fn send(&mut self, content: &str) -> io::Result<()> {
        // add a new line at the end of the content
        let content = format!("{content}\n\r");

        // send message
        self.stream.write_all(content.as_bytes())
    }
}