Update main.py

main.py

@@ -409,7 +409,47 @@ async def markdown_code(client, message):
 
     if has_code_entity(message):
         logging.info(f"has_code_entity: 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()
 
     try:
         if is_blocked_markdown_code(message.text.markdown or ""):