[adminutils] membersteal

adminutils/adminutils.py

@@ -1,7 +1,7 @@
 import re
 from asyncio import TimeoutError as AsyncTimeoutError
 from random import choice
-from typing import Optional
+from typing import Optional, Union
 
 import aiohttp
 import discord
@@ -171,7 +171,7 @@ async def emoji_add(self, ctx, name: str, url: str, *roles: discord.Role):
             await ctx.send(chat.error(_("Unable to get emoji from provided url: {}").format(e)))
             return
         try:
-            await ctx.guild.create_custom_emoji(
+            e = await ctx.guild.create_custom_emoji(
                 name=name,
                 image=data,
                 roles=roles,
@@ -189,32 +189,58 @@ async def emoji_add(self, ctx, name: str, url: str, *roles: discord.Role):
         except discord.HTTPException as e:
             await ctx.send(chat.error(_("An error occured on adding an emoji: {}").format(e)))
         else:
+            await ctx.send(_(f"{e} created with the name `{e.name}`."))
             await ctx.tick()
 
-    @emoji.command(name="message", aliases=["steal"])
-    async def emote_steal(self, ctx, name: str, message_id: discord.Message, *roles: discord.Role):
+    @emoji.command(name="import")
+    async def emoji_import(
+        self,
+        ctx,
+        name: str,
+        location: Union[discord.Message, discord.Member, discord.Emoji, discord.PartialEmoji],
+        *roles: discord.Role,
+    ):
         """
-        Add an emoji from a specified message
+        Add an emoji from either a member's status, a message, or an emoji from another server.
+
         Use double quotes if role name has spaces
 
         Examples:
-            `[p]emoji message Example 162379234070467641`
-            `[p]emoji message RoleBased 162379234070467641 EmojiRole`
+            `[p]emoji import 236598364265634245`
+            `[p]emoji import Username#0000 peepoDance`
+            `[p]emoji import :thonk: thonk emojiRole`
         """
-        # TrusyJaid NotSoBot converter
-        # https://github.com/TrustyJAID/Trusty-cogs/blob/a3e931bc6227645007b37c3f4f524c9fc9859686/notsobot/converter.py#L30-L36
-        emoji = EMOJI_RE.search(message_id.content)
-        if not emoji:
-            await ctx.send(chat.error(_("No emojis found specified message.")))
-            return
-        url = (
-            "https://cdn.discordapp.com/emojis/"
-            f"{emoji.group(3)}.{'gif' if emoji.group(2) else 'png'}?v=1"
-        )
-        async with self.session.get(url) as r:
-            data = await r.read()
+        if isinstance(location, (discord.Emoji, discord.PartialEmoji)):
+            async with self.session.get(str(location.url)) as r:
+                data = await r.read()
+        elif isinstance(location, discord.Message):
+            emoji = EMOJI_RE.search(location.content)
+            if not emoji:
+                await ctx.send(chat.error(_("No emojis found in the specified message.")))
+                return
+            url = (
+                "https://cdn.discordapp.com/emojis/"
+                f"{emoji.group(3)}.{'gif' if emoji.group(2) else 'png'}?v=1"
+            )
+            async with self.session.get(url) as r:
+                data = await r.read()
+        elif isinstance(location, discord.Member):
+            emoji = None
+            if (
+                location.activity
+                and location.activity.emoji
+                and location.activity.emoji.is_custom_emoji()
+            ):
+                emoji = location.activity.emoji
+            if not emoji:
+                await ctx.send(
+                    chat.error(_("This user does not have a custom emoji in their status."))
+                )
+                return
+            async with self.session.get(str(emoji.url)) as r:
+                data = await r.read()
         try:
-            await ctx.guild.create_custom_emoji(
+            em = await ctx.guild.create_custom_emoji(
                 name=name,
                 image=data,
                 roles=roles,
@@ -227,7 +253,6 @@ async def emote_steal(self, ctx, name: str, message_id: discord.Message, *roles:
                     else None,
                 ),
             )
-            await ctx.tick()
         except discord.InvalidArgument:
             await ctx.send(
                 _(
@@ -237,6 +262,9 @@ async def emote_steal(self, ctx, name: str, message_id: discord.Message, *roles:
             return
         except discord.HTTPException as e:
             await ctx.send(chat.error(_("An error occurred on adding an emoji: {}").format(e)))
+        else:
+            await ctx.send(_(f"{em} created with the name `{em.name}`."))
+            await ctx.tick()
 
     @emoji.command(name="rename")
     async def emoji_rename(self, ctx, emoji: discord.Emoji, name: str, *roles: discord.Role):
@@ -253,7 +281,7 @@ async def emoji_rename(self, ctx, emoji: discord.Emoji, name: str, *roles: disco
             await ctx.send_help()
             return
         try:
-            await emoji.edit(
+            em = await emoji.edit(
                 name=name,
                 roles=roles,
                 reason=get_audit_reason(
@@ -267,6 +295,8 @@ async def emoji_rename(self, ctx, emoji: discord.Emoji, name: str, *roles: disco
             )
         except discord.Forbidden:
             await ctx.send(chat.error(_("I can't edit this emoji")))
+        else:
+            await ctx.send(_(f"{em} edited to `{em.name}`."))
         await ctx.tick()
 
     @emoji.command(name="remove")