Websocket Support (#159)

src-tauri/yaak-ws/src/lib.rs

@@ -0,0 +1,37 @@
+mod cmd;
+mod connect;
+mod error;
+mod manager;
+mod render;
+
+use crate::cmd::{
+    close, connect, delete_connection, delete_connections, delete_request, list_connections,
+    list_events, list_requests, send, upsert_request,
+};
+use crate::manager::WebsocketManager;
+use tauri::plugin::{Builder, TauriPlugin};
+use tauri::{generate_handler, Manager, Runtime};
+use tokio::sync::Mutex;
+
+pub fn init<R: Runtime>() -> TauriPlugin<R> {
+    Builder::new("yaak-ws")
+        .invoke_handler(generate_handler![
+            close,
+            connect,
+            delete_connection,
+            delete_connections,
+            delete_request,
+            list_connections,
+            list_events,
+            list_requests,
+            send,
+            upsert_request,
+        ])
+        .setup(|app, _api| {
+            let manager = WebsocketManager::new();
+            app.manage(Mutex::new(manager));
+
+            Ok(())
+        })
+        .build()
+}