import asyncio
import typing

import lavalink
from discord.channel import TextChannel
from discord.ext import commands

from bot import TuneBot
from cogs.music import helper
from cogs.music import track_embed
from cogs.music.interactions import MusicCommands
from cogs.music.interactions import QUEUE_SIZE
from cogs.music.voice_client import LavalinkVoiceClient
from utils.classes import BaseCog
from utils.log import logger

if typing.TYPE_CHECKING:
    from bot import TuneBot


class MusicCog(BaseCog, name="Music"):
    def __init__(self, bot: TuneBot):
        super().__init__(bot)
        if not hasattr(bot, "lavalink"):
            self.bot.lavalink = self.bot.create_lavalink(bot.user.id)

        self.bot.lavalink.add_event_hook(self.track_hook)

    """	
    @commands.Cog.listener()
    async def on_ready(self):
        while not self.is_lavalink_ready():
            await asyncio.sleep(1)

        redis_result = await self.bot.global_autojoin.fetch_channels()
        for guild_id, (voicechannel_id, textchannel_id) in redis_result.items():
            try:
                player = self.bot.lavalink.player_manager.create(guild_id)
                player.store("channel", textchannel_id)
                voice_channel = await self.bot.fetch_channel(voicechannel_id)
                await voice_channel.connect(cls=LavalinkVoiceClient)
                if not player.is_playing:
                    await helper.fill_player_queue(self.bot, player, QUEUE_SIZE)
                    await player.play()
            except:
                logger.error(f"Failed to autojoin guild {guild_id}.")
    """

    async def track_hook(self, event: lavalink.Event):
        if isinstance(event, lavalink.events.QueueEndEvent):
            guild_id = event.player.guild_id
            guild = self.bot.get_guild(guild_id)
            await guild.voice_client.disconnect(force=True)
        elif isinstance(event, lavalink.events.TrackEndEvent):
            channel_id = int(event.player.fetch("channel"))
            channel: TextChannel = self.bot.get_channel(channel_id)
            await helper.fill_player_queue(self.bot, event.player, 1)
            event.player.append_history(event.track)

            if event.reason == "FINISHED":
                queue = event.player.queue[1:]
                current = event.player.queue[0]
            else:
                queue = event.player.queue
                current = event.player.current

            embed = track_embed.create_track_embed(current, queue, event.player.history)
            await channel.send(embed=embed)


async def setup(bot: "TuneBot"):
    await bot.add_cog(MusicCog(bot))
    bot.tree.add_command(MusicCommands(bot), override=True)