Add support for direct messages

A little update :)

Author: KurimuzonAkuma

compiler/docs/compiler.py

@@ -300,6 +300,7 @@ def get_title_list(s: str) -> list:
             delete_chat_photo
             set_chat_title
             set_chat_description
+            set_chat_direct_messages_group
             set_chat_permissions
             pin_chat_message
             unpin_chat_message
@@ -312,6 +313,8 @@ def get_title_list(s: str) -> list:
             get_chat_members_count
             get_dialogs
             get_dialogs_count
+            get_direct_messages_topics_by_id
+            get_direct_messages_topics
             set_chat_username
             archive_chats
             unarchive_chats
@@ -693,6 +696,8 @@ def get_title_list(s: str) -> list:
             PaidMediaPreview
             PaidMessagesRefunded
             PaidMessagesPriceChanged
+            DirectMessagePriceChanged
+            DirectMessagesTopic
             PaymentForm
             ChatBoost
             ContactRegistered

compiler/errors/source/400_BAD_REQUEST.tsv

@@ -60,12 +60,13 @@ BROADCAST_CALLS_DISABLED	Broadcast calls disabled.
 BROADCAST_ID_INVALID	The channel is invalid.
 BROADCAST_PUBLIC_VOTERS_FORBIDDEN	Polls with public voters cannot be sent in channels.
 BROADCAST_REQUIRED	This method can only be used with a channel.
