Merge branch 'main' of github.com:CommanderBot-Dev/commanderbot-ext into main

Author: Arcensoth

CHANGELOG.md

@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [Unreleased]
+
+### Added
+
+- Added `quote` and `quotem` commands
+
 ## [0.5.0] - 2021-01-06
 
 ### Changed

commanderbot_ext/quote/__init__.py

@@ -0,0 +1,6 @@
+from discord.ext import commands
+from commanderbot_ext.quote.quote_cog import QuoteCog
+
+
+def setup(bot: commands.Bot):
+    bot.add_cog(QuoteCog(bot))

commanderbot_ext/quote/quote_cog.py

@@ -0,0 +1,33 @@
+import discord
+from commanderbot_lib.logging import Logger, get_clogger
+from discord.ext.commands import Bot, Cog, command
+
+
+class QuoteCog(Cog):
+    def __init__(self, bot: Bot):
+        self.bot: Bot = bot
+        self._log: Logger = get_clogger(self)
+
+    async def construct_embed(self, msg_link: str, quotem: bool = False):
+        message_link = msg_link.split("/")
+        message_channel = self.bot.get_channel(int(message_link[-2]))
+        message = await message_channel.fetch_message(int(message_link[-1]))
+        
+        # message.author is name + discrim, icon_url is authors avatar
+        quote_embed = discord.Embed(description=message.clean_content, timestamp=message.created_at)
+        quote_embed.set_author(name=str(message.author), icon_url=str(message.author.avatar_url))
+        quote_embed.set_footer(text=f"#{message_channel.name}")
+
+        if quotem:
+            return (quote_embed, message.author)
+        return quote_embed
+
+    @command(name="quote")
+    async def cmd_quote(self, ctx: discord.Message, msg_link: str):
+        quote_embed = await self.construct_embed(msg_link)
+        await ctx.send(f"{ctx.author.mention}\n{msg_link}", embed=quote_embed)
+
+    @command(name="quotem")
+    async def cmd_quotem(self, ctx: discord.Message, msg_link: str):
+        quote_embed = await self.construct_embed(msg_link, quotem=True)
+        await ctx.send(f"{ctx.author.mention} {quote_embed[1].mention}\n{msg_link}", embed=quote_embed[0])