initial commit

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>

cmds/perm.py

@@ -0,0 +1,41 @@
+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))