from contextlib import asynccontextmanager
from fastapi import FastAPI, WebSocket
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import JSONResponse
import uvicorn

from config import settings
from utils import setup_logging
from routers import (
    market_router,
    sentiment_router,
    momentum_router,
    highlow_router,
    stock_router,
    news_router,
)
from websocket import websocket_handler, manager
from adapters import DataAdapterFactory
from models import ApiResponse


setup_logging()


@asynccontextmanager
async def lifespan(app: FastAPI):
    await manager.start_broadcast_loop()
    yield
    await manager.stop_broadcast_loop()
    await DataAdapterFactory.close_all()


app = FastAPI(
    title=settings.APP_NAME,
    description="A股智投后端API服务，提供市场数据、情绪指标、版块动量、AI分析等功能",
    version="1.0.0",
    lifespan=lifespan,
)


app.add_middleware(
    CORSMiddleware,
    allow_origins=settings.cors_origins_list,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


app.include_router(market_router)
app.include_router(sentiment_router)
app.include_router(momentum_router)
app.include_router(highlow_router)
app.include_router(stock_router)
app.include_router(news_router)


@app.get("/", response_model=ApiResponse, tags=["系统"])
async def root():
    return ApiResponse(
        success=True,
        data={
            "name": settings.APP_NAME,
            "version": "1.0.0",
            "status": "running",
        },
        message="欢迎使用A股智投API服务",
    )


@app.get("/health", response_model=ApiResponse, tags=["系统"])
async def health_check():
    return ApiResponse(
        success=True,
        data={"status": "healthy"},
    )


@app.websocket("/ws/realtime")
async def websocket_endpoint(websocket: WebSocket):
    await websocket_handler(websocket)


@app.exception_handler(Exception)
async def global_exception_handler(request, exc):
    return JSONResponse(
        status_code=500,
        content={
            "success": False,
            "error": str(exc),
            "message": "服务器内部错误",
        },
    )


if __name__ == "__main__":
    uvicorn.run(
        "main:app",
        host=settings.APP_HOST,
        port=settings.APP_PORT,
        reload=settings.APP_DEBUG,
        log_level=settings.LOG_LEVEL.lower(),
    )