Rune Harlyk
47 lines
1.5 KiB
C++
47 lines
1.5 KiB
C++
#ifndef Socket_h
|
|
#define Socket_h
|
|
|
|
#include <PsychicHttp.h>
|
|
#include <template/stateful_service.h>
|
|
#include <list>
|
|
#include <map>
|
|
#include <vector>
|
|
|
|
enum message_type_t { CONNECT = 0, DISCONNECT = 1, EVENT = 2, PING = 3, PONG = 4, BINARY_EVENT = 5 };
|
|
|
|
typedef std::function<void(JsonVariant &root, int originId)> EventCallback;
|
|
typedef std::function<void(const String &originId, bool sync)> SubscribeCallback;
|
|
|
|
class EventSocket {
|
|
public:
|
|
EventSocket(PsychicHttpServer &server, const char *route = "/api/ws");
|
|
|
|
void begin();
|
|
|
|
bool hasSubscribers(const char *event);
|
|
|
|
void onEvent(String event, EventCallback callback);
|
|
|
|
void onSubscribe(String event, SubscribeCallback callback);
|
|
|
|
void emit(const char *event, JsonVariant &payload, const char *originId = "", bool onlyToSameOrigin = false);
|
|
|
|
private:
|
|
PsychicWebSocketHandler _socket;
|
|
PsychicHttpServer &_server;
|
|
const char *_route;
|
|
|
|
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, JsonVariant &jsonObject, int originId);
|
|
void send(PsychicWebSocketClient *client, const char *data, size_t len);
|
|
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
|