maxapi-python 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

pymax/payloads.py

@@ -0,0 +1,175 @@
+from typing import Any, Literal
+
+from pydantic import BaseModel, Field
+
+from pymax.static import AttachType, AuthType
+
+
+def to_camel(string: str) -> str:
+    parts = string.split("_")
+    return parts[0] + "".join(word.capitalize() for word in parts[1:])
+
+
+class CamelModel(BaseModel):
+    model_config = {
+        "alias_generator": to_camel,
+        "populate_by_name": True,
+    }
+
+
+class BaseWebSocketMessage(BaseModel):
+    ver: int = 11
+    cmd: int
+    seq: int
+    opcode: int
+    payload: dict[str, Any]
+
+
+class RequestCodePayload(CamelModel):
+    phone: str
+    type: AuthType = AuthType.START_AUTH
+    language: str = "ru"
+
+
+class SendCodePayload(CamelModel):
+    token: str
+    verify_code: str
+    auth_token_type: AuthType = AuthType.CHECK_CODE
+
+
+class SyncPayload(CamelModel):
+    interactive: bool = True
+    token: str
+    chats_sync: int = 0
+    contacts_sync: int = 0
+    presence_sync: int = 0
+    drafts_sync: int = 0
+    chats_count: int = 40
+
+
+class ReplyLink(CamelModel):
+    type: str = "REPLY"
+    message_id: str
+
+
+class UploadPhotoPayload(CamelModel):
+    count: int = 1
+
+
+class AttachPhotoPayload(CamelModel):
+    type: AttachType = Field(AttachType.PHOTO, alias="_type")
+    photo_token: str
+
+
+class SendMessagePayloadMessage(CamelModel):
+    text: str
+    cid: int
+    elements: list[Any]
+    attaches: list[dict[str, Any]]
+    link: ReplyLink | None = None
+
+
+class SendMessagePayload(CamelModel):
+    chat_id: int
+    message: SendMessagePayloadMessage
+    notify: bool = False
+
+
+class EditMessagePayload(CamelModel):
+    chat_id: int
+    message_id: int
+    text: str
+    elements: list[Any]
+    attaches: list[Any]
+
+
+class DeleteMessagePayload(CamelModel):
+    chat_id: int
+    message_ids: list[int]
+    for_me: bool = False
+
+
+class FetchContactsPayload(CamelModel):
+    contact_ids: list[int]
+
+
+class FetchHistoryPayload(CamelModel):
+    chat_id: int
+    from_time: int = Field(alias="from")
+    forward: int
+    backward: int = 200
+    get_messages: bool = True
+
+
+class ChangeProfilePayload(CamelModel):
+    first_name: str
+    last_name: str | None = None
+    description: str | None = None
+
+
+class ResolveLinkPayload(CamelModel):
+    link: str
+
+
+class PinMessagePayload(CamelModel):
+    chat_id: int
+    notify_pin: bool
+    pin_message_id: int
+
+
+class CreateGroupAttach(CamelModel):
+    type: Literal["CONTROL"] = Field("CONTROL", alias="_type")
+    event: str = "new"
+    chat_type: str = "CHAT"
+    title: str
+    user_ids: list[int]
+
+
+class CreateGroupMessage(CamelModel):
+    cid: int
+    attaches: list[CreateGroupAttach]
+
+
+class CreateGroupPayload(CamelModel):
+    message: CreateGroupMessage
+    notify: bool = True
+
+
+class InviteUsersPayload(CamelModel):
+    chat_id: int
+    user_ids: list[int]
+    show_history: bool
+    operation: str = "add"
+
+
+class RemoveUsersPayload(CamelModel):
+    chat_id: int
+    user_ids: list[int]
+    operation: str = "remove"
+    clean_msg_period: int
+
+
+class ChangeGroupSettingsOptions(BaseModel):
+    ONLY_OWNER_CAN_CHANGE_ICON_TITLE: bool | None
+    ALL_CAN_PIN_MESSAGE: bool | None
+    ONLY_ADMIN_CAN_ADD_MEMBER: bool | None
+    ONLY_ADMIN_CAN_CALL: bool | None
+    MEMBERS_CAN_SEE_PRIVATE_LINK: bool | None
+
+
+class ChangeGroupSettingsPayload(CamelModel):
+    chat_id: int
+    options: ChangeGroupSettingsOptions
+
+
+class ChangeGroupProfilePayload(CamelModel):
+    chat_id: int
+    theme: str | None
+    description: str | None
+
+
+class GetGroupMembersPayload(CamelModel):
+    type: str = "MEMBER"
+    marker: int
+    chat_id: int
+    count: int

