Update docs and tests for close event

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "wasm-sockets"
-version = "0.2.2"
+version = "0.3.0"
 authors = ["scratchyone <[email protected]>"]
 edition = "2018"
 license = "MIT"
@@ -14,12 +14,12 @@ description = "A WASM-only websocket library"
 repository = "https://github.com/scratchyone/wasm-sockets"
 readme = "Readme.md"
 
-[lib]
-crate-type = ["cdylib", "rlib"]
-
 [package.metadata.docs.rs]
 targets = ["wasm32-unknown-unknown"]
 
+[lib]
+crate-type = ["cdylib", "rlib"]
+
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 

Readme.md

@@ -29,7 +29,7 @@ fn main() -> Result<(), WebSocketError> {
         client.send_string("Hello, World!").unwrap();
         client.send_binary(vec![20]).unwrap();
     })));
-    client.set_on_close(Some(Box::new(|| {
+    client.set_on_close(Some(Box::new(|_evt| {
         info!("Connection closed");
     })));
     client.set_on_message(Some(Box::new(

examples/event.rs

@@ -21,7 +21,7 @@ fn main() -> Result<(), WebSocketError> {
         client.send_string("Hello, World!").unwrap();
         client.send_binary(vec![20]).unwrap();
     })));
-    client.set_on_close(Some(Box::new(|| {
+    client.set_on_close(Some(Box::new(|_evt| {
         info!("Connection closed");
     })));
     client.set_on_message(Some(Box::new(

src/lib.rs

@@ -24,7 +24,7 @@
 //!         client.send_string("Hello, World!").unwrap();
 //!         client.send_binary(vec![20]).unwrap();
 //!     })));
-//!     client.set_on_close(Some(Box::new(|| {
+//!     client.set_on_close(Some(Box::new(|_evt| {
 //!         info!("Connection closed");
 //!     })));
 //!     client.set_on_message(Some(Box::new(
@@ -203,6 +203,24 @@ impl PollingClient {
     pub fn send_binary(&self, message: Vec<u8>) -> Result<(), JsValue> {
         self.event_client.send_binary(message)
     }
+
+    /// Close the connection
+    /// ```
+    /// client.close()?;
+    /// ```
+    pub fn close(&self) -> Result<(), JsValue> {
+        self.event_client.close()
+    }
+    /// Close the connection with a custom close code and, optionally, a reason string
+    ///
+    /// The reason string must be at most 123 bytes long.
+    ///
+    /// ```
+    /// client.close_with(1001, Some("going away"))?;
+    /// ```
+    pub fn close_with(&self, code: u16, reason: Option<&str>) -> Result<(), JsValue> {
+        self.event_client.close_with(code, reason)
+    }
 }
 
 #[derive(Debug, Clone, Error)]
@@ -412,7 +430,7 @@ impl EventClient {
     /// This will overwrite the previous handler.
     /// You can set [None](std::option) to disable the on_close handler.
     /// ```
-    /// client.set_on_close(Some(Box::new(|| {
+    /// client.set_on_close(Some(Box::new(|_evt| {
     ///     info!("Closed");
     /// })));
     /// ```
@@ -436,4 +454,28 @@ impl EventClient {
             .borrow()
             .send_with_u8_array(message.as_slice())
     }
+
+    /// Close the connection
+    /// ```
+    /// client.close()?;
+    /// ```
+    pub fn close(&self) -> Result<(), JsValue> {
+        self.connection.borrow().close()
+    }
+    /// Close the connection with a custom close code and, optionally, a reason string
+    ///
+    /// The reason string must be at most 123 bytes long.
+    ///
+    /// ```
+    /// client.close_with(1001, Some("going away"))?;
+    /// ```
+    pub fn close_with(&self, code: u16, reason: Option<&str>) -> Result<(), JsValue> {
+        match reason {
+            Some(reason) => self
+                .connection
+                .borrow()
+                .close_with_code_and_reason(code, reason),
+            None => self.connection.borrow().close_with_code(code),
+        }
+    }
 }

src/tests.rs

@@ -28,7 +28,7 @@ fn event() {
         client.send_string("Hello, World!").unwrap();
         client.send_binary(vec![20]).unwrap();
     })));
-    client.set_on_close(Some(Box::new(|| {
+    client.set_on_close(Some(Box::new(|_evt| {
         info!("Connection closed");
     })));
     client.set_on_message(Some(Box::new(