from typing import TYPE_CHECKING

from pyrogram import Client, enums

if TYPE_CHECKING:
    from userbot.modules.capture import CaptureContext


DEVICE_MODEL = "Beavergram"
DEVICE_MODEL_LIMIT = 32


class PyroClient(Client):
    def __init__(
        self,
        name: str,
        *,
        workdir: str = "sessions",
        device_model: str | None = None,
        load_handlers: bool = True,
    ) -> None:
        super().__init__(
            name,
            workdir=workdir,
            api_id=2040,
            api_hash="b18441a1ff607e10a989891a5462e627",
            device_model=device_model or DEVICE_MODEL,
            system_version="Windows 11 x64",
            app_version="7.0.8 x64",
            lang_pack="tdesktop",
            lang_code="en",
            system_lang_code="en-US",
            client_platform=enums.ClientPlatform.DESKTOP,
            max_concurrent_transmissions=4,
        )
        self.capture: CaptureContext | None = None

        if load_handlers:
            from userbot import handlers  # noqa: PLC0415

            for handler in handlers.handlers:
                self.add_handler(*handler)


__all__ = ["DEVICE_MODEL", "DEVICE_MODEL_LIMIT", "PyroClient"]