main.py

"""QJinEra — 拟人化赛博群友 QQBot 入口。"""

from __future__ import annotations

import asyncio

from alicebot import Bot

from config import config
from logger import get_logger, setup_logging, shutdown_logging
from services.scheduler import start_scheduler_loop
from services.storage.database import init_db
from utils import wait_all_tasks

logger = get_logger("Main")


async def main() -> None:
    # 1. 初始化日志
    setup_logging(level=config.log.level, log_dir=config.log.log_dir)
    logger.info("启动 %s (%s) ...", config.bot.name, config.bot.english_name)

    # 2. 初始化数据库
    init_db()
    logger.info("数据库就绪: %s", config.storage.database_file)

    # 3. 启动 AliceBot
    bot = Bot()

    @bot.bot_run_hook
    async def start_background_tasks(b: Bot) -> None:
        """注册并启动后台定时任务。"""
        asyncio.create_task(start_scheduler_loop(b))

    try:
        await bot.run_async()
    finally:
        await wait_all_tasks()
        shutdown_logging()


if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\n[System] Bot 已被手动停止。")