Add music

lynxie/commands/__init__.py

@@ -0,0 +1,7 @@
+from .hello import Hello
+from .music import Music
+
+__all__ = [
+    "Hello",
+    "Music",
+]

lynxie/commands/hello.py

@@ -1,4 +1,3 @@
-import discord
 from discord.ext import commands
 
 
@@ -7,6 +6,5 @@ class Hello(commands.Cog):
         self.bot = bot
 
     @commands.command()
-    async def hello(self, interaction: discord.Interaction):
-        """Says hello!"""
-        await interaction.response.send_message(f"Balls, {interaction.user.mention}")
+    async def hello(self, ctx):
+        await ctx.send(f"Hello {ctx.author.mention}!")

lynxie/commands/music.py

@@ -0,0 +1,91 @@
+import asyncio
+import youtube_dl
+import discord
+from discord.ext import commands
+
+
+youtube_dl.utils.bug_reports_message = lambda: ""
+ytdl = youtube_dl.YoutubeDL({
+    "format": "bestaudio/best",
+    "outtmpl": "%(extractor)s-%(id)s-%(title)s.%(ext)s",
+    "restrictfilenames": True,
+    "noplaylist": True,
+    "nocheckcertificate": True,
+    "ignoreerrors": False,
+    "logtostderr": False,
+    "quiet": True,
+    "no_warnings": True,
+    "default_search": "auto",
+    "source_address": "0.0.0.0",
+})
+
+
+class YTDLSource(discord.PCMVolumeTransformer):
+    def __init__(self, source, *, data, volume=0.5):
+        super().__init__(source, volume)
+
+        self.data = data
+
+        self.title = data.get("title")
+        self.url = data.get("url")
+
+    @classmethod
+    async def from_url(cls, url, *, loop=None, stream=False):
+        loop = loop or asyncio.get_event_loop()
+        data = await loop.run_in_executor(
+            None, lambda: ytdl.extract_info(url, download=not stream)
+        )
+
+        if "entries" in data:
+            # take first item from a playlist
+            data = data["entries"][0]
+
+        filename = data["url"] if stream else ytdl.prepare_filename(data)
+        return cls(discord.FFmpegPCMAudio(filename, options='-vn'), data=data)
+
+
+class Music(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.command()
+    async def move(self, ctx, *, channel: discord.VoiceChannel):
+        """Joins a voice channel"""
+
+        if ctx.voice_client is not None:
+            return await ctx.voice_client.move_to(channel)
+
+        await channel.connect()
+
+    @commands.command()
+    async def play(self, ctx, *, url):
+        """Streams from a url (same as yt, but doesn't predownload)"""
+
+        async with ctx.typing():
+            try:
+                player = await YTDLSource.from_url(url, loop=self.bot.loop, stream=True)
+                ctx.voice_client.play(
+                    player, after=lambda e: print(f"Player error: {e}") if e else None
+                )
+            except Exception:
+                await ctx.send("Sorry, I can't play that video.")
+                return
+
+        await ctx.send(f"Now playing: {player.title}")
+
+    @commands.command()
+    async def stop(self, ctx):
+        """Stops and disconnects the bot from voice"""
+
+        await ctx.voice_client.disconnect()
+
+    @play.before_invoke
+    async def ensure_voice(self, ctx):
+        if ctx.voice_client is None:
+            if ctx.author.voice:
+                await ctx.author.voice.channel.connect()
+            else:
+                await ctx.send("You are not connected to a voice channel.")
+                raise commands.CommandError("Author not connected to a voice channel.")
+        elif ctx.voice_client.is_playing():
+            ctx.voice_client.stop()