From 9598e6416b2ccea84095b7ee7babaae889b26228 Mon Sep 17 00:00:00 2001
From: sosokker <sirin_29@hotmail.co.th>
Date: Sun, 12 May 2024 03:02:54 +0700
Subject: [PATCH] Add web socket to stream camera video

---
 StreamServer/src/routers/video.py | 39 +++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/StreamServer/src/routers/video.py b/StreamServer/src/routers/video.py
index 3cea716..2daf074 100644
--- a/StreamServer/src/routers/video.py
+++ b/StreamServer/src/routers/video.py
@@ -168,3 +168,42 @@ async def disconnect_camera(camera_id: int) -> dict:
             save_to_config(key="cameras", value=cameras)
             return {"message": f"Camera {camera_id} disconnected successfully"}
     raise HTTPException(status_code=404, detail=f"Camera {camera_id} not found")
+
+
+@router.websocket("/ws/{camera_id}")
+async def websocket_endpoint(camera_id: int, websocket: WebSocket):
+    camera = next((c for c in cameras if c.camera_id == camera_id), None)
+    if not camera:
+        await websocket.close(code=1000)
+        return
+    
+    if not camera.status:
+        await websocket.close(code=1000)
+        return
+    
+    cap = VideoCapture(camera.link)
+    if not cap.isOpened():
+        await websocket.close(code=1000)
+        return
+    
+    try:
+        await websocket.accept()
+
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                await websocket.close(code=1000)
+                break
+            
+            ret, buffer = imencode('.png', frame)
+            if not ret:
+                await websocket.close(code=1000)
+                break
+            await websocket.send_bytes(buffer.tobytes())
+    except WebSocketDisconnect:
+        print("WebSocket disconnected")
+    except Exception as e:
+        print(f"Error sending frame: {e}")
+    finally:
+        cap.release()
+        await websocket.close(code=1000)
\ No newline at end of file