Update bot/handlers.py

bot/handlers.py

@@ -1,10 +1,10 @@
 import os
 import asyncio
 import random
-import re # ⚡ Added for error extraction
+import re
 import datetime
 from pyrogram import filters, enums
-from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
+from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, LinkPreviewOptions
 from pyrogram.errors import MessageNotModified
 from bot.client import bot
 from bot.state import state
@@ -13,12 +13,12 @@ from bot.scheduler import scheduler, schedule_next, get_pkt_time
 from utils.logger import log, get_recent_logs
 from bot.sniper import fleet
 from bot.proxy_manager import proxy_mgr
-from pyrogram.types import LinkPreviewOptions
 
 from bot.ui.keyboards import (
     markup_cancel, markup_main, markup_settings, 
     markup_claim_settings, markup_fragment_settings, 
-    markup_channel_accounts, markup_db_manager, markup_confirm
+    markup_channel_accounts, markup_db_manager, markup_confirm,
+    markup_bot_query
 )
 from bot.ui.text_views import (
     text_main, get_settings_text, text_dashboard, 
@@ -61,7 +61,8 @@ def is_admin(_, __, update):
 
 admin_filter = filters.create(is_admin)
 
-def get_account_by_index(idx: int): return next((a for a in state["account_pool"] if a.index == idx), None)
+def get_account_by_index(idx: int): 
+    return next((a for a in state["account_pool"] if a.index == idx), None)
 
 async def input_timeout(user_id: int, expected_state: str, msg):
     await asyncio.sleep(60)
@@ -209,7 +210,7 @@ async def cmd_addsnipe(client, message):
         try: await message.react("⚡")
         except Exception: pass
     else:
-        await message.reply_text(f"⚠️ <b>@{target}</b> is already in the snipe queue.", parse_mode=enums.ParseMode.HTML)
+        await message.reply_text(f"⚠️️ <b>@{target}</b> is already in the snipe queue.", parse_mode=enums.ParseMode.HTML)
 
 @bot.on_message(filters.command("rmsnipe") & admin_filter)
 async def cmd_rmsnipe(client, message):
@@ -311,10 +312,11 @@ async def cmd_fragcheck(client, message):
 
 @bot.on_message(filters.command("unfrag") & admin_filter)
 async def cmd_unfrag(client, message):
+    user_id = message.from_user.id
     if len(message.command) < 2:
         state["waiting_for_input"][user_id] = "unfrag_input"
         msg = await message.reply_text("Send the username to remove from Fragment Scraper:", reply_markup=markup_cancel(), parse_mode=enums.ParseMode.HTML)
-        asyncio.create_task(input_timeout(message.from_user.id, "unfrag_input", msg))
+        asyncio.create_task(input_timeout(user_id, "unfrag_input", msg))
         return
         
     target = message.command[1].replace("@", "").strip().lower()
@@ -338,7 +340,6 @@ async def cmd_proxy_mapping_dashboard(client, message):
     if not state.get("account_pool"):
         return await message.reply_text("❌ System error: Account pool is empty.")
         
-    # ⚡ UI UPGRADE: Wrapped in Expandable Blockquote to save screen space
     dashboard_lines = [
         "<b>🛡️ Cluster Network Egress Mapping</b>\n",
         f"<b>Proxy Engine:</b> {'✅ ACTIVE' if proxy_mgr.is_enabled else '❌ DISABLED'}",
@@ -523,7 +524,6 @@ async def handle_media_and_text_input(client, message):
         state["waiting_for_input"][user_id] = None
         return
     
-    # ⚡ ENGINE UPGRADE: Fully Automated Secure Transfer Initiation 
     if waiting_for and waiting_for.startswith("trans_target|"):
         _, idx_str, chat_id_str = waiting_for.split("|")
         target = message.text.strip()
@@ -551,7 +551,6 @@ async def handle_media_and_text_input(client, message):
                 else:
                     await msg.edit_text("⏳ <b>Telegram Security Lock:</b> Session must be active for 24 hours before transferring ownership.", parse_mode=enums.ParseMode.HTML)
             elif "PASSWORD_REQUIRED" in result or "PASSWORD_HASH_INVALID" in result:
-                # Fallback to manual password input if the user changed the default 2FA password
                 state["waiting_for_input"][user_id] = f"trans_2fa|{idx_str}|{chat_id_str}|{target}"
                 await msg.edit_text(f"🔒 <b>2FA Password Required</b>\n\nAccount #{idx_str} has a custom Two-Step Verification password.\nPlease send the password:", reply_markup=markup_cancel(), parse_mode=enums.ParseMode.HTML)
                 asyncio.create_task(input_timeout(user_id, state["waiting_for_input"][user_id], msg))
@@ -576,7 +575,7 @@ async def handle_media_and_text_input(client, message):
         if success:
             await msg.edit_text(f"✅ <b>Ownership transferred to {target}.</b>", parse_mode=enums.ParseMode.HTML)
         else:
-             await msg.edit_text(f"❌ Transfer Failed: <b>{result}</b>", parse_mode=enums.ParseMode.HTML)
+            await msg.edit_text(f"❌ Transfer Failed: <b>{result}</b>", parse_mode=enums.ParseMode.HTML)
         return
 
     if waiting_for == "watchdog_add":
@@ -642,9 +641,64 @@ async def callback_router(client, callback_query):
     data = callback_query.data
     user_id = callback_query.from_user.id
     try:
+        if data == "ignore":
+            return await callback_query.answer()
+
+        if data == "menu_bot_query":
+            await callback_query.message.edit_text(
+                "<b>🪙 Account Bot Query Menu</b>\nSelect an account to send <code>/coins</code> and fetch its balance:",
+                reply_markup=markup_bot_query(),
+                parse_mode=enums.ParseMode.HTML
+            )
+            return
+
+        if data.startswith("query_coins|"):
+            acc_idx = int(data.split("|")[1])
+            acc = get_account_by_index(acc_idx)
+            if not acc:
+                return await callback_query.answer("Account unavailable.", show_alert=True)
+
+            await callback_query.message.edit_text(f"⏳ Sending <code>/coins</code> from Account #{acc_idx}...", parse_mode=enums.ParseMode.HTML)
+            
+            from bot.checker import send_and_await_reply
+            target_bot = "@info_id_bot"
+            reply = await send_and_await_reply(acc, target_bot, "/coins", timeout=8)
+
+            text = (
+                f"<b>🪙 Balance Report (Acc #{acc_idx})</b>\n"
+                f"Target Bot: <code>{target_bot}</code>\n\n"
+                f"<blockquote expandable>{reply}</blockquote>"
+            )
+            buttons = [
+                [InlineKeyboardButton("🔄 Re-check", callback_data=f"query_coins|{acc_idx}", style=enums.ButtonStyle.PRIMARY)],
+                [InlineKeyboardButton("⬅️ Back", callback_data="menu_bot_query", style=enums.ButtonStyle.PRIMARY)]
+            ]
+            await callback_query.message.edit_text(text, reply_markup=InlineKeyboardMarkup(buttons), parse_mode=enums.ParseMode.HTML)
+            return
+
+        if data == "query_coins_all":
+            await callback_query.message.edit_text("⏳ Querying <code>/coins</code> across all active accounts...", parse_mode=enums.ParseMode.HTML)
+            
+            from bot.checker import send_and_await_reply
+            target_bot = "@info_id_bot"
+            active_accounts = [a for a in state["account_pool"] if a.status in ["active", "paused"] and a.client]
+
+            results = []
+            for acc in active_accounts:
+                masked = f"****{acc.phone[-4:]}" if len(acc.phone) >= 4 else acc.phone
+                res = await send_and_await_reply(acc, target_bot, "/coins", timeout=6)
+                first_line = res.split("\n")[0] if res else "No response"
+                results.append(f"<b>Acc #{acc.index} ({masked}):</b>\n{first_line}")
+
+            summary = "\n\n".join(results) if results else "No active accounts found."
+            text = f"<b>🪙 Fleet Balance Summary</b>\n\n<blockquote expandable>{summary}</blockquote>"
+            buttons = [[InlineKeyboardButton("⬅️ Back", callback_data="menu_bot_query", style=enums.ButtonStyle.PRIMARY)]]
+            await callback_query.message.edit_text(text, reply_markup=InlineKeyboardMarkup(buttons), parse_mode=enums.ParseMode.HTML)
+            return
+
         if data == "toggle_proxy":
             if len(proxy_mgr.proxies) == 0:
-                return await callback_query.answer("⚠️️ No proxies available! Check WEBSHARE_API_KEY.", show_alert=True)
+                return await callback_query.answer("⚠️ No proxies available! Check WEBSHARE_API_KEY.", show_alert=True)
             
             proxy_mgr.is_enabled = not proxy_mgr.is_enabled
             proxy_text = "🟢 Proxy: ON" if proxy_mgr.is_enabled else "🔴 Proxy: OFF"
@@ -701,7 +755,7 @@ async def callback_router(client, callback_query):
             
             text = f"🔍 <b>@{target}</b>\n\n<blockquote>Snipe: {in_snipe}\nCheck: {in_check}\nWatchdog: {in_watchdog}\nFragment: {in_frag}\nUnavailable DB: {in_unavail}</blockquote>"
             buttons = []
-            if target in state["snipe_queue"]: buttons.append([InlineKeyboardButton("🗑️ Remove from Snipe", callback_data=f"rm_q|s|{target}", style=enums.ButtonStyle.DANGER)])
+            if target in state["snipe_queue"]: buttons.append([InlineKeyboardButton("🗑️️ Remove from Snipe", callback_data=f"rm_q|s|{target}", style=enums.ButtonStyle.DANGER)])
             if target in state["watchdog_queue"]: buttons.append([InlineKeyboardButton("🗑️ Remove from Watchdog", callback_data=f"rm_q|w|{target}", style=enums.ButtonStyle.DANGER)])
             if target in state["fragcheck_queue"]: buttons.append([InlineKeyboardButton("🗑️ Remove from Fragment", callback_data=f"rm_q|f|{target}", style=enums.ButtonStyle.DANGER)])
             buttons.append([InlineKeyboardButton("❌ Dismiss", callback_data="cancel_input", style=enums.ButtonStyle.DEFAULT)])
@@ -786,7 +840,6 @@ async def callback_router(client, callback_query):
             markup = InlineKeyboardMarkup([[InlineKeyboardButton("📊 Live Dashboard", callback_data="start_live_dash", style=enums.ButtonStyle.PRIMARY)], [InlineKeyboardButton("⬅️ Back", callback_data="main_panel", style=enums.ButtonStyle.PRIMARY)]])
             await callback_query.message.edit_text(text_dashboard(), reply_markup=markup, parse_mode=enums.ParseMode.HTML)
 
-        # ⚡ UI UPGRADE: Expandable Blockquote
         elif data in ["view_logs", "refresh_logs"]:
             logs = get_recent_logs(20)
             log_text = "\n".join(logs) if logs else "No logs yet."
@@ -1370,7 +1423,7 @@ async def callback_router(client, callback_query):
                     state["unavailable_set"].discard(username)
                     if hasattr(db, "remove_unavailable"): await db.remove_unavailable(username)
                     await purge_target_from_queues(username)
-                    await callback_query.message.edit_text(f"🗑️️ Channel deleted. <b>@{username}</b> released and purged from active queues.", parse_mode=enums.ParseMode.HTML)
+                    await callback_query.message.edit_text(f"🗑️ Channel deleted. <b>@{username}</b> released and purged from active queues.", parse_mode=enums.ParseMode.HTML)
             except Exception as e:
                 if "FROZEN" in str(e).upper(): account.status = "frozen"
                 await callback_query.message.edit_text(f"❌ Undo Failed: <b>{e}</b>", parse_mode=enums.ParseMode.HTML)