pymax/static.py

@@ -3,27 +3,144 @@ from enum import Enum, IntEnum
 
 class Opcode(IntEnum):
     PING = 1
-    STATS = 5
-    HANDSHAKE = 6
+    DEBUG = 2
+    RECONNECT = 3
+    LOG = 5
+    SESSION_INIT = 6
     PROFILE = 16
-    REQUEST_CODE = 17
-    SEND_CODE = 18
-    SYNC = 19
+    AUTH_REQUEST = 17
+    AUTH = 18
+    LOGIN = 19
+    LOGOUT = 20
+    SYNC = 21
+    CONFIG = 22
+    AUTH_CONFIRM = 23
+    PRESET_AVATARS = 25
+    ASSETS_GET = 26
     UNKNOWN_26 = 26
-    SYNC_STICKERS_EMOJIS = 27
-    GET_EMOJIS_BY_ID = 28
-    GET_CONTACTS_INFO = 32
-    GET_LAST_SEEN = 35
-    GET_CHATS_DATA = 48
-    FETCH_HISTORY = 49
-
-    GET_HISTORY = 79
-
-    SEND_MESSAGE = 64
-    EDIT_MESSAGE = 67
-    DELETE_MESSAGE = 68
-
-    NEW_MESSAGE = 128
+    ASSETS_UPDATE = 27
+    ASSETS_GET_BY_IDS = 28
+    ASSETS_ADD = 29
+    SEARCH_FEEDBACK = 31
+    CONTACT_INFO = 32
+    CONTACT_ADD = 33
+    CONTACT_UPDATE = 34
+    CONTACT_PRESENCE = 35
+    CONTACT_LIST = 36
+    CONTACT_SEARCH = 37
+    CONTACT_MUTUAL = 38
+    CONTACT_PHOTOS = 39
+    CONTACT_SORT = 40
+    CONTACT_VERIFY = 42
+    REMOVE_CONTACT_PHOTO = 43
+    CONTACT_INFO_BY_PHONE = 46
+    CHAT_INFO = 48
+    CHAT_HISTORY = 49
+    CHAT_MARK = 50
+    CHAT_MEDIA = 51
+    CHAT_DELETE = 52
+    CHATS_LIST = 53
+    CHAT_CLEAR = 54
+    CHAT_UPDATE = 55
+    CHAT_CHECK_LINK = 56
+    CHAT_JOIN = 57
+    CHAT_LEAVE = 58
+    CHAT_MEMBERS = 59
+    PUBLIC_SEARCH = 60
+    CHAT_CLOSE = 61
+    CHAT_CREATE = 63
+    MSG_SEND = 64
+    MSG_TYPING = 65
+    MSG_DELETE = 66
+    MSG_EDIT = 67
+    CHAT_SEARCH = 68
+    MSG_SHARE_PREVIEW = 70
+    MSG_GET = 71
+    MSG_SEARCH_TOUCH = 72
+    MSG_SEARCH = 73
+    MSG_GET_STAT = 74
+    CHAT_SUBSCRIBE = 75
+    VIDEO_CHAT_START = 76
+    CHAT_MEMBERS_UPDATE = 77
+    VIDEO_CHAT_HISTORY = 79
+    PHOTO_UPLOAD = 80
+    STICKER_UPLOAD = 81
+    VIDEO_UPLOAD = 82
+    VIDEO_PLAY = 83
+    CHAT_PIN_SET_VISIBILITY = 86
+    FILE_UPLOAD = 87
+    FILE_DOWNLOAD = 88
+    LINK_INFO = 89
+    MSG_DELETE_RANGE = 92
+    SESSIONS_INFO = 96
+    SESSIONS_CLOSE = 97
+    PHONE_BIND_REQUEST = 98
+    PHONE_BIND_CONFIRM = 99
+    CONFIRM_PRESENT = 101
+    GET_INBOUND_CALLS = 103
+    EXTERNAL_CALLBACK = 105
+    AUTH_VALIDATE_PASSWORD = 107
+    AUTH_VALIDATE_HINT = 108
+    AUTH_VERIFY_EMAIL = 109
+    AUTH_CHECK_EMAIL = 110
+    AUTH_SET_2FA = 111
+    AUTH_CREATE_TRACK = 112
+    AUTH_LOGIN_CHECK_PASSWORD = 115
+    CHAT_COMPLAIN = 117
+    MSG_SEND_CALLBACK = 118
+    SUSPEND_BOT = 119
+    LOCATION_STOP = 124
+    LOCATION_SEND = 125
+    LOCATION_REQUEST = 126
+    GET_LAST_MENTIONS = 127
+    NOTIF_MESSAGE = 128
+    NOTIF_TYPING = 129
+    NOTIF_MARK = 130
+    NOTIF_CONTACT = 131
+    NOTIF_PRESENCE = 132
+    NOTIF_CONFIG = 134
+    NOTIF_CHAT = 135
+    NOTIF_ATTACH = 136
+    NOTIF_CALL_START = 137
+    NOTIF_CONTACT_SORT = 139
+    NOTIF_MSG_DELETE_RANGE = 140
+    NOTIF_MSG_DELETE = 142
+    NOTIF_CALLBACK_ANSWER = 143
+    CHAT_BOT_COMMANDS = 144
+    BOT_INFO = 145
+    NOTIF_LOCATION = 147
+    NOTIF_LOCATION_REQUEST = 148
+    NOTIF_ASSETS_UPDATE = 150
+    NOTIF_DRAFT = 152
+    NOTIF_DRAFT_DISCARD = 153
+    NOTIF_MSG_DELAYED = 154
+    NOTIF_MSG_REACTIONS_CHANGED = 155
+    NOTIF_MSG_YOU_REACTED = 156
+    CALLS_TOKEN = 158
+    NOTIF_PROFILE = 159
+    WEB_APP_INIT_DATA = 160
+    DRAFT_SAVE = 176
+    DRAFT_DISCARD = 177
+    MSG_REACTION = 178
+    MSG_CANCEL_REACTION = 179
+    MSG_GET_REACTIONS = 180
+    MSG_GET_DETAILED_REACTIONS = 181
+    STICKER_CREATE = 193
+    STICKER_SUGGEST = 194
+    VIDEO_CHAT_MEMBERS = 195
+    CHAT_HIDE = 196
+    CHAT_SEARCH_COMMON_PARTICIPANTS = 198
+    PROFILE_DELETE = 199
+    PROFILE_DELETE_TIME = 200
+    ASSETS_REMOVE = 259
+    ASSETS_MOVE = 260
+    ASSETS_LIST_MODIFY = 261
+    FOLDERS_GET = 272
+    FOLDERS_GET_BY_ID = 273
+    FOLDERS_UPDATE = 274
+    FOLDERS_REORDER = 275
+    FOLDERS_DELETE = 276
+    NOTIF_FOLDERS = 277
 
 
 class ChatType(str, Enum):
