from __future__ import annotations

import logging
from contextlib import asynccontextmanager

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from app.api.chart import router as chart_router
from app.api.metrics import router as metrics_router
from app.api.news import router as news_router
from app.api.predict import router as predict_router
from app.api.refresh import router as refresh_router
from app.api.symbols import router as symbols_router
from app.config import settings
from app.db.connection import ping as db_ping
from app.fetch import dart as dart_mod
from app.fetch import kis as kis_mod
from app.pipelines.scheduler import shutdown_scheduler, start_scheduler

logging.basicConfig(
    level=settings.log_level,
    format="%(asctime)s %(levelname)s %(name)s: %(message)s",
)
logger = logging.getLogger(__name__)


@asynccontextmanager
async def lifespan(_: FastAPI):
    # 스케줄러는 옵션. CI/테스트에서 disable 하고 싶으면 SCHEDULER_DISABLED 같은 env 추가 가능.
    try:
        start_scheduler()
    except Exception:  # noqa: BLE001
        logger.exception("scheduler start failed")
    yield
    shutdown_scheduler()


app = FastAPI(title="stock_chart_site", version="0.1.0", lifespan=lifespan)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(refresh_router)
app.include_router(symbols_router)
app.include_router(chart_router)
app.include_router(predict_router)
app.include_router(metrics_router)
app.include_router(news_router)


def _resolved_device() -> str:
    if settings.model_device != "auto":
        return settings.model_device
    try:
        import torch  # noqa: WPS433
        return "cuda" if torch.cuda.is_available() else "cpu"
    except Exception:  # noqa: BLE001
        return "cpu"


@app.get("/health")
def health() -> dict[str, object]:
    return {"ok": True, "device": _resolved_device(), "version": "0.1.0"}


@app.get("/health/db")
def health_db() -> dict[str, object]:
    return {"ok": True, **db_ping()}


@app.get("/health/keys")
def health_keys() -> dict[str, object]:
    """등록된 외부 키들 ping (key 값은 노출하지 않음)."""
    return {
        "kis": kis_mod.ping(),
        "dart": dart_mod.ping(),
        # huggingface 는 모델 다운로드 시점에 확인 (별도 ping 호출 비용 회피)
    }