"""
Zhenxun Bot - 核心服务模块

主要服务包括：
- 数据库上下文 (db_context): 提供数据库模型基类和连接管理。
- 日志服务 (log): 提供增强的、带上下文的日志记录器。
- LLM服务 (llm): 提供与大语言模型交互的统一API。
- 插件生命周期管理 (plugin_init): 支持插件安装和卸载时的钩子函数。
- 定时任务调度器 (scheduler): 提供持久化的、可管理的定时任务服务。
"""

import asyncio

import nonebot
from nonebot import require

require("nonebot_plugin_apscheduler")
require("nonebot_plugin_alconna")
require("nonebot_plugin_session")
require("nonebot_plugin_htmlrender")
require("nonebot_plugin_uninfo")
require("nonebot_plugin_waiter")

from .avatar_service import avatar_service
from .db_context import Model, disconnect, with_db_timeout
from .group_settings_service import group_settings_service
from .llm import (
    AI,
    AIConfig,
    CommonOverrides,
    LLMContentPart,
    LLMException,
    LLMGenerationConfig,
    LLMMessage,
    chat,
    clear_model_cache,
    code,
    create_multimodal_message,
    embed,
    generate,
    generate_structured,
    get_cache_stats,
    get_model_instance,
    list_available_models,
    list_embedding_models,
    search,
    set_global_default_model_name,
)
from .log import logger
from .plugin_init import PluginInit, PluginInitManager
from .renderer import renderer_service
from .scheduler import (
    ExecutionPolicy,
    ScheduleContext,
    Trigger,
    scheduler_manager,
)

__all__ = [
    "AI",
    "AIConfig",
    "CommonOverrides",
    "ExecutionPolicy",
    "LLMContentPart",
    "LLMException",
    "LLMGenerationConfig",
    "LLMMessage",
    "Model",
    "PluginInit",
    "PluginInitManager",
    "ScheduleContext",
    "Trigger",
    "avatar_service",
    "chat",
    "clear_model_cache",
    "code",
    "create_multimodal_message",
    "disconnect",
    "embed",
    "generate",
    "generate_structured",
    "get_cache_stats",
    "get_model_instance",
    "group_settings_service",
    "list_available_models",
    "list_embedding_models",
    "logger",
    "renderer_service",
    "scheduler_manager",
    "search",
    "set_global_default_model_name",
    "with_db_timeout",
]


async def cancel_pending_tasks():
    """仅取消 zhenxun 业务协程，避免干扰框架的关机流程。"""
    loop = asyncio.get_running_loop()
    current = asyncio.current_task(loop=loop)
    pending = []
    for task in asyncio.all_tasks(loop):
        if task is current or task.done():
            continue
        coro = task.get_coro()
        module = getattr(coro, "__module__", "")
        if module.startswith("zhenxun"):
            pending.append(task)

    if not pending:
        return

    for task in pending:
        task.cancel()

    await asyncio.gather(*pending, return_exceptions=True)


driver = nonebot.get_driver()
# 先取消可能在跑的任务，再断开数据库，避免 pool closing 异常
driver.on_shutdown(cancel_pending_tasks)