trimmed down music.py

music/music_wrapper.py

@@ -0,0 +1,103 @@
+import wavelink
+import discord
+import asyncio
+import json
+import logging
+from pathlib import Path
+from discord.ext import commands
+
+from .shared.constants import C_MAIN
+from .shared.state import GuildState
+from .player.commands import PlayerCommands
+from .player.queue import QueueCommands
+from .player.events import MusicEvents
+from .radio.commands import RadioCommands
+from .radio.events import RadioEvents
+
+log = logging.getLogger(__name__)
+
+
+class Music(PlayerCommands, QueueCommands, MusicEvents, RadioCommands, RadioEvents, commands.Cog):
+    """
+    Main Music cog. Inherits all command/event mixins:
+      - PlayerCommands  (play, pause, resume, skip, stop, seek, loop, autoplay, nowplaying)
+      - QueueCommands   (queue, remove, move, shuffle, clear, history)
+      - MusicEvents     (on_ready, on_voice_state_update, on_wavelink_track_end/start)
+      - RadioCommands   (radio, _play_radio_stream, _notify_radio_drop)
+      - RadioEvents     (on_wavelink_track_exception, on_wavelink_track_stuck)
+    """
+
+    def __init__(self, bot: commands.Bot):
+        self.bot = bot
+
+        # Per-guild state and infrastructure
+        self._states:         dict[int, GuildState]    = {}
+        self._voice_channels: dict[int, int]           = {}  # guild_id → voice channel id (for status cleanup)
+        self._radio_channels: dict[int, int]           = {}  # guild_id → text channel id (for drop notifications)
+        self._locks:          dict[int, asyncio.Lock]  = {}
+
+        # Load radio stations
+        stations_path = Path(__file__).parent / "radio_stations.json"
+        with open(stations_path, "r", encoding="utf-8") as f:
+            self._radio_stations: dict[str, list[dict]] = json.load(f)
+        log.info(
+            f"Loaded radio stations: "
+            f"{sum(len(v) for v in self._radio_stations.values())} stations "
+            f"in {len(self._radio_stations)} categories"
+        )
+
+    # ── shared helpers (used by all mixins via self) ──────────────────────────
+    def state(self, guild_id: int) -> GuildState:
+        if guild_id not in self._states:
+            self._states[guild_id] = GuildState()
+        return self._states[guild_id]
+
+    def _get_player(self, ctx: commands.Context) -> wavelink.Player | None:
+        """Get the wavelink Player for a context, or None."""
+        vc = ctx.voice_client
+        if isinstance(vc, wavelink.Player):
+            return vc
+        return None
+
+    def _lock(self, guild_id: int) -> asyncio.Lock:
+        """Get or create a per-guild asyncio.Lock."""
+        if guild_id not in self._locks:
+            self._locks[guild_id] = asyncio.Lock()
+        return self._locks[guild_id]
+
+    def _is_current_track(self, st: GuildState, event_track: wavelink.Playable | None) -> bool:
+        """True if event_track is the same track currently stored in guild state."""
+        if st.current is None or event_track is None:
+            return False
+        return event_track.identifier == st.current.identifier
+
+    # ── !music ────────────────────────────────────────────────────────────────
+    @commands.command()
+    async def music(self, ctx: commands.Context):
+        """List all music commands."""
+        embed = discord.Embed(title="🎵  Music Commands", color=C_MAIN)
+        embed.add_field(name="Playback", value=(
+            "`!play <query/url>` · `!p`\n"
+            "`!pause` · `!resume` · `!skip` · `!stop`\n"
+            "`!seek <1:30>` · `!looptrack` · `!loopqueue`\n"
+            "`!autoplay`"
+        ), inline=False)
+        embed.add_field(name="Radio", value=(
+            "`!radio` — Browse & play radio stations\n"
+            "Use `!stop` to stop either music or radio"
+        ), inline=False)
+        embed.add_field(name="Queue", value=(
+            "`!queue` · `!q`\n"
+            "`!remove <pos>` · `!move <from> <to>`\n"
+            "`!shuffle` · `!clear`"
+        ), inline=False)
+        embed.add_field(name="Info", value=(
+            "`!nowplaying` · `!np`\n"
+            "`!history` · `!h`"
+        ), inline=False)
+        embed.set_footer(text="🎧 DJ role required for control commands")
+        await ctx.reply(embed=embed)
+
+
+async def setup(bot: commands.Bot):
+    await bot.add_cog(Music(bot))