maxapi-python 2.3.1__py3-none-any.whl → 2.4.1__py3-none-any.whl

pymax/_data/rootca_ssl_rsa2022.crt

@@ -0,0 +1,33 @@
+-----BEGIN CERTIFICATE-----
+MIIFwjCCA6qgAwIBAgICEAAwDQYJKoZIhvcNAQELBQAwcDELMAkGA1UEBhMCUlUx
+PzA9BgNVBAoMNlRoZSBNaW5pc3RyeSBvZiBEaWdpdGFsIERldmVsb3BtZW50IGFu
+ZCBDb21tdW5pY2F0aW9uczEgMB4GA1UEAwwXUnVzc2lhbiBUcnVzdGVkIFJvb3Qg
+Q0EwHhcNMjIwMzAxMjEwNDE1WhcNMzIwMjI3MjEwNDE1WjBwMQswCQYDVQQGEwJS
+VTE/MD0GA1UECgw2VGhlIE1pbmlzdHJ5IG9mIERpZ2l0YWwgRGV2ZWxvcG1lbnQg
+YW5kIENvbW11bmljYXRpb25zMSAwHgYDVQQDDBdSdXNzaWFuIFRydXN0ZWQgUm9v
+dCBDQTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAMfFOZ8pUAL3+r2n
+qqE0Zp52selXsKGFYoG0GM5bwz1bSFtCt+AZQMhkWQheI3poZAToYJu69pHLKS6Q
+XBiwBC1cvzYmUYKMYZC7jE5YhEU2bSL0mX7NaMxMDmH2/NwuOVRj8OImVa5s1F4U
+zn4Kv3PFlDBjjSjXKVY9kmjUBsXQrIHeaqmUIsPIlNWUnimXS0I0abExqkbdrXbX
+YwCOXhOO2pDUx3ckmJlCMUGacUTnylyQW2VsJIyIGA8V0xzdaeUXg0VZ6ZmNUr5Y
+Ber/EAOLPb8NYpsAhJe2mXjMB/J9HNsoFMBFJ0lLOT/+dQvjbdRZoOT8eqJpWnVD
+U+QL/qEZnz57N88OWM3rabJkRNdU/Z7x5SFIM9FrqtN8xewsiBWBI0K6XFuOBOTD
+4V08o4TzJ8+Ccq5XlCUW2L48pZNCYuBDfBh7FxkB7qDgGDiaftEkZZfApRg2E+M9
+G8wkNKTPLDc4wH0FDTijhgxR3Y4PiS1HL2Zhw7bD3CbslmEGgfnnZojNkJtcLeBH
+BLa52/dSwNU4WWLubaYSiAmA9IUMX1/RpfpxOxd4Ykmhz97oFbUaDJFipIggx5sX
+ePAlkTdWnv+RWBxlJwMQ25oEHmRguNYf4Zr/Rxr9cS93Y+mdXIZaBEE0KS2iLRqa
+OiWBki9IMQU4phqPOBAaG7A+eP8PAgMBAAGjZjBkMB0GA1UdDgQWBBTh0YHlzlpf
+BKrS6badZrHF+qwshzAfBgNVHSMEGDAWgBTh0YHlzlpfBKrS6badZrHF+qwshzAS
+BgNVHRMBAf8ECDAGAQH/AgEEMA4GA1UdDwEB/wQEAwIBhjANBgkqhkiG9w0BAQsF
+AAOCAgEAALIY1wkilt/urfEVM5vKzr6utOeDWCUczmWX/RX4ljpRdgF+5fAIS4vH
+tmXkqpSCOVeWUrJV9QvZn6L227ZwuE15cWi8DCDal3Ue90WgAJJZMfTshN4OI8cq
+W9E4EG9wglbEtMnObHlms8F3CHmrw3k6KmUkWGoa+/ENmcVl68u/cMRl1JbW2bM+
+/3A+SAg2c6iPDlehczKx2oa95QW0SkPPWGuNA/CE8CpyANIhu9XFrj3RQ3EqeRcS
+AQQod1RNuHpfETLU/A2gMmvn/w/sx7TB3W5BPs6rprOA37tutPq9u6FTZOcG1Oqj
+C/B7yTqgI7rbyvox7DEXoX7rIiEqyNNUguTk/u3SZ4VXE2kmxdmSh3TQvybfbnXV
+4JbCZVaqiZraqc7oZMnRoWrXRG3ztbnbes/9qhRGI7PqXqeKJBztxRTEVj8ONs1d
+WN5szTwaPIvhkhO3CO5ErU2rVdUr89wKpNXbBODFKRtgxUT70YpmJ46VVaqdAhOZ
+D9EUUn4YaeLaS8AjSF/h7UkjOibNc4qVDiPP+rkehFWM66PVnP1Msh93tc+taIfC
+EYVMxjh8zNbFuoc7fzvvrFILLe7ifvEIUqSVIC/AzplM/Jxw7buXFeGP1qVCBEHq
+391d/9RAfaZ12zkwFsl+IKwE/OZxW8AHa9i1p4GO0YSNuczzEm4=
+-----END CERTIFICATE-----

