Upload join_request.py

chatbot/plugins/join_request.py

@@ -3,52 +3,44 @@ import time
 import json
 import asyncio
 import io
-import os
 import re
 import logging
 import string
 import random
 import requests
-from pyrogram import *
-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
+    Message,
+    ChatJoinRequest,
+    ChatType
 )
+from pyrogram.enums import ChatMemberStatus, ChatMembersFilter
+from pyrogram.errors import ChatAdminRequired
 
 from PIL import Image, ImageDraw, ImageFont, ImageFilter
 
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-user_list = []
 captcha_texts = {}
 captcha_modes = {}
 
 def generate_math_captcha():
     operations = ['+', '-', '*']
     operation = random.choice(operations)
-
     num1 = random.randint(1, 20)
     num2 = random.randint(1, 20)
-    
+
     if operation == '+':
         correct_answer = num1 + num2
     elif operation == '-':
         correct_answer = num1 - num2
     else:
         correct_answer = num1 * num2
-    
     captcha_text = f"{num1} {operation} {num2} = ?"
-    
     choices = [str(correct_answer)]
     while len(choices) < 3:
         wrong_answer = correct_answer + random.choice([-5, -3, -2, 2, 3, 5])
@@ -89,7 +81,7 @@ def generate_math_captcha():
         choice_x = 50 + idx * 80
         choice_y = height - choice_height - 20
         d.text((choice_x, choice_y), choice_text, font=choice_font, fill=(0, 0, 0))
-
+    
     for _ in range(5):
         start = (random.randint(0, width), random.randint(0, height))
         end = (random.randint(0, width), random.randint(0, height))
@@ -97,8 +89,10 @@ def generate_math_captcha():
         d.line([start, end], fill=line_color, width=2)
 
     img = img.filter(ImageFilter.BLUR)
+
     img_path = f"captcha_math_{captcha_text}.png"
     img.save(img_path)
+    
     return captcha_text, img_path, choices, correct_answer
 
 def generate_text_captcha():
@@ -155,6 +149,7 @@ def generate_text_captcha():
     img = img.filter(ImageFilter.BLUR)
     img_path = f"captcha_text_{captcha_text}.png"
     img.save(img_path)
+
     return captcha_text, img_path, choices, captcha_text
 
 def generate_captcha(user_id, mode='math'):
@@ -206,20 +201,20 @@ async def mode_callback(client: Client, cb: CallbackQuery):
     user_id = cb.from_user.id
     data = cb.data.split("_")
     if len(data) != 2:
-        await cb.answer("Invalid selection.", show_alert=True)
+        await cb.answer("❌ Invalid selection.", show_alert=True)
         return
     mode = data[1]
     if mode not in ['math', 'text']:
-        await cb.answer("Invalid mode.", show_alert=True)
+        await cb.answer("❌ Invalid mode.", show_alert=True)
         return
     captcha_modes[user_id] = mode
-    await cb.edit_message_text(f"CAPTCHA mode has been set to **{'Mathematics' if mode == 'math' else 'Text'}**.")
+    await cb.edit_message_text(f"✅ CAPTCHA mode has been set to **{'Mathematics' if mode == 'math' else 'Text'}**.")
     await cb.answer("The mode has been changed.", show_alert=False)
     logger.info(f"User {user_id} set CAPTCHA mode to {mode}.")
 
 @Client.on_callback_query(filters.regex("^cancel_mode$"))
 async def cancel_mode_callback(client: Client, cb: CallbackQuery):
-    await cb.edit_message_text("CAPTCHA mode setting has been canceled.")
+    await cb.edit_message_text("❌ CAPTCHA mode setting has been canceled.")
     await cb.answer("Cancellation successful.", show_alert=False)
     logger.info(f"User {cb.from_user.id} canceled CAPTCHA mode setting.")
 
@@ -228,27 +223,32 @@ async def join_request(client: Client, event: ChatJoinRequest):
     member = await client.get_chat_member(event.chat.id, "me")
     if member.status != ChatMemberStatus.ADMINISTRATOR:
         return await client.send_message(event.chat.id, text="I am not an administrator in this group.")
-    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)
+
     try:
         chat_link = await client.export_chat_invite_link(event.chat.id)
     except ChatAdminRequired:
         await client.send_message(event.chat.id, text="I need to be an administrator to perform this action.")
         return
+
     mode = captcha_modes.get(event.from_user.id, "math")
     captcha_text, img_path, choices, correct_answer = generate_captcha(event.from_user.id, mode)
-    captcha_texts[event.from_user.id] = (captcha_text, correct_answer)
-    captcha_texts["chat_id"] = event.chat.id
-    captcha_texts["chat_link"] = chat_link
-    buttons = []
-    for choice in choices:
-        buttons.append([InlineKeyboardButton(choice, callback_data=f"verify_{event.from_user.id}_{choice}")])
+    captcha_texts[event.from_user.id] = {
+        'captcha_text': captcha_text,
+        'correct_answer': correct_answer,
+        'chat_id': event.chat.id,
+        'chat_link': chat_link
+    }
+
+    buttons = [
+        [InlineKeyboardButton(choice, callback_data=f"verify_{event.from_user.id}_{choice}")]
+        for choice in choices
+    ]
     buttons.append([
         InlineKeyboardButton("🔄 Refresh CAPTCHA", callback_data="refresh_captcha"),
         InlineKeyboardButton("❌ Cancel", callback_data="cancel_captcha")
     ])
     keyboard = InlineKeyboardMarkup(buttons)
