Update 2025-05-21_08:58:06

list_streams.py

@@ -2,14 +2,63 @@
 
 from fastapi import APIRouter
 from pathlib import Path
+from fastapi.responses import StreamingResponse
+import asyncio
 
 router = APIRouter()
 DATA_ROOT = Path("./data")
 
-@router.get("/streams")
+@router.get("/streams-sse")
+def streams_sse():
+    return list_streams_sse()
+
+import json
+
+import datetime
+
+def list_streams_sse():
+    async def event_generator():
+        txt_path = Path("./public_streams.txt")
+        if not txt_path.exists():
+            print(f"[{datetime.datetime.now()}] [SSE] No public_streams.txt found")
+            yield f"data: {json.dumps({'end': True})}\n\n"
+            return
+        try:
+            with txt_path.open("r") as f:
+                for line in f:
+                    line = line.strip()
+                    if not line:
+                        continue
+                    try:
+                        stream = json.loads(line)
+                        print(f"[{datetime.datetime.now()}] [SSE] Yielding stream: {stream}")
+                        yield f"data: {json.dumps(stream)}\n\n"
+                        await asyncio.sleep(0)  # Yield control to event loop
+                    except Exception as e:
+                        print(f"[{datetime.datetime.now()}] [SSE] JSON decode error: {e}")
+                        continue  # skip malformed lines
+            print(f"[{datetime.datetime.now()}] [SSE] Yielding end event")
+            yield f"data: {json.dumps({'end': True})}\n\n"
+        except Exception as e:
+            print(f"[{datetime.datetime.now()}] [SSE] Exception: {e}")
+            yield f"data: {json.dumps({'end': True, 'error': True})}\n\n"
+    return StreamingResponse(event_generator(), media_type="text/event-stream")
+
 def list_streams():
-    streams = []
-    for user_dir in DATA_ROOT.iterdir():
-        if user_dir.is_dir() and (user_dir / "stream.opus").exists():
-            streams.append(user_dir.name)
-    return {"streams": streams}
+    txt_path = Path("./public_streams.txt")
+    if not txt_path.exists():
+        return {"streams": []}
+    try:
+        streams = []
+        with txt_path.open("r") as f:
+            for line in f:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    streams.append(json.loads(line))
+                except Exception:
+                    continue  # skip malformed lines
+        return {"streams": streams}
+    except Exception:
+        return {"streams": []}