Rune Harlyk
57 lines
1.7 KiB
C++
57 lines
1.7 KiB
C++
#ifndef Socket_h
|
|
#define Socket_h
|
|
|
|
#include <PsychicHttp.h>
|
|
#include <SecurityManager.h>
|
|
#include <StatefulService.h>
|
|
#include <list>
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
#define EVENT_SERVICE_PATH "/ws/events"
|
|
|
|
enum message_type_t {
|
|
CONNECT = 0,
|
|
DISCONNECT = 1,
|
|
EVENT = 2,
|
|
PING = 3,
|
|
PONG = 4,
|
|
BINARY_EVENT = 5
|
|
};
|
|
|
|
typedef std::function<void(JsonObject &root, int originId)> EventCallback;
|
|
typedef std::function<void(const String &originId, bool sync)> SubscribeCallback;
|
|
|
|
class EventSocket
|
|
{
|
|
public:
|
|
EventSocket(PsychicHttpServer *server, SecurityManager *_securityManager, AuthenticationPredicate authenticationPredicate = AuthenticationPredicates::IS_AUTHENTICATED);
|
|
|
|
void begin();
|
|
|
|
void onEvent(String event, EventCallback callback);
|
|
|
|
void onSubscribe(String event, SubscribeCallback callback);
|
|
|
|
void emit(const char *event, const char *payload, const char *originId = "", bool onlyToSameOrigin = false);
|
|
// if onlyToSameOrigin == true, the message will be sent to the originId only, otherwise it will be broadcasted to all clients except the originId
|
|
|
|
private:
|
|
PsychicHttpServer *_server;
|
|
PsychicWebSocketHandler _socket;
|
|
SecurityManager *_securityManager;
|
|
AuthenticationPredicate _authenticationPredicate;
|
|
|
|
std::map<String, std::list<int>> client_subscriptions;
|
|
std::map<String, std::list<EventCallback>> event_callbacks;
|
|
std::map<String, std::list<SubscribeCallback>> subscribe_callbacks;
|
|
void handleEventCallbacks(String event, JsonObject &jsonObject, int originId);
|
|
void handleSubscribeCallbacks(String event, const String &originId);
|
|
|
|
void onWSOpen(PsychicWebSocketClient *client);
|
|
void onWSClose(PsychicWebSocketClient *client);
|
|
esp_err_t onFrame(PsychicWebSocketRequest *request, httpd_ws_frame *frame);
|
|
};
|
|
|
|
#endif
|