from contextlib import asynccontextmanager

from fastapi import FastAPI

from app.core.logger import log


@asynccontextmanager
async def lifespan(app: FastAPI):
    """应用生命周期管理：初始化和释放所有连接资源"""
    from app.core.database import init_db, close_db
    from app.core.redis import init_redis, close_redis
    from pathlib import Path

    # 启动：初始化资源
    try:
        log.info("初始化数据库连接")
        await init_db()
    except Exception as e:
        log.error(f"数据库连接初始化失败: {e}")
        raise e

    try:
        log.info("初始化redis连接")
        await init_redis()
    except Exception as e:
        log.warning(f"Redis 连接初始化失败: {e}")
        raise e
    log.info("所有资源已初始化，应用启动完成")

    # 打印 banner
    banner_path = Path(__file__).parent.parent / "banner.txt"
    if banner_path.exists():
        print(banner_path.read_text(encoding="utf-8"))



    yield

    # 关闭：释放资源
    await close_db()
    await close_redis()
    log.info("所有连接资源已释放")