Add clients

7bbced2d · scratchyone · 7bbced2d · 7bbced2d · 7bbced2d · 7bbced2d
Unverified Commit 7bbced2d authored 4 years ago by scratchyone
--- a/.cargo/config
+++ b/.cargo/config
+[build]
+target = "wasm32-unknown-unknown"
\ No newline at end of file
--- a/.gitignore
+++ b/.gitignore
+/target
+Cargo.lock
--- a/Cargo.toml
+++ b/Cargo.toml
+[package]
+name = "wasm-sockets"
+version = "0.1.0"
+authors = ["scratchyone <scratchywon@gmail.com>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+wasm-bindgen = "0.2"
+js-sys = "0.3"
+log = "0.4.11"
+console_log = "0.2.0"
+console_error_panic_hook = "0.1.6"
+
+[dependencies.web-sys]
+version = "0.3.22"
+features = [
+  "BinaryType",
+  "Blob",
+  "ErrorEvent",
+  "FileReader",
+  "MessageEvent",
+  "ProgressEvent",
+  "WebSocket",
+]
--- a/examples/basic.rs
+++ b/examples/basic.rs
+use console_error_panic_hook;
+use log::{debug, error, info, trace, warn, Level};
+use std::cell::RefCell;
+use std::panic;
+use std::rc::Rc;
+use wasm_bindgen::prelude::*;
+use wasm_sockets;
+
+fn main() -> Result<(), JsValue> {
+    panic::set_hook(Box::new(console_error_panic_hook::hook));
+    console_log::init_with_level(Level::Trace).expect("Failed to enable logging");
+    info!("Creating connection");
+
+    let mut client = wasm_sockets::EventClient::new("ws://localhost:9001")?;
+    client.set_on_error(Some(Box::new(|e| {
+        error!("{:#?}", e);
+    })));
+    client.set_on_connection(Some(Box::new(
+        |c: Rc<RefCell<wasm_sockets::EventClient>>, e| {
+            info!("Connected: {:#?}", e);
+            info!("{:#?}", &c.borrow_mut().status);
+            info!("Sending message...");
+            c.borrow().send_string("test...").unwrap();
+        },
+    )));
+
+    info!("Connection successfully created");
+    Ok(())
+}
--- a/examples/index.html
+++ b/examples/index.html
+<html>
+  <head>
+    <meta charset="UTF-8" />
+    <style>
+      body {
+        background: linear-gradient(
+          135deg,
+          white 0%,
+          white 49%,
+          black 49%,
+          black 51%,
+          white 51%,
+          white 100%
+        );
+        background-repeat: repeat;
+        background-size: 20px 20px;
+        margin: 0px;
+      }
+      canvas {
+        background-color: white;
+      }
+    </style>
+  </head>
+  <script type="module">
+    import init from '../target/basic.js';
+    init();
+  </script>
+</html>
--- a/src/lib.rs
+++ b/src/lib.rs
+use log::{debug, error, info, trace, warn, Level};
+use std::cell::RefCell;
+use std::rc::Rc;
+use std::sync::{Arc, Condvar, Mutex};
+use wasm_bindgen::prelude::*;
+use wasm_bindgen::JsCast;
+use web_sys::{ErrorEvent, MessageEvent, WebSocket};
+
+pub fn start_websocket() -> Result<(), JsValue> {
+    // Connect to an echo server
+    let ws: web_sys::WebSocket = WebSocket::new("ws://localhost:9001")?;
+    // For small binary messages, like CBOR, Arraybuffer is more efficient than Blob handling
+    ws.set_binary_type(web_sys::BinaryType::Arraybuffer);
+    // create callback
+    let cloned_ws = ws.clone();
+    let onmessage_callback = Closure::wrap(Box::new(move |e: MessageEvent| {
+        // Handle difference Text/Binary,...
+        if let Ok(abuf) = e.data().dyn_into::<js_sys::ArrayBuffer>() {
+            info!("message event, received arraybuffer: {:?}", abuf);
+            let array = js_sys::Uint8Array::new(&abuf);
+            let len = array.byte_length() as usize;
+            info!("Arraybuffer received {}bytes: {:?}", len, array.to_vec());
+            // here you can for example use Serde Deserialize decode the message
+            // for demo purposes we switch back to Blob-type and send off another binary message
+            cloned_ws.set_binary_type(web_sys::BinaryType::Blob);
+            match cloned_ws.send_with_u8_array(&vec![5, 6, 7, 8]) {
+                Ok(_) => info!("binary message successfully sent"),
+                Err(err) => info!("error sending message: {:?}", err),
+            }
+        } else if let Ok(blob) = e.data().dyn_into::<web_sys::Blob>() {
+            info!("message event, received blob: {:?}", blob);
+            // better alternative to juggling with FileReader is to use https://crates.io/crates/gloo-file
+            let fr = web_sys::FileReader::new().unwrap();
+            let fr_c = fr.clone();
+            // create onLoadEnd callback
+            let onloadend_cb = Closure::wrap(Box::new(move |_e: web_sys::ProgressEvent| {
+                let array = js_sys::Uint8Array::new(&fr_c.result().unwrap());
+                let len = array.byte_length() as usize;
+                info!("Blob received {}bytes: {:?}", len, array.to_vec());
+                // here you can for example use the received image/png data
+            })
+                as Box<dyn FnMut(web_sys::ProgressEvent)>);
+            fr.set_onloadend(Some(onloadend_cb.as_ref().unchecked_ref()));
+            fr.read_as_array_buffer(&blob).expect("blob not readable");
+            onloadend_cb.forget();
+        } else if let Ok(txt) = e.data().dyn_into::<js_sys::JsString>() {
+            info!("message event, received Text: {:?}", txt);
+        } else {
+            info!("message event, received Unknown: {:?}", e.data());
+        }
+    }) as Box<dyn FnMut(MessageEvent)>);
+    // set message event handler on WebSocket
+    ws.set_onmessage(Some(onmessage_callback.as_ref().unchecked_ref()));
+    // forget the callback to keep it alive
+    onmessage_callback.forget();
+
+    let onerror_callback = Closure::wrap(Box::new(move |e: ErrorEvent| {
+        info!("error event: {:?}", e);
+    }) as Box<dyn FnMut(ErrorEvent)>);
+    ws.set_onerror(Some(onerror_callback.as_ref().unchecked_ref()));
+    onerror_callback.forget();
+
+    let cloned_ws = ws.clone();
+    let onopen_callback = Closure::wrap(Box::new(move |_| {
+        info!("socket opened");
+        match cloned_ws.send_with_str("ping") {
+            Ok(_) => info!("message successfully sent"),
+            Err(err) => info!("error sending message: {:?}", err),
+        }
+        // send off binary message
+        match cloned_ws.send_with_u8_array(&vec![0, 1, 2, 3]) {
+            Ok(_) => info!("binary message successfully sent"),
+            Err(err) => info!("error sending message: {:?}", err),
+        }
+    }) as Box<dyn FnMut(JsValue)>);
+    ws.set_onopen(Some(onopen_callback.as_ref().unchecked_ref()));
+    onopen_callback.forget();
+
+    Ok(())
+}
+#[derive(Debug)]
+pub enum ConnectionStatus {
+    Connecting,
+    Connected,
+    Error(ErrorEvent),
+}
+
+#[derive(Debug)]
+pub enum Message {
+    Text(String),
+}
+pub struct BlockingClient {
+    pub url: String,
+    pub connection: Rc<RefCell<web_sys::WebSocket>>,
+    pub status: Rc<RefCell<ConnectionStatus>>,
+    pub data: Rc<RefCell<Vec<String>>>,
+}
+// TODO: Replace unwraps and JsValue with custom error type
+impl BlockingClient {
+    pub fn connect_with_url(url: &str) -> Result<Self, JsValue> {
+        // Create connection
+        let mut client = EventClient::new(url)?;
+        let data = Rc::new(RefCell::new(vec![]));
+        let data_ref = data.clone();
+        let status = Rc::new(RefCell::new(ConnectionStatus::Connecting));
+        let status_ref = status.clone();
+
+        client.set_on_connection(Some(Box::new(move |c, e| {
+            *status_ref.borrow_mut() = ConnectionStatus::Connected;
+        })));
+
+        let status_ref = status.clone();
+
+        client.set_on_error(Some(Box::new(move |e| {
+            *status_ref.borrow_mut() = ConnectionStatus::Error(e);
+        })));
+
+        Ok(Self {
+            url: url.to_string(),
+            connection: client.connection,
+            status,
+            data,
+        })
+    }
+}
+pub struct EventClient {
+    pub url: Rc<RefCell<String>>,
+    pub connection: Rc<RefCell<web_sys::WebSocket>>,
+    pub status: Rc<RefCell<ConnectionStatus>>,
+    pub on_error: Rc<RefCell<Option<Box<dyn Fn(ErrorEvent) -> ()>>>>,
+    pub on_connection: Rc<RefCell<Option<Box<dyn Fn(Rc<RefCell<EventClient>>, JsValue) -> ()>>>>,
+}
+impl EventClient {
+    pub fn new(url: &str) -> Result<Self, JsValue> {
+        // Create connection
+        let ws: web_sys::WebSocket = WebSocket::new(url)?;
+        // For small binary messages, like CBOR, Arraybuffer is more efficient than Blob handling
+        ws.set_binary_type(web_sys::BinaryType::Arraybuffer);
+
+        let status = Rc::new(RefCell::new(ConnectionStatus::Connecting));
+        let ref_status = status.clone();
+
+        let on_error: Rc<RefCell<Option<Box<dyn Fn(ErrorEvent) -> ()>>>> =
+            Rc::new(RefCell::new(None));
+        let on_error_ref = on_error.clone();
+
+        let onerror_callback = Closure::wrap(Box::new(move |e: ErrorEvent| {
+            *ref_status.borrow_mut() = ConnectionStatus::Error(e.clone());
+            if let Some(f) = &*on_error_ref.borrow() {
+                f.as_ref()(e);
+            }
+        }) as Box<dyn FnMut(ErrorEvent)>);
+        ws.set_onerror(Some(onerror_callback.as_ref().unchecked_ref()));
+        onerror_callback.forget();
+
+        let on_connection: Rc<
+            RefCell<Option<Box<dyn Fn(Rc<RefCell<EventClient>>, JsValue) -> ()>>>,
+        > = Rc::new(RefCell::new(None));
+        let on_connection_ref = on_connection.clone();
+
+        let ref_status = status.clone();
+
+        let connection = Rc::new(RefCell::new(ws));
+        let connection_ref = connection.clone();
+
+        let test = Rc::new(RefCell::new(Self {
+            url: Rc::new(RefCell::new(url.to_string())),
+            connection: connection.clone(),
+            on_error: on_error.clone(),
+            on_connection: on_connection.clone(),
+            status: status.clone(),
+        }));
+
+        let onopen_callback = Closure::wrap(Box::new(move |v| {
+            *ref_status.borrow_mut() = ConnectionStatus::Connected;
+            if let Some(f) = &*on_connection_ref.borrow() {
+                f.as_ref()(test.clone(), v);
+            }
+        }) as Box<dyn FnMut(JsValue)>);
+        connection
+            .borrow_mut()
+            .set_onopen(Some(onopen_callback.as_ref().unchecked_ref()));
+        onopen_callback.forget();
+
+        Ok(Self {
+            url: Rc::new(RefCell::new(url.to_string())),
+            connection,
+            on_error,
+            on_connection,
+            status: status,
+        })
+    }
+
+    pub fn set_on_error(&mut self, f: Option<Box<dyn Fn(ErrorEvent) -> ()>>) {
+        *self.on_error.borrow_mut() = f;
+    }
+    pub fn set_on_connection(
+        &mut self,
+        f: Option<Box<dyn Fn(Rc<RefCell<EventClient>>, JsValue) -> ()>>,
+    ) {
+        *self.on_connection.borrow_mut() = f;
+    }
+
+    pub fn send_string(&self, message: &str) -> Result<(), JsValue> {
+        self.connection.borrow_mut().send_with_str(message)
+    }
+}