add music client

Author: muzaffarmhd

pyproject.toml

@@ -25,7 +25,7 @@ dependencies = [
   "pytest",
   "Pillow>=9.0.0",  
   "requests>=2.20.0", 
-  "aiohttp>=3.8.0"    
+  "aiohttp>=3.8.0",    
 ]
 
 [project.scripts]

src/zuffer/cli.py

@@ -1,8 +1,8 @@
 import click
-from .commands import login #, login, config, create_rooms #, token
+from .commands import login 
 from .commands import create_channels
 from .commands import embed
-from .commands import welcome, list
+from .commands import welcome, list, play_music
 
 @click.group(invoke_without_command=True)
 @click.version_option(package_name="zuffer")
@@ -22,3 +22,4 @@ def main(ctx: click.Context):
 main.add_command(create_channels.create_private)
 main.add_command(welcome.welcome_group)
 main.add_command(list.list_command)
+main.add_command(play_music.play_music)

src/zuffer/commands/play_music.py

@@ -0,0 +1,15 @@
+import click
+import questionary
+import discord
+from ..core import clients
+from ..core import auth
+
+@click.command(name="play-music")
+@click.argument("playlist", required=True, type=click.Path(exists=True))
+def play_music(playlist):
+    token = auth.get_token()
+    if (token is not None):
+        intents = discord.Intents.all()
+        client = clients.MusicClient(playlist, intents=intents)
+        client.run(token)
+

src/zuffer/core/clients.py

@@ -2,6 +2,7 @@
 import json
 import os
 import requests
+import random
 from io import BytesIO
 from discord import Role, Member, Object
 from typing import Mapping
@@ -243,3 +244,41 @@ async def on_member_join(self, member: discord.Member):
         else:
             print(f"Welcome channel with ID {welcome_channel_id} not found in guild {member.guild.name} or globally for bot.")
 
+class MusicClient(discord.Client):
+    def __init__(self, path_to_playlist, intents: discord.Intents):
+        super().__init__(intents=intents)
+        self.voice_recorders = {}
+        self.playlist = path_to_playlist
+
+    async def on_ready(self):
+        if (self.user is not None):
+            print(f"Logged in as {self.user} (ID: {self.user.id})")
+            print("------")
+            await self.change_presence(activity=discord.Activity(type=discord.ActivityType.listening, name="Music"))
+    async def on_message(self, message: discord.Message):
+        if message.author == self.user:
+            return
+        if message.content.startswith("!join"):
+            if isinstance(message.author, discord.Member) and message.author.voice and message.author.voice.channel:
+                voice_channel = message.author.voice.channel
+                try:
+                    voice_client = await voice_channel.connect()
+                    await message.channel.send(f"Joined {voice_channel.name}!")
+                    
+                    if (os.path.isdir(self.playlist)):
+                        files = [os.path.join(self.playlist, f) for f in os.listdir(self.playlist) if f.endswith(".mp3")]
+                        def after_callback(error):
+                            if error:
+                                print(f'Player error: {error}')
+                            else:
+                                audio = discord.FFmpegPCMAudio(random.choice(files))
+                                voice_client.play(audio, after=after_callback)
+                        
+                        audio_source = discord.FFmpegPCMAudio(random.choice(files))  
+                        voice_client.play(audio_source, after=after_callback)
+                    
+                    await message.channel.send("Now playing music on loop!")
+                except Exception as e:
+                    await message.channel.send(f"Error joining voice channel: {str(e)}")
+            else:
+                await message.channel.send("You need to be in a voice channel first!")