from discord.ext import commands
from utils.EmbedGenerator import EmbedGenerator
from bot import ChristmasBot
from discord.ext.commands import Context


class SettingsCog(commands.Cog, name="Settings"):
    def __init__(self, bot: ChristmasBot):
        self.bot = bot

    @commands.group(aliases=["aj"], invoke_without_command=True)
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def autojoin(self, ctx: Context):
        await EmbedGenerator.Message(
            ctx,
            "Autojoin",
            f"Usage:\n\n`{ctx.prefix}autojoin set`\n`{ctx.prefix}autojoin unset`",
        )

    @autojoin.command(name="set")
    @commands.has_permissions(manage_channels=True)
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def autojoin_set(self, ctx: Context):
        vc = ctx.author.voice.channel
        await AutoJoin.update_channel(self.bot, ctx.guild.id, vc.id)
        await EmbedGenerator.Message(ctx, "Autojoin", "`enabled`")

    @autojoin.command(name="unset")
    @commands.has_permissions(manage_channels=True)
    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
    async def autojoin_del(self, ctx: Context):
        vc = ctx.author.voice.channel
        await AutoJoin.del_channel(self.bot, ctx.guild.id)
        await EmbedGenerator.Message(ctx, "Autojoin", "`disabled`")


def setup(bot: ChristmasBot):
    bot.add_cog(SettingsCog(bot))