+
     if event.chat.type == ChatType.SUPERGROUP:
         try:
             await client.send_message(
@@ -271,70 +271,6 @@ async def join_request(client: Client, event: ChatJoinRequest):
             )
             logger.error(str(e))
 
-@Client.on_callback_query(filters.regex("^cancel_captcha$"))
-async def cancel_captcha_callback(client: Client, cb: CallbackQuery):
-    user_id = cb.from_user.id
-    if user_id in captcha_texts:
-        del captcha_texts[user_id]
-        logger.info(f"User {user_id} has canceled CAPTCHA verification.")
-        await cb.edit_message_text(
-            "❌ CAPTCHA verification has been canceled. If you wish to try again,",
-            disable_web_page_preview=True
-        )
-        await cb.answer("CAPTCHA verification canceled.", show_alert=False)
-    else:
-        await cb.answer("No active CAPTCHA verification found.", show_alert=True)
-
-@Client.on_callback_query(filters.regex("^close$"))
-async def close_final(client: Client, cb: CallbackQuery):
-    await cb.message.delete()
-
-def create_button_join_group(chat_link):
-    return InlineKeyboardMarkup(
-        [
-            [InlineKeyboardButton("👁️ Join chat", url=chat_linkp)],
-            [InlineKeyboardButton("🔘 Close", callback_data="close")],
-        ]
-    )
-
-def create_button_userinfo(user_id, username):
-    return InlineKeyboardMarkup(
-        [
-            [InlineKeyboardButton("👤 Chmod +W $USER", url=f"tg://user?id={user_id}")],
-            [InlineKeyboardButton("🔔 Check human Bot", url=f"https://t.me/{username}")],
-        ]
-    ) 
-
-@Client.on_callback_query(filters.regex("^refresh_captcha$"))
-async def refresh_captcha_callback(client: Client, cb: CallbackQuery):
-    user_id = cb.from_user.id
-    mode = captcha_modes.get(user_id, 'math')
-    if user_id in captcha_texts:
-        del captcha_texts[user_id]
-        logger.info(f"Old CAPTCHA for user {user_id} has been removed.")
-    captcha_text, img_path, choices, correct_answer = generate_captcha(user_id, mode)
-    captcha_texts[user_id] = (captcha_text, correct_answer)
-    logger.info(f"Generated new {mode} CAPTCHA for user {user_id}: {captcha_text}")
-    buttons = []
-    for choice in choices:
-        buttons.append([InlineKeyboardButton(choice, callback_data=f"verify_{user_id}_{choice}")])
-    buttons.append([
-        InlineKeyboardButton("🔄 Refresh CAPTCHA", callback_data="refresh_captcha"),
-        InlineKeyboardButton("❌ Cancel", callback_data="cancel_captcha")
-    ])
-    keyboard = InlineKeyboardMarkup(buttons)
-    try:
-        await cb.edit_message_media(
-            media=InputMediaPhoto(img_path),
-            reply_markup=keyboard
-        )
-        logger.info(f"Updated CAPTCHA image for user {user_id}.")
-    except Exception as e:
-        await cb.answer("❌ Failed to update CAPTCHA.", show_alert=True)
-        logger.error(f"Error refreshing CAPTCHA for user {user_id}: {e}")
-    os.remove(img_path)
-    await cb.answer("🔄 CAPTCHA has been updated!", show_alert=False)
-
 @Client.on_callback_query(filters.regex("^verify_"))
 async def verify_captcha_callback(client: Client, cb: CallbackQuery):
     data = cb.data.split("_")
@@ -355,9 +291,15 @@ async def verify_captcha_callback(client: Client, cb: CallbackQuery):
         await cb.answer("❗️ No active CAPTCHA verification.", show_alert=True)
         logger.warning(f"User {user_id} mencoba memverifikasi CAPTCHA tanpa aktif.")
         return
-    captcha_text, correct_answer = captcha_texts.get(user_id)
+    captcha_data = captcha_texts.get(user_id)
+    captcha_text = captcha_data['captcha_text']
+    correct_answer = captcha_data['correct_answer']
+    chat_id = captcha_data['chat_id']
+    chat_link = captcha_data['chat_link']
+
     failed_image = failed_hacker_by_randydev()
     hacker_image = thanks_hacker_by_randydev()
+
     try:
         if str(user_choice) == str(correct_answer):
             await cb.edit_message_media(
@@ -365,16 +307,14 @@ async def verify_captcha_callback(client: Client, cb: CallbackQuery):
                     hacker_image,
                     caption="✅ CAPTCHA verification successful!"
                 ),
