Fix stream video and Disconnect camera endpoint + new auto generated id mechanism

StreamServer/src/routers/video.py

@@ -1,29 +1,82 @@
-from fastapi import APIRouter, HTTPException
+import asyncio
+import cv2
+import datetime
+import os
+
+from threading import Thread
+
+from cv2 import VideoCapture, VideoWriter, imencode
+
+from fastapi import APIRouter, BackgroundTasks, HTTPException
+from fastapi.responses import StreamingResponse
+
 from scheme import Camera
+from database import minio_client
+from config import VIDEO_BUCKET, TEMP_VIDEO_FILE
+
 from typing import List
 
 router = APIRouter()
 
 cameras: List[Camera] = []
 
+# --- UTILITY FUNCTIONS ---
+
+def generate_camera_id() -> int:
+    if not cameras:
+        return 1
+    cameras.sort(key=lambda x: x.camera_id)
+    return cameras[-1].camera_id + 1
+
+# --- ROUTER ENDPOINTS ---
+
 @router.post("/add_camera", response_model=dict)
 async def add_camera(rtsp_link: str):
     if rtsp_link:
-        camera = Camera(rtsp_link=rtsp_link)
+        id = generate_camera_id()
+        camera = Camera(camera_id=id, link=rtsp_link)
         cameras.append(camera)
-        return {"message": "Camera added successfully"}
+        return {"message": "Camera added successfully", "camera_id": id}
     else:
         raise HTTPException(status_code=400, detail="Invalid RTSP link")
 
-@router.get("/stream_video")
-async def stream_video():
-    # TODO: Implement video streaming
-    return {"message": "Camera disconnected successfully"}
+
+@router.get("/list_cameras", response_model=List[Camera])
+async def list_cameras() -> List[Camera]:
+    return cameras
+
+
+@router.get("/stream_video/{camera_id}")
+async def stream_video(camera_id: int) -> StreamingResponse:
+    available_cameras = [camera.camera_id for camera in cameras]
+    if camera_id in available_cameras:
+        for camera in cameras:
+            if camera.camera_id == camera_id:
+                link = camera.link
+                break
+        cap = VideoCapture(link)
+        if not cap.isOpened():
+            raise HTTPException(status_code=404, detail="Camera is closed or not available")
+        
+        def generate_frames():
+            while cap.isOpened():
+                ret, frame = cap.read()
+                if not ret:
+                    break
+                ret, buffer = imencode('.jpg', frame)
+                if not ret:
+                    break
+                yield b'--frame\r\n' b'Content-Type: image/jpeg\r\n\r\n' + buffer.tobytes() + b'\r\n'
+
+        return StreamingResponse(generate_frames(), media_type="multipart/x-mixed-replace; boundary=frame")
+    else:
+        raise HTTPException(status_code=404, detail="No cameras available")
+
 
 @router.delete("/disconnect_camera/{camera_id}", response_model=dict)
-async def disconnect_camera(camera_id: int):
-    if 0 <= camera_id < len(cameras):
-        del cameras[camera_id]
-        return {"message": "Camera disconnected successfully"}
-    else:
-        raise HTTPException(status_code=404, detail="Camera not found")
+async def disconnect_camera(camera_id: int) -> dict:
+    for camera in cameras:
+        if camera.camera_id == camera_id:
+            cameras.remove(camera)
+            return {"message": f"Camera {camera_id} disconnected successfully"}
+    raise HTTPException(status_code=404, detail=f"Camera {camera_id} not found")