Initial commit

plugins/handlers/user_join.py

@@ -0,0 +1,108 @@
+import asyncio
+import contextlib
+import logging
+from datetime import datetime, timedelta
+
+from pyrogram import filters
+from pyrogram.types import (
+    ChatPermissions,
+    InlineKeyboardButton,
+    InlineKeyboardMarkup,
+    Message,
+)
+
+from classes.pyroclient import PyroClient
+from classes.pyrouser import PyroUser
+from modules.database import col_schedule
+from modules.kicker import kick_unstarted
+
+logger = logging.getLogger(__name__)
+
+
+@PyroClient.on_message(filters.new_chat_members & ~filters.me)
+async def handler_user_join(app: PyroClient, message: Message):
+    if (
+        app.config["whitelist"]["enabled"]
+        and message.chat.id not in app.config["whitelist"]["groups"]
+    ):
+        logger.info(
+            "User %s has joined the group %s, but it's not whitelisted, ignoring.",
+            message.from_user.id,
+            message.chat.id,
+        )
+        return
+
+    logger.info(
+        "User %s has joined the group %s", message.from_user.id, message.chat.id
+    )
+
+    await message.delete()
+
+    # If user has already failed the test and joined once more
+    with contextlib.suppress(KeyError):
+        user = await app.find_user(message.from_user, message.chat.id)
+        if user.failed is True:
+            logger.info(
+                "User %s has previously failed the captcha, kicking and banning him",
+                user.id,
+            )
+            banned = await app.ban_chat_member(message.chat.id, user.id)
+
+            if isinstance(banned, Message):
+                await banned.delete()
+
+            return
+
+    await app.restrict_chat_member(
+        chat_id=message.chat.id,
+        user_id=message.from_user.id,
+        permissions=ChatPermissions(can_send_messages=False),
+    )
+
+    user = PyroUser.create_if_not_exists(message.from_user.id, message.chat.id)
+
+    if user.mistakes > 0 or user.score > 0:
+        user.set_score(0)
+        user.set_mistakes(0)
+
+    await asyncio.sleep(2)
+    verification_request = await app.send_message(
+        chat_id=message.chat.id,
+        text=app._(
+            "welcome",
+            "messages",
+        ).format(mention=message.from_user.mention),
+        reply_markup=InlineKeyboardMarkup(
+            [
+                [
+                    InlineKeyboardButton(
+                        app._(
+                            "verify",
+                            "buttons",
+                        ),
+                        callback_data=f"verify;{message.from_user.id}",
+                    )
+                ],
+                [
+                    InlineKeyboardButton(
+                        app._(
+                            "ban",
+                            "buttons",
+                        ),
+                        callback_data=f"ban;{message.from_user.id}",
+                    )
+                ],
+            ],
+        ),
+    )
+
+    if app.scheduler is not None:
+        job = app.scheduler.add_job(
+            kick_unstarted,
+            "date",
+            [app, user.id, verification_request.chat.id, verification_request.id],
+            run_date=datetime.now() + timedelta(seconds=app.config["timeouts"]["join"]),
+        )
+        col_schedule.insert_one(
+            {"user": message.from_user.id, "group": message.chat.id, "job_id": job.id}
+        )