from discord.ext import commands
from embeds.MatrixRolesEmbed import MatrixRolesEmbed
from embeds.LanguageRolesEmbed import LanguageRolesEmbed
from embeds.FrameworkRolesEmbed import FrameworkRolesEmbed
from libs.Channels import Channels
from libs.Guilds import Guilds
from libs.Cog import Cog
from views.MatrixButtons import MatrixButtons
from views.LanguageButtons import LanguageButtons
from views.FrameworkButtons import FrameworkButtons

class RolesCog(Cog):

    def __init__(self, bot):
        super().__init__(bot)
        self.bot = bot
        self.messages_deletable = True

    @commands.Cog.listener()
    async def on_ready(self):
        guild = await Guilds().get_guild(self.bot)
        channel = await Channels().get_channel(guild, "add-roles")

        if channel is not None:
            await channel.purge()
            await channel.send(embed=MatrixRolesEmbed(), view=MatrixButtons())
            await channel.send(embed=LanguageRolesEmbed(), view=LanguageButtons(self.bot))
            await channel.send(embed=FrameworkRolesEmbed(), view=FrameworkButtons(self.bot))


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