from typing import TYPE_CHECKING

from pyrogram import Client, enums

if TYPE_CHECKING:
    from userbot.modules.capture import CaptureContext


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

        if load_handlers:
            from userbot import handlers  # noqa: PLC0415

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


__all__ = ["PyroClient"]