lan-mouse/src/client.rs

use std::{net::SocketAddr, collections::{HashSet, hash_set::Iter}, fmt::Display, time::{Instant, Duration}, iter::Cloned};

use serde::{Serialize, Deserialize};

#[derive(Debug, Eq, Hash, PartialEq, Clone, Copy, Serialize, Deserialize)]
pub enum Position {
    Left,
    Right,
    Top,
    Bottom,
}

impl Position {
    pub fn opposite(&self) -> Self {
        match self {
            Position::Left => Self::Right,
            Position::Right => Self::Left,
            Position::Top => Self::Bottom,
            Position::Bottom => Self::Top,
        }
    }
}

impl Display for Position {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", match self {
            Position::Left => "left",
            Position::Right => "right",
            Position::Top => "top",
            Position::Bottom => "bottom",
        })
    }
}

#[derive(Debug, Eq, PartialEq, Clone, Serialize, Deserialize)]
pub struct Client {
    /// handle to refer to the client.
    /// This way any event consumer / producer backend does not
    /// need to know anything about a client other than its handle.
    pub handle: ClientHandle,
    /// `active` address of the client, used to send data to.
    /// This should generally be the socket address where data
    /// was last received from.
    pub active_addr: Option<SocketAddr>,
    /// all socket addresses associated with a particular client
    /// e.g. Laptops usually have at least an ethernet and a wifi port
    /// which have different ip addresses
    pub addrs: HashSet<SocketAddr>,
    /// position of a client on screen
    pub pos: Position,
}

pub enum ClientEvent {
    Create(ClientHandle, Position),
    Destroy(ClientHandle),
    UpdatePos(ClientHandle, Position),
    AddAddr(ClientHandle, SocketAddr),
    RemoveAddr(ClientHandle, SocketAddr),
}

pub type ClientHandle = u32;

pub struct ClientManager {
    /// probably not beneficial to use a hashmap here
    clients: Vec<Client>,
    last_ping: Vec<(ClientHandle, Option<Instant>)>,
    last_seen: Vec<(ClientHandle, Option<Instant>)>,
    last_replied:  Vec<(ClientHandle, Option<Instant>)>,
    next_client_id: u32,
}

impl ClientManager {
    pub fn new() -> Self {
        Self {
            clients: vec![],
            next_client_id: 0,
            last_ping: vec![],
            last_seen: vec![],
            last_replied: vec![],
        }
    }

    /// add a new client to this manager
    pub fn add_client(&mut self, addrs: HashSet<SocketAddr>, pos: Position) -> ClientHandle {
        let handle = self.next_id();
        // we dont know, which IP is initially active
        let active_addr = None;

        // store the client
        let client = Client { handle, active_addr, addrs, pos };
        self.clients.push(client);
        self.last_ping.push((handle, None));
        self.last_seen.push((handle, None));
        self.last_replied.push((handle, None));
        handle
    }

    /// add a socket address to the given client
    pub fn add_addr(&mut self, client: ClientHandle, addr: SocketAddr) {
        if let Some(client) = self.get_mut(client) {
            client.addrs.insert(addr);
        }
    }

    /// remove socket address from the given client
    pub fn remove_addr(&mut self, client: ClientHandle, addr: SocketAddr) {
        if let Some(client) = self.get_mut(client) {
            client.addrs.remove(&addr);
        }
    }

    pub fn set_default_addr(&mut self, client: ClientHandle, addr: SocketAddr) {
        if let Some(client) = self.get_mut(client) {
            client.active_addr = Some(addr)
        }
    }

    /// update the position of a client
    pub fn update_pos(&mut self, client: ClientHandle, pos: Position) {
        if let Some(client) = self.get_mut(client) {
            client.pos = pos;
        }
    }

    pub fn get_active_addr(&self, client: ClientHandle) -> Option<SocketAddr> {
        self.get(client)?.active_addr
    }

    pub fn get_addrs(&self, client: ClientHandle) -> Option<Cloned<Iter<'_, SocketAddr>>> {
        Some(self.get(client)?.addrs.iter().cloned())
    }

    pub fn last_ping(&self, client: ClientHandle) -> Option<Duration> {
        let last_ping = self.last_ping
            .iter()
            .find(|(c,_)| *c == client)?.1;
        last_ping.map(|p| p.elapsed())
    }

    pub fn last_seen(&self, client: ClientHandle) -> Option<Duration> {
        let last_seen = self.last_seen
            .iter()
            .find(|(c, _)| *c == client)?.1;
        last_seen.map(|t| t.elapsed())
    }

    pub fn last_replied(&self, client: ClientHandle) -> Option<Duration> {
        let last_replied = self.last_replied
            .iter()
            .find(|(c, _)| *c == client)?.1;
        last_replied.map(|t| t.elapsed())
    }

    pub fn reset_last_ping(&mut self, client: ClientHandle) {
        if let Some(c) = self.last_ping
            .iter_mut()
            .find(|(c, _)| *c == client) {
                c.1 = Some(Instant::now());
            }
    }

    pub fn reset_last_seen(&mut self, client: ClientHandle) {
        if let Some(c) = self.last_seen
            .iter_mut()
            .find(|(c, _)| *c == client) {
                c.1 = Some(Instant::now());
            }
    }

    pub fn reset_last_replied(&mut self, client: ClientHandle) {
        if let Some(c) = self.last_replied
            .iter_mut()
            .find(|(c, _)| *c == client) {
                c.1 = Some(Instant::now());
            }
    }

    pub fn get_client(&self, addr: SocketAddr) -> Option<ClientHandle> {
        self.clients
            .iter()
            .find(|c| c.addrs.contains(&addr))
            .map(|c| c.handle)
    }

    pub fn remove_client(&mut self, client: ClientHandle) {
        if let Some(i) = self.clients.iter().position(|c| c.handle == client) {
            self.clients.remove(i);
            self.last_ping.remove(i);
            self.last_seen.remove(i);
            self.last_replied.remove(i);
        }
    }

    fn next_id(&mut self) -> ClientHandle {
        let handle = self.next_client_id;
        self.next_client_id += 1;
        handle
    }

    fn get<'a>(&'a self, client: ClientHandle) -> Option<&'a Client> {
        self.clients
            .iter()
            .find(|c| c.handle == client)
    }

    fn get_mut<'a>(&'a mut self, client: ClientHandle) -> Option<&'a mut Client> {
        self.clients
            .iter_mut()
            .find(|c| c.handle == client)
    }
}