wasm_extra 0.1.6

Extra utilities for WASM.
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

use wasm_bindgen::prelude::*;
use web_sys::ReadableStream;

#[wasm_bindgen]
#[cfg(all(
    feature = "ReadableStreamDefaultReader",
    feature = "ReadableStreamDefaultController"
))]
extern "C" {
    #[wasm_bindgen(js_name = ReadableStream, extends = ReadableStream)]
    type ReadableStreamClass;

    #[wasm_bindgen(constructor, js_class = "ReadableStream")]
    fn new(source: &JsValue) -> ReadableStreamClass;
}

pub trait ReadableStreamExt {
    #[cfg(all(
        feature = "ReadableStreamDefaultReader",
        feature = "ReadableStreamDefaultController"
    ))]
    fn from_u8_vec(vec: Vec<u8>) -> Result<web_sys::ReadableStream, JsValue>;
}

impl ReadableStreamExt for web_sys::ReadableStream {
    #[cfg(all(
        feature = "ReadableStreamDefaultReader",
        feature = "ReadableStreamDefaultController"
    ))]
    fn from_u8_vec(vec: Vec<u8>) -> Result<web_sys::ReadableStream, JsValue> {
        let source = js_sys::Object::new();
        {
            let f = Closure::<dyn Fn(JsValue)>::new(move |controller: JsValue| {
                let controller = controller
                    .dyn_into::<web_sys::ReadableStreamDefaultController>()
                    .unwrap();
                let arr = js_sys::Uint8Array::from(vec.as_slice());
                controller.enqueue_with_chunk(&arr).unwrap();
                controller.close().unwrap();
            });
            js_sys::Reflect::set(
                &source,
                &JsValue::from_str("pull"),
                f.as_ref().unchecked_ref(),
            )
            .unwrap();

            f.forget();
        }
        Ok(ReadableStreamClass::new(&source).into())
    }
}