Update main.py

main.py

@@ -7,6 +7,7 @@
 # All rights reserved. See COPYING, AUTHORS.
 #
 
+
 import re
 import asyncio
 import logging
@@ -42,6 +43,96 @@ def has_code_entity(message):
             return True
     return False
 
+@bot.on_callback_query(filters.regex(r"^unwarn_(\d+)$"))
+async def unwarn_button(client, callback_query):
+    user_id = int(callback_query.match.group(1))
+    chat_id = callback_query.message.chat.id
+    from_user = callback_query.from_user
+    try:
+        member = await client.get_chat_member(chat_id, from_user.id)
+        if member.status.value not in ("administrator", "creator"):
+            await callback_query.answer("❌ Hanya admin yang bisa menghapus peringatan!", show_alert=True)
+            return
+    except Exception:
+        await callback_query.answer("⚠️ Gagal memverifikasi status admin.", show_alert=True)
+        return
+
+    await db.warns.delete_one({"chat_id": chat_id, "user_id": user_id})
+    await callback_query.answer(f"✅ Peringatan untuk <code>{user_id}</code> telah dihapus.", show_alert=True)
+
+@bot.on_callback_query(filters.regex(r"^warn_(\d+)$"))
+async def warn_button(client, callback_query):
+    user_id = int(callback_query.match.group(1))
+    chat_id = callback_query.message.chat.id
+    from_user = callback_query.from_user
+    try:
+        member = await client.get_chat_member(chat_id, from_user.id)
+        if member.status.value not in ("administrator", "creator"):
+            await callback_query.answer("❌ Hanya admin yang bisa memberikan peringatan!", show_alert=True)
+            return
+    except Exception:
+        await callback_query.answer("⚠️ Gagal memverifikasi status admin.", show_alert=True)
+        return
+
+    warn_data = await db.warns.find_one({"chat_id": chat_id, "user_id": user_id})
+    warn_count = 1
+
+    if warn_data:
+        warn_count = warn_data.get("count", 0) + 1
+        await db.warns.update_one(
+            {"chat_id": chat_id, "user_id": user_id},
+            {"$set": {"count": warn_count}},
+            upsert=True
+        )
+    else:
+        await db.warns.insert_one({"chat_id": chat_id, "user_id": user_id, "count": warn_count})
+
+    if warn_count >= 3:
+        try:
+            await client.ban_chat_member(
+                chat_id,
+                user_id,
+                reply_markup=InlineKeyboardMarkup(
+                    [
+                        [
+                            InlineKeyboardButton(
+                                "Unban", callback_data=f"unban_{user_id}"
+                            )
+                        ]
+                    ]
+                )
+            )
+            await callback_query.message.reply_text(
+                f"🚫 User <code>{user_id}</code> telah dihapus setelah 3 kali peringatan."
+            )
+            await db.warns.delete_one({"chat_id": chat_id, "user_id": user_id})
+        except Exception as e:
+            await callback_query.message.reply_text(f"Gagal menghapus user: {e}")
+    else:
+        await callback_query.answer(f"⚠️ Peringatan ke {warn_count} diberikan.", show_alert=True)
+
+@bot.on_callback_query(filters.regex(r"^unban_(\d+)$"))
+async def unban_button(client, callback_query):
+    user_id = int(callback_query.match.group(1))
+    chat_id = callback_query.message.chat.id
+    from_user = callback_query.from_user
+    try:
+        member = await client.get_chat_member(chat_id, from_user.id)
+        if member.status.value not in ("administrator", "creator"):
+            await callback_query.answer("❌ Hanya admin yang bisa menghapus larangan!", show_alert=True)
+            return
+    except Exception:
+        await callback_query.answer("⚠️ Gagal memverifikasi status admin.", show_alert=True)
+        return
+
+    try:
+        await client.unban_chat_member(chat_id, user_id)
+        await callback_query.message.reply_text(
+            f"✅ User <code>{user_id}</code> telah dihapus dari larangan."
+        )
+    except Exception as e:
+        await callback_query.message.reply_text(f"Gagal menghapus larangan: {e}")
+
 @bot.on_message(filters.command("start") & filters.private)
 async def start_command(client, message):
     reply_markup = InlineKeyboardMarkup( # type: ignore
@@ -323,9 +414,50 @@ async def markdown_code(client, message):
     try:
         if is_blocked_markdown_code(message.text.markdown or ""):
             logging.info(f"is_blocked_markdown_code: Blocked message from {message.from_user.first_name} in {message.chat.title}")
-            return await message.delete()
+            warn = await db.warns.find_one({"chat_id": message.chat.id, "user_id": message.from_user.id})
+            if warn:
+                warn_count = warn.get("count", 0) + 1
+                await db.warns.update_one(
+                    {"chat_id": message.chat.id, "user_id": message.from_user.id},
+                    {"$set": {"count": warn_count}},
+                    upsert=True
+                )
+                if warn_count >= 3:
+                    await client.ban_chat_member(
+                        message.chat.id,
+                        message.from_user.id,
+                        reply_markup=InlineKeyboardMarkup(
+                            [
+                                [
+                                    InlineKeyboardButton(
+                                        "Unban", callback_data=f"unban_{message.from_user.id}"
+                                    )
+                                ]
+                            ]
+                        )
+                    )
+                    await message.reply_text(
+                        f"🚫 User <code>{message.from_user.id}</code> telah dihapus setelah 3 kali peringatan."
+                    )
+                    await db.warns.delete_one({"chat_id": message.chat.id, "user_id": message.from_user.id})
+            else:
+                await db.warns.insert_one({"chat_id": message.chat.id, "user_id": message.from_user.id, "count": 1})
+                await message.reply_text(
+                    f"⚠️ Peringatan pertama untuk <code>{message.from_user.id}</code>.",
+                    reply_markup=InlineKeyboardMarkup(
+                        [
+                            [
+                                InlineKeyboardButton(
+                                    "Unwarn", callback_data=f"unwarn_{message.from_user.id}"
+                                )
+                            ]
+                        ]
+                    )
+                )
+                await message.delete()
     except AttributeError:
         pass
+
     if check_anti_word_by_ryzenth(message.text):
         logging.info(f"check_anti_word_by_ryzenth: Blocked message from {message.from_user.first_name} in {message.chat.title}")
         return await message.delete()