merge pull request #42 from matthewflegg/feature/servericon-command

added ~servericon command
v0idzdev · Apr 10, 2022 · 95a8d85 · 95a8d85
2 parents 24c9058 + 19e0fae
commit 95a8d85
Show file tree

Hide file tree

Showing 3 changed files with 49 additions and 2 deletions.
diff --git a/cogs/info/info_commands.py b/cogs/info/info_commands.py
@@ -151,3 +151,14 @@ async def avatar_callback(
     embed = discord.Embed(title=f"💡 {member.name}'s Avatar")
     embed.set_image(url=member.avatar.url)
     return await send_embed(is_interaction, embed, ctx)
+
+
+async def servericon_callback(ctx: discord.Interaction | commands.Context):
+    is_interaction = isinstance(ctx, discord.Interaction)
+
+    if is_interaction:
+        await ctx.response.defer()
+
+    embed = discord.Embed(title=f"💡 {ctx.guild.name}'s Icon")
+    embed.set_image(url=ctx.guild.icon.url)
+    return await send_embed(is_interaction, embed, ctx)
diff --git a/cogs/info/prefix_cog/info_cog.py b/cogs/info/prefix_cog/info_cog.py
@@ -95,9 +95,9 @@ async def botinfo(self, ctx: commands.Context):
     @commands.command()
     async def avatar(self, ctx: commands.Context, *, member: discord.Member = None):
         """
-        💡 Shows a member's avatar. If no member is specified, it shows yours.
+        💡 Shows a member's avatar.
 
-        ❓ This will change depending on whether the bot is self-hosted.
+        ❓ If no member is specified, it shows yours.
 
         Usage:
         ```
@@ -110,6 +110,24 @@ async def avatar(self, ctx: commands.Context, *, member: discord.Member = None):
         """
         await avatar_callback(ctx, member)
 
+    @commands.command()
+    async def servericon(self, ctx: commands.Context):
+        """
+        💡 Shows the server's icon.
+
+        ❓ This command is also available as a slash command.
+
+        Usage:
+        ```
+        ~avatar [@member]
+        ```
+        Or:
+        ```
+        /avatar [@member]
+        ```
+        """
+        await servericon_callback(ctx)
+
 
 async def setup(client: commands.Bot):
     """

diff --git a/cogs/info/slash_cog/info_slash_cog.py b/cogs/info/slash_cog/info_slash_cog.py
@@ -116,6 +116,24 @@ async def avatar(
         """
         await avatar_callback(interaction, member)
 
+    @app_commands.command()
+    async def servericon(self, interaction: discord.Interaction):
+        """
+        💡 Shows the server's icon.
+
+        ❓ This command is also available as a prefix command.
+
+        Usage:
+        ```
+        ~avatar [@member]
+        ```
+        Or:
+        ```
+        /avatar [@member]
+        ```
+        """
+        await servericon_callback(interaction)
+
 
 async def setup(client: commands.Bot):
     """