life-echo/api/app/main.py

"""
FastAPI 应用入口（app 内主入口，符合架构计划）
"""

from pathlib import Path

from app.core.logging import setup_logging

setup_logging()

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from loguru import logger

from app.core.config import settings
from app.core.errors import register_exception_handlers
from app.core.middleware import RequestIdMiddleware
from app.core.openapi import custom_openapi
from app.features.auth.router import router as auth_router
from app.features.content.router import router as content_router
from app.features.conversation.router import router as conversation_router
from app.features.conversation.ws.router import websocket_endpoint
from app.features.memoir.router import router as memoir_router
from app.features.payment.router import router as payment_router
from app.features.plan.router import router as plan_router
from app.features.quota.router import router as quota_router
from app.features.tasks.router import router as tasks_router
from app.features.user.router import feedback_router as user_feedback_router
from app.features.user.router import router as user_router

# 聚合注册所有 feature 的 model 到 Base.metadata（供 Alembic 等使用）
from app.features.auth import models as _auth_models  # noqa: F401
from app.features.conversation import models as _conv_models  # noqa: F401
from app.features.memory import models as _memory_models  # noqa: F401
from app.features.memoir import models as _memoir_models  # noqa: F401
from app.features.payment import models as _payment_models  # noqa: F401
from app.features.story import models as _story_models  # noqa: F401
from app.features.user import models as _user_models  # noqa: F401

app = FastAPI(
    title="Life Echo API",
    version="1.0.0",
    docs_url="/docs" if settings.enable_docs else None,
    redoc_url="/redoc" if settings.enable_docs else None,
    openapi_url="/openapi.json" if settings.enable_docs else None,
)

# OpenAPI 全局增强
app.openapi = lambda: custom_openapi(app)  # type: ignore[assignment]

# Middleware（注册顺序：LIFO，先注册的后执行）
app.add_middleware(RequestIdMiddleware)
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 全局异常处理
register_exception_handlers(app)


def _run_alembic_upgrade() -> None:
    """在子线程中执行 alembic upgrade head，初始化/更新数据库表结构。"""
    from pathlib import Path

    from alembic.config import Config
    from alembic.command import upgrade

    api_dir = Path(__file__).resolve().parent.parent
    cfg = Config(str(api_dir / "alembic.ini"))
    upgrade(cfg, "head")


@app.on_event("startup")
async def startup_event():
    """应用启动事件。启动时自动执行 alembic upgrade head 初始化/更新数据库表结构。"""
    import asyncio

    logger.info("Life Echo API 正在启动...")

    try:
        await asyncio.to_thread(_run_alembic_upgrade)
        logger.info("Alembic 迁移已就绪")
    except Exception as e:
        logger.warning(
            "Alembic 迁移失败（可能数据库未启动或 DATABASE_URL 未配置）: {}", e
        )

    try:
        from app.core.redis import redis_service

        await redis_service.get_client()
        logger.info("Redis 连接已建立")
    except Exception as e:
        logger.warning("Redis 连接失败（会话存储将不可用）: {}", e)

    try:
        from app.core.dependencies import get_asr_provider

        provider = get_asr_provider()
        ensure_ready = getattr(provider, "ensure_ready", None)
        if callable(ensure_ready):
            asr_ready = await asyncio.to_thread(ensure_ready)
        else:
            asr_ready = True
        if asr_ready:
            from app.core.config import settings

            name = (
                "腾讯云一句话识别"
                if settings.asr_provider == "tencent"
                else "本地 Whisper"
            )
            logger.info("ASR 服务已就绪（{}）", name)
        else:
            logger.warning("ASR 服务未就绪，语音转写将不可用")
    except Exception as e:
        logger.warning("ASR 初始化检查失败: {}", e)

    try:

        def _init_wechat_pay_client():
            from app.features.payment.deps import get_payment_service

            svc = get_payment_service()
            if svc.is_method_available("wechat"):
                _ = svc.wechat_client

        await asyncio.to_thread(_init_wechat_pay_client)
        logger.info("微信支付客户端已预初始化")
    except Exception as e:
        logger.warning(f"微信支付预初始化失败（首次下单时再初始化）: {e}")


@app.on_event("shutdown")
async def shutdown_event():
    """应用关闭事件"""
    logger.info("Life Echo API 正在关闭...")

    try:
        from app.core.redis import redis_service

        await redis_service.close()
        logger.info("Redis 连接已关闭")
    except Exception as e:
        logger.warning("关闭 Redis 连接失败: {}", e)


# ── Feature routers ──────────────────────────────────────────
app.include_router(auth_router)
app.websocket("/ws/conversation/{conversation_id}")(websocket_endpoint)
app.include_router(conversation_router)
app.include_router(memoir_router)
app.include_router(user_router)
app.include_router(user_feedback_router)
app.include_router(plan_router)
app.include_router(payment_router)
app.include_router(quota_router)
app.include_router(tasks_router)
app.include_router(content_router)

# static 在 api/ 下，app/main.py 在 api/app/ 下
_static_dir = Path(__file__).resolve().parent.parent / "static"
app.mount("/static", StaticFiles(directory=_static_dir), name="static")


@app.get("/health", include_in_schema=False)
async def health():
    return {"status": "ok"}