import asyncio
from discord.ext.commands import Context
from discord import Client
from Controllers.AbstractController import AbstractController
from Controllers.ControllerResponse import ControllerResponse
from Music.Downloader import Downloader
from Utils.Utils import Utils
from Parallelism.ProcessManager import ProcessManager


class QueueController(AbstractController):
    def __init__(self, ctx: Context, bot: Client) -> None:
        super().__init__(ctx, bot)
        self.__down = Downloader()

    async def run(self) -> ControllerResponse:
        # Retrieve the process of the guild
        process = ProcessManager()
        processContext = process.getRunningPlayerContext(self.guild)
        if not processContext:  # If no process return empty list
            embed = self.embeds.EMPTY_QUEUE()
            return ControllerResponse(self.ctx, embed)

        # Acquire the Lock to manipulate the playlist
        with processContext.getLock():
            playlist = processContext.getPlaylist()

            if playlist.isLoopingOne():
                song = playlist.getCurrentSong()
                embed = self.embeds.ONE_SONG_LOOPING(song.info)
                return ControllerResponse(self.ctx, embed)

            songs_preload = playlist.getSongsToPreload()
            if len(songs_preload) == 0:
                embed = self.embeds.EMPTY_QUEUE()
                return ControllerResponse(self.ctx, embed)

            asyncio.create_task(self.__down.preload(songs_preload))

            if playlist.isLoopingAll():
                title = self.messages.ALL_SONGS_LOOPING
            else:
                title = self.messages.QUEUE_TITLE

            total_time = Utils.format_time(sum([int(song.duration if song.duration else 0)
                                                for song in songs_preload]))
            total_songs = len(playlist.getSongs())

            text = f'📜 Queue length: {total_songs} | ⌛ Duration: `{total_time}` downloaded  \n\n'

            for pos, song in enumerate(songs_preload, start=1):
                song_name = song.title if song.title else self.messages.SONG_DOWNLOADING
                text += f"**`{pos}` - ** {song_name} - `{Utils.format_time(song.duration)}`\n"

            embed = self.embeds.QUEUE(title, text)
            return ControllerResponse(self.ctx, embed)