-                reply_markup=create_button_join_group(captcha_texts.get("chat_link"))
+                reply_markup=create_button_join_group(chat_link)
             )
             logger.info(f"User {user_id} berhasil memverifikasi CAPTCHA.")
             await client.approve_chat_join_request(
-                chat_id=captcha_texts.get("chat_id"),
+                chat_id=chat_id,
                 user_id=user_id
             )
             del captcha_texts[user_id]
-            del captcha_texts["chat_id"]
-            del captcha_texts["chat_link"]
         else:
             await cb.edit_message_media(
                 media=InputMediaPhoto(
@@ -383,21 +323,49 @@ async def verify_captcha_callback(client: Client, cb: CallbackQuery):
                 )
             )
             await client.decline_chat_join_request(
-                chat_id=captcha_texts.get("chat_id"),
+                chat_id=chat_id,
                 user_id=user_id
             )
             logger.info(f"User {user_id} gagal memverifikasi CAPTCHA.")
             del captcha_texts[user_id]
-            del captcha_texts["chat_id"]
-            del captcha_texts["chat_link"]
     except Exception as e:
         await cb.answer(f"Error CAPTCHA: {e}", show_alert=True)
 
-@Client.on_callback_query(filters.regex("^cancel_mode$"))
-async def cancel_mode_callback(client: Client, cb: CallbackQuery):
-    await cb.edit_message_text("❌ CAPTCHA mode setting has been canceled.")
-    await cb.answer("Pembatalan berhasil.", show_alert=False)
-    logger.info(f"User {cb.from_user.id} canceled CAPTCHA mode setting.")
+@Client.on_callback_query(filters.regex("^refresh_captcha$"))
+async def refresh_captcha_callback(client: Client, cb: CallbackQuery):
+    user_id = cb.from_user.id
+    mode = captcha_modes.get(user_id, 'math')
+    if user_id in captcha_texts:
+        del captcha_texts[user_id]
+        logger.info(f"Old CAPTCHA for user {user_id} has been removed.")
+    captcha_text, img_path, choices, correct_answer = generate_captcha(user_id, mode)
+    captcha_texts[user_id] = {
+        'captcha_text': captcha_text,
+        'correct_answer': correct_answer,
+        'chat_id': captcha_texts[user_id]['chat_id'],
+        'chat_link': captcha_texts[user_id]['chat_link']
+    }
+    logger.info(f"Generated new {mode} CAPTCHA for user {user_id}: {captcha_text}")
+    buttons = [
+        [InlineKeyboardButton(choice, callback_data=f"verify_{user_id}_{choice}")]
+        for choice in choices
+    ]
+    buttons.append([
+        InlineKeyboardButton("🔄 Refresh CAPTCHA", callback_data="refresh_captcha"),
+        InlineKeyboardButton("❌ Cancel", callback_data="cancel_captcha")
+    ])
+    keyboard = InlineKeyboardMarkup(buttons)
+    try:
+        await cb.edit_message_media(
+            media=InputMediaPhoto(open(img_path, "rb")),
+            reply_markup=keyboard
+        )
+        logger.info(f"Updated CAPTCHA image for user {user_id}.")
+    except Exception as e:
+        await cb.answer("❌ Failed to update CAPTCHA.", show_alert=True)
+        logger.error(f"Error refreshing CAPTCHA for user {user_id}: {e}")
+    os.remove(img_path)
+    await cb.answer("🔄 CAPTCHA has been updated!", show_alert=False)
 
 @Client.on_callback_query(filters.regex("^cancel_captcha$"))
 async def cancel_captcha_callback(client: Client, cb: CallbackQuery):
@@ -413,12 +381,23 @@ async def cancel_captcha_callback(client: Client, cb: CallbackQuery):
         await cb.answer("❗️ No active CAPTCHA verification found.", show_alert=True)
         logger.warning(f"User {user_id} attempted to cancel CAPTCHA without active verification.")
 
-@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", show_alert=True)
-        return
-    await cb.message.reply_text("🔍 **Please enter the CAPTCHA text exactly as shown in the image:**")
+@Client.on_callback_query(filters.regex("^close$"))
+async def close_final(client: Client, cb: CallbackQuery):
+    await cb.message.delete()
     await cb.answer()
-    logger.info(f"User {user_id} is attempting to verify CAPTCHA.")
+
+def create_button_join_group(chat_link):
+    return InlineKeyboardMarkup(
+        [
+            [InlineKeyboardButton("👁️ Join chat", url=chat_link)],
+            [InlineKeyboardButton("🔘 Close", callback_data="close")],
+        ]
+    )
+
+def create_button_userinfo(user_id, username):
+    return InlineKeyboardMarkup(
+        [
+            [InlineKeyboardButton("👤 Chmod +W $USER", url=f"tg://user?id={user_id}")],
+            [InlineKeyboardButton("🔔 Check human Bot", url=f"https://t.me/{username}")],
+        ]
+    )