# /api/runs routes
from typing import Dict
from uuid import UUID

from fastapi import APIRouter, Depends, HTTPException
from pydantic import BaseModel

from ..deps import get_db
from ..models.ag_db import Message, Run

router = APIRouter()


class CreateRunRequest(BaseModel):
    session_id: int
    user_id: str


# @router.post("/")
# async def create_run(
#     request: CreateRunRequest,
#     db=Depends(get_db),
# ) -> Dict:
#     """Create a new run with initial state"""
#     session_response = db.get(
#         Session,
#         filters={"id": request.session_id, "user_id": request.user_id},
#         return_json=False,
#     )
#     if not session_response.status or not session_response.data:
#         raise HTTPException(status_code=404, detail="Session not found")

#     try:
#         # Create run with default state
#         run = db.upsert(
#             Run(
#                 session_id=request.session_id,
#                 status=RunStatus.CREATED,
#                 task=None,  # Will be set when run starts
#                 team_result=None,
#             ),
#             return_json=False,
#         )
#         return {"status": run.status, "data": {"run_id": str(run.data.id)}}
#     except Exception as e:
#         raise HTTPException(status_code=500, detail=str(e)) from e


# We might want to add these endpoints:


@router.get("/{run_id}")
async def get_run(run_id: UUID, db=Depends(get_db)) -> Dict:
    """Get run details including task and result"""
    run = db.get(Run, filters={"id": run_id}, return_json=False)
    if not run.status or not run.data:
        raise HTTPException(status_code=404, detail="Run not found")

    return {"status": True, "data": run.data[0]}


@router.get("/{run_id}/messages")
async def get_run_messages(run_id: UUID, db=Depends(get_db)) -> Dict:
    """Get all messages for a run"""
    messages = db.get(
        Message, filters={"run_id": run_id}, order="created_at asc", return_json=False
    )

    return {"status": True, "data": messages.data}
