from uuid import UUID

from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy import func, select
from sqlalchemy.orm import Session

from app.db.session import get_db
from app.models import RetrievalRun
from app.schemas.catalog import Page
from app.schemas.retrieval import RetrievalRequest, RetrievalResponse
from app.schemas.vectors import VectorRetrievalResponse
from app.services.auth import require_admin
from app.services.retrieval import retrieve

router = APIRouter(prefix="/retrieval", tags=["retrieval"])


@router.post("", response_model=RetrievalResponse)
def keyword_retrieval(payload: RetrievalRequest, db: Session = Depends(get_db)):
    """只检索已核验知识；保存原始检索结果供复核，不调用大模型。"""
    return retrieve(db, payload)


@router.get(
    "/runs",
    response_model=Page[RetrievalResponse | VectorRetrievalResponse],
    dependencies=[Depends(require_admin)],
)
def list_runs(
    page: int = Query(1, ge=1),
    page_size: int = Query(20, ge=1, le=100),
    db: Session = Depends(get_db),
):
    rows = db.scalars(
        select(RetrievalRun)
        .order_by(RetrievalRun.created_at.desc(), RetrievalRun.id)
        .offset((page - 1) * page_size)
        .limit(page_size)
    )
    return {
        "items": [row.response for row in rows],
        "total": db.scalar(select(func.count()).select_from(RetrievalRun)),
        "page": page,
        "page_size": page_size,
    }


@router.get(
    "/runs/{run_id}",
    response_model=RetrievalResponse | VectorRetrievalResponse,
    dependencies=[Depends(require_admin)],
)
def read_run(run_id: UUID, db: Session = Depends(get_db)):
    row = db.get(RetrievalRun, str(run_id))
    if row is None:
        raise HTTPException(404, "检索记录不存在")
    return row.response