@@ -69,6 +186,13 @@ class DeviceType(str, Enum):
     IOS = "IOS"
 
 
+class AttachType(str, Enum):
+    PHOTO = "PHOTO"
+    VIDEO = "VIDEO"
+    FILE = "FILE"
+    STICKER = "STICKER"
+
+
 class Constants(Enum):
     PHONE_REGEX = r"^\+?\d{10,15}$"
     WEBSOCKET_URI = "wss://ws-api.oneme.ru/websocket"

pymax/types.py

@@ -1,10 +1,83 @@
-from typing import Any
+from typing import Any, override
 
-from .static import AccessType, ChatType, ElementType, MessageStatus, MessageType
+from .static import (
+    AccessType,
+    AttachType,
+    ChatType,
+    ElementType,
+    MessageStatus,
+    MessageType,
+)
+
+
+class Names:
+    def __init__(
+        self, name: str, first_name: str, last_name: str | None, type: str
+    ) -> None:
+        self.name = name
+        self.first_name = first_name
+        self.last_name = last_name
+        self.type = type
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "Names":
+        return cls(
+            name=data["name"],
+            first_name=data["firstName"],
+            last_name=data.get("lastName"),
+            type=data["type"],
+        )
+
+    @override
+    def __repr__(self) -> str:
+        return f"Names(name={self.name!r}, first_name={self.first_name!r}, last_name={self.last_name!r}, type={self.type!r})"
+
+    @override
+    def __str__(self) -> str:
+        return self.name
+
+
+class Me:
+    def __init__(
+        self,
+        id: int,
+        account_status: int,
+        phone: str,
+        names: list[Names],
+        update_time: int,
+        options: list[str] | None = None,
+    ) -> None:
+        self.id = id
+        self.account_status = account_status
+        self.phone = phone
+        self.update_time = update_time
+        self.options = options
+        self.names = names
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "Me":
+        return cls(
+            id=data["id"],
+            account_status=data["accountStatus"],
+            phone=data["phone"],
+            names=[Names.from_dict(n) for n in data["names"]],
+            update_time=data["updateTime"],
+            options=data.get("options"),
+        )
+
+    @override
+    def __repr__(self) -> str:
+        return f"Me(id={self.id!r}, account_status={self.account_status!r}, phone={self.phone!r}, names={self.names!r}, update_time={self.update_time!r}, options={self.options!r})"
+
+    @override
+    def __str__(self) -> str:
+        return f"Me {self.id}: {', '.join(str(n) for n in self.names)}"
 
 
 class Element:
