main.py

"""Telegram Bot Template — main entry point with dynamic handler loading.

Replace BOT_NAME references and customize handlers/ for your project.
"""

import importlib
import logging
import sys
import os

from telegram.ext import Application

import config
from shared.db import init_pool, close_pool

logging.basicConfig(
    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
    level=logging.INFO,
)
logger = logging.getLogger(__name__)

# ── Handler modules to load ──
# Each module must implement: def register(app: Application)
# Organized by phase for incremental deployment.
HANDLER_MODULES = [
    # Phase 1 — Core
    "handlers.help",
    "handlers.welcome",
    "handlers.logging_handler",
    "handlers.tickets",
    "handlers.announce",
    # Phase 2 — Engagement
    "handlers.leveling",
    "handlers.giveaway",
    # Add more handlers as needed:
    # "handlers.your_handler_name",
]


async def post_init(application: Application):
    """Called after bot is initialized but before polling starts."""
    await init_pool(config.DATABASE_URL)
    logger.info("Database pool initialized")


async def post_shutdown(application: Application):
    """Called after polling stops."""
    await close_pool()
    logger.info("Database pool closed")


def main():
    if not config.BOT_TOKEN:
        logger.error("BOT_TOKEN not set in .env")
        sys.exit(1)

    app = (
        Application.builder()
        .token(config.BOT_TOKEN)
        .post_init(post_init)
        .post_shutdown(post_shutdown)
        .build()
    )

    # Dynamically load handler modules
    for module_name in HANDLER_MODULES:
        try:
            mod = importlib.import_module(module_name)
            if hasattr(mod, "register"):
                mod.register(app)
                logger.info(f"Loaded: {module_name}")
            else:
                logger.warning(f"No register() in {module_name}, skipped")
        except Exception as e:
            logger.error(f"Failed to load {module_name}: {e}")

    logger.info("Starting bot...")
    app.run_polling(drop_pending_updates=True)


if __name__ == "__main__":
    main()