kagura 0.8.2

Frontend frame-work for wasm on Rust.
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use crate::dom::component::Messenger;
use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;
use wasm_bindgen::prelude::*;
use wasm_bindgen::JsCast;

thread_local!(static WEB_SOCKETS: RefCell<HashMap<&'static str, Rc<web_sys::WebSocket>>> = RefCell::new(HashMap::new()));

fn web_socket(url: &'static str) -> Rc<web_sys::WebSocket> {
    WEB_SOCKETS.with(|web_sockets| {
        if let Some(ws) = web_sockets.borrow().get(url) {
            Rc::clone(ws)
        } else {
            let ws = Rc::new(web_sys::WebSocket::new(url).unwrap());
            web_sockets.borrow_mut().insert(url, Rc::clone(&ws));
            ws
        }
    })
}

pub fn on_close<Msg>(
    url: &'static str,
    mut msg_gen: impl FnMut() -> Msg + 'static,
) -> Box<dyn FnOnce(Messenger<Msg>)>
where
    Msg: 'static,
{
    Box::new(move |mut messenger: Messenger<Msg>| {
        let socket = web_socket(url);
        let a = Closure::wrap(Box::new(move || {
            messenger(msg_gen());
        }) as Box<dyn FnMut()>);
        socket.set_onclose(Some(a.as_ref().unchecked_ref()));
        a.forget();
    })
}

pub fn on_error<Msg>(
    url: &'static str,
    mut msg_gen: impl FnMut() -> Msg + 'static,
) -> Box<dyn FnOnce(Messenger<Msg>)>
where
    Msg: 'static,
{
    Box::new(move |mut messenger: Messenger<Msg>| {
        let socket = web_socket(url);
        let a = Closure::wrap(Box::new(move || {
            messenger(msg_gen());
        }) as Box<dyn FnMut()>);
        socket.set_onerror(Some(a.as_ref().unchecked_ref()));
        a.forget();
    })
}

pub fn on_message<Msg>(
    url: &'static str,
    mut msg_gen: impl FnMut() -> Msg + 'static,
) -> Box<dyn FnOnce(Messenger<Msg>)>
where
    Msg: 'static,
{
    Box::new(move |mut messenger: Messenger<Msg>| {
        let socket = web_socket(url);
        let a = Closure::wrap(Box::new(move || {
            messenger(msg_gen());
        }) as Box<dyn FnMut()>);
        socket.set_onmessage(Some(a.as_ref().unchecked_ref()));
        a.forget();
    })
}

pub fn on_open<Msg>(
    url: &'static str,
    mut msg_gen: impl FnMut() -> Msg + 'static,
) -> Box<dyn FnOnce(Messenger<Msg>)>
where
    Msg: 'static,
{
    Box::new(move |mut messenger: Messenger<Msg>| {
        let socket = web_socket(url);
        let a = Closure::wrap(Box::new(move || {
            messenger(msg_gen());
        }) as Box<dyn FnMut()>);
        socket.set_onopen(Some(a.as_ref().unchecked_ref()));
        a.forget();
    })
}