Update chatbot/plugins/join_request.py

chatbot/plugins/join_request.py

@@ -1,22 +1,4 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# Copyright 2020-2024 (c) Randy W @xtdevs, @xtsea
-#
-# from : https://github.com/TeamKillerX
-# Channel : @RendyProjects
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU Affero 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 Affero General Public License for more details.
-#
-# You should have received a copy of the GNU Affero General Public License
-# along with this program.  If not, see <https://www.gnu.org/licenses/>.
-
+import os
 import time
 import json
 import asyncio
@@ -24,88 +6,44 @@ import io
 import os
 import re
 import logging
+import string
+import random
 from pyrogram import *
-from pyrogram.enums import ChatMemberStatus, ChatMembersFilter, ChatType
-from pyrogram import Client, filters
 from pyrogram.types import *
 from pyrogram.errors import *
 from database import db
 from logger import LOGS
 
+from pyrogram import Client, filters
+from pyrogram.enums import ChatMemberStatus, ChatMembersFilter, ChatType
+from pyrogram.types import (
+    CallbackQuery,
+    InlineKeyboardMarkup,
+    InlineKeyboardButton,
+    Message
+)
+
+from PIL import Image, ImageDraw, ImageFont
+
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 user_list = []
+captcha_texts = {}
 
-@Client.on_callback_query(filters.regex("^joinrequest_"))
-async def joinrequest_callback(client: Client, cb: CallbackQuery):
-    data = cb.data.split("_")
-    if len(data) != 2:
-        await cb.answer("Invalid data format.", True)
-        return
+def generate_captcha():
+    letters = string.ascii_uppercase + string.digits
+    captcha_text = ''.join(random.choice(letters) for _ in range(5))
+    img = Image.new('RGB', (150, 60), color = (255, 255, 255))
+    d = ImageDraw.Draw(img)
     try:
-        user_id = int(data[1])
-    except ValueError:
-        await cb.answer("Invalid user ID.", True)
-        return
-    try:
-        if cb.from_user.id in user_list:
-            await client.approve_chat_join_request(
-                chat_id=cb.message.chat.id,
-                user_id=user_id,
-            )
-            await cb.edit_message_text(
-                f"User {user_id} joined the request in the group",
-                disable_web_page_preview=True,
-            )
-        else:
-            await cb.answer("Only the Admins can perform this action.", True)
-    except Exception as e:
-        await cb.answer(f"Error: {e}", True)
-
-@Client.on_callback_query(filters.regex("^declinerequest_"))
-async def declinerequest_callback(client: Client, cb: CallbackQuery):
-    data = cb.data.split("_")
-    if len(data) != 2:
-        await cb.answer("Invalid data format.", True)
-        return
-    try:
-        user_id = int(data[1])
-    except ValueError:
-        await cb.answer("Invalid user ID.", True)
-        return
-    try:
-        if cb.from_user.id in user_list:
-            await client.decline_chat_join_request(
-                chat_id=cb.message.chat.id,
-                user_id=user_id,
-            )
-            await cb.edit_message_text(
-                f"User {user_id} declined to join the support group",
-                disable_web_page_preview=True,
-            )
-        else:
-            await cb.answer("Only the Admins can perform this action.", True)
-    except Exception as e:
-        await cb.answer(f"Error: {e}", True)
-        
-def create_button_join_request(user_id):
-    return InlineKeyboardMarkup(
-        [
-            [
-                InlineKeyboardButton(
-                    text="✅ Approve",
-                    callback_data=f"joinrequest_{user_id}"
-                )
-            ],
-            [
-                InlineKeyboardButton(
-                    text="❌ Decline",
-                    callback_data=f"declinerequest_{user_id}"
-                )
-            ]
-        ]
-    )
+        font = ImageFont.truetype("arial.ttf", 40)
+    except IOError:
+        font = ImageFont.load_default()
+    d.text((10,10), captcha_text, font=font, fill=(0,0,0))
+    img_path = f"captcha_{captcha_text}.png"
+    img.save(img_path)
+    return captcha_text, img_path
 
 @Client.on_chat_join_request(filters.chat("KillerXSupport"))
 async def join_request(client: Client, event: ChatJoinRequest):
