cards/client/src/components/websocket.rs

use std::rc::Rc;

use leptos::*;
use leptos_use::{core::ConnectionReadyState, use_websocket, UseWebsocketReturn};
use lib::models::*;
use serde_json::from_str;

#[derive(Clone)]
pub struct WebSocketContext {
    pub ready_state: Signal<ConnectionReadyState>,
    // pub message: Signal<Option<String>>,
    pub send: Rc<dyn Fn(&str)>,
    pub open: Rc<dyn Fn()>,
    pub close: Rc<dyn Fn()>,
}

impl WebSocketContext {
    pub fn new(
        ready_state: Signal<ConnectionReadyState>,
        // message: Signal<Option<String>>,
        send: Rc<dyn Fn(&str)>,
        open: Rc<dyn Fn()>,
        close: Rc<dyn Fn()>,
    ) -> Self {
        Self {
            ready_state,
            // message,
            send,
            open,
            close,
        }
    }

    #[inline(always)]
    pub fn send(&self, message: &str) {
        (self.send)(message)
    }

    #[inline(always)]
    pub fn open(&self) {
        (self.open)()
    }

    #[inline(always)]
    pub fn close(&self) {
        (self.close)()
    }

}

#[component]
pub fn Websocket() -> impl IntoView {
    let UseWebsocketReturn {
        ready_state,
        message,
        send,
        open,
        close,
        ..
    } = use_websocket("ws://0.0.0.0:3030/websocket");

    provide_context(WebSocketContext::new(
        ready_state,
        // message,
        Rc::new(send.clone()),
        Rc::new(open.clone()),
        Rc::new(close.clone()),
    ));

    // Contexts for message handler
    let (state_summary, set_state_summary) =
        create_signal::<Option<ServerStateSummary>>(Option::None);
    let (game_object, set_game_object) = create_signal::<Option<Game>>(Option::None);
    let (chat_message, set_chat_message) = create_signal::<Option<ChatMessage>>(Option::None);

    provide_context::<ReadSignal<Option<Game>>>(game_object);
    provide_context::<ReadSignal<Option<ChatMessage>>>(chat_message);
    provide_context::<ReadSignal<Option<ServerStateSummary>>>(state_summary);

    // Message handler
    create_effect(move |_| {
        message.with(move |message_raw| {
            // Send all messages as strings into chat box

            if let Some(message) = message_raw {
                if let Ok(game) = from_str::<Game>(message) {
                    set_game_object(Some(game));
                } else if let Ok(state_summary) =
                    from_str::<ServerStateSummary>(message)
                {
                    set_state_summary(Some(state_summary));
                } else if let Ok(chat_message) = from_str::<ChatMessage>(message) {
                    set_chat_message(Some(chat_message));
                } else {
                    logging::log!("Unhandled message: {}", message);
                }
            }
        })
    });
}