checkers/src/network/client.rs

use std::net::SocketAddr;

use log::{debug, error, info};
use tokio::sync::mpsc::{self, Receiver, Sender};

use crate::{
    logic::{board::pawn::CapturePath, math::position::Position},
    network::{
        connection::{Connection, ConnectionType},
        message::{
            ClientMessage::{self},
            Message, ServerMessage,
        },
    },
    settings::{client_settings::ClientSettings, general_settings::SettingsLike},
};

pub enum ClientCommands {
    SendCapture(CapturePath),
    SendMove { from: Position, to: Position },
    SendText(String),
    SendReady,
}

pub struct Client {
    conn_reciever_incoming: Option<Receiver<(SocketAddr, Message<ServerMessage>)>>,
    conn_sender_outgoing: Sender<Message<ClientMessage>>,
    settings: ClientSettings,

    update_sender: Sender<ServerMessage>,
    update_receiver: Option<Receiver<ServerMessage>>,
}

impl Client {
    pub async fn new() -> Option<Client> {
        let settings = ClientSettings::new();

        let (conn_sender, conn_reciever) =
            mpsc::channel::<(SocketAddr, Message<ServerMessage>)>(1024);
        let mut conn: Connection<ServerMessage, ClientMessage> =
            Connection::new(ConnectionType::Server, conn_sender);
        let conn_sender_outgoing = conn.get_sender();

        let (update_sender, update_receiver) = mpsc::channel(1024);

        if !conn.connect_to_server(settings.server_url).await {
            error!("Could not connect with the server: {}", settings.server_url);
            return None;
        }

        tokio::spawn(async move {
            conn.start_listening().await;
        });

        Some(Client {
            conn_reciever_incoming: Some(conn_reciever),
            conn_sender_outgoing,

            settings,

            update_sender,
            update_receiver: Some(update_receiver),
        })
    }

    pub fn update(&mut self) {
        let Some(mut conn_reciever_incoming) = self.conn_reciever_incoming.take() else {
            error!("Tried updating, but there's no conn_reciever of incoming messages!");
            return;
        };

        let conn_sender = self.conn_sender_outgoing.clone();

        let settings = self.settings.clone();
        let update_sender = self.update_sender.clone();

        tokio::spawn(async move {
            loop {
                tokio::select! {
                    oldest_msg = conn_reciever_incoming.recv() => {
                        if let Some(oldest_msg) = oldest_msg {
                        let (addr, msg) = oldest_msg;
                        debug!("Got message from: {:?}", addr);

                        match msg.content {
                            ServerMessage::RequestHandshake => {
                                info!("Server requested handshake!");
                                let msg = Message::new(ClientMessage::AnswerHandshake {
                                    player_name: settings.name.clone(),
                                });

                                let _ = conn_sender.send(msg.unwrap()).await;
                            }

                            ServerMessage::AcceptHandshake => {
                                info!("Server accepted us! Yay :D");
                            }

                            ServerMessage::DeclineHandshake { reason } => {
                                error!("Server declined connection: {}", reason);
                                break;
                            }

                            _ => { // Pass all messages that require player intervention to the
                                   // local client
                                let _ = update_sender.send(msg.content).await;
                            }
                            }
                        } else {
                            error!("Connection broken!");
                            break;
                        }
                    }
                }
            }
        });
    }

    async fn send_message(&mut self, msg: Result<Message<ClientMessage>, postcard::Error>) {
        debug!("Sending message to the server!");
        let _ = match msg {
            Err(e) => {
                error!("There was an error while creating the message: {}", e);
                return;
            }
            Ok(msg) => {
                if let Err(e) = self.conn_sender_outgoing.send(msg).await {
                    error!("Error while sending a message to connection thread: {}", e);
                    return;
                }
            }
        };
    }

    pub async fn send_capture(&mut self, capture_path: CapturePath) {
        debug!("Requesting sending capture: {:?}", capture_path);

        let msg = Message::new(ClientMessage::RequestCapture { capture_path });
        self.send_message(msg).await;
    }

    pub async fn send_move(&mut self, from: Position, to: Position) {
        debug!("Requested sending move from: {} to: {}", from, to);
        let msg = Message::new(ClientMessage::RequestMove { from, to });
        self.send_message(msg).await;
    }

    pub async fn send_text_message(&mut self, content: String) {
        debug!("Sending text message with content: [{}]", content);
        let msg = Message::new(ClientMessage::TextMessage(content));
        self.send_message(msg).await;
    }

    pub async fn signal_readiness(&mut self) {
        debug!("Signaling readiness to the server...");
        let msg = Message::new(ClientMessage::SignalReadiness);
        self.send_message(msg).await;
    }

    pub fn get_update_receiver(&mut self) -> Option<Receiver<ServerMessage>> {
        self.update_receiver.take()
    }
}