from fastapi import APIRouter, HTTPException, Query from pydantic import BaseModel, Field from app.services.session_service import SessionService router = APIRouter() class SelectedEpisodesAddRequest(BaseModel): items: list[dict] = Field(default_factory=list) class SelectedEpisodesReorderRequest(BaseModel): from_index: int = Field(ge=0) to_index: int = Field(ge=0) class SelectedFilesAddRequest(BaseModel): items: list[dict] = Field(default_factory=list) class SelectedFilesReorderRequest(BaseModel): from_index: int = Field(ge=0) to_index: int = Field(ge=0) def _normalize_session_id(session_id: str) -> str: normalized = session_id.strip() if not normalized: raise HTTPException(status_code=400, detail="session_id must not be empty") return normalized @router.get("/selected-episodes") def get_selected_episodes(session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) items = service.list_selected_episodes(normalized_session_id) return {"session_id": normalized_session_id, "items": items} @router.post("/selected-episodes") def add_selected_episodes( payload: SelectedEpisodesAddRequest, session_id: str = Query("default", min_length=1), ): service = SessionService() normalized_session_id = _normalize_session_id(session_id) items = service.add_selected_episodes(normalized_session_id, payload.items) return {"session_id": normalized_session_id, "items": items} @router.delete("/selected-episodes") def clear_selected_episodes(session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) service.clear_selected_episodes(normalized_session_id) return {"session_id": normalized_session_id, "items": []} @router.delete("/selected-episodes/{selection_id}") def remove_selected_episode(selection_id: int, session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) items = service.remove_selected_episode(normalized_session_id, selection_id) return {"session_id": normalized_session_id, "items": items} @router.post("/selected-episodes/reorder") def reorder_selected_episodes( payload: SelectedEpisodesReorderRequest, session_id: str = Query("default", min_length=1), ): service = SessionService() normalized_session_id = _normalize_session_id(session_id) try: items = service.reorder_selected_episodes( normalized_session_id, payload.from_index, payload.to_index, ) except ValueError as exc: raise HTTPException(status_code=400, detail=str(exc)) return {"session_id": normalized_session_id, "items": items} @router.get("/selected-files") def get_selected_files(session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) items = service.list_selected_files(normalized_session_id) return {"session_id": normalized_session_id, "items": items} @router.post("/selected-files") def add_selected_files( payload: SelectedFilesAddRequest, session_id: str = Query("default", min_length=1), ): service = SessionService() normalized_session_id = _normalize_session_id(session_id) items = service.add_selected_files(normalized_session_id, payload.items) return {"session_id": normalized_session_id, "items": items} @router.delete("/selected-files") def clear_selected_files(session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) service.clear_selected_files(normalized_session_id) return {"session_id": normalized_session_id, "items": []} @router.delete("/selected-files/{selection_id}") def remove_selected_file(selection_id: int, session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) items = service.remove_selected_file(normalized_session_id, selection_id) return {"session_id": normalized_session_id, "items": items} @router.post("/selected-files/reorder") def reorder_selected_files( payload: SelectedFilesReorderRequest, session_id: str = Query("default", min_length=1), ): service = SessionService() normalized_session_id = _normalize_session_id(session_id) try: items = service.reorder_selected_files( normalized_session_id, payload.from_index, payload.to_index, ) except ValueError as exc: raise HTTPException(status_code=400, detail=str(exc)) return {"session_id": normalized_session_id, "items": items} @router.get("/mapping-preview") def get_mapping_preview(session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) try: return service.build_mapping_preview(normalized_session_id) except ValueError as exc: raise HTTPException(status_code=400, detail=str(exc)) @router.get("/filename-preview") def get_filename_preview(session_id: str = Query("default", min_length=1)): service = SessionService() normalized_session_id = _normalize_session_id(session_id) try: return service.build_filename_preview(normalized_session_id) except ValueError as exc: raise HTTPException(status_code=400, detail=str(exc)) @router.post("/rename-execute") def rename_execute( session_id: str = Query("default", min_length=1), confirm: bool = Query(False), ): service = SessionService() normalized_session_id = _normalize_session_id(session_id) try: return service.execute_rename(normalized_session_id, confirm=confirm) except ValueError as exc: raise HTTPException(status_code=400, detail=str(exc))