use futures_util::{SinkExt, StreamExt}; use salvo::extra::ws::{Message, WebSocket}; use crate::{LiveViewError, LiveViewSocket}; /// Convert a salvo websocket into a LiveViewSocket /// /// This is required to launch a LiveView app using the warp web framework pub fn salvo_socket(ws: WebSocket) -> impl LiveViewSocket { ws.map(transform_rx) .with(transform_tx) .sink_map_err(|_| LiveViewError::SendingFailed) } fn transform_rx(message: Result) -> Result { let as_bytes = message.map_err(|_| LiveViewError::SendingFailed)?; let msg = String::from_utf8(as_bytes.into_bytes()).map_err(|_| LiveViewError::SendingFailed)?; Ok(msg) } async fn transform_tx(message: String) -> Result { Ok(Message::text(message)) }