[SKY-6] Backend: Enable 2FA code detection without TOTP credentials (#4786)

skyvern/forge/sdk/notification/local.py

@@ -0,0 +1,45 @@
+"""In-process notification registry using asyncio queues (single-pod only)."""
+
+import asyncio
+from collections import defaultdict
+
+import structlog
+
+from skyvern.forge.sdk.notification.base import BaseNotificationRegistry
+
+LOG = structlog.get_logger()
+
+
+class LocalNotificationRegistry(BaseNotificationRegistry):
+    """In-process fan-out pub/sub using asyncio queues. Single-pod only."""
+
+    def __init__(self) -> None:
+        self._subscribers: dict[str, list[asyncio.Queue[dict]]] = defaultdict(list)
+
+    def subscribe(self, organization_id: str) -> asyncio.Queue[dict]:
+        queue: asyncio.Queue[dict] = asyncio.Queue()
+        self._subscribers[organization_id].append(queue)
+        LOG.info("Notification subscriber added", organization_id=organization_id)
+        return queue
+
+    def unsubscribe(self, organization_id: str, queue: asyncio.Queue[dict]) -> None:
+        queues = self._subscribers.get(organization_id)
+        if queues:
+            try:
+                queues.remove(queue)
+            except ValueError:
+                pass
+            if not queues:
+                del self._subscribers[organization_id]
+        LOG.info("Notification subscriber removed", organization_id=organization_id)
+
+    def publish(self, organization_id: str, message: dict) -> None:
+        queues = self._subscribers.get(organization_id, [])
+        for queue in queues:
+            try:
+                queue.put_nowait(message)
+            except asyncio.QueueFull:
+                LOG.warning(
+                    "Notification queue full, dropping message",
+                    organization_id=organization_id,
+                )