Icons

1 files changed, 10 insertions, 3 deletions

diff --git a/src/ws.h b/src/ws.h
index ecc88b3..058f69b 100644
--- a/src/ws.h
+++ b/src/ws.h
@@ -4,6 +4,7 @@
 using json = nlohmann::json;

 

 #include <string>

+#include <functional>

 

 namespace ws

 {

@@ -12,10 +13,14 @@ namespace ws
 

   bool done = false;

 

+  std::function<void()> onConnect;

+

   static void cb(struct mg_connection *c, int ev, void *ev_data, void *fn_data) {

     if (ev == MG_EV_WS_OPEN) {

+      puts("Open");

     } 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);

       std::string jsonStr(wm->data.ptr, wm->data.len);

       auto msg = json::parse(jsonStr);

       int op = msg["op"].get<int>();

@@ -32,16 +37,18 @@ namespace ws
       }

       else if (op == 2)

       {

-        MessageBoxA(NULL, "hura", "connected", MB_OK);

+        puts("Connected");

+        if (onConnect)

+          onConnect();

       }

     }

 

     if (ev == MG_EV_ERROR) {

-      MessageBoxA(NULL, "", "Error", MB_OK);

+      puts("Error");

       done = true;

     }

     if (ev == MG_EV_CLOSE) {

-      MessageBoxA(NULL, "", "Close", MB_OK);

+      puts("Close");

       done = true;

     }

   }