import asyncio,re
from discord.ext import commands
def uid(s):
    m=re.fullmatch(r"<@!?(\d+)>|(\d+)",s.strip())
    return int(m[1] or m[2]) if m else None
class perm(commands.Cog):
    def __init__(s,bot):
        s.bot=bot;s.names={};s.sem=asyncio.Semaphore(2)
    async def cog_check(s,ctx):return ctx.author.id==s.bot.user.id
    async def name(s,i):
        if i in s.names:return s.names[i]
        u=s.bot.get_user(i)
        if u is None:
            async with s.sem:
                try:u=await s.bot.fetch_user(i)
                except Exception:u=None
                await asyncio.sleep(.6)
        n=(u.global_name or u.name) if u else "unknown"
        if u:s.names[i]=n
        return n
    @commands.group(invoke_without_command=True)
    async def perm(s,ctx):await ctx.send("`;perm list|add <id|mention>|remove <id|mention>`")
    @perm.command(name="list")
    async def _list(s,ctx):
        ids=sorted(s.bot.wl.ids)
        if not ids:return await ctx.send("whitelist empty")
        n=await asyncio.gather(*(s.name(i) for i in ids))
        await ctx.send("\n".join(f"{i} - {x}" for i,x in zip(ids,n)))
    @perm.command(name="add")
    async def _add(s,ctx,target:str):
        i=uid(target)
        if i is None:return await ctx.send("invalid id")
        await s.bot.wl.mutate(i,True);s.bot.hook.put(f"whitelist: {i} added by {ctx.author.id}")
        await ctx.send(f"{i} added ({await s.name(i)})")
    @perm.command(name="remove")
    async def _remove(s,ctx,target:str):
        i=uid(target)
        if i is None:return await ctx.send("invalid id")
        await s.bot.wl.mutate(i,False);s.bot.hook.put(f"whitelist: {i} removed by {ctx.author.id}")
        await ctx.send(f"{i} removed")
async def setup(bot):await bot.add_cog(perm(bot))