pymax/api/auth/enums.py

@@ -6,6 +6,7 @@ class AuthType(str, Enum):
     CHECK_CODE = "CHECK_CODE"
     REGISTER = "REGISTER"
     RESEND = "RESEND"
+    LOGIN = "LOGIN"
 
 
 class ProfileOptions(int, Enum):

pymax/api/auth/payloads.py

@@ -10,12 +10,13 @@ from .enums import AuthType, TwoFactorAction
 class RequestCodePayload(CamelModel):
     phone: str
     type: AuthType = AuthType.START_AUTH
-    language: str = "ru"
+    mode: bytes | None
 
 
 class SendCodePayload(CamelModel):
     token: str
     verify_code: str
+
     auth_token_type: AuthType = AuthType.CHECK_CODE
 
 
@@ -42,19 +43,21 @@ class WebSyncPayload(CamelModel):
         cls,
         token: str,
         sync: SyncState,
+        interactive: bool = True,
     ) -> "WebSyncPayload":
         return cls(
             token=token,
             chats_sync=sync.chats_sync,
             contacts_sync=sync.contacts_sync,
             drafts_sync=sync.drafts_sync,
+            interactive=interactive,
         )
 
 
 class SyncPayload(CamelModel):
     user_agent: MobileUserAgentPayload
     token: str
-    chat_hash_fingerprint: str | None = None
+    chat_cache_fingerprint: bytes | None = None
     chats_count: int | None = None
     chats_sync: int = -1
     contacts_sync: int = -1
@@ -70,15 +73,19 @@ class SyncPayload(CamelModel):
         user_agent: MobileUserAgentPayload,
         token: str,
         sync: SyncState,
+        chat_cache_fingerprint: bytes | None = None,
+        interactive: bool = True,
     ) -> "SyncPayload":
         return cls(
             user_agent=user_agent,
             token=token,
+            chat_cache_fingerprint=chat_cache_fingerprint,
             chats_sync=sync.chats_sync,
             contacts_sync=sync.contacts_sync,
             drafts_sync=sync.drafts_sync,
             presence_sync=sync.presence_sync,
             config_hash=sync.config_hash,
+            interactive=interactive,
         )
 
 
@@ -138,3 +145,22 @@ class ConfirmRegistrationPayload(CamelModel):
     last_name: str | None = None
     token: str
     token_type: AuthType = AuthType.REGISTER
+
+
+class Login2Payload(CamelModel):
+    need_profile: bool
+    contacts_sync: int
+    config_hash: ConfigHash
+
+    @classmethod
+    def from_sync_state(
+        cls,
+        sync: SyncState,
+        profile_enabled: bool,
+        contact_enabled: bool,
+    ) -> "Login2Payload":
+        return cls(
+            need_profile=profile_enabled,
+            contacts_sync=sync.contacts_sync if contact_enabled else -1,
+            config_hash=sync.config_hash,
+        )

pymax/api/auth/service.py

@@ -13,6 +13,7 @@ from pymax.auth import EmailCodeProvider
 from pymax.auth.providers import ConsoleEmailCodeProvider
 from pymax.logging import get_logger
 from pymax.protocol import Opcode
