jsglue 0.1.0

A Rust framework for building frontends with JS/TS!
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use futures_util::{stream::SplitSink, Sink, SinkExt};
use std::{fmt::Debug, net::SocketAddr, ops::ControlFlow};
use tokio_tungstenite::tungstenite::Message;

pub async fn process_message<T, E>(
    socket: &mut SplitSink<T, Message>,
    who: SocketAddr,
    message: Message,
) -> ControlFlow<(), ()>
where
    T: Sink<Message, Error = E>,
    E: Debug,
{
    match message {
        Message::Text(txt) => {
            socket.send(Message::Text(txt)).await.unwrap();
        }

        Message::Binary(bin) => {
            socket.send(Message::Binary(bin)).await.unwrap();
        }

        Message::Close(c) => {
            if let Some(cf) = c.clone() {
                info!(
                    "{} sent close with code {} and reason `{}`.",
                    who, cf.code, cf.reason
                );
            } else {
                info!("{who} somehow sent close message without CloseFrame!");
            }

            socket.send(Message::Close(c)).await.unwrap();
            socket.close().await.unwrap();

            return ControlFlow::Break(());
        }

        Message::Pong(v) => {
            socket.send(Message::Pong(v)).await.unwrap();
        }

        Message::Ping(v) => {
            socket.send(Message::Ping(v)).await.unwrap();
        }

        Message::Frame(_) => unreachable!(),
    }

    ControlFlow::Continue(())
}