Create `stop` command

1 files changed, 40 insertions, 0 deletions

diff --git a/code/cogs/stop.py b/code/cogs/stop.py
new file mode 100644
index 0000000..eb3341b
--- /dev/null
+++ b/code/cogs/stop.py
@@ -0,0 +1,40 @@
+import discord
+import datetime
+from discord import app_commands
+from discord.ext import commands
+from cogs.music import Music
+
+from global_variables import BOT_COLOR
+
+
+class Stop(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+
+    @app_commands.command()
+    @app_commands.check(Music.create_player)
+    async def stop(self, interaction: discord.Interaction):
+        "Disconnects the bot from the voice channel and clears the queue"
+        player = self.bot.lavalink.player_manager.get(interaction.guild.id)
+
+        player.queue.clear()
+        await player.stop()
+        await interaction.guild.voice_client.disconnect(force=True)
+
+        embed = discord.Embed(
+            title="Queue Cleared and Music Stopped",
+            description=f"Thank you for using Guava :wave:\n\nIssued by: {interaction.user.mention}",
+            color=BOT_COLOR,
+        )
+        embed.set_footer(
+            text=datetime.datetime.now(datetime.timezone.utc).strftime(
+                "%Y-%m-%d %H:%M:%S"
+            )
+            + " UTC"
+        )
+        await interaction.response.send_message(embed=embed)
+
+
+async def setup(bot):
+    await bot.add_cog(Stop(bot))