from pyrogram import *
from pyrogram import Client as ren, filters
from pyrogram.types import *
from akn.utils.database import db as db_client

@ren.on_callback_query(filters.regex(r"^statusub_(\d+)$"))
async def check_request_statusub(bot: Client, cb: CallbackQuery):
    user_id = int(cb.matches[0].group(1))
    request = await db_client.session.find_one({"user_id": user_id})
    if not request:
        await cb.answer("No active requests found", show_alert=True)
        return

    status_icon = (
        "🟢"
        if request["user_client"][0]["status"] == "approved"
        else "🔴"
        if request["user_client"][0]["status"] == "rejected"
        else "⚠️"
        if request["user_client"][0]["status"] == "stopped"
        else "🟡"
    )
    await cb.answer(
        f"Request Status: {status_icon} {request['user_client'][0]['status'].capitalize()}\n"
        f"Submitted: {request['user_client'][0].get('timestamp', 'Not available')}\n",
        show_alert=True
    )

@ren.on_callback_query(filters.regex(r"^statusdl_(\d+)$"))
async def check_request_status(bot: Client, cb: CallbackQuery):
    user_id = int(cb.matches[0].group(1))
    request = await db_client.alldl_bot.find_one({"user_id": user_id})
    if not request:
        await cb.answer("No active requests found", show_alert=True)
        return
    status_icon = "🟢" if request["bots"][0]["status"] == "approved" else "🔴" if request["bots"][0]["status"] == "rejected" else "🟡"
    await cb.answer(
        f"Request Status: {status_icon} {request['bots'][0]['status'].capitalize()}\n"
        f"Submitted: {request['bots'][0].get('timestamp', 'Not available')}\n",
        show_alert=True
    )

@ren.on_callback_query(filters.regex(r"^statussesi_(\d+)$"))
async def check_request_status_sesi(bot: Client, cb: CallbackQuery):
    user_id = int(cb.matches[0].group(1))
    request = await db_client.session_bot.find_one({"user_id": user_id})
    if not request:
        await cb.answer("No active requests found", show_alert=True)
        return
    status_icon = "🟢" if request["status"] == "approved" else "🔴" if request["status"] == "rejected" else "🟡"
    await cb.answer(
        f"Request Status: {status_icon} {request['status'].capitalize()}\n"
        f"Submitted: {request.get('timestamp', 'Not available')}\n",
        show_alert=True
    )

@ren.on_callback_query(filters.regex(r"^statuscapt_(\d+)$"))
async def check_request_status_capt(bot: Client, cb: CallbackQuery):
    user_id = int(cb.matches[0].group(1))
    request = await db_client.captcha_bot.find_one({"user_id": user_id})
    if not request:
        await cb.answer("No active requests found", show_alert=True)
        return
    status_icon = "🟢" if request["status"] == "approved" else "🔴" if request["status"] == "rejected" else "🟡"
    await cb.answer(
        f"Request Status: {status_icon} {request['status'].capitalize()}\n"
        f"Submitted: {request.get('timestamp', 'Not available')}\n",
        show_alert=True
    )

@ren.on_callback_query(filters.regex(r"^statusmagic_(\d+)$"))
async def check_request_status_capt(bot: Client, cb: CallbackQuery):
    user_id = int(cb.matches[0].group(1))
    request = await db_client.magic_bot.find_one({"user_id": user_id})
    if not request:
        await cb.answer("No active requests found", show_alert=True)
        return
    status_icon = "🟢" if request["status"] == "approved" else "🔴" if request["status"] == "rejected" else "🟡"
    await cb.answer(
        f"Request Status: {status_icon} {request['status'].capitalize()}\n"
        f"Submitted: {request.get('timestamp', 'Not available')}\n",
        show_alert=True
    )

@ren.on_callback_query(filters.regex(r"^statusyt_(\d+)$"))
async def check_request_status_yt(bot: Client, cb: CallbackQuery):
    user_id = int(cb.matches[0].group(1))
    request = await db_client.youtube_bot.find_one({"user_id": user_id})
    if not request:
        await cb.answer("No active requests found", show_alert=True)
        return
    status_icon = "🟢" if request["status"] == "approved" else "🔴" if request["status"] == "rejected" else "🟡"
    await cb.answer(
        f"Request Status: {status_icon} {request['status'].capitalize()}\n"
        f"Submitted: {request.get('timestamp', 'Not available')}\n",
        show_alert=True
    )

@ren.on_callback_query(filters.regex(r"^statusgm_(\d+)$"))
async def check_request_status_yt(bot: Client, cb: CallbackQuery):
    user_id = int(cb.matches[0].group(1))
    request = await db_client.gemini_bot.find_one({"user_id": user_id})
    if not request:
        await cb.answer("No active requests found", show_alert=True)
        return
    status_icon = "🟢" if request["status"] == "approved" else "🔴" if request["status"] == "rejected" else "🟡"
    await cb.answer(
        f"Request Status: {status_icon} {request['status'].capitalize()}\n"
        f"Submitted: {request.get('timestamp', 'Not available')}\n",
        show_alert=True
    )