from pyrogram.enums import ChatMemberStatus
from pyrogram.errors import (
    ChatAdminRequired,
    FloodWait,
    PeerIdInvalid,
    UsernameInvalid,
    UserNotParticipant,
)
from pyrogram.types import Chat

from pagermaid.listener import listener
from pagermaid.enums import Client, Message
from pagermaid.services import bot
from pagermaid.utils import lang


async def check_uid(chat: Chat, uid: str):
    member = None
    try:
        uid = int(uid)
        member = await bot.get_chat_member(chat.id, uid)
    except ValueError:
        try:
            chat = await bot.get_chat(uid)
            uid = chat.id
        except PeerIdInvalid:
            member = await bot.get_chat_member(chat.id, uid)
            uid = member.user.id
    if not member:
        member = await bot.get_chat_member(chat.id, uid)
    return uid, member


async def get_uid(chat: Chat, message: Message):
    uid = None
    member = None
    if reply := message.reply_to_message:
        if sender := reply.from_user:
            uid = sender.id
        if sender := reply.sender_chat:
            uid = sender.id
        member = await bot.get_chat_member(chat.id, uid)
    elif len(message.parameter) == 1:
        uid, member = await check_uid(chat, message.arguments)
    return uid, member


@listener(
    command="unban",
    description="解除封禁一位用户",
    need_admin=True,
    groups_only=True,
    parameters="[reply|id|username]",
)
async def unban(client: Client, message: Message):
    chat = message.chat
    try:
        uid, member = await get_uid(chat, message)
    except (ValueError, PeerIdInvalid, UsernameInvalid, FloodWait):
        return await message.edit(lang("arg_error"))
    except UserNotParticipant:
        return await message.edit("此用户未被限制。")
    if not uid:
        return await message.edit(lang("arg_error"))
    if uid == chat.id:
        return await message.edit(lang("arg_error"))
    try:
        if not member:
            member = await bot.get_chat_member(chat.id, uid)
        if member.status in [ChatMemberStatus.RESTRICTED, ChatMemberStatus.BANNED]:
            await client.unban_chat_member(chat.id, uid)
        else:
            return await message.edit("此用户未被限制。")
    except UserNotParticipant:
        return await message.edit("此用户未被限制。")
    except ChatAdminRequired:
        return await message.edit(lang("sb_no_per"))
    except Exception as e:
        return await message.edit(f"出现错误：{e}")
    await message.edit("已解封此用户。")