mirror of
https://github.com/xpltdco/media-rip.git
synced 2026-04-03 02:53:58 -06:00
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
"""Download management API routes.
|
|
|
|
POST /downloads — enqueue a new download job
|
|
GET /downloads — list jobs for the current session
|
|
DELETE /downloads/{job_id} — cancel a job
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import os
|
|
|
|
from fastapi import APIRouter, Depends, Request
|
|
from fastapi.responses import JSONResponse
|
|
|
|
from app.core.database import delete_job, get_job, get_jobs_by_session
|
|
from app.dependencies import get_session_id
|
|
from app.models.job import Job, JobCreate
|
|
|
|
logger = logging.getLogger("mediarip.api.downloads")
|
|
|
|
router = APIRouter(tags=["downloads"])
|
|
|
|
|
|
@router.post("/downloads", response_model=Job, status_code=201)
|
|
async def create_download(
|
|
job_create: JobCreate,
|
|
request: Request,
|
|
session_id: str = Depends(get_session_id),
|
|
) -> Job:
|
|
"""Submit a URL for download."""
|
|
logger.debug("POST /downloads session=%s url=%s", session_id, job_create.url)
|
|
download_service = request.app.state.download_service
|
|
job = await download_service.enqueue(job_create, session_id)
|
|
return job
|
|
|
|
|
|
@router.get("/downloads", response_model=list[Job])
|
|
async def list_downloads(
|
|
request: Request,
|
|
session_id: str = Depends(get_session_id),
|
|
) -> list[Job]:
|
|
"""List all download jobs for the current session."""
|
|
logger.debug("GET /downloads session=%s", session_id)
|
|
jobs = await get_jobs_by_session(request.app.state.db, session_id)
|
|
return jobs
|
|
|
|
|
|
@router.delete("/downloads/{job_id}")
|
|
async def cancel_download(
|
|
job_id: str,
|
|
request: Request,
|
|
) -> dict:
|
|
"""Delete a download job and remove its file."""
|
|
logger.debug("DELETE /downloads/%s", job_id)
|
|
db = request.app.state.db
|
|
|
|
# Fetch the job first to get its session_id and filename
|
|
job = await get_job(db, job_id)
|
|
if job is None:
|
|
return {"status": "not_found"}
|
|
|
|
# Delete the downloaded file if it exists
|
|
if job.filename:
|
|
output_dir = request.app.state.config.downloads.output_dir
|
|
filepath = os.path.join(output_dir, job.filename)
|
|
try:
|
|
if os.path.isfile(filepath):
|
|
os.remove(filepath)
|
|
logger.info("Deleted file: %s", filepath)
|
|
except OSError:
|
|
logger.warning("Failed to delete file: %s", filepath)
|
|
|
|
# Remove job from database
|
|
await delete_job(db, job_id)
|
|
|
|
# Notify any SSE clients watching this session
|
|
request.app.state.broker.publish(
|
|
job.session_id,
|
|
{"event": "job_removed", "data": {"job_id": job_id}},
|
|
)
|
|
|
|
return {"status": "deleted"}
|
|
|
|
|
|
@router.post("/url-info")
|
|
async def url_info(
|
|
request: Request,
|
|
body: dict,
|
|
) -> dict:
|
|
"""Extract metadata about a URL (title, playlist detection, audio-only detection)."""
|
|
url = body.get("url", "").strip()
|
|
if not url:
|
|
return JSONResponse(status_code=400, content={"detail": "URL required"})
|
|
download_service = request.app.state.download_service
|
|
return await download_service.get_url_info(url)
|