feat: Add checkban command for owner to check and unban themselves

DragMusic/plugins/tools/dev.py

@@ -11,7 +11,8 @@ from pyrogram import filters
 from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message, CallbackQuery
 
 from DragMusic import app
-from config import OWNER_ID
+from config import OWNER_ID, BANNED_USERS
+from DragMusic.utils.database import is_banned_user, remove_banned_user, is_gbanned_user, remove_gban_user
 
 
 async def aexec(code, client, message):
@@ -113,3 +114,28 @@ async def close_command(_, CallbackQuery: CallbackQuery):
     
     await CallbackQuery.message.delete()
     await CallbackQuery.answer()
+
+
+@app.on_message(filters.command("checkban") & filters.user(OWNER_ID))
+async def check_ban_status(client, message):
+    """Check if owner is banned and unban if needed."""
+    user_id = message.from_user.id
+    
+    # Check if user is banned locally
+    is_banned = await is_banned_user(user_id)
+    # Check if user is globally banned
+    is_gbanned = await is_gbanned_user(user_id)
+    
+    if is_banned:
+        await remove_banned_user(user_id)
+        await message.reply_text("✅ You were locally banned. You have been unbanned.")
+    elif is_gbanned:
+        await remove_gban_user(user_id)
+        await message.reply_text("✅ You were globally banned. You have been unbanned.")
+    else:
+        await message.reply_text("✅ You are not banned.")
+    
+    # Also remove from BANNED_USERS set if present
+    if user_id in BANNED_USERS:
+        BANNED_USERS.remove(user_id)
+        await message.reply_text("✅ Removed from BANNED_USERS set.")