ctms/backend/app/api/v1/data_queries.py

import uuid
from datetime import date

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

from app.core.deps import get_current_user, get_db_session, require_study_member, require_study_roles
from app.crud import data_query as dq_crud
from app.crud import audit as audit_crud
from app.crud import study as study_crud
from app.schemas.data_query import DataQueryCreate, DataQueryRead, DataQueryUpdate

router = APIRouter()


def _is_overdue(dq: DataQueryRead) -> bool:
    return bool(dq.due_date and date.today() > dq.due_date and dq.status != "CLOSED")


async def _ensure_study_exists(db: AsyncSession, study_id: uuid.UUID):
    study = await study_crud.get(db, study_id)
    if not study:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="Study not found")
    return study


@router.post(
    "/",
    response_model=DataQueryRead,
    status_code=status.HTTP_201_CREATED,
    dependencies=[Depends(require_study_roles(["PM", "CRA"]))],
)
async def create_query(
    study_id: uuid.UUID,
    query_in: DataQueryCreate,
    db: AsyncSession = Depends(get_db_session),
    current_user=Depends(get_current_user),
) -> DataQueryRead:
    await _ensure_study_exists(db, study_id)
    try:
        dq = await dq_crud.create_query(db, study_id, query_in, created_by=current_user.id)
    except ValueError as exc:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)) from exc
    await audit_crud.log_action(
        db,
        study_id=study_id,
        entity_type="data_query",
        entity_id=dq.id,
        action="CREATE_DATA_QUERY",
        detail=f"DataQuery {dq.id} created",
        operator_id=current_user.id,
        operator_role=current_user.role,
    )
    obj = DataQueryRead.model_validate(dq)
    obj.is_overdue = _is_overdue(obj)
    return obj


@router.get(
    "/",
    response_model=list[DataQueryRead],
    dependencies=[Depends(require_study_member())],
)
async def list_queries(
    study_id: uuid.UUID,
    status_filter: str | None = None,
    site_id: uuid.UUID | None = None,
    subject_id: uuid.UUID | None = None,
    assigned_to: uuid.UUID | None = None,
    overdue: bool | None = None,
    category: str | None = None,
    priority: str | None = None,
    skip: int = 0,
    limit: int = 100,
    db: AsyncSession = Depends(get_db_session),
) -> list[DataQueryRead]:
    await _ensure_study_exists(db, study_id)
    queries = await dq_crud.list_queries(
        db,
        study_id,
        status=status_filter,
        site_id=site_id,
        subject_id=subject_id,
        assigned_to=assigned_to,
        overdue=overdue,
        category=category,
        priority=priority,
        skip=skip,
        limit=limit,
    )
    result: list[DataQueryRead] = []
    for q in queries:
        obj = DataQueryRead.model_validate(q)
        obj.is_overdue = _is_overdue(obj)
        result.append(obj)
    return result


@router.get(
    "/{query_id}",
    response_model=DataQueryRead,
    dependencies=[Depends(require_study_member())],
)
async def get_query(
    study_id: uuid.UUID,
    query_id: uuid.UUID,
    db: AsyncSession = Depends(get_db_session),
) -> DataQueryRead:
    await _ensure_study_exists(db, study_id)
    dq = await dq_crud.get_query(db, query_id)
    if not dq or dq.study_id != study_id:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="DataQuery not found")
    obj = DataQueryRead.model_validate(dq)
    obj.is_overdue = _is_overdue(obj)
    return obj


@router.patch(
    "/{query_id}",
    response_model=DataQueryRead,
    dependencies=[Depends(require_study_roles(["PM", "CRA"]))],
)
async def update_query(
    study_id: uuid.UUID,
    query_id: uuid.UUID,
    query_in: DataQueryUpdate,
    db: AsyncSession = Depends(get_db_session),
    current_user=Depends(get_current_user),
) -> DataQueryRead:
    await _ensure_study_exists(db, study_id)
    dq = await dq_crud.get_query(db, query_id)
    if not dq or dq.study_id != study_id:
        raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, detail="DataQuery not found")
    old_status = dq.status
    updated = await dq_crud.update_query(db, dq, query_in)
    detail = None
    action = "UPDATE_DATA_QUERY"
    if query_in.status and query_in.status != old_status:
        detail = f"DataQuery {query_id} status {old_status} -> {query_in.status}"
        action = "DATA_QUERY_STATUS_CHANGE"
    await audit_crud.log_action(
        db,
        study_id=study_id,
        entity_type="data_query",
        entity_id=query_id,
        action=action,
        detail=detail or "DataQuery updated",
        operator_id=current_user.id,
        operator_role=current_user.role,
    )
    obj = DataQueryRead.model_validate(updated)
    obj.is_overdue = _is_overdue(obj)
    return obj