"""
Title, space, and agent selection API routes.
"""

from __future__ import annotations

from fastapi import APIRouter, Request
from fastapi.responses import JSONResponse
from starlette.responses import Response

from ..providers import is_provider_supported
from ._request_secrets import get_llm_api_key
from ..services.generation import generate_title_space_and_agent

router = APIRouter(tags=["title-space-agent"])


@router.post("/title-space-agent")
async def title_space_agent(request: Request) -> Response:
    body = await request.json()
    provider = body.get("provider")
    message = body.get("message")
    spaces_with_agents = body.get("spacesWithAgents") or []
    api_key = get_llm_api_key(request)
    base_url = body.get("baseUrl")
    model = body.get("model")
    user_timezone = body.get("userTimezone")
    user_locale = body.get("userLocale")

    if not provider or not message:
        return JSONResponse(
            status_code=400,
            content={"error": "Missing required fields: provider, message"},
        )
    if not is_provider_supported(provider):
        return JSONResponse(status_code=400, content={"error": f"Unsupported provider: {provider}"})

    result = await generate_title_space_and_agent(
        provider=provider,
        first_message=message,
        spaces_with_agents=spaces_with_agents,
        api_key=api_key,
        base_url=base_url,
        model=model,
        user_timezone=user_timezone,
        user_locale=user_locale,
    )
    return JSONResponse(content=result)