"""Admin audio extraction API endpoints."""

from typing import Annotated

from fastapi import APIRouter, Depends, HTTPException, status
from sqlmodel.ext.asyncio.session import AsyncSession

from app.core.database import get_db
from app.core.dependencies import get_admin_user
from app.models.user import User
from app.services.extraction import ExtractionService
from app.services.extraction_processor import extraction_processor

router = APIRouter(prefix="/extractions", tags=["admin-extractions"])


async def get_extraction_service(
    session: Annotated[AsyncSession, Depends(get_db)],
) -> ExtractionService:
    """Get the extraction service."""
    return ExtractionService(session)


@router.get("/status")
async def get_extraction_processor_status(
    current_user: Annotated[User, Depends(get_admin_user)],  # noqa: ARG001
) -> dict:
    """Get the status of the extraction processor. Admin only."""
    return extraction_processor.get_status()


@router.delete("/{extraction_id}")
async def delete_extraction(
    extraction_id: int,
    current_user: Annotated[User, Depends(get_admin_user)],  # noqa: ARG001
    extraction_service: Annotated[ExtractionService, Depends(get_extraction_service)],
) -> dict[str, str]:
    """Delete any extraction and its associated sound and files. Admin only."""
    try:
        deleted = await extraction_service.delete_extraction(extraction_id, None)

        if not deleted:
            raise HTTPException(
                status_code=status.HTTP_404_NOT_FOUND,
                detail=f"Extraction {extraction_id} not found",
            )

    except HTTPException:
        # Re-raise HTTPExceptions without wrapping them
        raise
    except Exception as e:
        raise HTTPException(
            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
            detail=f"Failed to delete extraction: {e!s}",
        ) from e
    else:
        return {
            "message": f"Extraction {extraction_id} deleted successfully",
        }