import logging

from fastapi import APIRouter, HTTPException

from app.core.dependencies import get_prompt_service
from app.models.prompts import (
    ProfileListResponse,
    ProfileItem,
    PromptSetResponse,
    PromptUpdateRequest,
    PromptBatchUpdateRequest,
    ResetToDefaultsRequest,
)

logger = logging.getLogger(__name__)

router = APIRouter(prefix="/api/v1/prompts", tags=["prompts"])

_VALID_NAMES = {"A", "B", "C"}
_VALID_STEPS = {"decompose", "filter", "generate"}


def _ensure_valid_name(name: str) -> None:
    if name not in _VALID_NAMES:
        raise HTTPException(status_code=400, detail=f"Invalid profile name '{name}'. Must be one of A, B, C.")


def _ensure_valid_step(step: str) -> None:
    if step not in _VALID_STEPS:
        raise HTTPException(status_code=400, detail=f"Invalid step '{step}'. Must be one of decompose, filter, generate.")


@router.get("/profiles", response_model=ProfileListResponse)
def list_profiles():
    svc = get_prompt_service()
    profiles = [ProfileItem(**p) for p in svc.list_profiles()]
    return ProfileListResponse(profiles=profiles)


@router.get("/profiles/{name}", response_model=PromptSetResponse)
def get_profile_prompts(name: str):
    _ensure_valid_name(name)
    svc = get_prompt_service()
    prompts = svc.get_profile_prompts(name)
    return PromptSetResponse(profile_name=name, prompts=prompts)


@router.put("/profiles/{name}/activate")
def activate_profile(name: str):
    _ensure_valid_name(name)
    svc = get_prompt_service()
    svc.activate_profile(name)
    return {"status": "ok", "active_profile": name}


@router.put("/profiles/{name}/all")
def update_all_prompts(name: str, body: PromptBatchUpdateRequest):
    _ensure_valid_name(name)
    svc = get_prompt_service()
    svc.update_all_prompts(name, body.prompts)
    return {"status": "ok", "profile": name}


@router.put("/profiles/{name}/reset")
def reset_to_defaults(name: str, body: ResetToDefaultsRequest | None = None):
    _ensure_valid_name(name)
    step = body.step if body else None
    if step is not None:
        _ensure_valid_step(step)
    svc = get_prompt_service()
    svc.reset_to_defaults(name, step=step)
    return {"status": "ok", "profile": name, "reset_step": step or "all"}


@router.put("/profiles/{name}/{step}")
def update_prompt(name: str, step: str, body: PromptUpdateRequest):
    _ensure_valid_name(name)
    _ensure_valid_step(step)
    svc = get_prompt_service()
    svc.update_prompt(name, step, body.template)
    return {"status": "ok", "profile": name, "step": step}