Merge commit '3eaef5dc90b716cc0fa391cbe2394be56d5f6041' as 'userbot'

userbot/modules/misc/autofwd.py

@@ -0,0 +1,165 @@
+#  Moon-Userbot - telegram userbot
+#  Copyright (C) 2020-present Moon Userbot Organization
+#
+#  This program is free software: you can redistribute it and/or modify
+#  it under the terms of the GNU General Public License as published by
+#  the Free Software Foundation, either version 3 of the License, or
+#  (at your option) any later version.
+
+#  This program 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 General Public License for more details.
+
+#  You should have received a copy of the GNU General Public License
+#  along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+from pyrogram import Client, filters
+from pyrogram.errors import MessageTooLong
+from pyrogram.types import Message
+from utils.db import db
+from utils.misc import modules_help, prefix
+
+
+def addtrg(channel_id):
+    channel_ids = db.get('custom.autofwd', 'chatto', default=[])
+    if channel_id not in channel_ids:
+        channel_ids.append(channel_id)
+        db.set('custom.autofwd', 'chatto', channel_ids)
+
+
+def rmtrg(channel_id):
+    channel_ids = db.get('custom.autofwd', 'chatto', default=[])
+    if channel_id in channel_ids:
+        channel_ids.remove(channel_id)
+        db.set('custom.autofwd', 'chatto', channel_ids)
+
+
+def addsrc(channel_id):
+    channel_ids = db.get('custom.autofwd', 'chatsrc', default=[])
+    if channel_id not in channel_ids:
+        channel_ids.append(channel_id)
+        db.set('custom.autofwd', 'chatsrc', channel_ids)
+
+
+def rmsrc(channel_id):
+    channel_ids = db.get('custom.autofwd', 'chatsrc', default=[])
+    if channel_id in channel_ids:
+        channel_ids.remove(channel_id)
+        db.set('custom.autofwd', 'chatsrc', channel_ids)
+
+
+def getfwd_data():
+    source_chats = db.get('custom.autofwd', 'chatsrc')
+    target_chats = db.get('custom.autofwd', 'chatto')
+    return source_chats, target_chats
+
+
+@Client.on_message(filters.command(['addfwd_src', 'addfwd_to'], prefix) & filters.me)
+async def addfwd(_, message: Message):
+    if message.command[0] == 'addfwd_src':
+        if len(message.command) > 1:
+            channel_id = message.text.split(maxsplit=1)[1]
+            if not channel_id.startswith('-100'):
+                channel_id = '-100' + channel_id
+            # chat id should be integer
+            if not channel_id.isdigit():
+                try:
+                    channel_id = int(channel_id)
+                except Exception:
+                    return await message.edit_text('Chat id should be in integer')
+            addsrc(channel_id=channel_id)
+            await message.edit_text(f'Auto Forwarding Enabled for Chat with id: <code>{channel_id}</code>')
+        else:
+            await message.edit_text('Chat id not provided!')
+            return
+    elif message.command[0] == 'addfwd_to':
+        if len(message.command) > 1:
+            channel_id = message.text.split(maxsplit=1)[1]
+            if not channel_id.startswith('-100'):
+                channel_id = '-100' + channel_id
+            # chat id should be integer
+            if not channel_id.isdigit():
+                try:
+                    channel_id = int(channel_id)
+                except Exception:
+                    return await message.edit_text('Chat id should be in integer')
+            addtrg(channel_id=channel_id)
+            await message.edit_text(f'Auto Forwarding Enabled to Chat with id: <code>{channel_id}</code>')
+        else:
+            await message.edit_text('Chat id not provided!')
+            return
+
+
+@Client.on_message(filters.command(['delfwd_src', 'delfwd_to'], prefix) & filters.me)
+async def delfwd(_, message: Message):
+    if message.command[0] == 'delfwd_src':
+        if len(message.command) > 1:
+            channel_id = message.text.split(maxsplit=1)[1]
+            if not channel_id.startswith('-100'):
+                channel_id = '-100' + channel_id
+            # chat id should be integer
+            if not channel_id.isdigit():
+                try:
+                    channel_id = int(channel_id)
+                except Exception:
+                    return await message.edit_text('Chat id should be in integer')
+            rmsrc(channel_id=channel_id)
+            await message.edit_text(f'Auto Forwarding Disabled for Chat with id: <code>{channel_id}</code>')
+        else:
+            await message.edit_text('Chat id not provided!')
+            return
+    elif message.command[0] == 'delfwd_to':
+        if len(message.command) > 1:
+            channel_id = message.text.split(maxsplit=1)[1]
+            if not channel_id.startswith('-100'):
+                channel_id = '-100' + channel_id
+            # chat id should be integer
+            if not channel_id.isdigit():
+                try:
+                    channel_id = int(channel_id)
+                except Exception:
+                    return await message.edit_text('Chat id should be in integer')
+            rmtrg(channel_id=channel_id)
+            await message.edit_text(f'Auto Forwarding Disabled to Chat with id: <code>{channel_id}</code>')
+        else:
+            await message.edit_text('Chat id not provided!')
+            return
+
+
+@Client.on_message(filters.command('autofwd', prefix) & filters.me)
+async def autofwd(_, message: Message):
+    source_chats, target_chats = getfwd_data()
+    return await message.edit_text(f'Source Chats: {source_chats}\nTarget Chats: {target_chats}')
+
+
+@Client.on_message(filters.channel | filters.group)
+async def autofwd_main(client: Client, message: Message):
+    chat_id = message.chat.id
+    source_chats = db.get('custom.autofwd', 'chatsrc')
+    target_chats = db.get('custom.autofwd', 'chatto')
+
+    if source_chats is not None and chat_id in source_chats and target_chats is not None:
+            for chat in target_chats:
+                try:
+                    await message.copy(chat)
+                except Exception as e:
+                    try:
+                        await client.send_message(
+                            'me',
+                            f'Auto Forwarding Failed for Chat with id: <code>{chat_id}</code> to <code>{chat}</code>\n\n{e}',
+                        )
+                    except MessageTooLong:
+                        await client.send_message(
+                            'me',
+                            f'Auto Forwarding Failed for Chat with id: <code>{chat_id}</code> to <code>{chat}</code>, Please check logs!',
+                        )
+
+
+modules_help['autofwd'] = {
+    'autofwd': 'Retrieve Data of auto fwd',
+    'addfwd_src [channel_id]*': 'Enable auto forwarding for a channel',
+    'addfwd_to [channel_id]*': 'Enable auto forwarding to a channel',
+    'delfwd_src [channel_id]*': 'Disable auto forwarding for a channel',
+    'delfwd_to [channel_id]*': 'Disable auto forwarding to a channel',
+}