1 files changed, 92 insertions, 0 deletions

diff --git a/components/wisp_relay/sub_manager.h b/components/wisp_relay/sub_manager.h
new file mode 100644
index 0000000..64afb04
--- /dev/null
+++ b/components/wisp_relay/sub_manager.h
@@ -0,0 +1,92 @@
+#ifndef SUB_MANAGER_H
+#define SUB_MANAGER_H
+
+#include <stdbool.h>
+#include <stdint.h>
+#include "esp_err.h"
+#include "freertos/FreeRTOS.h"
+#include "freertos/semphr.h"
+#include "relay_types.h"
+
+#define SUB_MAX_TOTAL         64
+#define SUB_MAX_PER_CONN      8
+#define SUB_MAX_FILTERS       4
+#define SUB_MAX_ID_LEN        64
+
+#define SUB_MAX_FILTER_IDS    20
+#define SUB_MAX_FILTER_AUTHORS 20
+#define SUB_MAX_FILTER_KINDS  20
+#define SUB_MAX_FILTER_ETAGS  20
+#define SUB_MAX_FILTER_PTAGS  20
+
+typedef enum {
+    SUB_OK = 0,
+    SUB_ERR_INVALID,
+    SUB_ERR_TOO_MANY_FILTERS,
+    SUB_ERR_MEMORY,
+    SUB_ERR_NOT_FOUND,
+} sub_error_t;
+
+typedef struct {
+    char *ids[SUB_MAX_FILTER_IDS];
+    size_t ids_count;
+    char *authors[SUB_MAX_FILTER_AUTHORS];
+    size_t authors_count;
+    int32_t kinds[SUB_MAX_FILTER_KINDS];
+    size_t kinds_count;
+    char *e_tags[SUB_MAX_FILTER_ETAGS];
+    size_t e_tags_count;
+    char *p_tags[SUB_MAX_FILTER_PTAGS];
+    size_t p_tags_count;
+    int64_t since;
+    int64_t until;
+    int limit;
+} sub_filter_t;
+
+typedef struct {
+    char sub_id[SUB_MAX_ID_LEN + 1];
+    int conn_fd;
+    sub_filter_t filters[SUB_MAX_FILTERS];
+    uint8_t filter_count;
+    uint16_t events_sent;
+    bool active;
+} subscription_t;
+
+typedef struct sub_manager {
+    subscription_t subs[SUB_MAX_TOTAL];
+    SemaphoreHandle_t lock;
+    uint16_t active_count;
+} sub_manager_t;
+
+typedef struct {
+    int conn_fd;
+    char sub_id[SUB_MAX_ID_LEN + 1];
+} sub_match_entry_t;
+
+typedef struct {
+    sub_match_entry_t matches[SUB_MAX_TOTAL];
+    uint8_t count;
+} sub_match_result_t;
+
+esp_err_t sub_manager_init(sub_manager_t *mgr);
+void sub_manager_destroy(sub_manager_t *mgr);
+
+sub_error_t sub_manager_add(sub_manager_t *mgr, int conn_fd,
+                            const char *sub_id,
+                            const sub_filter_t *filters,
+                            size_t filter_count);
+
+sub_error_t sub_manager_remove(sub_manager_t *mgr, int conn_fd,
+                               const char *sub_id);
+
+void sub_manager_remove_all(sub_manager_t *mgr, int conn_fd);
+
+void sub_manager_match_json(sub_manager_t *mgr, const char *event_json,
+                            size_t event_len, int event_kind,
+                            const char *event_pubkey_hex,
+                            uint64_t event_created_at,
+                            sub_match_result_t *result);
+
+uint8_t sub_manager_count(sub_manager_t *mgr, int conn_fd);
+
+#endif