Upload auto_update_sse.py

auto_update_sse.py

@@ -0,0 +1,55 @@
+"""Auto-update SSE endpoint for VNEWS - pushes updates when new posts/shorts published."""
+import asyncio
+import json
+import time
+from fastapi import Request
+from fastapi.responses import StreamingResponse
+
+# Connected clients queue
+_clients = []
+_lock = asyncio.Lock()
+
+async def _notify_clients(event_type: str, data: dict):
+    """Send notification to all SSE clients."""
+    if not _clients:
+        return
+    msg = f"data: {json.dumps({'type': event_type, 'data': data, 'ts': int(time.time())})}\n\n"
+    async with _lock:
+        dead = []
+        for q in _clients:
+            try:
+                await q.put_nowait(msg)
+            except asyncio.QueueFull:
+                pass
+            except:
+                dead.append(q)
+        for q in dead:
+            if q in _clients:
+                _clients.remove(q)
+
+# Public functions to call from other modules
+notify_new_post = lambda post: asyncio.create_task(_notify_clients("new_post", post)) if post else None
+notify_new_short = lambda post: asyncio.create_task(_notify_clients("new_short", post)) if post else None
+
+async def sse_events(request: Request):
+    """SSE endpoint for real-time updates on homepage."""
+    q = asyncio.Queue(maxsize=10)
+    _clients.append(q)
+    
+    async def event_generator():
+        try:
+            # Send initial connection message
+            yield "data: {\"type\":\"connected\",\"ts\":null}\n\n"
+            while not await request.is_disconnected():
+                try:
+                    msg = await asyncio.wait_for(q.get(), timeout=25.0)
+                    yield msg
+                except asyncio.TimeoutError:
+                    yield ":keepalive\n\n"
+        except:
+            pass
+        finally:
+            if q in _clients:
+                _clients.remove(q)
+    
+    return StreamingResponse(event_generator(), media_type="text/event-stream")