Create the new DMRelay cog.

Includes the `!dmrelay` command, allowing moderators to relay
direct messages between the bot and other users.

1 files changed, 59 insertions, 0 deletions

diff --git a/bot/exts/moderation/dm_relay.py b/bot/exts/moderation/dm_relay.py
new file mode 100644
index 000000000..2bf2391a4
--- /dev/null
+++ b/bot/exts/moderation/dm_relay.py
@@ -0,0 +1,59 @@
+import logging
+import textwrap
+
+import discord
+from discord.ext.commands import Cog, Context, command
+
+from bot.bot import Bot
+from bot.constants import Emojis
+from bot.utils.services import send_to_paste_service
+
+log = logging.getLogger(__name__)
+
+
+class DMRelay(Cog):
+    """Relay direct messages from the bot."""
+
+    def __init__(self, bot: Bot):
+        self.bot = bot
+
+    @command(aliases=("relay", "dr"))
+    async def dmrelay(self, ctx: Context, user: discord.User, limit: int = 100) -> None:
+        """Relays the direct message history between the bot and given user."""
+        log.trace(f"Relaying DMs with {user.name} ({user.id})")
+
+        if not user.dm_channel:
+            await ctx.send(f"{Emojis.cross_mark} No direct message history with {user.mention}.")
+            return
+
+        output = textwrap.dedent(f"""\
+            User: {user} ({user.id})
+            Channel ID: {user.dm_channel.id}\n
+        """)
+
+        async for msg in user.history(limit=limit, oldest_first=True):
+            created_at = msg.created_at.strftime(r"%Y-%m-%d %H:%M")
+
+            # Metadata (author, created_at, id)
+            output += f"{msg.author} [{created_at}] ({msg.id}): "
+
+            # Content
+            if msg.content:
+                output += msg.content + "\n"
+
+            # Embeds
+            if (embeds := len(msg.embeds)) > 0:
+                output += f"<{embeds} embed{'s' if embeds > 1 else ''}>\n"
+
+            # Attachments
+            attachments = "\n".join(a.url for a in msg.attachments)
+            if attachments:
+                output += attachments + "\n"
+
+        paste_link = await send_to_paste_service(output, extension="txt")
+        await ctx.send(paste_link)
+
+
+def setup(bot: Bot) -> None:
+    """Load the DMRelay cog."""
+    bot.add_cog(DMRelay(bot))