Upload join_request.py

chatbot/plugins/join_request.py

@@ -31,31 +31,85 @@ logger = logging.getLogger(__name__)
 
 user_list = []
 captcha_texts = {}
+captcha_modes = {}
 
-def thanks_hacker_by_randydev():
-    url = "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcR0u8UqJ2JDhfmPOCb_zAHjUQG2NYMjTwLbkq_sQhCQOxX8hn66YbaGFvLL&s=10"
-    response = requests.get(url)
-    image_hacker = "hacker.png"
-    with open(image_hacker, "wb") as f:
-        f.write(response.content)
-    return image_hacker
+def generate_math_captcha():
+    operations = ['+', '-', '*']
+    operation = random.choice(operations)
 
-async def remove_captcha_after_timeout(client, user_id, delay=300):
-    await asyncio.sleep(delay)
-    if user_id in captcha_texts:
-        del captcha_texts[user_id]
-        await client.send_message(user_id, "⏰ Your CAPTCHA verification has expired. Please try again.")
-        logger.info(f"CAPTCHA for user {user_id} has expired and been removed.")
+    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])
+        if str(wrong_answer) not in choices:
+            choices.append(str(wrong_answer))
+    random.shuffle(choices)
+
+    width, height = 250, 100
+    background_color = (random.randint(200, 255), random.randint(200, 255), random.randint(200, 255))  # Warna pastel
+    img = Image.new('RGB', (width, height), color=background_color)
+    d = ImageDraw.Draw(img)
+
+    for _ in range(1000):
+        x = random.randint(0, width)
+        y = random.randint(0, height)
+        noise_color = (random.randint(150, 200), random.randint(150, 200), random.randint(150, 200))
+        d.point((x, y), fill=noise_color)
+
+    try:
+        font = ImageFont.truetype("arial.ttf", 40)
+    except IOError:
+        font = ImageFont.load_default()
 
-def generate_captcha_multiple_choice():
+    text_width, text_height = d.textsize(captcha_text, font=font)
+    text_x = (width - text_width) / 2
+    text_y = (height - text_height) / 2 - 20
+
+    text_image = Image.new('RGBA', (text_width, text_height), (255, 255, 255, 0))
+    text_draw = ImageDraw.Draw(text_image)
+    text_draw.text((0, 0), captcha_text, font=font, fill=(0, 0, 0))
+    rotated_text = text_image.rotate(random.randint(-25, 25), expand=1)
+    img.paste(rotated_text, (int(text_x), int(text_y)), rotated_text)
+
+    choice_font = ImageFont.truetype("arial.ttf", 30) if font else ImageFont.load_default()
+    for idx, choice in enumerate(choices):
+        choice_text = f"{idx + 1}. {choice}"
+        choice_width, choice_height = d.textsize(choice_text, font=choice_font)
+        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))
+        line_color = (random.randint(0, 150), random.randint(0, 150), random.randint(0, 150))
+        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():
     letters = string.ascii_uppercase + string.digits
     captcha_text = ''.join(random.choice(letters) for _ in range(5))
+
     choices = [captcha_text]
-    for _ in range(2):
+    while len(choices) < 3:
         wrong_choice = ''.join(random.choice(letters) for _ in range(5))
-        while wrong_choice in choices:
-            wrong_choice = ''.join(random.choice(letters) for _ in range(5))
-        choices.append(wrong_choice)
+        if wrong_choice not in choices:
+            choices.append(wrong_choice)
     random.shuffle(choices)
 
     width, height = 200, 80
@@ -68,7 +122,7 @@ def generate_captcha_multiple_choice():
         y = random.randint(0, height)
         noise_color = (random.randint(150, 200), random.randint(150, 200), random.randint(150, 200))
         d.point((x, y), fill=noise_color)
-    
+
     try:
         font = ImageFont.truetype("arial.ttf", 45)
     except IOError:
@@ -77,25 +131,99 @@ def generate_captcha_multiple_choice():
     text_width, text_height = d.textsize(captcha_text, font=font)
     text_x = (width - text_width) / 2
     text_y = (height - text_height) / 2
-    
+
     text_image = Image.new('RGBA', (text_width, text_height), (255, 255, 255, 0))
     text_draw = ImageDraw.Draw(text_image)
     text_draw.text((0, 0), captcha_text, font=font, fill=(0, 0, 0))
     rotated_text = text_image.rotate(random.randint(-25, 25), expand=1)
     img.paste(rotated_text, (int(text_x), int(text_y)), rotated_text)
 
