Update akn/manage/builderclone_bots.py

akn/manage/builderclone_bots.py

@@ -10,6 +10,52 @@ from akn.utils.database import db as db_client
 from config import PRIVATE_LOGS
 from akn.utils.logger import LOGS
 
+from typing import Dict
+
+storage_rm_token: Dict[int, str] = {}
+
+@ren.on_callback_query(filters.regex("^rmdeploybots$"))
+async def bot_token_removal(bot: Client, cb: CallbackQuery):
+    user_id = cb.from_user.id
+    
+    try:
+        if user_id not in storage_rm_token:
+            await cb.answer("❌ No token found. Please restart the removal process.", show_alert=True)
+            return
+
+        bot_token = storage_rm_token[user_id]
+        
+        token_exists = await db_client.alldl_bot.find_one(
+            {"user_id": user_id, "bots.bot_token": bot_token}
+        )
+        
+        if not token_exists:
+            await cb.answer("⚠️ Token not found in your registered bots", show_alert=True)
+            return
+
+        result = await db_client.alldl_bot.update_one(
+            {"user_id": user_id},
+            {"$pull": {"bots": {"bot_token": bot_token}}}
+        )
+
+        if result.modified_count > 0:
+            storage_rm_token.pop(user_id, None)
+            
+            await cb.message.edit_text(
+                "✅ Bot token successfully removed from your account.\n\n"
+                "For security reasons:\n"
+                "1. Revoke the token via @BotFather\n"
+                "2. Delete any remaining session files"
+            )
+        else:
+            await cb.answer("⚠️ No changes made. Token may have been already removed.", show_alert=True)
+
+    except FloodWait as e:
+        await cb.answer(f"⏳ Please wait {e.value} seconds before trying again", show_alert=True)
+    except Exception as e:
+        LOGS.error(f"Token removal failed for {user_id}: {str(e)}")
+        await cb.answer("🚨 Server error during removal. Contact support.", show_alert=True)
+
 @ren.on_callback_query(filters.regex("^alldl_bot$"))
 async def new_alldlbot_clone(bot: Client, cb: CallbackQuery):
     user_id = cb.from_user.id
@@ -63,13 +109,15 @@ async def new_alldlbot_clone(bot: Client, cb: CallbackQuery):
             {"bots.bot_token": bot_token}
         )
         if token_exists:
+            storage_rm_token[user_id] = bot_token
             await bot.send_message(
                 chat_id,
                 "⚠️ This bot token is already registered in our system!\n\n"
                 "Please create a new bot via @BotFather if you want to deploy another instance.",
                 reply_markup=InlineKeyboardMarkup([
                     [InlineKeyboardButton("🆕 Create New Bot", url="https://t.me/BotFather")],
-                    [InlineKeyboardButton("🔄 Try Again", callback_data="alldl_bot")],
+                    [InlineKeyboardButton("🔄 Try Again", callback_data="alldl_bot"),
+                     InlineKeyboardButton("⚠️ Delete Token", callback_data="rmdeploybots")],
                     [InlineKeyboardButton("❌ Cancel", callback_data="customzie_bot")]
                 ])
             )