from contextlib import asynccontextmanager
from agentscope_runtime.engine import AgentApp
from database import init_db, async_engine
from modules.auth.router import router as auth_router
from modules.org.router import router as org_router
from modules.rbac.router import router as rbac_router
from modules.wecom.router import router as wecom_router
from modules.task.router import router as task_router
from modules.monitor.router import router as monitor_router
from modules.mcp_registry.router import router as mcp_router
from modules.flow_engine.router import router as flow_router
from modules.flow_engine.gateway import gateway_router
from modules.audit.router import router as audit_router
from modules.document.router import router as document_router
from modules.notification.router import router as notification_router
from modules.system.router import router as system_router
from modules.rag.router import router as rag_router
from modules.chat.router import router as chat_router
from modules.custom_tool.router import router as custom_tool_router
from modules.memory.router import router as memory_router
from modules.memory.manager import init_memory_manager, get_memory_manager
from modules.model_provider.router import router as model_provider_router
from websocket_manager import ws_manager
from middleware.rbac_middleware import rbac_middleware
from middleware.rate_limiter import rate_limit_middleware
from middleware.cache_manager import cache_manager
from database import AsyncSessionLocal


@asynccontextmanager
async def lifespan(app: AgentApp):
    """应用生命周期管理器，启动时初始化数据库和缓存，关闭时清理资源。"""
    await init_db()
    await cache_manager.connect()
    await init_memory_manager(AsyncSessionLocal)
    yield
    await cache_manager.disconnect()
    try:
        mm = get_memory_manager()
        await mm.redis.close()
    except Exception:
        pass
    await async_engine.dispose()


app = AgentApp(
    app_name="Enterprise AI Platform",  # 应用名称
    app_description="企业级 AI Agent 平台 - 双RBAC/企微集成/无代码流编排",  # 应用描述
    lifespan=lifespan,
    docs_url="/docs",
    redoc_url=None,
)

# 注册全局 HTTP 中间件
app.middleware("http")(rate_limit_middleware)  # 速率限制中间件
app.middleware("http")(rbac_middleware)  # RBAC 权限中间件

# 注册所有业务模块的路由
app.include_router(auth_router)  # 认证模块
app.include_router(org_router)  # 组织架构模块
app.include_router(rbac_router)  # 权限管理模块
app.include_router(wecom_router)  # 企业微信模块
app.include_router(task_router)  # 任务管理模块
app.include_router(monitor_router)  # 监控模块
app.include_router(mcp_router)  # MCP 服务注册模块
app.include_router(flow_router)  # 流程定义管理模块
app.include_router(gateway_router)  # 流程 API 网关模块
app.include_router(audit_router)  # 审计日志模块
app.include_router(document_router)  # 文档管理模块
app.include_router(notification_router)  # 通知模块
app.include_router(system_router)  # 系统设置模块
app.include_router(rag_router)  # 知识库模块
app.include_router(chat_router)  # 对话模块
app.include_router(custom_tool_router)  # 自定义工具模块
app.include_router(memory_router)  # 记忆管理模块
app.include_router(model_provider_router)  # 模型供应商管理模块