"""Route definitions for the read-only query API."""

from __future__ import annotations

from fastapi import APIRouter, Query, Depends

from aucourt_ingest.api.dependencies import get_query_service

router = APIRouter(prefix="/api/v1")


@router.get("/health")
async def health(svc=Depends(get_query_service)):
    return await svc.health()


@router.get("/personas")
async def list_personas(svc=Depends(get_query_service)):
    return svc.list_personas()


@router.get("/cases")
async def list_cases(
    court: str | None = Query(None, description="Filter by court code"),
    limit: int = Query(50, ge=1, le=200),
    offset: int = Query(0, ge=0),
    svc=Depends(get_query_service),
):
    return await svc.list_cases(court=court, limit=limit, offset=offset)


@router.get("/cases/search")
async def search_cases(
    q: str = Query(..., description="Search query text"),
    limit: int = Query(10, ge=1, le=50),
    svc=Depends(get_query_service),
):
    return await svc.search_cases(q=q, limit=limit)


@router.get("/cases/{case_mnc}/juror/{persona}")
async def get_juror_context(
    case_mnc: str,
    persona: str,
    max_tokens: int | None = Query(None, ge=100, le=16000),
    svc=Depends(get_query_service),
):
    try:
        return await svc.get_juror_context(
            case_mnc, persona, max_tokens,
        )
    except KeyError as e:
        from fastapi import HTTPException
        raise HTTPException(status_code=404, detail=str(e))


@router.get("/cases/{case_mnc}")
async def get_case_graph(case_mnc: str, svc=Depends(get_query_service)):
    result = await svc.get_case_graph(case_mnc)
    if result is None:
        from fastapi import HTTPException
        raise HTTPException(status_code=404, detail=f"Case not found: {case_mnc}")
    return result


@router.get("/search")
async def vector_search(
    q: str = Query(..., description="Query text"),
    top_k: int = Query(10, ge=1, le=50),
    chunk_types: str | None = Query(None, description="Comma-separated chunk types"),
    doc_ids: str | None = Query(None, description="Comma-separated doc IDs"),
    svc=Depends(get_query_service),
):
    types = chunk_types.split(",") if chunk_types else None
    docs = doc_ids.split(",") if doc_ids else None
    return svc.vector_search(q, top_k=top_k, chunk_types=types, doc_ids=docs)


@router.get("/hybrid")
async def hybrid_search(
    q: str = Query(..., description="Query text"),
    persona: str | None = Query(None, description="Juror persona name"),
    top_k: int = Query(10, ge=1, le=50),
    max_tokens: int | None = Query(None, ge=100, le=16000),
    svc=Depends(get_query_service),
):
    return await svc.hybrid_search(q, persona_name=persona, top_k=top_k, max_tokens=max_tokens)