feat: add realtime capturing

backend/src/userbot/modules/client.py

@@ -1,8 +1,15 @@
+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") -> None:
+    def __init__(
+        self, name: str, *, workdir: str = "sessions", load_handlers: bool = True
+    ) -> None:
         super().__init__(
             name,
             workdir=workdir,
@@ -14,6 +21,13 @@ class PyroClient(Client):
             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"]