from __future__ import annotations

from typing import Annotated

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

from app.api.deps import get_db
from app.schemas.settings import (
    ModelConnectivityTestRead,
    ModelConnectivityTestRequest,
    SettingsRead,
    SettingsWrite,
)
from app.services.model_connectivity import probe_model_connectivity
from app.services.settings import SettingsService

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


@router.get("", response_model=SettingsRead)
def get_settings(db: DbSession) -> SettingsRead:
    return SettingsService(db).get_settings_snapshot()


@router.put("", response_model=SettingsRead)
def update_settings(payload: SettingsWrite, db: DbSession) -> SettingsRead:
    try:
        return SettingsService(db).save_settings_snapshot(payload)
    except ValueError as exc:
        raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)) from exc


@router.post("/model-connectivity", response_model=ModelConnectivityTestRead)
def test_model_connectivity(payload: ModelConnectivityTestRequest, db: DbSession) -> ModelConnectivityTestRead:
    resolved_payload = payload

    if not payload.api_key and payload.slot:
        stored_api_key = SettingsService(db).load_saved_model_api_key(payload.slot)
        if stored_api_key:
            resolved_payload = payload.model_copy(update={"api_key": stored_api_key})

    return probe_model_connectivity(resolved_payload)