1 files changed, 41 insertions, 0 deletions

diff --git a/components/wisp_relay/ws_server.h b/components/wisp_relay/ws_server.h
new file mode 100644
index 0000000..4fe616e
--- /dev/null
+++ b/components/wisp_relay/ws_server.h
@@ -0,0 +1,41 @@
+#ifndef WS_SERVER_H
+#define WS_SERVER_H
+
+#include <stdint.h>
+#include <stdbool.h>
+#include <stddef.h>
+#include "esp_http_server.h"
+#include "freertos/FreeRTOS.h"
+#include "freertos/semphr.h"
+
+#define WS_MAX_CONNECTIONS     8
+#define WS_MAX_FRAME_SIZE      65536
+#define WS_IP_ADDR_MAX_LEN     48
+
+typedef struct {
+    int fd;
+    bool active;
+    uint32_t connected_at;
+    uint32_t last_activity;
+    char remote_ip[WS_IP_ADDR_MAX_LEN];
+} ws_connection_t;
+
+typedef struct {
+    httpd_handle_t server;
+    ws_connection_t connections[WS_MAX_CONNECTIONS];
+    SemaphoreHandle_t lock;
+    uint8_t connection_count;
+} ws_server_t;
+
+typedef void (*ws_message_cb_t)(int fd, const char *data, size_t len);
+typedef void (*ws_disconnect_cb_t)(int fd);
+
+esp_err_t ws_server_init(ws_server_t *server, uint16_t port, ws_message_cb_t on_message);
+void ws_server_set_disconnect_cb(ws_disconnect_cb_t cb);
+void ws_server_stop(ws_server_t *server);
+bool ws_server_is_running(ws_server_t *server);
+esp_err_t ws_server_send(ws_server_t *server, int fd, const char *data, size_t len);
+esp_err_t ws_server_broadcast(ws_server_t *server, const char *data, size_t len);
+void ws_server_close_connection(ws_server_t *server, int fd);
+
+#endif