CSocketIO.h

#pragma once
 
#include "pch.h"
 
#include "HttpRequest/IHttpRequest.h"
#include "SocketIO/IWebsocket.h"
#include <functional>
 
namespace Skylicht
{
    namespace Network
    {
        class CSocketIO
        {
        public:
            enum EConnectState
            {
                None,
                RequestSocketIO,
                UpdateSocket,
                Closed
            };
 
        protected:
            std::string m_url;
            std::string m_wsURL;
            std::string m_wsOriginURL;
 
            IHttpRequest* m_httpRequest;
 
            std::queue<std::string> m_sendData;
            float m_sendTimeout;
 
            EConnectState m_state;
 
            char* m_bufferData;
 
            IWebsocket* m_ws;
 
            bool m_connected;
 
            std::string m_socketID;
 
        public:
 
            std::function<void()> OnConnected;
            std::function<void()> OnDisconnected;
            std::function<void()> OnConnectFailed;
            std::function<void(const std::string&)> OnMessage;
            std::function<void(const std::string&, int)> OnMessageAsk;
 
        public:
            CSocketIO(const char* url);
 
            virtual ~CSocketIO();
 
            void update();
 
            void forceSend();
 
            void init();
 
            bool startServices();
 
            void setWSURL(const char* ws)
            {
                m_wsURL = ws;
            }
 
            void setOriginURL(const char* ws)
            {
                m_wsOriginURL = ws;
            }
 
            void setSocketID(const char* id)
            {
                m_socketID = id;
            }
 
            const char* getSocketID()
            {
                return m_socketID.c_str();
            }
 
            void sendMessage(const char* message);
 
            void emit(const char* type, bool ack, int askID = 1);
            void emit(const char* type, const char* param, const char* value, bool ack, int askID = 1);
            void emit(const char* type, const char* param, const std::string& value, bool ack, int askID = 1);
            void emit(const char* type, std::map<std::string, std::string>& params, bool ack, int askID = 1);
 
            bool isConnected();
 
            void onConnected();
            void onDisconnected();
            void onMessage(const std::string& msg);
            void onMessageAsk(const std::string& msg, int id);
 
            void updateRequest();
 
            void close();
 
            std::string toStringParam(const char* s);
            std::string toStringParam(const std::string& s);
 
            inline IWebsocket* getWebSocket()
            {
                return m_ws;
            }
        };
    }
}