import contextlib

from asyncio import sleep

from pyrogram.errors import FloodWait
from pyrogram.raw.functions.contacts import GetBlocked
from pyrogram.raw.types.contacts import BlockedSlice

from pagermaid.listener import listener
from pagermaid.enums import Client, Message


async def clear_blocked_func(client: Client, message: Message):
    offset = 0
    success, failed, skipped = 0, 0, 0
    while True:
        blocked = await client.invoke(GetBlocked(offset=offset, limit=100))
        if not blocked.users:
            break
        for user in blocked.users:
            if (user.bot or user.scam or user.fake) and not user.deleted:
                skipped += 1
                continue
            try:
                await client.unblock_user(user.id)
                success += 1
            except FloodWait as e:
                with contextlib.suppress(Exception):
                    await message.edit(
                        f"🧹 Clearing blocked users...\n\nWill run after {e.value} seconds."
                    )
                await sleep(e.value + 1)
                with contextlib.suppress(Exception):
                    await message.edit("🧹 Clearing blocked users...")
                await client.unblock_user(user.id)
                success += 1
            except Exception:
                failed += 1
        offset += 100
        if (
            isinstance(blocked, BlockedSlice) and offset > blocked.count
        ) or not isinstance(blocked, BlockedSlice):
            break
    return success, failed, skipped


@listener(command="clear_blocked", description="Clear blocked users.", need_admin=True)
async def clear_blocked(client: Client, message: Message):
    """Clear blocked users."""
    message: Message = await message.edit("🧹 Clearing blocked users...")
    try:
        success, failed, skipped = await clear_blocked_func(client, message)
    except Exception as e:
        await message.edit(f"❌ Failed to clear blocked users: {e}")
        return
    await message.edit(
        f"🧹 Clear blocked users complete. \n"
        f"Success: {success}, Failed: {failed}, Skipped: {skipped}"
    )