-    def __init__(self, type: ElementType | str, length: int, from_: int | None = None) -> None:
+    def __init__(
+        self, type: ElementType | str, length: int, from_: int | None = None
+    ) -> None:
         self.type = type
         self.length = length
         self.from_ = from_
@@ -13,9 +86,13 @@ class Element:
     def from_dict(cls, data: dict[Any, Any]) -> "Element":
         return cls(type=data["type"], length=data["length"], from_=data.get("from"))
 
+    @override
     def __repr__(self) -> str:
-        return f"Element(type={self.type!r}, length={self.length!r}, from_={self.from_!r})"
+        return (
+            f"Element(type={self.type!r}, length={self.length!r}, from_={self.from_!r})"
+        )
 
+    @override
     def __str__(self) -> str:
         return f"{self.type}({self.length})"
 
@@ -60,12 +137,14 @@ class Message:
             reaction_info=data.get("reactionInfo"),
         )
 
+    @override
     def __repr__(self) -> str:
         return (
             f"Message(id={self.id!r}, sender={self.sender!r}, text={self.text!r}, "
             f"type={self.type!r}, status={self.status!r}, elements={self.elements!r})"
         )
 
+    @override
     def __str__(self) -> str:
         return f"Message {self.id} from {self.sender}: {self.text}"
 
@@ -130,9 +209,11 @@ class Dialog:
             participants=data["participants"],
         )
 
+    @override
     def __repr__(self) -> str:
         return f"Dialog(id={self.id!r}, owner={self.owner!r}, type={self.type!r}, last_message={self.last_message!r})"
 
+    @override
     def __str__(self) -> str:
         return f"Dialog {self.id} ({self.type})"
 
@@ -199,10 +280,14 @@ class Chat:
     @classmethod
     def from_dict(cls, data: dict[Any, Any]) -> "Chat":
         raw_admins = data.get("adminParticipants", {}) or {}
-        admin_participants: dict[int, dict[Any, Any]] = {int(k): v for k, v in raw_admins.items()}
+        admin_participants: dict[int, dict[Any, Any]] = {
+            int(k): v for k, v in raw_admins.items()
+        }
         raw_participants = data.get("participants", {}) or {}
         participants: dict[int, int] = {int(k): v for k, v in raw_participants.items()}
-        last_msg = Message.from_dict(data["lastMessage"]) if data.get("lastMessage") else None
+        last_msg = (
+            Message.from_dict(data["lastMessage"]) if data.get("lastMessage") else None
+        )
         return cls(
             participants_count=data.get("participantsCount", 0),
             access=AccessType(data.get("access", AccessType.PUBLIC.value)),
@@ -233,44 +318,25 @@ class Chat:
             cid=data.get("cid", 0),
         )
 
+    @override
     def __repr__(self) -> str:
         return f"Chat(id={self.id!r}, title={self.title!r}, type={self.type!r})"
 
