From 81f259240d157ddcb0cb692826855d7505f4a22f Mon Sep 17 00:00:00 2001
From: Patrick
Date: Tue, 3 Oct 2023 20:38:21 +0200
Subject: Why didnt I commit these

---
 src/ws.h | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

(limited to 'src/ws.h')

diff --git a/src/ws.h b/src/ws.h
index 4bebc1f..d3acb27 100644
--- a/src/ws.h
+++ b/src/ws.h
@@ -11,13 +11,16 @@ namespace ws
   mg_mgr mgr;
   mg_connection *c = nullptr;
 
-  bool done = false;
+  bool isConnected = false;
 
   std::function<void()> onConnect;
+  std::function<void()> onClose;
+  std::function<void()> onError;
 
   static void cb(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {
     if (ev == MG_EV_WS_OPEN) {
       puts("Open");
+      isConnected = true;
     } else if (ev == MG_EV_WS_MSG) {
       struct mg_ws_message *wm = (struct mg_ws_message *) ev_data;
       printf("Msg: %.*s\n", (int)wm->data.len, wm->data.ptr);
@@ -45,11 +48,15 @@ namespace ws
 
     if (ev == MG_EV_ERROR) {
       printf("Error: %s\n", (char*)ev_data);
-      done = true;
+      isConnected = false;
+      if (onError)
+          onError();
     }
     if (ev == MG_EV_CLOSE) {
       puts("Close");
-      done = true;
+      isConnected = false;
+      if (onClose)
+        onClose();
     }
   }
 
@@ -65,7 +72,7 @@ namespace ws
 
   void update()
   {
-    if (c && !done)
+    if (c)
       mg_mgr_poll(&mgr, 10);
   }
 
-- 
cgit v1.2.3