0.3.4

kb_app/src/lib.rs

@@ -10,6 +10,7 @@
 #![warn(missing_docs)]
 
 mod splash;
+mod demo_ws;
 
 pub use crate::splash::SplashOrder;
 use tauri::Emitter;
@@ -34,6 +35,7 @@ impl KbWsRuntimeState {
 struct KbAppState {
     config: kb_lib::KbConfig,
     ws_runtime: tokio::sync::Mutex<KbWsRuntimeState>,
+    demo_ws_runtime: std::sync::Arc<tokio::sync::Mutex<crate::demo_ws::KbDemoWsRuntimeState>>,
 }
 
 /// Runs the desktop application.
@@ -73,12 +75,24 @@ pub fn run() {
     let app_state = KbAppState {
         config: config.clone(),
         ws_runtime: tokio::sync::Mutex::new(KbWsRuntimeState::new()),
+        demo_ws_runtime: std::sync::Arc::new(tokio::sync::Mutex::new(
+            crate::demo_ws::KbDemoWsRuntimeState::new(),
+        )),
     };
     let tracing_builder = tauri_plugin_tracing::Builder::new();
     let mut tauri_builder = tauri::Builder::default();
     tauri_builder = tauri_builder.manage(app_state);
-    tauri_builder =
-        tauri_builder.invoke_handler(tauri::generate_handler![start_ws_clients, stop_ws_clients]);
+    tauri_builder = tauri_builder.invoke_handler(tauri::generate_handler![
+        start_ws_clients,
+        stop_ws_clients,
+        crate::demo_ws::open_demo_ws_window,
+        crate::demo_ws::demo_ws_list_endpoints,
+        crate::demo_ws::demo_ws_get_status,
+        crate::demo_ws::demo_ws_connect,
+        crate::demo_ws::demo_ws_disconnect,
+        crate::demo_ws::demo_ws_subscribe,
+        crate::demo_ws::demo_ws_unsubscribe_current
+    ]);
     tauri_builder = tauri_builder.plugin(tracing_builder.build::<tauri::Wry>());
     tauri_builder = tauri_builder.setup(|app| {
         let app_handle = app.handle().clone();