+    @override
     def __str__(self) -> str:
         return f"{self.title} ({self.type})"
 
 
 class Channel(Chat):
+    @override
     def __repr__(self) -> str:
         return f"Channel(id={self.id!r}, title={self.title!r})"
 
+    @override
     def __str__(self) -> str:
         return f"Channel: {self.title}"
 
 
-class Names:
-    def __init__(self, name: str, first_name: str, last_name: str | None, type: str) -> None:
-        self.name = name
-        self.first_name = first_name
-        self.last_name = last_name
-        self.type = type
-
-    @classmethod
-    def from_dict(cls, data: dict[str, Any]) -> "Names":
-        return cls(
-            name=data["name"],
-            first_name=data["firstName"],
-            last_name=data.get("lastName"),
-            type=data["type"],
-        )
-
-    def __repr__(self) -> str:
-        return f"Names(name={self.name!r}, first_name={self.first_name!r}, last_name={self.last_name!r}, type={self.type!r})"
-
-    def __str__(self) -> str:
-        return self.name
-
-
 class User:
     def __init__(
         self,
@@ -320,8 +386,47 @@ class User:
             menu_button=data.get("menuButton"),
         )
 
+    @override
     def __repr__(self) -> str:
         return f"User(id={self.id!r}, names={self.names!r}, status={self.account_status!r})"
 
+    @override
     def __str__(self) -> str:
         return f"User {self.id}: {', '.join(str(n) for n in self.names)}"
+
+
+class Attach:
+    def __init__(
+        self,
+        _type: AttachType,
+        video_id: int | None = None,
+        photo_token: str | None = None,
+        file_id: int | None = None,
+        token: str | None = None,
+    ) -> None:
+        self.type = _type
+        self.video_id = video_id
+        self.photo_token = photo_token
+        self.file_id = file_id
+        self.token = token
+
+    @classmethod
+    def from_dict(cls, data: dict[str, Any]) -> "Attach":
+        return cls(
+            _type=AttachType(data["type"]),
+            video_id=data.get("videoId"),
+            photo_token=data.get("photoToken"),
+            file_id=data.get("fileId"),
+            token=data.get("token"),
+        )
+
+    @override
+    def __repr__(self) -> str:
+        return (
+            f"Attach(type={self.type!r}, video_id={self.video_id!r}, "
+            f"photo_token={self.photo_token!r}, file_id={self.file_id!r}, token={self.token!r})"
+        )
+
+    @override
+    def __str__(self) -> str:
+        return f"Attach: {self.type}"

pymax/utils.py

@@ -0,0 +1,38 @@
+from typing import Any
+
+import lz4.block
+import msgpack
+
+
+def unpack_packet(data: bytes) -> None | dict[str, Any]:
+    ver = int.from_bytes(data[0:1], "big")
+    cmd = int.from_bytes(data[1:3], "big")
+    seq = int.from_bytes(data[3:4], "big")
+    opcode = int.from_bytes(data[4:6], "big")
+    packed_len = int.from_bytes(data[6:10], "big", signed=False)
+    comp_flag = packed_len >> 24
+    payload_length = packed_len & 0xFFFFFF
+    payload_bytes = data[10 : 10 + payload_length]
+    if comp_flag != 0:
+        compressed_data = payload_bytes
+        try:
+            payload_bytes = lz4.block.decompress(compressed_data, uncompressed_size=255)
+        except lz4.block.LZ4BlockError:
+            return None
+    payload = msgpack.unpackb(payload_bytes, raw=False)
+    return {"ver": ver, "cmd": cmd, "seq": seq, "opcode": opcode, "payload": payload}
+
+
+# ToDo: Add lz4 compression
+def pack_packet(
+    ver: int, cmd: int, seq: int, opcode: int, payload: dict[str, Any]
+) -> bytes:
+    ver_b = ver.to_bytes(1, "big")
+    cmd_b = cmd.to_bytes(2, "big")
+    seq_b = seq.to_bytes(1, "big")
+    opcode_b = opcode.to_bytes(2, "big")
+    payload_bytes = msgpack.packb(payload)
+    if payload_bytes is None:
+        payload_bytes = b""
+    payload_len_b = len(payload_bytes).to_bytes(4, "big")
+    return ver_b + cmd_b + seq_b + opcode_b + payload_len_b + payload_bytes