-BUSINESS_BOT_MISSING	Business bot missing.
+BUSINESS_BOT_MISSING	The business bot is missing.
 BUSINESS_CONNECTION_INVALID	The business connection is invalid.
 BUSINESS_CONNECTION_NOT_ALLOWED	This business connection is not allowed.
 BUSINESS_PEER_INVALID	Messages can't be set to the specified peer through the current [business connection](https://core.telegram.org/api/business#connected-bots).
 BUSINESS_PEER_USAGE_MISSING	You cannot send a message to a user through a [business connection](https://core.telegram.org/api/business#connected-bots) if the user hasn't recently contacted us.
 BUSINESS_RECIPIENTS_EMPTY	You didn't set any flag in inputBusinessBotRecipients, thus the bot cannot work with *any* peer.
+BUSINESS_RECIPIENTS_INVALID	The specified inputBusinessBotRecipients is invalid.
 BUSINESS_WORK_HOURS_EMPTY	No work hours were specified.
 BUSINESS_WORK_HOURS_PERIOD_INVALID	The specified work hours are invalid, see [here »](https://core.telegram.org/api/business#opening-hours) for the exact requirements.
 BUTTON_COPY_TEXT_INVALID	The specified [keyboardButtonCopy](https://core.telegram.org/constructor/keyboardButtonCopy).`copy_text` is invalid.
@@ -91,6 +92,7 @@ CHANNEL_BANNED	The channel is banned.
 CHANNEL_FORUM_MISSING	This supergroup is not a forum.
 CHANNEL_ID_INVALID	The specified supergroup ID is invalid.
 CHANNEL_INVALID	The provided channel is invalid.
+CHANNEL_MONOFORUM_UNSUPPORTED	You cannot use this method in direct chats.
 CHANNEL_PARICIPANT_MISSING	The current user is not in the channel.
 CHANNEL_PRIVATE	You haven't joined this channel/supergroup.
 CHANNEL_TOO_BIG	This channel is too big.
@@ -150,6 +152,7 @@ DATE_EMPTY	The date argument is empty.
 DC_ID_INVALID	The provided dc_id is invalid.
 DH_G_A_INVALID	The g_a parameter is invalid.
 DOCUMENT_INVALID	The specified document is invalid.
+EFFECT_ID_INVALID	The specified effect id is invalid.
 EMAIL_HASH_EXPIRED	The email hash expired and cannot be used to verify it.
 EMAIL_INVALID	The specified email is invalid.
 EMAIL_NOT_ALLOWED	The specified email cannot be used to complete the operation.
@@ -219,6 +222,7 @@ FORUM_ENABLED	You can't execute the specified action because the group is a [for
 FRESH_CHANGE_ADMINS_FORBIDDEN	You can't change administrator settings in this chat because your session was logged-in recently.
 FROM_MESSAGE_BOT_DISABLED	Bots can't use fromMessage min constructors.
 FROM_PEER_INVALID	The specified from_id is invalid.
+FROZEN_PARTICIPANT_MISSING	The account is frozen and cannot access the chat participant.
 GAME_BOT_INVALID	You cannot send that game with the current bot.
 GENERAL_MODIFY_ICON_FORBIDDEN	You can't modify the icon of the "General" topic.
 GEO_POINT_INVALID	Invalid geo point provided.
@@ -354,7 +358,7 @@ PEER_TYPES_INVALID	The passed [keyboardButtonSwitchInline](https://core.telegram
 PERSISTENT_TIMESTAMP_EMPTY	The pts argument is empty.
 PERSISTENT_TIMESTAMP_INVALID	The persistent timestamp is invalid.
 PHONE_CODE_EMPTY	The phone code is missing.
-PHONE_CODE_EXPIRED	The provided confirmation code has expired.
+PHONE_CODE_EXPIRED	The provided confirmation code has expired. Make sure you haven't sent it to anyone else and use the code on the same connection.
 PHONE_CODE_HASH_EMPTY	The phone code hash is missing.
 PHONE_CODE_INVALID	The provided confirmation code is invalid.
 PHONE_HASH_EXPIRED	An invalid or expired phone code hash was provided.
@@ -389,6 +393,7 @@ POLL_OPTION_INVALID	IA poll option used invalid data (the data may be too long).
 POLL_QUESTION_INVALID	The poll question is invalid.
 POLL_UNSUPPORTED	This layer does not support polls in the invoked method.
 POLL_VOTE_REQUIRED	Cast a vote in the poll before calling this method.
+PRECHECKOUT_FAILED	Precheckout failed.
 PREMIUM_ACCOUNT_REQUIRED	A premium account is required to execute this action.
 PREMIUM_GIFTCODE_WAS_REFUNDED	This gift code can't be redeemed because the giveaway organizer requested a refund.
 PRICING_CHAT_INVALID	The pricing for the [subscription](https://core.telegram.org/api/subscriptions) is invalid, the maximum price is specified in the [`stars_subscription_amount_max` config key »](https://core.telegram.org/api/config#stars-subscription-amount-max).
@@ -425,6 +430,7 @@ REPLY_MARKUP_TOO_LONG	The specified reply markup is too long.
 REPLY_MESSAGES_TOO_MUCH	Each shortcut can contain a maximum of [appConfig.`quick_reply_messages_limit`](https://core.telegram.org/api/config#quick-reply-messages-limit) messages, the limit was reached.
 REPLY_MESSAGE_ID_INVALID	The specified reply message id is invalid.
 REPLY_TO_INVALID	The specified `reply_to` field is invalid.
+REPLY_TO_MONOFORUM_PEER_INVALID	The specified `reply_to` field is invalid.
 REPLY_TO_USER_INVALID	The replied-to user is invalid.
 REQUEST_TOKEN_INVALID	The master DC did not accept the `request_token` from the CDN DC. Continue downloading the file from the master DC using upload.getFile.
 RESET_REQUEST_MISSING	No password reset is in progress.
@@ -439,6 +445,7 @@ RINGTONE_INVALID	The specified ringtone is invalid.
 RINGTONE_MIME_INVALID	The MIME type for the ringtone is invalid.
 RSA_DECRYPT_FAILED	Internal RSA decryption failed.
 SAVED_DIALOGS_UNSUPPORTED	You cannot use this method.
+SAVED_PEER_INVALID	The saved peer is invalid.
 SCHEDULE_BOT_NOT_ALLOWED	Bots are not allowed to schedule messages.
 SCHEDULE_DATE_INVALID	Invalid schedule date provided.
 SCHEDULE_DATE_TOO_LATE	The date you tried to schedule is too far in the future (more than one year).
@@ -471,6 +478,7 @@ STARGIFT_ALREADY_CONVERTED	The provided star gift already converted to stars.
 STARGIFT_ALREADY_UPGRADED	This star gift was already upgraded before.
 STARGIFT_CONVERT_TOO_OLD	This gift can no longer be converted into stars.
 STARGIFT_INVALID	The passed [inputInvoiceStarGift](https://core.telegram.org/constructor/inputInvoiceStarGift) is invalid.
+STARGIFT_MESSAGE_INVALID	The provided message for this gift is invalid.
 STARGIFT_NOT_FOUND	The specified star gift was not found.
 STARGIFT_NOT_OWNER	You can't control this gift because it doesn't belong to you.
 STARGIFT_RESELL_NOT_ALLOWED	You can't buy this gift. Someone else may have already bought it.
@@ -480,8 +488,11 @@ STARGIFT_UPGRADE_UNAVAILABLE	Gift is not available for upgrade yet.
 STARGIFT_USAGE_LIMITED	The gift is sold out.
 STARREF_AWAITING_END	The previous referral program was terminated less than 24 hours ago: further changes can be made after the date specified in userFull.starref_program.end_date.
 STARREF_HASH_REVOKED	The specified affiliate link was already revoked.
+STARREF_MONTHS_TOO_LOW	The `duration_months` parameter must be between 0 and 36.
 STARREF_PERMILLE_INVALID	The specified commission_permille is invalid: the minimum and maximum values for this parameter are contained in the [starref_min_commission_permille](https://core.telegram.org/api/config#starref-min-commission-permille) and [starref_max_commission_permille](https://core.telegram.org/api/config#starref-max-commission-permille) client configuration parameters.
 STARREF_PERMILLE_TOO_LOW	The specified commission_permille is too low: the minimum and maximum values for this parameter are contained in the [starref_min_commission_permille](https://core.telegram.org/api/config#starref-min-commission-permille) and [starref_max_commission_permille](https://core.telegram.org/api/config#starref-max-commission-permille) client configuration parameters.
+STARS_AMOUNT_INVALID	The specified `paid_stars` amount is invalid.
+STARS_FORM_AMOUNT_MISMATCH	The stars form amount mismatch.
 STARS_INVOICE_INVALID	The specified Telegram Star invoice is invalid.
 STARS_PAYMENT_REQUIRED	To import this chat invite link, you must first [pay for the associated Telegram Star subscription »](https://core.telegram.org/api/subscriptions#channel-subscriptions).
 START_PARAM_EMPTY	The start parameter is empty.

compiler/errors/source/403_FORBIDDEN.tsv

@@ -2,6 +2,7 @@ id	message
 ALLOW_PAYMENT_REQUIRED_X	The user has enabled paid messages with {value} stars.
 ANONYMOUS_REACTIONS_DISABLED	Sorry, anonymous administrators cannot leave reactions or participate in polls.
 BOT_ACCESS_FORBIDDEN	You can't access this bot.
+BOT_OWNER_REQUIRED	You must be the owner of the bot to perform this action.
 BROADCAST_FORBIDDEN	This request can't be used in channels.
 CHANNEL_PUBLIC_GROUP_NA	The channel/supergroup is not available.
 CHAT_ACTION_FORBIDDEN	You cannot execute this action.

pyrogram/enums/chat_type.py

@@ -39,5 +39,8 @@ class ChatType(AutoName):
     CHANNEL = auto()
     "Chat is a channel"
 
+    FORUM = auto()
+    "Chat is a forum"
+
     DIRECT = auto()
     "Chat is a direct with a channel"

pyrogram/enums/message_service_type.py

@@ -174,5 +174,8 @@ class MessageServiceType(AutoName):
     PAID_MESSAGES_REFUNDED = auto()
     "Refunded paid messages"
 
-    PAID_MESSAGES_PRICE = auto()
+    PAID_MESSAGES_PRICE_CHANGED = auto()
     "Paid messages price"
+
+    DIRECT_MESSAGE_PRICE_CHANGED = auto()
+    "Direct message price"

pyrogram/filters.py

@@ -549,7 +549,7 @@ async def private_filter(_, __, m: Message):
 
 # region group_filter
 async def group_filter(_, __, m: Message):
-    return bool(m.chat and m.chat.type in {enums.ChatType.GROUP, enums.ChatType.SUPERGROUP})
+    return bool(m.chat and m.chat.type in {enums.ChatType.GROUP, enums.ChatType.SUPERGROUP, enums.ChatType.FORUM})
 
 
 group = create(group_filter)
@@ -567,6 +567,17 @@ async def channel_filter(_, __, m: Message):
 """Filter messages sent in channels."""
 
 
+# endregion
+
+# region direct_filter
+async def direct_filter(_, __, m: Message):
+    return bool(m.chat and m.chat.type == enums.ChatType.DIRECT)
+
+
+direct = create(direct_filter)
+"""Filter messages sent in direct."""
+
+
 # endregion
 
 # region forum_filter

pyrogram/methods/bots/send_inline_bot_result.py

@@ -33,6 +33,7 @@ async def send_inline_bot_result(
         result_id: str,
         disable_notification: Optional[bool] = None,
         message_thread_id: Optional[int] = None,
+        direct_messages_chat_topic_id: int = None,
         reply_parameters: Optional["types.ReplyParameters"] = None,
         paid_message_star_count: int = None,
         schedule_date: datetime = None,
@@ -68,7 +69,11 @@ async def send_inline_bot_result(
 
             message_thread_id (``int``, *optional*):
                 Unique identifier of a message thread to which the message belongs.
-                For supergroups only.
+                For forums only.
+
+            direct_messages_chat_topic_id (``int``, *optional*):
+                Unique identifier of the topic in a channel direct messages chat administered by the current user.
+                For directs only only.
 
             reply_parameters (:obj:`~pyrogram.types.ReplyParameters`, *optional*):
                 Describes reply parameters for the message that is being sent.
@@ -153,7 +158,8 @@ async def send_inline_bot_result(
                 reply_to=await utils.get_reply_to(
                     self,
                     reply_parameters,
-                    message_thread_id
+                    message_thread_id,
+                    direct_messages_chat_topic_id
                 ),
                 schedule_date=utils.datetime_to_timestamp(schedule_date),
                 allow_paid_stars=paid_message_star_count

pyrogram/methods/chats/__init__.py

@@ -43,6 +43,8 @@
 from .get_similar_channels import GetSimilarChannels
 from .get_suitable_discussion_chats import GetSuitableDiscussionChats
 from .get_dialogs import GetDialogs
+from .get_direct_messages_topics_by_id import GetDirectMessagesTopicsByID
+from .get_direct_messages_topics import GetDirectMessagesTopics
 from .get_dialogs_count import GetDialogsCount
 from .get_folders import GetFolders
 from .get_forum_topics import GetForumTopics
@@ -60,6 +62,7 @@
 from .restrict_chat_member import RestrictChatMember
 from .set_administrator_title import SetAdministratorTitle
 from .set_chat_description import SetChatDescription
+from .set_chat_direct_messages_group import SetChatDirectMessagesGroup
 from .set_chat_permissions import SetChatPermissions
 from .set_chat_discussion_group import SetChatDiscussionGroup
 from .set_chat_photo import SetChatPhoto
@@ -102,6 +105,7 @@ class Chats(
     SetChatTitle,
     SetChatTTL,
     SetChatDescription,
+    SetChatDirectMessagesGroup,
     PinChatMessage,
     UnpinChatMessage,
     PinForumTopic,
@@ -110,6 +114,8 @@ class Chats(
     UpdateColor,
     EditFolder,
     GetDialogs,
+    GetDirectMessagesTopicsByID,
+    GetDirectMessagesTopics,
     GetChatMembersCount,
     SetChatUsername,
     SetChatPermissions,

pyrogram/methods/chats/get_direct_messages_topics.py

@@ -0,0 +1,109 @@
+#  Pyrogram - Telegram MTProto API Client Library for Python
+#  Copyright (C) 2017-present Dan <https://github.com/delivrance>
+#
+#  This file is part of Pyrogram.
+#
+#  Pyrogram is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU Lesser General Public License as published
+#  by the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+#
+#  Pyrogram is distributed in the hope that it will be useful,
+#  but WITHOUT ANY WARRANTY; without even the implied warranty of
+#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#  GNU Lesser General Public License for more details.
+#
+#  You should have received a copy of the GNU Lesser General Public License
+#  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+from typing import AsyncGenerator, Optional, Union
+
+import pyrogram
+from pyrogram import raw, types, utils
+
+
+class GetDirectMessagesTopics:
+    async def get_direct_messages_topics(
+        self: "pyrogram.Client",
+        chat_id: Union[int, str],
+        limit: int = 0,
+        exclude_pinned: Optional[bool] = None
+    ) -> AsyncGenerator["types.DirectMessagesTopic", None]:
+        """Get one or more topic from a direct messages channel chat.
+
+        .. include:: /_includes/usable-by/users.rst
+
+        Parameters:
+            chat_id (``int`` | ``str``):
+                Unique identifier (int) or username (str) of the target chat.
+
+            limit (``int``, *optional*):
+                Limits the number of topics to be retrieved.
+                By default, no limit is applied and all topics are returned.
+
+            exclude_pinned (``bool``, *optional*):
+                Exclude pinned topics.
+
+        Returns:
+            ``Generator``: A generator yielding :obj:`~pyrogram.types.DirectMessagesTopic` objects.
+
+        Example:
+            .. code-block:: python
+
+                # Iterate through all topics
+                async for topic in app.get_direct_messages_topics(chat_id):
+                    print(topic)
+        """
+        current = 0
+        total = limit or (1 << 31) - 1
+        limit = min(100, total)
+
+        offset_date = 0
+        offset_id = 0
+        offset_peer = raw.types.InputPeerEmpty()
+
+        while True:
+            r = await self.invoke(
+                raw.functions.messages.GetSavedDialogs(
+                    offset_date=offset_date,
+                    offset_id=offset_id,
+                    offset_peer=offset_peer,
+                    limit=limit,
+                    hash=0,
+                    exclude_pinned=exclude_pinned,
+                    parent_peer=await self.resolve_peer(chat_id)
+                )
+            )
+
+            users = {i.id: i for i in r.users}
+            chats = {i.id: i for i in r.chats}
+
+            messages = {}
+
+            for message in r.messages:
+                if isinstance(message, raw.types.MessageEmpty):
+                    continue
+
+                messages[message.id] = await types.Message._parse(self, message, users, chats)
+
+            topics = []
+
+            for topic in r.dialogs:
+                topics.append(types.DirectMessagesTopic._parse(topic, messages, users, chats))
+
+            if not topics:
+                return
+
+            last = topics[-1]
+
+            offset_id = last.last_message.id
+            offset_date = utils.datetime_to_timestamp(last.last_message.date)
+            offset_peer = await self.resolve_peer(last.id)
+
+            for topic in topics:
+                yield topic
+
+                current += 1
+
+                if current >= total:
+                    return