+    choice_font = ImageFont.truetype("arial.ttf", 30) if font else ImageFont.load_default()
+    for idx, choice in enumerate(choices):
+        choice_text = f"{idx + 1}. {choice}"
+        choice_width, choice_height = d.textsize(choice_text, font=choice_font)
+        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))
         line_color = (random.randint(0, 150), random.randint(0, 150), random.randint(0, 150))
         d.line([start, end], fill=line_color, width=2)
-    
+
     img = img.filter(ImageFilter.BLUR)
-    img_path = f"captcha_{captcha_text}.png"
+    img_path = f"captcha_text_{captcha_text}.png"
     img.save(img_path)
-    return captcha_text, img_path, choices
+    return captcha_text, img_path, choices, captcha_text
+
+def generate_captcha(user_id, mode='math'):
+    if mode == 'math':
+        return generate_math_captcha()
+    else:
+        return generate_text_captcha()
+
+def thanks_hacker_by_randydev():
+    url = "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcR0u8UqJ2JDhfmPOCb_zAHjUQG2NYMjTwLbkq_sQhCQOxX8hn66YbaGFvLL&s=10"
+    response = requests.get(url)
+    image_hacker = "hacker.png"
+    with open(image_hacker, "wb") as f:
+        f.write(response.content)
+    return image_hacker
+
+def failed_hacker_by_randydev():
+    url = "https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcTM0vb4s59H9F1-_7FyELiLU04e8bCHy7o6KQV2mG3DFRLnzP547KjckKG2&s=10"
+    response = requests.get(url)
+    image_hacker = "failed_hacker.png"
+    with open(image_hacker, "wb") as f:
+        f.write(response.content)
+    return image_hacker
+
+async def remove_captcha_after_timeout(client, user_id, delay=300):
+    await asyncio.sleep(delay)
+    if user_id in captcha_texts:
+        del captcha_texts[user_id]
+        await client.send_message(user_id, "⏰ Your CAPTCHA verification has expired. Please try again.")
+        logger.info(f"CAPTCHA for user {user_id} has expired and been removed.")
+
+@Client.on_message(filters.command("settingmode") & filters.private)
+async def setting_mode(client: Client, message: Message):
+    user_id = message.from_user.id
+    keyboard = InlineKeyboardMarkup(
+        [
+            [InlineKeyboardButton("🔢 Mathematics", callback_data="mode_math")],
+            [InlineKeyboardButton("🔤 Text", callback_data="mode_text")],
+            [InlineKeyboardButton("❌ Cancel", callback_data="cancel_mode")]
+        ]
+    )
+    await message.reply_text(
+        "📍 Select the CAPTCHA mode you want:",
+        reply_markup=keyboard
+    )
+
+@Client.on_callback_query(filters.regex("^mode_"))
+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)
+        return
+    mode = data[1]
+    if mode not in ['math', 'text']:
+        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.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.answer("Cancellation successful.", show_alert=False)
+    logger.info(f"User {cb.from_user.id} canceled CAPTCHA mode setting.")
 
-@Client.on_chat_join_request()
+@Client.on_chat_join_request(filters.chat("KillerXSupport"))
 async def join_request(client: Client, event: ChatJoinRequest):
     member = await client.get_chat_member(event.chat.id, "me")
     if member.status != ChatMemberStatus.ADMINISTRATOR:
@@ -108,19 +236,19 @@ async def join_request(client: Client, event: ChatJoinRequest):
     except ChatAdminRequired:
         await client.send_message(event.chat.id, text="I need to be an administrator to perform this action.")
         return
