from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager
from sqlalchemy import text

from app.config import settings
from app.database import engine
from app.routes import router
from app.logger import get_logger, LoggerManager


# 初始化日志系统
LoggerManager.setup_logging()
logger = get_logger("main")


@asynccontextmanager
async def lifespan(app: FastAPI):
    # 启动时初始化
    logger.info("🚀 应用启动中...")

    try:
        # 验证数据库连接
        logger.info("📊 验证数据库连接...")
        try:
            async with engine.begin() as conn:
                # 执行简单的查询来验证连接
                result = await conn.execute(text("SELECT 1"))
                connection_test = result.scalar()
                if connection_test == 1:
                    logger.info("✅ 数据库连接验证成功")
                else:
                    raise Exception("数据库连接测试失败")
        except Exception as db_error:
            logger.error(f"❌ 数据库连接验证失败: {db_error}")
            raise

        logger.info(f"🎉 {settings.app_name} 启动完成!")
        yield

    except Exception as e:
        logger.error(f"❌ 应用启动失败: {e}")
        raise

    finally:
        # 关闭时清理
        logger.info("🛑 应用关闭中...")
        logger.info("👋 应用已关闭")


app = FastAPI(
    title=settings.app_name,
    version="1.0.0",
    lifespan=lifespan,
    docs_url=(
        "/docs"
        if not settings.disable_docs and settings.environment != "production"
        else None
    ),
    redoc_url=(
        "/redoc"
        if not settings.disable_docs and settings.environment != "production"
        else None
    ),
    openapi_url="/openapi.json" if not settings.disable_docs else None,
)

# 添加CORS中间件
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 注册路由
app.include_router(router)


@app.get("/")
async def root():
    # 检查环境，生产环境下禁用根接口
    if settings.environment == "production":
        logger.warning("🚫 生产环境下禁止访问根接口")
        from fastapi import HTTPException

        raise HTTPException(status_code=404, detail="Not Found")

    logger.info("📝 根接口被访问")
    return {"message": f"Welcome to {settings.app_name}"}


@app.get("/health")
async def health_check():
    # 检查环境，生产环境下禁用健康检查接口
    if settings.environment == "production":
        logger.warning("🚫 生产环境下禁止访问健康检查接口")
        from fastapi import HTTPException

        raise HTTPException(status_code=404, detail="Not Found")

    logger.debug("💓 健康检查接口被访问")
    return {"status": "healthy"}


if __name__ == "__main__":
    import uvicorn

    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=settings.debug)