Rune Harlyk
57 lines
1.4 KiB
C++
57 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include <esp_http_server.h>
|
|
#include <map>
|
|
#include <list>
|
|
#include <functional>
|
|
|
|
namespace websocket {
|
|
|
|
struct WebSocketClient {
|
|
int fd;
|
|
uint64_t last_seen;
|
|
};
|
|
|
|
typedef std::function<void(int fd)> ClientCallback;
|
|
typedef std::function<esp_err_t(httpd_req_t *req, httpd_ws_frame_t *frame)> FrameCallback;
|
|
|
|
class WebSocketServer {
|
|
public:
|
|
WebSocketServer();
|
|
~WebSocketServer();
|
|
|
|
void setOpenCallback(ClientCallback callback);
|
|
void setCloseCallback(ClientCallback callback);
|
|
void setFrameCallback(FrameCallback callback);
|
|
|
|
esp_err_t handleWebSocket(httpd_req_t *req);
|
|
|
|
void addClient(int fd);
|
|
void removeClient(int fd);
|
|
WebSocketClient *getClient(int fd);
|
|
|
|
esp_err_t sendText(int fd, const char *data, size_t len);
|
|
esp_err_t sendBinary(int fd, const uint8_t *data, size_t len);
|
|
esp_err_t sendToAll(httpd_ws_type_t type, const uint8_t *data, size_t len);
|
|
|
|
bool hasClients() const { return !_clients.empty(); }
|
|
size_t clientCount() const { return _clients.size(); }
|
|
|
|
public:
|
|
void setServer(httpd_handle_t server) { _server = server; }
|
|
|
|
private:
|
|
std::map<int, WebSocketClient> _clients;
|
|
ClientCallback _onOpen;
|
|
ClientCallback _onClose;
|
|
FrameCallback _onFrame;
|
|
SemaphoreHandle_t _mutex;
|
|
httpd_handle_t _server;
|
|
|
|
friend esp_err_t websocket_handler(httpd_req_t *req);
|
|
};
|
|
|
|
esp_err_t websocket_handler(httpd_req_t *req);
|
|
|
|
} // namespace websocket
|