-    captcha_text, img_path, choices = generate_captcha_multiple_choice()
-    captcha_texts[event.from_user.id] = captcha_text
+    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
-    keyboard = InlineKeyboardMarkup(
-        [
-            [InlineKeyboardButton(choice, callback_data=f"verify_{event.from_user.id}_{choice}")]
-            for choice in choices
-        ] + [
-            [InlineKeyboardButton("🔄 Refresh CAPTCHA", callback_data="refresh_captcha")],
-            [InlineKeyboardButton("❌ Cancel", callback_data="cancel_captcha")]
-        ]
-    )
+    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)
     if event.chat.type == ChatType.SUPERGROUP:
         try:
             await client.send_message(
@@ -164,7 +292,7 @@ async def close_final(client: Client, cb: CallbackQuery):
 def create_button_join_group(chat_link):
     return InlineKeyboardMarkup(
         [
-            [InlineKeyboardButton("👁️ Join chat", url=chat_link)],
+            [InlineKeyboardButton("👁️ Join chat", url=chat_link")],
             [InlineKeyboardButton("🔘 Close", callback_data="close")],
         ]
     )
@@ -180,45 +308,57 @@ def create_button_userinfo(user_id, 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]
-    captcha_text, img_path, choices = generate_captcha_multiple_choice()
-    captcha_texts[user_id] = captcha_text
-    keyboard = InlineKeyboardMarkup(
-        [
-            [InlineKeyboardButton("🔄 Refresh CAPTCHA", callback_data="refresh_captcha")],
-            [InlineKeyboardButton("✅ Verify", callback_data="verify_captcha")],
-            [InlineKeyboardButton("❌ Cancel", callback_data="cancel_captcha")]
-        ]
-    )
-    await cb.edit_message_media(
-        media=InputMediaPhoto(
-            img_path,
-            caption=f"❗️ **Verify that you are human!**\n\n❔ Please select the correct CAPTCHA text shown in the image below.",
-        ),
-        reply_markup=keyboard
-    )
+        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(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 refreshed!", show_alert=False)
+    await cb.answer("🔄 CAPTCHA has been updated!", show_alert=False)
 
 @Client.on_callback_query(filters.regex("^verify_"))
-async def verify_captcha_multiple_choice_callback(client: Client, cb: CallbackQuery):
+async def verify_captcha_callback(client: Client, cb: CallbackQuery):
     data = cb.data.split("_")
     if len(data) != 3:
-        await cb.answer("Invalid data format.", show_alert=True)
+        await cb.answer("❌ Format data tidak valid.", show_alert=True)
         return
     _, user_id_str, user_choice = data
     try:
         user_id = int(user_id_str)
     except ValueError:
-        await cb.answer("Invalid user ID.", show_alert=True)
+        await cb.answer("❌ Invalid user ID.", show_alert=True)
+        return
+    if cb.from_user.id != user_id:
+        await cb.answer("❌ You have no right to do this.", show_alert=True)
+        logger.warning(f"User {cb.from_user.id} mencoba memverifikasi CAPTCHA milik user {user_id}.")
         return
     if user_id not in captcha_texts:
-        await cb.answer("❗️ Please start the CAPTCHA verification first.", show_alert=True)
+        await cb.answer("❗️ No active CAPTCHA verification.", show_alert=True)
+        logger.warning(f"User {user_id} mencoba memverifikasi CAPTCHA tanpa aktif.")
         return
-    correct_captcha = captcha_texts.get(user_id)
-    if user_choice == correct_captcha:
-        hacker_image = thanks_hacker_by_randydev()
+    captcha_text, correct_answer = captcha_texts.get(user_id)
+    failed_image = failed_hacker_by_randydev()
+    hacker_image = thanks_hacker_by_randydev()
+    if str(user_choice) == str(correct_answer):
         await cb.edit_message_media(
             media=InputMediaPhoto(
                 hacker_image,
@@ -226,7 +366,7 @@ async def verify_captcha_multiple_choice_callback(client: Client, cb: CallbackQu
             ),
             reply_markup=create_button_join_group(captcha_texts.get("chat_link"))
         )
-        logger.info(f"User {user_id} successfully verified CAPTCHA.")
+        logger.info(f"User {user_id} berhasil memverifikasi CAPTCHA.")
         await client.approve_chat_join_request(
             chat_id=captcha_texts.get("chat_id"),
             user_id=user_id
@@ -235,14 +375,41 @@ async def verify_captcha_multiple_choice_callback(client: Client, cb: CallbackQu
         del captcha_texts["chat_id"]
         del captcha_texts["chat_link"]
     else:
-        await cb.edit_message_text("❌ Incorrect CAPTCHA. Please try again")
+        await cb.edit_message_media(
+            media=InputMediaPhoto(
+                failed_image,
+                caption="❌ **CAPTCHA verification failed.**\n\nPlease try again."
+            )
+        )
         await client.decline_chat_join_request(
             chat_id=captcha_texts.get("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"]
+    await cb.answer()
+
+@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("^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_caption(
+            caption="❌ **CAPTCHA verification has been canceled.**\n\nIf you want to try again.",
+        )
+        await cb.answer("CAPTCHA verification canceled.", show_alert=False)
+    else:
+        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):