"""Main entry point for Telegram bot with MCP integration."""
import asyncio
import logging
import signal
import sys
from logger import setup_logging
from mcp_manager import MCPManager
from bot import TelegramBot
from scheduler import PeriodicTaskMonitor
logger = logging.getLogger(__name__)
class Application:
"""Main application orchestrator."""
def __init__(self):
self.mcp_manager: MCPManager = None
self.bot: TelegramBot = None
self.scheduler: PeriodicTaskMonitor = None
self.shutdown_event = asyncio.Event()
async def startup(self) -> None:
"""Initialize and start all components."""
logger.info("Application startup initiated")
self.mcp_manager = MCPManager()
logger.info("Connecting to MCP server")
self.mcp_context = self.mcp_manager.connect()
await self.mcp_context.__aenter__()
self.bot = TelegramBot(self.mcp_manager)
self.bot.initialize()
await self.bot.run()
# Initialize and start scheduler after bot is running
logger.info("Initializing periodic task monitor")
self.scheduler = PeriodicTaskMonitor(self.bot.application.bot, self.mcp_manager)
await self.scheduler.start()
logger.info("Application startup completed")
async def shutdown(self) -> None:
"""Gracefully shutdown all components."""
logger.info("Application shutdown initiated")
if self.scheduler:
logger.info("Stopping periodic task monitor")
await self.scheduler.stop()
if self.bot:
await self.bot.stop()
if self.mcp_context:
await self.mcp_context.__aexit__(None, None, None)
logger.info("Application shutdown completed")
async def run(self) -> None:
"""Run the application until shutdown signal."""
try:
await self.startup()
await self.shutdown_event.wait()
except Exception as e:
logger.error(f"Application error: {e}", exc_info=True)
finally:
await self.shutdown()
def signal_handler(self, signum, frame) -> None:
"""Handle shutdown signals."""
logger.info(f"Received signal {signum}, initiating shutdown")
self.shutdown_event.set()
async def main() -> None:
"""Application entry point."""
setup_logging(level=logging.INFO)
logger.info("=== Telegram Task Tracker Bot Starting ===")
app = Application()
signal.signal(signal.SIGINT, lambda s, f: app.signal_handler(s, f))
signal.signal(signal.SIGTERM, lambda s, f: app.signal_handler(s, f))
try:
await app.run()
except KeyboardInterrupt:
logger.info("Keyboard interrupt received")
except Exception as e:
logger.error(f"Fatal error: {e}", exc_info=True)
sys.exit(1)
logger.info("=== Telegram Task Tracker Bot Stopped ===")
if __name__ == "__main__":
asyncio.run(main())
