from __future__ import annotations

from typing import Annotated

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

from app.api.deps import get_db
from app.schemas.agent_run import AgentRunRead
from app.services.agent_runs import AgentRunService

router = APIRouter(prefix="/agent-runs")
DbSession = Annotated[Session, Depends(get_db)]


@router.get("", response_model=list[AgentRunRead])
def list_agent_runs(
    db: DbSession,
    agent: str | None = Query(default=None),
    status_value: str | None = Query(default=None, alias="status"),
    source: str | None = Query(default=None),
    limit: int = Query(default=20, ge=1, le=100),
) -> list[AgentRunRead]:
    return AgentRunService(db).list_runs(
        agent=agent, status=status_value, source=source, limit=limit
    )


@router.get("/{run_id}", response_model=AgentRunRead)
def get_agent_run(run_id: str, db: DbSession) -> AgentRunRead:
    run = AgentRunService(db).get_run(run_id)
    if run is None:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Run not found")
    return run