+from pymax.types.domain import Login2Flags, Login2Response
 from pymax.types.domain.auth import (
     CheckCodeResponse,
     CheckPasswordResponse,
@@ -31,6 +32,7 @@ from .payloads import (
     ConfirmQrPayload,
     ConfirmRegistrationPayload,
     CreateAuthTrackPayload,
+    Login2Payload,
     MobileUserAgentPayload,
     RemoveTwoFactorPayload,
     RequestCodePayload,
@@ -58,7 +60,38 @@ class AuthService:
 
     async def request_code(self, phone: str) -> StartAuthResponse:
         logger.info("requesting sms code phone_set=%s", bool(phone))
-        frame = RequestCodePayload(phone=phone)
+
+        if not self.app.handshake_response:
+            logger.error("request_code requested without handshake response")
+            raise RuntimeError("No handshake response available for request_code")
+
+        device_id = (
+            self.app.session.device_id if self.app.session else self.app.config.device.device_id
+        )
+
+        if self.app.config.device.user_agent.device_type != DeviceType.WEB:
+            if self.app.handshake_response.calls_seed is None:
+                raise ValueError(
+                    "Unexpected internal state: handshake_response.calls_seed is missing "
+                    + "in AuthService.request_code. Please report this issue to the developer."
+                )
+
+            if self.app.fingerprint_generator is None:
+                raise ValueError(
+                    "Unexpected internal state: self.app.fingerprint_generator is missing and "
+                    + "DeviceType != WEB"
+                    + "in AuthService.request_code. Please report this issue to the developer."
+                )
+
+            mode = self.app.fingerprint_generator.generate_fingerprint(
+                device_id=device_id,
+                calls_seed=self.app.handshake_response.calls_seed,
+                arch=self.app.config.device.user_agent.arch or "arm64-v8a",
+            )
+        else:
+            mode = None
+
+        frame = RequestCodePayload(phone=phone, mode=mode)
         response = await self.app.invoke(Opcode.AUTH_REQUEST, frame.to_payload())
         logger.debug(
             "sms code request accepted payload_keys=%s",
@@ -72,7 +105,11 @@ class AuthService:
             bool(token),
             bool(verify_code),
         )
-        frame = SendCodePayload(token=token, verify_code=verify_code)
+
+        frame = SendCodePayload(
+            token=token,
+            verify_code=verify_code,
+        )
         response = await self.app.invoke(Opcode.AUTH, frame.to_payload())
         logger.debug(
             "sms code response payload_keys=%s",
@@ -117,11 +154,41 @@ class AuthService:
             raise RuntimeError("No session available for login")
 
         logger.info("logging in")
+
+        if not self.app.handshake_response:
+            logger.error("login requested without handshake response")
+            raise RuntimeError("No handshake response available for login")
+
+        device_id = (
+            self.app.session.device_id if self.app.session else self.app.config.device.device_id
+        )
+
+        if self.app.handshake_response.calls_seed is None:
+            raise ValueError(
+                "Unexpected internal state: handshake_response.calls_seed is missing "
+                + "in AuthService.mobile_login. Please report this issue to the developer."
+            )
+
+        if self.app.fingerprint_generator is None:
+            raise ValueError(
+                "Unexpected internal state: self.app.fingerprint_generator is missing and "
+                + "DeviceType != WEB"
+                + "in AuthService.request_code. Please report this issue to the developer."
+            )
+
+        ccf = self.app.fingerprint_generator.generate_fingerprint(
+            device_id=device_id,
+            calls_seed=self.app.handshake_response.calls_seed,
+            arch=self.app.config.device.user_agent.arch or "arm64-v8a",
+        )
+
         sync = self.app.config.sync.resolve(session.sync)
         frame = SyncPayload.from_sync_state(
             user_agent=self.app.config.device.user_agent,
             token=session.token,
             sync=sync,
+            chat_cache_fingerprint=ccf,
+            interactive=self.app.config.interactive,
         )
         response = await self.app.invoke(Opcode.LOGIN, frame.to_payload())
 
@@ -134,6 +201,30 @@ class AuthService:
         await self._update_session(login_response)
         return login_response
 
+    async def mobile_login2(self, flags: Login2Flags) -> Login2Response:
+        session = self.app.session
+        if session is None:
+            logger.error("login2 requested without session")
+            raise RuntimeError("No session available for login2")
+
+        sync = self.app.config.sync.resolve(session.sync)
+
+        frame = Login2Payload.from_sync_state(
+            sync,
+            profile_enabled=flags.profile_enabled,
+            contact_enabled=flags.contact_enabled,
+        )
+
+        response = await self.app.invoke(Opcode.LOGIN2, frame.to_payload())
+        logger.debug("login2 response payload_keys=%s", payload_keys(response))
+
+        login2_response = bind_api_model(
+            self.app,
+            require_payload_model(response, Login2Response),
+        )
+        await self._update_session(login2_response)
+        return login2_response
+
     async def web_login(self) -> LoginResponse:
         session = self.app.session
         if session is None:
@@ -146,6 +237,7 @@ class AuthService:
         frame = WebSyncPayload.from_sync_state(
             token=session.token,
             sync=sync,
+            interactive=self.app.config.interactive,
         )
         response = await self.app.invoke(Opcode.LOGIN, frame.to_payload())
 
@@ -177,7 +269,7 @@ class AuthService:
 
         return require_payload_model(response, CheckCodeResponse)
 
-    async def _update_session(self, response: LoginResponse) -> None:
+    async def _update_session(self, response: LoginResponse | Login2Response) -> None:
         session = self.app.session
         if session is None:
             return
@@ -377,3 +469,8 @@ class AuthService:
         response = await self.app.invoke(Opcode.AUTH_CONFIRM, frame.to_payload())
 
         return require_payload_model(response, ConfirmRegistrationResponse)
+
+    def is_update_available(self) -> bool:
+        return bool(
+            self.app.handshake_response.app_update_type if self.app.handshake_response else False
+        )

pymax/api/chats/__init__.py

@@ -1,4 +1,5 @@
 from .enums import (
+    ChannelPermissions,
     ChatLinkPrefix,
     ChatMemberOperation,
     ChatOption,

pymax/api/chats/enums.py

@@ -22,7 +22,22 @@ class ChatPayloadKey(str, Enum):
     CHAT = "chat"
     CHATS = "chats"
     MEMBERS = "members"
+    MARKER = "marker"
 
 
 class ChatLinkPrefix(str, Enum):
     JOIN = "join/"
+
+
+class ChannelPermissions(int, Enum):
+    ADD_REMOVE_MEMBER = 2
+    ADD_ADMIN = 4
+    CHANGE_CHAT_INFO = 8
+    PIN_MESSAGE = 16
+    POST_MESSAGE = 256
+    EDIT_MESSAGE = 512
+    DELETE_MESSAGE = 1024
+
+
+class PermType(str, Enum):
+    ADMIN = "ADMIN"

pymax/api/chats/payloads.py

@@ -6,7 +6,7 @@ from pymax.api.models import CamelModel
 from pymax.types.domain.attachments.enums import AttachmentType
 from pymax.types.domain.enums import ChatType
 
-from .enums import ChatMemberOperation, ChatOption, ControlEvent
+from .enums import ChatMemberOperation, ChatOption, ControlEvent, PermType
 
 
 class CreateGroupAttach(CamelModel):
@@ -74,8 +74,9 @@ class ChangeGroupSettingsPayload(CamelModel):
 
 class ChangeGroupProfilePayload(CamelModel):
     chat_id: int
-    theme: str | None
+    theme: str | None = None
     description: str | None = None
+    photo_token: str | None = None
 
 
 class JoinChatPayload(CamelModel):
@@ -95,6 +96,13 @@ class GetChatInfoPayload(CamelModel):
     chat_ids: list[int]
 
 
+class GetChatMembersPayload(CamelModel):
+    type: str = "MEMBER"  # TODO: ENUMM!!!
+    chat_id: int
+    marker: int
+    count: int = 50
+
+
 class LeaveChatPayload(CamelModel):
     chat_id: int
 
@@ -121,3 +129,11 @@ class DeleteChatPayload(CamelModel):
     chat_id: int
     last_event_time: int
     for_all: bool = True
+
+
+class AddAdminPayload(CamelModel):
+    chat_id: int
+    user_ids: list[int]
+    type: PermType = PermType.ADMIN
+    operation: str = "add"
+    permissions: int

pymax/api/chats/service.py

@@ -1,22 +1,27 @@
 from __future__ import annotations
 
 import time
+from functools import reduce
+from operator import or_
 from typing import TYPE_CHECKING
 
 from pymax.api.binding import bind_api_model
 from pymax.api.response import (
     parse_payload_item_model,
     parse_payload_list,
+    payload_item,
     require_payload_item_model,
     require_payload_model,
 )
 from pymax.exceptions import PyMaxError
+from pymax.files import Photo
 from pymax.logging import get_logger
 from pymax.protocol import Opcode
 from pymax.types.domain import Chat, Member, Message
 
-from .enums import ChatLinkPrefix, ChatMemberOperation, ChatPayloadKey
+from .enums import ChannelPermissions, ChatLinkPrefix, ChatMemberOperation, ChatPayloadKey
 from .payloads import (
+    AddAdminPayload,
     ChangeGroupProfilePayload,
     ChangeGroupSettingsOptions,
     ChangeGroupSettingsPayload,
@@ -27,6 +32,7 @@ from .payloads import (
     FetchChatsPayload,
     FetchJoinRequests,
     GetChatInfoPayload,
+    GetChatMembersPayload,
     InviteUsersPayload,
     JoinChatPayload,
     JoinRequestActionPayload,
@@ -206,13 +212,19 @@ class ChatService:
         chat_id: int,
         name: str | None,
         description: str | None = None,
+        photo: Photo | None = None,
     ) -> None:
+
         frame = ChangeGroupProfilePayload(
             chat_id=chat_id,
             theme=name,
             description=description,
         )
 
+        if photo:
+            photo_payload = await self.app.api.uploads.upload_photo(photo)
+            frame.photo_token = photo_payload.photo_token
+
         response = await self.app.invoke(Opcode.CHAT_UPDATE, frame.to_payload())
         chat = parse_payload_item_model(response, ChatPayloadKey.CHAT, Chat)
         if chat:
@@ -266,6 +278,22 @@ class ChatService:
 
         return [cached[chat_id] for chat_id in chat_ids if chat_id in cached]
 
+    async def get_chat_members(
+        self,
+        chat_id: int,
+        marker: int | None = None,
+        count: int = 50,
+    ) -> tuple[list[Member], int]:
+        frame = GetChatMembersPayload(chat_id=chat_id, marker=marker or 0, count=count)
+        response = await self.app.invoke(Opcode.CHAT_MEMBERS, frame.to_payload())
+
+        members = bind_api_model(
+            self.app,
+            parse_payload_list(response, ChatPayloadKey.MEMBERS, Member),
+        )
+        next_marker = payload_item(response, ChatPayloadKey.MARKER, int) or 0
+        return members, next_marker
+
     async def get_chat(self, chat_id: int) -> Chat:
         chats = await self.get_chats([chat_id])
         if not chats:
@@ -380,3 +408,17 @@ class ChatService:
 
         await self.app.invoke(Opcode.CHAT_DELETE, frame.to_payload())
         self._remove_cached_chat(chat_id)
+
+    async def add_admin(
+        self,
+        chat_id: int,
+        user_id: int,
+        permissions: list[ChannelPermissions],
+    ) -> None:
+        frame = AddAdminPayload(
+            chat_id=chat_id,
+            user_ids=[user_id],
+            permissions=reduce(or_, permissions),
+        )
+
+        await self.app.invoke(Opcode.CHAT_MEMBERS_UPDATE, frame.to_payload())

pymax/api/messages/__init__.py

@@ -1 +1 @@
-from .service import MessageService
+from .service import DateTimeUnion, MessageService

pymax/api/messages/payloads.py

@@ -8,6 +8,7 @@ from pymax.api.uploads.payloads import (
     AttachPhotoPayload,
     VideoAttachPayload,
 )
+from pymax.types.domain import Poll
 
 from .enums import ItemType, ReadAction
 
@@ -20,9 +21,9 @@ class GetMessagesPayload(CamelModel):
 class EditMessagePayload(CamelModel):
     chat_id: int
     message_id: int
-    text: str
+    text: str | None = None
     elements: list[Any]
-    attachments: list[AttachPhotoPayload | VideoAttachPayload | AttachFilePayload] = Field(
+    attachments: list[AttachPhotoPayload | VideoAttachPayload | AttachFilePayload | Poll] = Field(
         default_factory=list
     )
 
@@ -32,12 +33,18 @@ class ReplyLink(CamelModel):
     message_id: int
 
 
+class DelayedAttributes(CamelModel):
+    time_to_fire: int
+    notify_sender: bool
+
+
 class SendMessagePayloadMessage(CamelModel):
-    text: str
+    text: str | None = None
     cid: int
     elements: list[Any]
-    attaches: list[AttachPhotoPayload | VideoAttachPayload | AttachFilePayload]
+    attaches: list[AttachPhotoPayload | VideoAttachPayload | AttachFilePayload | Poll]
     link: ReplyLink | None = None
+    delayed_attributes: DelayedAttributes | None = None
 
 
 class SendMessagePayload(CamelModel):
@@ -110,18 +117,18 @@ class ReactionInfoPayload(CamelModel):
 
 class AddReactionPayload(CamelModel):
     chat_id: int
-    message_id: str
+    message_id: int
     reaction: ReactionInfoPayload
 
 
 class GetReactionsPayload(CamelModel):
     chat_id: int
-    message_ids: list[str]
+    message_ids: list[int]
 
 
 class RemoveReactionPayload(CamelModel):
     chat_id: int
-    message_id: str
+    message_id: int
 
 
 class ReadMessagesPayload(CamelModel):
@@ -129,3 +136,10 @@ class ReadMessagesPayload(CamelModel):
     chat_id: int
     message_id: str | int  # Сокет просит int а вс str
     mark: int
+
+
+class VotePollPayload(CamelModel):
+    chat_id: int
+    message_id: int
+    poll_id: int
+    answers_ids: list[int]