фиксы апи, добавли typing контекст менеджер ну и по мелочи

Author: ink-developer

src/pymax/interfaces.py

@@ -310,7 +310,7 @@ async def _handle_file_upload(self, data: dict[str, Any]) -> None:
                     self.logger.debug("Fulfilled file upload waiter for %s=%s", key, id_)
 
     async def _send_notification_response(self, chat_id: int, message_id: str) -> None:
-        if self._socket is None or not self.is_connected:
+        if self._socket is not None:
             return
         await self._queue_message(
             opcode=Opcode.NOTIF_MESSAGE,

src/pymax/mixins/socket.py

@@ -481,3 +481,21 @@ async def _get_chat(self, chat_id: int) -> Chat | None:
             if chat.id == chat_id:
                 return chat
         return None
+
+    async def _send_only(self, opcode: Opcode, payload: dict[str, Any], cmd: int = 0) -> None:
+        async def send_task():
+            async with self._sock_lock:
+                if not self.is_connected or self._socket is None:
+                    return
+                msg = self._make_message(opcode, payload, cmd)
+                packet = self._pack_packet(
+                    msg["ver"],
+                    msg["cmd"],
+                    msg["seq"],
+                    msg["opcode"],
+                    msg["payload"],
+                )
+                loop = asyncio.get_running_loop()
+                await loop.run_in_executor(None, lambda: self._socket.sendall(packet))
+
+        asyncio.create_task(send_task())

src/pymax/mixins/user.py

@@ -1,3 +1,5 @@
+from collections.abc import AsyncGenerator
+from contextlib import asynccontextmanager
 from typing import Any, Literal
 
 from pymax.exceptions import Error, ResponseError, ResponseStructureError
@@ -7,9 +9,10 @@
     ContactPresencePayload,
     FetchContactsPayload,
     SearchByPhonePayload,
+    SetTypingPayload,
 )
 from pymax.protocols import ClientProtocol
-from pymax.static.enum import ContactAction, Opcode
+from pymax.static.enum import ContactAction, Opcode, TypingType
 from pymax.types import Contact, Presence, Session, User
 from pymax.utils import MixinsUtils
 
@@ -267,3 +270,28 @@ async def get_contact_presence(self, contact_ids: list[int]) -> list[Presence]:
             Presence(user_id=int(contact_id), last_seen=info.get("seen"))
             for contact_id, info in presence.items()
         ]
+
+    @asynccontextmanager
+    async def typing(
+        self, chat_id: int, typing_type: TypingType = TypingType.TEXT
+    ) -> AsyncGenerator[None, Any]:
+        """
+        Устанавливает состояние "печатает" для указанного чата.
+
+        :param chat_id: ID чата.
+        :type chat_id: int
+        :param typing_type: Тип состояния "печатает".
+        :type typing_type: TypingType
+        :yields: None
+        :rtype: None
+        """
+        self.logger.debug("Set typing for chat %s", chat_id)
+
+        payload = SetTypingPayload(
+            chat_id=chat_id,
+            type=typing_type,
+        ).model_dump(by_alias=True)
+        await self._send_only(opcode=Opcode.MSG_TYPING, payload=payload)
+        self.logger.debug("Set typing for chat %s success (maybe?)", chat_id)
+        yield
+        self.logger.debug("Set typing for chat %s done", chat_id)

src/pymax/mixins/websocket.py

@@ -152,3 +152,9 @@ async def _get_chat(self, chat_id: int) -> Chat | None:
             if chat.id == chat_id:
                 return chat
         return None
+
+    async def _send_only(self, opcode: Opcode, payload: dict[str, Any], cmd: int = 0) -> None:
+        msg = self._make_message(opcode, payload, cmd)
+        packet = json.dumps(msg)
+
+        asyncio.create_task(self.ws.send(packet))

src/pymax/payloads.py

@@ -15,7 +15,14 @@
     DEFAULT_TIMEZONE,
     DEFAULT_USER_AGENT,
 )
-from pymax.static.enum import AttachType, AuthType, Capability, ContactAction, ReadAction
+from pymax.static.enum import (
+    AttachType,
+    AuthType,
+    Capability,
+    ContactAction,
+    ReadAction,
+    TypingType,
+)
 
 
 def to_camel(string: str) -> str:
@@ -413,3 +420,8 @@ class ContactPresencePayload(CamelModel):
 
 class ApproveQrLoginPayload(CamelModel):
     qr_link: str
+
+
+class SetTypingPayload(CamelModel):
+    chat_id: int
+    typing_type: TypingType = Field(default=TypingType.TEXT, alias="type")

src/pymax/protocols.py

@@ -126,3 +126,7 @@ def _create_safe_task(
         self, coro: Awaitable[Any], name: str | None = None
     ) -> asyncio.Task[Any]:
         pass
+
+    @abstractmethod
+    async def _send_only(self, opcode: Opcode, payload: dict[str, Any], cmd: int = 0) -> None:
+        pass

src/pymax/static/constant.py

@@ -80,7 +80,7 @@
 DEFAULT_SCREEN: Final[str] = choice(SCREEN_SIZES)
 DEFAULT_OS_VERSION: Final[str] = choice(OS_VERSIONS)
 DEFAULT_USER_AGENT: Final[str] = ua_generator.generate().text
-DEFAULT_BUILD_NUMBER: Final[int] = 0x9E2A
+DEFAULT_BUILD_NUMBER: Final[int] = 40490
 DEFAULT_CLIENT_SESSION_ID: Final[int] = randint(1, 15)
 DEFAULT_TIMEZONE: Final[str] = choice(TIMEZONES)
 DEFAULT_CHAT_MEMBERS_LIMIT: Final[int] = 50

src/pymax/static/enum.py

@@ -231,3 +231,12 @@ class Capability(int, Enum):
     SECOND_FACTOR_PASSWORD_ENABLED = 2
     SECOND_FACTOR_HAS_EMAIL = 3
     SECOND_FACTOR_HAS_HINT = 4
+
+
+class TypingType(str, Enum):
+    TEXT = "TEXT"
+    AUDIO = "AUDIO"
+    FILE = "FILE"
+    PHOTO = "PHOTO"
+    STICKER = "STICKER"
+    VIDEO = "VIDEO"