@@ -115,17 +53,81 @@ async def join_request(client: Client, event: ChatJoinRequest):
     async for m in client.get_chat_members(event.chat.id, filter=ChatMembersFilter.ADMINISTRATORS):
         if not m.user.is_bot:
             user_list.append(m.user.id)
+    captcha_text, img_path = generate_captcha()
+    captcha_texts[user_id] = captcha_text
+    captcha_texts["chat_id"] = event.chat.id
+    keyboard = InlineKeyboardMarkup(
+        [
+            [InlineKeyboardButton("🔄 Refresh CAPTCHA", callback_data="refresh_captcha")],
+            [InlineKeyboardButton("✅ Verify", callback_data="verify_captcha")]
+        ]
+    )
     if event.chat.type == ChatType.SUPERGROUP:
         try:
-            user_request_text = f"UserID: {event.from_user.id} by admin soon"
-            await client.send_message(
-                event.chat.id,
-                user_request_text,
-                reply_markup=create_button_join_request(event.from_user.id)
+            await client.send_photo(
+                event.from_user.id,
+                photo=img_path,
+                caption=f"❗️ **Verify that you are human!**\n\n❔ Please enter the CAPTCHA text below the image.",
+                reply_markup=keyboard
             )
+            os.remove(img_path)
         except Exception as e:
             await client.send_message(
                 event.chat.id,
                 text=str(e)
             )
-            logger.error(str(e))
+            logger.error(str(e))
+
+@Client.on_callback_query(filters.regex("^refresh_captcha$"))
+async def refresh_captcha_callback(client: Client, cb: CallbackQuery):
+    user_id = cb.from_user.id
+    if user_id in captcha_texts:
+        del captcha_texts[user_id]
+    captcha_text, img_path = generate_captcha()
+    captcha_texts[user_id] = captcha_text
+    keyboard = InlineKeyboardMarkup(
+        [
+            [InlineKeyboardButton("🔄 Refresh CAPTCHA", callback_data="refresh_captcha")],
+            [InlineKeyboardButton("✅ Verify", callback_data="verify_captcha")]
+        ]
+    )
+    await cb.edit_message_media(
+        media=InputMediaPhoto(open(img_path, "rb")),
+        reply_markup=keyboard
+    )
+    os.remove(img_path)
+    await cb.answer("CAPTCHA refreshed!", show_alert=False)
+
+@Client.on_callback_query(filters.regex("^verify_captcha$"))
+async def verify_captcha_callback(client: Client, cb: CallbackQuery):
+    user_id = cb.from_user.id
+    if user_id not in captcha_texts:
+        await cb.answer("❗️ Please start the CAPTCHA verification first by join request.", show_alert=True)
+        return
+    await cb.message.reply_text("🔍 **Please enter the CAPTCHA text:**")
+    await cb.answer()
+
+@Client.on_message(filters.text & filters.private)
+async def handle_captcha_response(client: Client, message: Message):
+    user_id = message.from_user.id
+    user_response = message.text.strip().upper()
+    if user_id not in captcha_texts:
+        await message.reply("❗️ Please start the CAPTCHA verification first by join request again.")
+        return
+    correct_captcha = captcha_texts.get(user_id)
+    if user_response == correct_captcha:
+        await message.reply("✅ CAPTCHA verification successful!")
+        await client.approve_chat_join_request(
+            chat_id=captcha_texts.get("chat_id"),
+            user_id=user_id
+        )
+        del captcha_texts[user_id]
+        del captcha_texts["chat_id"]
+    else:
+        await message.reply("❌ Incorrect CAPTCHA. Please try again by join request again.")
+        await client.decline_chat_join_request(
+            chat_id=captcha_texts.get("chat_id"),
+            user_id=user_id
+        )
+        del captcha_texts[user_id]
+        del captcha_texts["chat_id"]