from pagermaid import bot, log
from pagermaid.listener import listener
from telethon.errors import rpcerrorlist
from asyncio import sleep            
from datetime import timedelta
from telethon.tl.types import ChannelParticipantsAdmins


@listener(incoming=True, outgoing=True, command="portball",
          description="回复你要临时禁言的人的消息来实现XX秒的禁言。",
          parameters="<理由> <时间 单位：秒>")
async def portball(context):
    reply = await context.get_reply_message()
    if context.is_group:
        if reply:
            action = context.arguments.split()
            if reply.sender.last_name == None:
                last_name = ''
            else:
                last_name = reply.sender.last_name
            if int(action[1])<60:
                notification = await bot.send_message(context.chat_id, '诶呀不要小于60秒啦')
                await sleep(10)
                await notification.delete()
                try:
                    await context.delete()
                except:
                    pass
                return False
            admins = await context.client.get_participants(context.chat, filter=ChannelParticipantsAdmins)
            if context.sender in admins:
                try:
                    await bot.edit_permissions(context.chat_id, reply.sender.id, timedelta(seconds=int(action[1].replace(' ',''))), send_messages=False)
                    await bot.send_message(
                        context.chat_id,
                        f'[{reply.sender.first_name}{last_name}](tg://user?id={reply.sender.id}) 由于 {action[0]} \n'
                        f'被暂时禁言{action[1]}秒',
                        reply_to = reply.id
                    )
                except rpcerrorlist.UserAdminInvalidError:
                    notification = await bot.send_message(context.chat_id, '错误：我没有管理员权限或我的权限比被封禁的人要小')
                    await sleep(10)
                    await notification.delete()
                except rpcerrorlist.ChatAdminRequiredError:
                    notification = await bot.send_message(context.chat_id, '错误：我没有管理员权限或我的权限比被封禁的人要小')
                    await sleep(10)
                    await notification.delete()
            else:
                notification = await bot.send_message(context.chat_id, '诶呀你不是管理员，不要给人家塞口球啊')
                await sleep(10)
                await notification.delete()
        else:
            notification = await bot.send_message(context.chat_id, '你好蠢诶，都没有回复人，我哪知道你要搞谁的事情……')
            await sleep(10)
            await notification.delete()
    else:
        notification = await bot.send_message(context.chat_id, '你好蠢诶，又不是群组，怎么禁言啦！')
        await sleep(10)
        await notification.delete()
    try:
        await context.delete()
    except:
        pass