forge-remote 0.1.13 → 0.1.15

package/src/webhook-watcher.js

@@ -0,0 +1,111 @@
+// Forge Remote Relay — Webhook Config Watcher
+// Copyright (c) 2025-2026 Iron Forge Apps
+// Created by Daniel Wendel, CEO/Founder of Iron Forge Apps
+
+import { getDb } from "./firebase.js";
+import * as log from "./logger.js";
+
+/**
+ * In-memory cache of webhook configurations.
+ * Map<webhookId, config>
+ */
+const webhookConfigs = new Map();
+
+/** Firestore unsubscribe function. */
+let unsubscribe = null;
+
+/**
+ * Watch Firestore for webhook config changes and maintain an in-memory cache.
+ *
+ * @param {string} desktopId
+ * @param {string} tunnelBaseUrl — Public tunnel URL to build webhook URLs from
+ */
+export function watchWebhookConfigs(desktopId, tunnelBaseUrl) {
+  const db = getDb();
+  const webhooksRef = db
+    .collection("desktops")
+    .doc(desktopId)
+    .collection("webhooks");
+
+  unsubscribe = webhooksRef.onSnapshot(
+    (snap) => {
+      for (const change of snap.docChanges()) {
+        const id = change.doc.id;
+        const data = change.doc.data();
+
+        if (change.type === "removed") {
+          webhookConfigs.delete(id);
+          log.info(`Webhook removed: ${id}`);
+          continue;
+        }
+
+        // added or modified
+        webhookConfigs.set(id, data);
+
+        if (change.type === "added") {
+          log.info(
+            `Webhook registered: ${id} (source: ${data.source || "custom"}, enabled: ${data.enabled !== false})`,
+          );
+        } else {
+          log.info(`Webhook updated: ${id}`);
+        }
+
+        // If the webhook doc doesn't have a webhookUrl yet, write it back.
+        if (!data.webhookUrl) {
+          const webhookUrl = `${tunnelBaseUrl}/hooks/${id}`;
+          webhooksRef
+            .doc(id)
+            .update({ webhookUrl })
+            .then(() => {
+              log.info(`Wrote webhook URL for ${id}: ${webhookUrl}`);
+            })
+            .catch((err) => {
+              log.error(
+                `Failed to write webhook URL for ${id}: ${err.message}`,
+              );
+            });
+        }
+      }
+    },
+    (err) => {
+      log.error(`Webhook watcher error: ${err.message}`);
+    },
+  );
+
+  log.info(`Watching webhook configs for desktop ${desktopId}`);
+}
+
+/**
+ * Get a webhook config from the in-memory cache.
+ *
+ * @param {string} webhookId
+ * @returns {object|null}
+ */
+export function getWebhookConfig(webhookId) {
+  return webhookConfigs.get(webhookId) || null;
+}
+
+/**
+ * Get the count of enabled webhooks.
+ *
+ * @returns {number}
+ */
+export function getActiveWebhookCount() {
+  let count = 0;
+  for (const config of webhookConfigs.values()) {
+    if (config.enabled !== false) count++;
+  }
+  return count;
+}
+
+/**
+ * Stop watching Firestore for webhook changes.
+ */
+export function stopWatching() {
+  if (unsubscribe) {
+    unsubscribe();
+    unsubscribe = null;
+    log.info("Stopped watching webhook configs");
+  }
+  webhookConfigs.clear();
+}