Merge pull request #47 from Infamous-Hydra/Infamous-Hydra-patch-3

Mikobot/plugins/fsub.py

@@ -0,0 +1,189 @@
+# <============================================== IMPORTS =========================================================>
+from telethon import Button, events, types
+from telethon.errors import ChatAdminRequiredError, UserNotParticipantError
+from telethon.tl.functions.channels import GetParticipantRequest
+
+from Database.mongodb import fsub_db as db
+from Mikobot import BOT_ID
+from Mikobot import DRAGONS as DEVS
+from Mikobot import OWNER_ID, tbot
+from Mikobot.events import register
+
+# <=======================================================================================================>
+
+# Constants
+F_SUBSCRIBE_COMMAND = "/(fsub|Fsub|forcesubscribe|Forcesub|forcesub|Forcesubscribe)"
+FORCESUBSCRIBE_ON = ["on", "yes", "y"]
+FORCESUBSCRIBE_OFF = ["off", "no", "n"]
+
+
+# <================================================ FUNCTION =======================================================>
+def fsk_ck(**args):
+    def decorator(func):
+        tbot.add_event_handler(func, events.CallbackQuery(**args))
+        return func
+
+    return decorator
+
+
+# Helper functions
+async def is_admin(chat_id, user_id):
+    try:
+        p = await tbot(GetParticipantRequest(chat_id, user_id))
+    except UserNotParticipantError:
+        return False
+    return isinstance(
+        p.participant, (types.ChannelParticipantAdmin, types.ChannelParticipantCreator)
+    )
+
+
+async def participant_check(channel, user_id):
+    try:
+        await tbot(GetParticipantRequest(channel, int(user_id)))
+        return True
+    except UserNotParticipantError:
+        return False
+    except Exception:
+        return False
+
+
+# Main command function
+@register(pattern=f"^{F_SUBSCRIBE_COMMAND} ?(.*)")
+async def force_subscribe(event):
+    """Handle the force subscribe command."""
+    if event.is_private:
+        return
+
+    if event.is_group:
+        perm = await event.client.get_permissions(event.chat_id, event.sender_id)
+        if not perm.is_admin:
+            return await event.reply("You need to be an admin to do this.")
+
+        if not perm.is_creator:
+            return await event.reply(
+                "❗ Group creator required\nYou have to be the group creator to do that."
+            )
+
+    try:
+        channel = event.text.split(None, 1)[1]
+    except IndexError:
+        channel = None
+
+    if not channel:
+        chat_db = db.fs_settings(event.chat_id)
+        if not chat_db:
+            await event.reply("Force subscribe is disabled in this chat.")
+        else:
+            await event.reply(
+                f"Force subscribe is currently enabled. Users are forced to join @{chat_db.channel} to speak here."
+            )
+    elif channel.lower() in FORCESUBSCRIBE_ON:
+        await event.reply("Please specify the channel username.")
+    elif channel.lower() in FORCESUBSCRIBE_OFF:
+        await event.reply("**Force subscribe is disabled successfully.**")
+        db.disapprove(event.chat_id)
+    else:
+        try:
+            channel_entity = await event.client.get_entity(channel)
+        except:
+            return await event.reply("Invalid channel username provided.")
+
+        channel = channel_entity.username
+        try:
+            if not channel_entity.broadcast:
+                return await event.reply("That's not a valid channel.")
+        except:
+            return await event.reply("That's not a valid channel.")
+
+        if not await participant_check(channel, BOT_ID):
+            return await event.reply(
+                f"**Not an admin in the channel**\nI am not an admin in the [channel](https://t.me/{channel}). Add me as an admin to enable force subscribe.",
+                link_preview=False,
+            )
+
+        db.add_channel(event.chat_id, str(channel))
+        await event.reply(f"Force subscribe is enabled to @{channel}.")
+
+
+# Event handler for new messages
+@tbot.on(events.NewMessage())
+async def force_subscribe_new_message(e):
+    """Handle new messages for force subscribe."""
+    if not db.fs_settings(e.chat_id):
+        return
+
+    if e.is_private or not e.from_id or e.sender_id in DEVS or e.sender_id == OWNER_ID:
+        return
+
+    if not e.chat.admin_rights or not e.chat.admin_rights.ban_users:
+        return
+
+    try:
+        channel = db.fs_settings(e.chat_id)["channel"]
+        check = await participant_check(channel, e.sender_id)
+    except (ChatAdminRequiredError, UserNotParticipantError):
+        return
+
+    if not check:
+        buttons = [
+            Button.url("Join Channel", f"t.me/{channel}"),
+            Button.inline("Unmute Me", data=f"fs_{e.sender_id}"),
+        ]
+
+        txt = f'<b><a href="tg://user?id={e.sender_id}">{e.sender.first_name}</a></b>, you have <b>not subscribed</b> to our <b><a href="t.me/{channel}">channel</a></b> yet. Please <b><a href="t.me/{channel}">join</a></b> and press the button below to unmute yourself.'
+        await e.reply(txt, buttons=buttons, parse_mode="html", link_preview=False)
+        await e.client.edit_permissions(e.chat_id, e.sender_id, send_messages=False)
+
+
+# Inline query handler
+@fsk_ck(pattern=r"fs(\_(.*))")
+async def unmute_force_subscribe(event):
+    """Handle inline query for unmuting force subscribe."""
+    user_id = int(((event.pattern_match.group(1)).decode()).split("_", 1)[1])
+
+    if not event.sender_id == user_id:
+        return await event.answer("This is not meant for you.", alert=True)
+
+    channel = db.fs_settings(event.chat_id)["channel"]
+    try:
+        check = await participant_check(channel, user_id)
+    except ChatAdminRequiredError:
+        check = False
+        return
+
+    if not check:
+        return await event.answer(
+            "You have to join the channel first, to get unmuted!", alert=True
+        )
+
+    try:
+        await event.client.edit_permissions(event.chat_id, user_id, send_messages=True)
+    except ChatAdminRequiredError:
+        pass
+
+    await event.delete()
+
+
+# <=================================================== HELP ====================================================>
+
+
+__help__ = """
+➠ *Dazai has the capability to hush members who haven't yet subscribed to your channel until they decide to hit that subscribe button.*
+➠ *When activated, I'll silence those who are not subscribed and provide them with an option to unmute. Once they click the button, I'll lift the mute.*
+
+➠ Configuration Process
+➠ Exclusively for Creators
+➠ Grant me admin privileges in your group
+➠ Designate me as an admin in your channel
+
+➠ *Commands*
+» /fsub channel_username - to initiate and customize settings for the channel.
+
+➠ *Kick things off with...*
+» /fsub - to review the current settings.
+» /fsub off - to deactivate the force subscription feature.
+
+➠ *If you disable fsub, you'll need to set it up again for it to take effect. Utilize* /fsub channel_username.
+"""
+__mod_name__ = "F-SUB"
+# <================================================ END =======================================================>