From a4aec6bf60f4ef7a96281a67a018b76a2c64df46 Mon Sep 17 00:00:00 2001
From: Zevaryx <zevaryx@gmail.com>
Date: Fri, 9 Sep 2022 22:32:54 -0600
Subject: [PATCH] Add Avatar context menu

---
 jarvis/cogs/util.py | 23 +++++++++++++++--------
 1 file changed, 15 insertions(+), 8 deletions(-)

diff --git a/jarvis/cogs/util.py b/jarvis/cogs/util.py
index f9ac647..43b547a 100644
--- a/jarvis/cogs/util.py
+++ b/jarvis/cogs/util.py
@@ -126,14 +126,6 @@ class UtilCog(Extension):
         else:
             await ctx.send(to_send, ephemeral=True)
 
-    @slash_command(name="avatar", description="Get a user avatar")
-    @slash_option(
-        name="user",
-        description="User to view avatar of",
-        opt_type=OptionTypes.USER,
-        required=False,
-    )
-    @cooldown(bucket=Buckets.USER, rate=1, interval=5)
     async def _avatar(self, ctx: InteractionContext, user: User = None) -> None:
         if not user:
             user = ctx.author
@@ -198,6 +190,21 @@ class UtilCog(Extension):
             )
             await ctx.send(embeds=embed, file=color_show, components=components)
 
+    @slash_command(name="avatar", description="Get a user avatar")
+    @slash_option(
+        name="user",
+        description="User to view avatar of",
+        opt_type=OptionTypes.USER,
+        required=False,
+    )
+    @cooldown(bucket=Buckets.USER, rate=1, interval=5)
+    async def _avatar_slash(self, ctx: InteractionContext, user: User = None) -> None:
+        await self._userinfo(ctx, user)
+
+    @context_menu(name="Avatar", context_type=CommandTypes.USER)
+    async def _avatar_menu(self, ctx: InteractionContext) -> None:
+        await self._avatar(ctx, ctx.target)
+
     async def _userinfo(self, ctx: InteractionContext, user: User = None) -> None:
         await ctx.defer()
         if not user: