Converted information cog

cogs/music/interactions.py

@@ -0,0 +1,110 @@
+import typing
+from discord import Interaction, app_commands
+
+from bot import config
+from utils.embed import create_embed
+from cogs.music import helper
+
+if typing.TYPE_CHECKING:
+    from bot import TuneBot
+
+QUEUE_SIZE = config["queue_buffer_size"] + 1
+
+
+class MusicCommands(app_commands.Group):
+    def __init__(self, bot: "TuneBot"):
+        super().__init__(name="radio", description="Never-ending stream of good vibes")
+        self.bot: "TuneBot" = bot
+
+    @app_commands.command(name="connect", description="Start the radio")
+    async def connect(self, ctx: Interaction):
+        embed = create_embed(ctx.user)
+        player = helper.get_player(ctx.client, ctx.guild_id)
+
+        try:
+            await helper.ensure_voice(
+                permissions=ctx.app_permissions,
+                player=player,
+                author=ctx.user,
+                voice_client=ctx.guild.voice_client,
+                channel_id=ctx.channel.id,
+            )
+        except Exception as exc:
+            embed.title = str(exc)
+            await ctx.response.send_message(embed=embed)
+            return
+
+        if player.is_connected:
+            embed.title = "Already connected."
+            await ctx.response.send_message(embed=embed)
+            return
+
+        await helper.fill_player_queue(ctx.client, player, QUEUE_SIZE)
+
+        if not player.is_playing:
+            await player.play()
+
+        embed.title = "*⃣ | Connected."
+        await ctx.response.send_message(embed=embed)
+
+    @app_commands.command(
+        name="disconnect", description="I've had enough music for a while"
+    )
+    async def disconnect(self, ctx: Interaction):
+        embed = create_embed(ctx.user)
+        player = helper.get_player(ctx.client, ctx.guild_id)
+
+        if not ctx.user.voice or (
+            player.is_connected and ctx.user.voice.channel.id != int(player.channel_id)
+        ):
+            embed.title = "You're not in my voicechannel!"
+            await ctx.response.send_message(embed=embed)
+            return
+
+        # Clear the queue to ensure old tracks don't start playing
+        # when someone else queues something.
+        player.queue.clear()
+        # Stop the current track so Lavalink consumes less resources.
+        await player.stop()
+        # Disconnect from the voice channel.
+        await ctx.guild.voice_client.disconnect(force=True)
+        embed.title = "*⃣ | Disconnected."
+        await ctx.response.send_message(embed=embed)
+
+    @app_commands.command(
+        name="current", description="Gives more details about what's currently playing."
+    )
+    async def current(self, ctx: Interaction):
+        player = helper.get_player(ctx.client, ctx.guild_id)
+        if not player.current:
+            embed = create_embed(ctx.user)
+            embed.title = "You're not listening to anything right now."
+            await ctx.response.send_message(embed=embed)
+            return
+
+        embed = helper.create_track_embed(player.current)
+        await ctx.response.send_message(embed=embed)
+
+    @app_commands.command(name="queue", description="See what's ahead")
+    async def queue(self, ctx: Interaction):
+        player = helper.get_player(ctx.client, ctx.guild_id)
+        embed = create_embed(ctx.user)
+        embed.title = "Coming up..."
+        if len(player.queue) > 0:
+            embed.description = "\n".join(
+                f"{index}. [{track.title}]({track.uri})"
+                for index, track in enumerate(player.queue, 1)
+            )
+        else:
+            embed.description = "We are still determining a playlist"
+        await ctx.response.send_message(embed=embed)
+
+    @app_commands.command(
+        name="skip", description="You might like the next track better"
+    )
+    async def skip(self, ctx: Interaction):
+        player = helper.get_player(ctx.client, ctx.guild_id)
+        embed = create_embed(ctx.user)
+        embed.title = "Skipped current song"
+        await player.skip()
+        await ctx.response.send_message(embed=embed)