from datetime import datetime
from uuid import UUID

from fastapi import APIRouter, Depends, HTTPException, Query
from psycopg import AsyncConnection

from app.infra.db.timescaledb.repositories.analysis import AnalysisResultsRepository

from .dependencies import get_timescale_connection

router = APIRouter()


@router.get("/timeseries/analysis/runs/{run_id}/nodes/{node_id}")
async def get_analysis_node_series(
    run_id: UUID,
    node_id: str,
    start_time: datetime = Query(...),
    end_time: datetime = Query(...),
    field: str = Query(...),
    conn: AsyncConnection = Depends(get_timescale_connection),
):
    try:
        return await AnalysisResultsRepository.get_node_series(
            conn, run_id, node_id, start_time, end_time, field
        )
    except ValueError as exc:
        raise HTTPException(status_code=400, detail=str(exc)) from exc


@router.get("/timeseries/analysis/runs/{run_id}/links/{link_id}")
async def get_analysis_link_series(
    run_id: UUID,
    link_id: str,
    start_time: datetime = Query(...),
    end_time: datetime = Query(...),
    field: str = Query(...),
    conn: AsyncConnection = Depends(get_timescale_connection),
):
    try:
        return await AnalysisResultsRepository.get_link_series(
            conn, run_id, link_id, start_time, end_time, field
        )
    except ValueError as exc:
        raise HTTPException(status_code=400, detail=str(exc)) from exc


@router.get("/timeseries/analysis/runs/{run_id}/values")
async def get_analysis_values_at_time(
    run_id: UUID,
    result_time: datetime = Query(...),
    element_type: str = Query(..., pattern="^(node|link)$"),
    field: str = Query(...),
    conn: AsyncConnection = Depends(get_timescale_connection),
):
    try:
        return await AnalysisResultsRepository.get_values_at_time(
            conn, run_id, element_type, result_time, field
        )
    except ValueError as exc:
        raise HTTPException(status_code=400, detail=str(exc)) from exc