Update storage/media_routers.py

storage/media_routers.py

@@ -667,4 +667,134 @@ def list_subtype_audios(
             "audio_name": latest_audio
         })
 
-    return results
+    return results
+
+
+@router.post("/upload_subtype_video", tags=["Media Manager"])
+async def upload_subtype_video(
+    sha1: str = Query(..., description="SHA1 associated to the media folder"),
+    subtype: str = Query(..., description="Subtype: HITL, MoE or Salamandra"),
+    video: UploadFile = File(...),
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    Upload a video to /data/media/<sha1>/<subtype>/.
+    Steps:
+    - Validate subtype.
+    - Create subtype folder if missing.
+    - Delete existing MP4 files.
+    - Save new MP4.
+    """
+    validate_token(token)
+
+    VALID_SUBTYPES = ("HITL", "MoE", "Salamandra")
+    if subtype not in VALID_SUBTYPES:
+        raise HTTPException(status_code=400, detail="Invalid subtype")
+
+    MEDIA_ROOT = Path("/data/media")
+    file_manager = FileManager(MEDIA_ROOT)
+
+    subtype_dir = MEDIA_ROOT / sha1 / subtype
+    subtype_dir.mkdir(parents=True, exist_ok=True)
+
+    # Remove old mp4 files
+    for f in subtype_dir.glob("*.mp4"):
+        f.unlink()
+
+    file_bytes = await video.read()
+    final_path = subtype_dir / video.filename
+
+    save_result = file_manager.upload_file(io.BytesIO(file_bytes), final_path)
+    if not save_result["operation_success"]:
+        raise HTTPException(status_code=500, detail=save_result["error"])
+
+    return {
+        "status": "ok",
+        "sha1": sha1,
+        "subtype": subtype,
+        "saved_to": str(final_path)
+    }
+
+
+@router.get("/download_subtype_video", tags=["Media Manager"])
+def download_subtype_video(
+    sha1: str,
+    subtype: str,
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    Download the video stored under /data/media/<sha1>/<subtype>.
+    Returns the first MP4 found.
+    """
+    validate_token(token)
+
+    VALID_SUBTYPES = ("HITL", "MoE", "Salamandra")
+    if subtype not in VALID_SUBTYPES:
+        raise HTTPException(status_code=400, detail="Invalid subtype")
+
+    MEDIA_ROOT = Path("/data/media")
+    file_manager = FileManager(MEDIA_ROOT)
+
+    subtype_dir = MEDIA_ROOT / sha1 / subtype
+
+    if not subtype_dir.exists() or not subtype_dir.is_dir():
+        raise HTTPException(status_code=404, detail="Subtype folder not found")
+
+    mp4_files = list(subtype_dir.glob("*.mp4"))
+    if not mp4_files:
+        raise HTTPException(status_code=404, detail="No MP4 files found")
+
+    video_path = mp4_files[0]
+    relative_path = video_path.relative_to(MEDIA_ROOT)
+
+    handler = file_manager.get_file(relative_path)
+    if handler is None:
+        raise HTTPException(status_code=404, detail="Video not accessible")
+
+    handler.close()
+
+    return FileResponse(
+        path=video_path,
+        media_type="video/mp4",
+        filename=video_path.name
+    )
+
+
+@router.get("/list_subtype_videos", tags=["Media Manager"])
+def list_subtype_videos(
+    sha1: str,
+    token: str = Query(..., description="Token required for authorization")
+):
+    """
+    List the most recent .mp4 video for each subtype (HITL, MoE, Salamandra)
+    inside /data/media/<sha1>.
+    Returns:
+    - sha1
+    - subtype
+    - video_name (latest mp4 or None)
+    """
+    validate_token(token)
+
+    MEDIA_ROOT = Path("/data/media")
+    VALID_SUBTYPES = ("HITL", "MoE", "Salamandra")
+
+    results = []
+
+    for subtype in VALID_SUBTYPES:
+        subtype_dir = MEDIA_ROOT / sha1 / subtype
+
+        latest_video = None
+
+        if subtype_dir.exists() and subtype_dir.is_dir():
+            files = list(subtype_dir.glob("*.mp4"))
+            if files:
+                files.sort(key=lambda f: f.stat().st_mtime, reverse=True)
+                latest_video = files[0].name
+
+        results.append({
+            "sha1": sha1,
+            "subtype": subtype,
+            "video_name": latest_video
+        })
+
+    return results