import asyncio

from fastapi import FastAPI, Request
from fastapi.responses import JSONResponse
from pydantic import BaseModel
from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor

from common.db import engine
from common.logging_config import configure_logging
from common.telemetry import setup_telemetry

from . import chaos
from .repository import is_match

SERVICE_NAME = "sanctions"
logger = configure_logging(SERVICE_NAME)
setup_telemetry(service_name=SERVICE_NAME)
app = FastAPI(title=SERVICE_NAME)

FastAPIInstrumentor.instrument_app(
    app,
    excluded_urls="/healthz",
    exclude_spans=["receive", "send"],
)

class LookupRequest(BaseModel):
    holder_name: str
    request_id: str = "unknown"


@app.get("/")
async def root():
    return {"service": SERVICE_NAME, "version": "0.1.0"}


@app.get("/healthz")
async def healthz(request: Request):
    if request.headers.get("X-Ledgerline-Probe") == "chatty-health":
        logger.info(f"received chatty-health healthz probe from gateway")
    return {"status": "ok", "service": SERVICE_NAME}


@app.get("/readyz")
async def readyz():
    try:
        async with engine.connect() as conn:
            await conn.exec_driver_sql("SELECT 1")
    except Exception as exc:
        logger.exception(f"readyz check failed: {exc}")
        return JSONResponse(status_code=503, content={"status": "unavailable"})
    return {"status": "ok"}


@app.post("/internal/lookup")
async def lookup(body: LookupRequest):
    logger.info(f"lookup for holder_name={body.holder_name!r} request_id={body.request_id}")

    if await chaos.sanctions_errors_enabled() and chaos.should_error():
        logger.info(f"sanctions_errors chaos triggered for request_id={body.request_id}")
        return JSONResponse(status_code=503, content={"error": "sanctions vendor unavailable"})

    if await chaos.slow_sanctions_enabled():
        delay = chaos.sample_delay_seconds()
        logger.info(f"slow_sanctions chaos sleeping {delay:.3f}s for request_id={body.request_id}")
        await asyncio.sleep(delay)

    match = await is_match(body.holder_name)
    if match:
        return {"matched": True, **match}
    return {"matched": False, "country": None, "list_source": None}