feat: download - fase 02

webui/backend/app/api/routes_files.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
-from fastapi import APIRouter, Depends, File, Form, Request, UploadFile
+from fastapi import APIRouter, Depends, File, Form, Query, Request, UploadFile
 from fastapi.responses import StreamingResponse
+from starlette.background import BackgroundTask
 
 from backend.app.api.schemas import DeleteRequest, DeleteResponse, FileInfoResponse, MkdirRequest, MkdirResponse, RenameRequest, RenameResponse, SaveRequest, SaveResponse, UploadResponse, ViewResponse
 from backend.app.dependencies import get_file_ops_service
@@ -63,15 +64,18 @@ async def info(
 
 @router.get("/download")
 async def download(
-    path: str,
+    path: list[str] = Query(...),
     service: FileOpsService = Depends(get_file_ops_service),
 ) -> StreamingResponse:
-    prepared = service.prepare_download(path=path)
-    return StreamingResponse(
+    prepared = service.prepare_download(paths=path)
+    response = StreamingResponse(
         prepared["content"],
         headers=prepared["headers"],
         media_type=prepared["content_type"],
     )
+    if prepared.get("cleanup"):
+        response.background = BackgroundTask(prepared["cleanup"])
+    return response
 
 
 @router.get("/video")