connect and heartbeat

1 files changed, 30 insertions, 0 deletions

diff --git a/discord/websocket.cpp b/discord/websocket.cpp
new file mode 100644
index 0000000..3590db3
--- /dev/null
+++ b/discord/websocket.cpp
@@ -0,0 +1,30 @@
+#include "websocket.hpp"
+#include <functional>
+#include <nlohmann/json.hpp>
+
+Websocket::Websocket() {}
+
+void Websocket::StartConnection(std::string url) {
+    m_websocket.setUrl(url);
+    m_websocket.setOnMessageCallback(std::bind(&Websocket::OnMessage, this, std::placeholders::_1));
+    m_websocket.start();
+}
+
+void Websocket::SetJSONCallback(JSONCallback_t func) {
+    m_json_callback = func;
+}
+
+void Websocket::Send(const std::string &str) {
+    m_websocket.sendText(str);
+}
+
+void Websocket::OnMessage(const ix::WebSocketMessagePtr &msg) {
+    switch (msg->type) {
+        case ix::WebSocketMessageType::Message:
+            printf("%s\n", msg->str.c_str());
+            auto obj = nlohmann::json::parse(msg->str);
+            if (m_json_callback)
+                m_json_callback(obj);
+            break;
+    }
+}