Adding Player module

vulkanbot/music/Player.py

@@ -0,0 +1,288 @@
+import discord
+from discord.ext import commands
+from config import config
+import datetime
+
+from vulkanbot.music.Downloader import Downloader
+from vulkanbot.music.Playlist import Playlist
+from vulkanbot.music.Searcher import Searcher
+from vulkanbot.music.Types import Provider
+from vulkanbot.music.utils import *
+
+
+class Player(commands.Cog):
+    def __init__(self, bot, guild):
+        self.__searcher: Searcher = Searcher()
+        self.__downloader: Downloader = Downloader()
+        self.__playlist: Playlist = Playlist()
+        self.__bot: discord.Client = bot
+        self.__guild: discord.Guild = guild
+
+        self.__playing = False
+
+        self.YDL_OPTIONS = {'format': 'bestaudio', 'noplaylist': 'True'}
+        self.FFMPEG_OPTIONS = {'before_options': '-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5',
+                               'options': '-vn'}
+
+    async def connect(self, ctx):
+        if not ctx.author.voice:
+            return {'success': False, 'reason': config.NO_CHANNEL}
+
+        if self.__guild.voice_client == None:
+            await ctx.author.voice.channel.connect(reconnect=True, timeout=None)
+            return {'success': True, 'reason': ''}
+
+    def __play_next(self, error, ctx):
+        song = self.__playlist.next_song()
+        if song != None:  # If there is not a song for the song
+            coro = self.__play_music(ctx, song)
+            self.__bot.loop.create_task(coro)
+        else:
+            self.__playing = False
+
+    async def __play_music(self, ctx, song):
+        self.__playing = True
+
+        player = discord.FFmpegPCMAudio(song.source, **self.FFMPEG_OPTIONS)
+        self.__guild.voice_client.play(
+            player, after=lambda e: self.__play_next(e, ctx))
+
+        await ctx.invoke(self.__bot.get_command('np'))
+
+        songs = self.__playlist.songs_to_preload
+        await self.__downloader.preload(songs)
+
+    async def play(self, ctx, *args) -> str:
+        user_input = " ".join(args)
+
+        try:
+            if self.__guild.voice_client == None:
+                voice_channel = ctx.author.voice.channel
+                await voice_channel.connect()
+        except Exception as e:
+            print(e)
+            return config.NO_CHANNEL
+        else:
+            songs_quant = 0
+            musics_identifiers, provider = self.__searcher.search(user_input)
+
+            if provider == Provider.Unknown:
+                return config.INVALID_INPUT
+
+            if provider == Provider.YouTube:
+                musics_identifiers = self.__downloader.extract_youtube_link(
+                    musics_identifiers[0])
+
+            for identifier in musics_identifiers:  # Creating songs
+                print('Creating Song')
+                last_song = self.__playlist.add_song(identifier)
+                songs_quant += 1
+
+            songs_preload = self.__playlist.songs_to_preload
+            await self.__downloader.preload(songs_preload)
+
+            if songs_quant == 1:  # If only one music downloaded
+                song = self.__downloader.download_one(
+                    last_song)  # Download the new music
+
+                if song == None:  # If song not downloaded
+                    embed = discord.Embed(
+                        description=config.DOWNLOADING_ERROR, colour=config.COLOURS['blue'])
+                    await ctx.send(embed=embed)
+
+                elif not self.__playing:  # If not playing
+                    text = f'You added the song **{song.title}** to the queue'
+                    embed = discord.Embed(
+                        description=text, colour=config.COLOURS['blue'])
+                    await ctx.send(embed=embed)
+
+                else:  # If playing
+                    title = config.SONG_ADDED
+                    embed = self.__format_embed(song.info, title=title)
+                    await ctx.send(embed=embed)
+            else:
+                text = f'You added {songs_quant} songs to the queue'
+                embed = discord.Embed(
+                    description=text, colour=config.COLOURS['blue'])
+                await ctx.send(embed=embed)
+
+            if not self.__playing:
+                try_another = True
+
+                while try_another:
+                    first = self.__playlist.next_song()
+                    if first == None:
+                        embed = discord.Embed(
+                            description=config.DOWNLOADING_ERROR, colour=config.COLOURS['blue'])
+                        await ctx.send(embed=embed)
+                        break
+
+                    while True:
+                        if first.source != None:  # If song got downloaded
+                            try_another = False
+                            break
+
+                        if first.problematic:  # If song got any error, try another one
+                            break
+
+                        else:  # The song is downloading, check again
+                            continue
+
+                if first != None:
+                    await self.__play_music(ctx, first)
+
+    async def queue(self) -> discord.Embed:
+        if self.__playlist.looping_one:
+            info = self.__playlist.current.info
+            title = 'Song Looping Now'
+            return self.__format_embed(info, title)
+
+        songs_preload = self.__playlist.songs_to_preload
+        await self.__downloader.preload(songs_preload)
+        total_time = format_time(sum([int(song.duration if song.duration else 0)
+                                 for song in songs_preload]))  # Sum the duration
+        total_songs = len(self.__playlist)
+        text = f'Total musics: {total_songs} | Duration: `{total_time}` downloaded  \n\n'
+
+        for pos, song in enumerate(songs_preload, start=1):
+            title = song.title if song.title else 'Downloading...'
+            text += f"**`{pos}` - ** {title} - `{format_time(song.duration)}`\n"
+
+        title = 'Songs in Queue'
+        if len(songs_preload) > 0:
+            if self.__playlist.looping_all:
+                title = 'Repeating All'
+        else:
+            text = 'There is no musics in queue'
+
+        embed = discord.Embed(
+            title=title,
+            description=text,
+            colour=config.COLOURS['blue']
+        )
+
+        return embed
+
+    async def skip(self) -> bool:
+        if self.__guild.voice_client != None:
+            self.__guild.voice_client.stop()
+            return True
+        else:
+            return False
+
+    async def stop(self) -> bool:
+        if self.__guild.voice_client == None:
+            return False
+
+        if self.__guild.voice_client.is_connected():
+            self.__playlist.clear()
+            self.__playlist.loop_off()
+            self.__guild.voice_client.stop()
+            await self.__guild.voice_client.disconnect()
+            return True
+
+    async def pause(self) -> bool:
+        if self.__guild.voice_client == None:
+            return False
+
+        if self.__guild.voice_client.is_playing():
+            self.__guild.voice_client.pause()
+            return True
+
+    async def resume(self) -> bool:
+        if self.__guild.voice_client == None:
+            return False
+
+        if self.__guild.voice_client.is_paused():
+            self.__guild.voice_client.resume()
+            return True
+
+    async def loop(self, args: str):
+        args = args.lower()
+        if args == 'one':
+            description = self.__playlist.loop_one()
+        elif args == 'all':
+            description = self.__playlist.loop_all()
+        elif args == 'off':
+            description = self.__playlist.loop_off()
+        else:
+            description = 'Comando Loop\nOne - Repete a música atual\nAll - Repete as músicas atuais\nOff - Desativa o loop'
+
+        return description
+
+    async def clear(self) -> None:
+        self.__playlist.clear()
+
+    async def now_playing(self) -> discord.Embed:
+        if self.__playlist.looping_one:
+            title = 'Song Looping Now'
+        else:
+            title = 'Song Playing Now'
+
+        current_song = self.__playlist.current
+        embed = self.__format_embed(current_song.info, title)
+
+        return embed
+
+    async def shuffle(self) -> str:
+        try:
+            self.__playlist.shuffle()
+            songs = self.__playlist.songs_to_preload
+
+            await self.__downloader.preload(songs)
+            return 'Musics shuffled successfully'
+        except:
+            return 'An error ocurred :/'
+
+    async def move(self, pos1, pos2='1') -> str:
+        try:
+            pos1 = int(pos1)
+            pos2 = int(pos2)
+
+        except:
+            return 'This command require a number'
+
+        result = self.__playlist.move_songs(pos1, pos2)
+
+        songs = self.__playlist.songs_to_preload
+        await self.__downloader.preload(songs)
+        return result
+
+    async def remove(self, position) -> str:
+        """Remove a song from the queue in the position"""
+        try:
+            position = int(position)
+
+        except Exception as e:
+            print(e)
+            return 'This command require a number'
+
+        result = self.__playlist.remove_song(position)
+        return result
+
+    def __format_embed(self, info, title='') -> discord.Embed:
+        """Configure the embed to show the song information"""
+        embedvc = discord.Embed(
+            title=title,
+            description=f"[{info['title']}]({info['original_url']})",
+            color=config.COLOURS['blue']
+        )
+
+        embedvc.add_field(name=config.SONGINFO_UPLOADER,
+                          value=info['uploader'],
+                          inline=False)
+
+        if 'thumbnail' in info.keys():
+            embedvc.set_thumbnail(url=info['thumbnail'])
+
+        if 'duration' in info.keys():
+            duration = str(datetime.timedelta(seconds=info['duration']))
+            embedvc.add_field(name=config.SONGINFO_DURATION,
+                              value=f"{duration}",
+                              inline=False)
+        else:
+            embedvc.add_field(name=config.SONGINFO_DURATION,
+                              value=config.SONGINFO_UNKNOWN_DURATION,
+                              inline=False)
+
+        return embedvc