import discord
import datetime
from discord import app_commands
from discord.ext import commands
from cogs.music import Music
from typing import Literal
from ai_recommendations import add_song_recommendations

from global_variables import BOT_COLOR


class Autoplay(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @app_commands.command()
    @app_commands.check(Music.create_player)
    @app_commands.describe(toggle="Turn autoplay ON or OFF")
    async def autoplay(
        self, interaction: discord.Interaction, toggle: Literal["ON", "OFF"]
    ):
        "Keep the music playing forever with music suggestions from OpenAI"
        if toggle == "OFF":
            self.bot.autoplay.remove(interaction.guild.id)

            embed = discord.Embed(
                title="Autoplay Off",
                description="Autoplay has been turned off. I will no longer automatically add new songs to the queue based on AI recommendations.",
                color=BOT_COLOR,
            )
            return await interaction.response.send_message(embed=embed)

        # Otherwise, toggle must be "ON", so enable autoplaying

        if interaction.guild.id in self.bot.autoplay:
            embed = discord.Embed(
                title="Autoplay Already Enabled",
                description="Autoplay is already enabled. If you would like to turn it off, choose the `OFF` option in the `/autoplay` command.",
                color=BOT_COLOR,
            )
            return await interaction.response.send_message(embed=embed, ephemeral=True)

        player = self.bot.lavalink.player_manager.get(interaction.guild.id)

        if len(player.queue) < 5:
            embed = discord.Embed(
                title="Not Enough Context",
                description="You must have at least 5 songs in the queue so that I can get a good understanding of what music I should continue to play. Add some more music to the queue, then try again.",
                color=BOT_COLOR,
            )
            embed.set_footer(
                text=datetime.datetime.now(datetime.timezone.utc).strftime(
                    "%Y-%m-%d %H:%M:%S"
                )
                + " UTC"
            )
            return await interaction.response.send_message(embed=embed, ephemeral=True)

        inputs = {}
        for song in player.queue[:10]:
            inputs[song.title] = song.author

        embed = discord.Embed(
            title="Getting AI Recommendations",
            description="Attempting to generate recommendations based on the current songs in your queue. Just a moment...",
            color=BOT_COLOR,
        )
        await interaction.response.send_message(embed=embed)

        if await add_song_recommendations(self.bot.user, player, 5, inputs):
            self.bot.autoplay.append(interaction.guild.id)
            embed = discord.Embed(
                title=":infinity: Autoplay Enabled :infinity:",
                description=f"I have added a few similar songs to the queue and will continue to do so once the queue gets low again. Now just sit back and enjoy the music!\n\nEnabled by: {interaction.user.mention}",
                color=BOT_COLOR,
            )
            await interaction.edit_original_response(embed=embed)

        else:
            embed = discord.Embed(
                title="Autoplay Error",
                description="Autoplay is an experimental feature, meaning sometimes it doesn't work as expected. I had an error when attempting to get similar songs for you, please try running the command again. If the issue persists, fill out a bug report with the </bug:1224840889906499626> command.",
                color=BOT_COLOR,
            )
            embed.set_footer(
                text=datetime.datetime.now(datetime.timezone.utc).strftime(
                    "%Y-%m-%d %H:%M:%S"
                )
                + " UTC"
            )
            await interaction.edit_original_response(embed=embed)


async def setup(bot):
    await bot.add_cog(Autoplay(bot))