from typing import Awaitable
from Config.Emojis import VEmojis
from discord import ButtonStyle, Interaction, Message, TextChannel
from discord.ui import Button
from Handlers.HandlerResponse import HandlerResponse
from Messages.MessagesCategory import MessagesCategory
from Messages.MessagesManager import MessagesManager
from Music.VulkanBot import VulkanBot


class CallbackButton(Button):
    """When clicked execute an callback passing the args and kwargs"""

    def __init__(self, bot: VulkanBot, cb: Awaitable, emoji: VEmojis, textChannel: TextChannel, guildID: int, category: MessagesCategory, label=None, *args, **kwargs):
        super().__init__(label=label, style=ButtonStyle.secondary, emoji=emoji)
        self.__channel = textChannel
        self.__guildID = guildID
        self.__category = category
        self.__messagesManager = MessagesManager()
        self.__bot = bot
        self.__args = args
        self.__kwargs = kwargs
        self.__callback = cb

    async def callback(self, interaction: Interaction) -> None:
        """Callback to when Button is clicked"""
        # Return to Discord that this command is being processed
        await interaction.response.defer()

        response: HandlerResponse = await self.__callback(*self.__args, **self.__kwargs)

        if response and response.view is not None:
            message: Message = await self.__channel.send(embed=response.embed, view=response.view)
        else:
            message: Message = await self.__channel.send(embed=response.embed)

        # Clear the last sended message in this category and add the new one
        await self.__messagesManager.addMessageAndClearPrevious(self.__guildID, self.__category, message)