from hydrogram import Client
from logging import getLogger
from logging.config import dictConfig
from .config import Telegram

LOGGER_CONFIG_JSON = {
    "version": 1,
    "formatters": {
        "default": {
            "format": "[%(asctime)s][%(name)s][%(levelname)s] -> %(message)s",
            "datefmt": "%d/%m/%Y %H:%M:%S",
        },
    },
    "handlers": {
        "stream_handler": {"class": "logging.StreamHandler", "formatter": "default"},
    },
    "loggers": {
        "uvicorn": {"level": "INFO", "handlers": ["stream_handler"]},
        "uvicorn.error": {"level": "WARNING", "handlers": ["stream_handler"]},
        "bot": {"level": "INFO", "handlers": ["stream_handler"]},
        "hydrogram": {"level": "INFO", "handlers": ["stream_handler"]},
    },
}

dictConfig(LOGGER_CONFIG_JSON)
logger = getLogger("bot")

if not Telegram.SESSION_STRING:
    raise RuntimeError("SESSION_STRING missing. Put it in HF Secrets/Variables.")

TelegramBot = Client(
    name="bot",
    api_id=Telegram.API_ID,
    api_hash=Telegram.API_HASH,
    session_string=Telegram.SESSION_STRING,
    plugins={"root": "bot/plugins"},
    sleep_threshold=-1,
    max_concurrent_transmissions=10,
)