Music Bot

.gitignore

@@ -4,3 +4,5 @@
 
 *.pyc
 /logs/
+
+jdk-13.0.2/

application.yml

@@ -0,0 +1,49 @@
+server: # REST and WS server
+  port: 2333
+  address: localhost
+lavalink:
+  server:
+    password: "youshallnotpass"
+    sources:
+      youtube: true
+      bandcamp: true
+      soundcloud: true
+      twitch: true
+      vimeo: true
+      http: true
+      local: false
+    bufferDurationMs: 400 # The duration of the NAS buffer. Higher values fare better against longer GC pauses
+    frameBufferDurationMs: 5000 # How many milliseconds of audio to keep buffered
+    youtubePlaylistLoadLimit: 6 # Number of pages at 100 each
+    playerUpdateInterval: 5 # How frequently to send player updates to clients, in seconds
+    youtubeSearchEnabled: true
+    soundcloudSearchEnabled: true
+    gc-warnings: true
+      #ratelimit:
+      #ipBlocks: ["1.0.0.0/8", "..."] # list of ip blocks
+      #excludedIps: ["...", "..."] # ips which should be explicit excluded from usage by lavalink
+      #strategy: "RotateOnBan" # RotateOnBan | LoadBalance | NanoSwitch | RotatingNanoSwitch
+      #searchTriggersFail: true # Whether a search 429 should trigger marking the ip as failing
+    #retryLimit: -1 # -1 = use default lavaplayer value | 0 = infinity | >0 = retry will happen this numbers times
+
+metrics:
+  prometheus:
+    enabled: false
+    endpoint: /metrics
+
+sentry:
+  dsn: ""
+  environment: ""
+#  tags:
+#    some_key: some_value
+#    another_key: another_value
+
+logging:
+  file:
+    max-history: 30
+    max-size: 1GB
+  path: ./logs/
+
+  level:
+    root: INFO
+    lavalink: INFO

cogs/musicbot.py

@@ -0,0 +1,129 @@
+# Inspired by https://github.com/afazio1/robotic-nation-proj/blob/main/projects/discord-bot/new-music-bot/music-yt.py
+
+import os
+import subprocess
+import time
+
+import discord
+import requests
+import wavelink
+from discord.ext import commands
+from dotenv import load_dotenv
+
+
+class MusicBot(commands.Cog):
+    voice_clients = {}
+
+    def __init__(self, client):
+        self.client = client
+
+    @commands.command()
+    async def join(self, ctx: commands.Context):
+        if not ctx.message.author.voice:
+            await ctx.send('You are not in a voice channel')
+            return
+        # Leave current channel
+        if ctx.voice_client is not None:
+            await ctx.voice_client.disconnect()
+        channel = ctx.message.author.voice.channel
+        self.voice_clients[ctx.guild.id] = await channel.connect(cls=wavelink.Player())
+        await ctx.send(f'Joined {channel}')
+
+    @commands.command()
+    async def leave(self, ctx: commands.Context):
+        if ctx.guild.id in self.voice_clients:
+            await self.voice_clients[ctx.guild.id].stop()
+            del self.voice_clients[ctx.guild.id]
+            await ctx.send('Left')
+        if ctx.voice_client is not None:
+            await ctx.voice_client.disconnect()
+
+    @commands.command()
+    async def play(self, ctx: commands.Context, *, search: wavelink.YouTubeTrack):
+        if ctx.guild.id not in self.voice_clients:
+            # Join the user's voice channel
+            await self.join(ctx)
+        if ctx.guild.id not in self.voice_clients:
+            return
+        voice = self.voice_clients[ctx.guild.id]
+        await voice.play(search)
+        embed = discord.Embed(
+            title=voice.source.title,
+            url=voice.source.uri,
+            description=f"Playing {voice.source.title} in {voice.channel}"
+        )
+        embed.set_image(url=voice.source.thumbnail)
+        if hasattr(ctx.author, 'avatar'):
+            embed.set_author(name=ctx.author.name, icon_url=ctx.author.avatar.url)
+        await ctx.send(embed=embed)
+
+    @commands.command()
+    async def stop(self, ctx: commands.Context):
+        if ctx.guild.id in self.voice_clients:
+            await self.voice_clients[ctx.guild.id].stop()
+            await ctx.send('Stopped')
+        else:
+            await ctx.send('Not in a voice channel')
+
+    @commands.command()
+    async def pause(self, ctx: commands.Context):
+        if ctx.guild.id in self.voice_clients:
+            await self.voice_clients[ctx.guild.id].pause()
+            await ctx.send('Paused')
+        else:
+            await ctx.send('Not in a voice channel')
+
+    @commands.command()
+    async def resume(self, ctx: commands.Context):
+        if ctx.guild.id in self.voice_clients:
+            await self.voice_clients[ctx.guild.id].resume()
+            await ctx.send('Resumed')
+        else:
+            await ctx.send('Not in a voice channel')
+
+    # Start the bot
+    @commands.Cog.listener()
+    async def on_ready(self):
+        print('Logged in as')
+        print(self.client.user.name)
+        print(self.client.user.id)
+        print('------')
+        # Try start Lavalink server
+        subprocess.Popen(["./jdk-13.0.2/bin/java", "-jar", "Lavalink.jar"])
+        # wait for port to open
+        while True:
+            try:
+                r = requests.get('http://localhost:2333')
+                break
+            except requests.exceptions.ConnectionError:
+                print("Waiting for lavalink to go live...")
+                time.sleep(1)
+                continue
+
+        async def connect_wavelink():
+            await self.client.wait_until_ready()
+            await wavelink.NodePool.create_node(
+                bot=self.client,
+                host='localhost',
+                port=2333,
+                password='youshallnotpass'
+            )
+
+        self.client.loop.create_task(connect_wavelink())
+
+    @commands.Cog.listener()
+    async def on_wavelink_node_ready(self, node: wavelink.Node):
+        print(f'Connected to wavefront! ID: {node.identifier}')
+
+
+async def setup(bot):
+    await bot.add_cog(MusicBot(bot))
+
+
+if __name__ == '__main__':
+    load_dotenv()
+    TOKEN = os.getenv('TOKEN')
+
+    client = commands.Bot(command_prefix='!')
+    client.add_cog(MusicBot(client))
+    client.run(TOKEN)