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

from utils.ai_recommendations import add_song_recommendations
from utils.config 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:1228216490386391052> 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.openai, self.bot.user, player, 5, inputs
        ):
            self.bot.autoplay.append(interaction.guild.id)
            embed = discord.Embed(
                title=":infinity: Autoplay Enabled :infinity:",
                description=(
                    "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:"
                    f" {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))