feat: Implement new bot architecture including handlers, services, analysis modules, and comprehensive tests.

2026-03-12 11:44:52 +08:00
parent c582015163
commit f4fea03f35
28 changed files with 1525 additions and 445 deletions
@@ -0,0 +1,93 @@
+from __future__ import annotations
+
+import os
+from typing import Any
+
+from loguru import logger  # type: ignore
+
+from src.bot.analysis.city_analysis_service import CityAnalysisService
+from src.bot.analysis.deb_analysis_service import DebAnalysisService
+from src.bot.command_guard import CommandGuard
+from src.bot.handlers.activity import ActivityHandler
+from src.bot.handlers.basic import BasicCommandHandler
+from src.bot.handlers.city import CityCommandHandler
+from src.bot.handlers.deb import DebCommandHandler
+from src.bot.io_layer import BotIOLayer
+from src.bot.runtime_coordinator import StartupCoordinator
+from src.bot.services.city_command_service import CityCommandService
+from src.bot.services.deb_command_service import DebCommandService
+from src.bot.services.entitlement_service import BotEntitlementService
+
+
+def _project_root() -> str:
+    return os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+
+def _register_handlers(
+    bot: Any,
+    io_layer: BotIOLayer,
+    guard: CommandGuard,
+    city_service: CityCommandService,
+    deb_service: DebCommandService,
+    startup_coordinator: StartupCoordinator,
+) -> None:
+    BasicCommandHandler(
+        bot=bot,
+        io_layer=io_layer,
+        runtime_status_provider=startup_coordinator.get_runtime_status,
+    ).register()
+    CityCommandHandler(bot=bot, guard=guard, city_service=city_service).register()
+    DebCommandHandler(bot=bot, guard=guard, deb_service=deb_service).register()
+    ActivityHandler(bot=bot, io_layer=io_layer).register()
+
+
+def start_bot() -> None:
+    import telebot  # type: ignore
+
+    from src.data_collection.weather_sources import WeatherDataCollector
+    from src.database.db_manager import DBManager
+    from src.utils.config_loader import load_config
+
+    config = load_config()
+    token = os.getenv("TELEGRAM_BOT_TOKEN")
+    if not token:
+        logger.error("未找到 TELEGRAM_BOT_TOKEN 环境变量")
+        return
+
+    bot = telebot.TeleBot(token)
+    db = DBManager()
+    weather = WeatherDataCollector(config)
+
+    io_layer = BotIOLayer(bot=bot, db=db)
+    city_analysis = CityAnalysisService(weather=weather)
+    deb_analysis = DebAnalysisService(project_root=_project_root())
+    entitlement_service = BotEntitlementService(db=db)
+    guard = CommandGuard(io_layer=io_layer, entitlement_service=entitlement_service)
+    city_service = CityCommandService(analysis=city_analysis)
+    deb_service = DebCommandService(analysis=deb_analysis)
+    startup_coordinator = StartupCoordinator(
+        bot=bot,
+        config=config,
+        entitlement_enabled=entitlement_service.enabled,
+        protected_commands=sorted(entitlement_service.protected_commands),
+    )
+
+    _register_handlers(
+        bot=bot,
+        io_layer=io_layer,
+        guard=guard,
+        city_service=city_service,
+        deb_service=deb_service,
+        startup_coordinator=startup_coordinator,
+    )
+    runtime_status = startup_coordinator.start_all()
+    started_count = sum(1 for loop in runtime_status.loops if loop.started)
+
+    logger.info(
+        "🤖 Bot 启动中... entitlement_enabled={} protected_commands={} loops_started={}/{}",
+        entitlement_service.enabled,
+        ",".join(sorted(entitlement_service.protected_commands)),
+        started_count,
+        len(runtime_status.loops),
+    )
+    bot.infinity_polling()