from discord import (
    ApplicationContext,
    Cog,
    Member,
    SlashCommandGroup,
    VoiceChannel,
    VoiceState,
    option,
    utils,
)
from discord.ext import commands

from functions import (
    appendLog,
    changeNomicPerms,
    createUserVoice,
    getMsg,
    guildConfGet,
    guildConfReset,
    guildConfSet,
    isUserVoice,
    isVoiceOfUser,
    loadJson,
    makeEmbed,
    removeUserVoice,
    strToColor,
)


# =========================================================================================================================
class CogChannel(commands.Cog):
    def __init__(self, client):
        self.client = client

    @Cog.listener()
    async def on_voice_state_update(
        self, member: Member, before: VoiceState, after: VoiceState
    ):
        config = loadJson("config.json")

        vc_from = before.channel
        vc_to = after.channel

        # If user left vc
        if vc_to is None:
            if isUserVoice(vc_from):
                if isVoiceOfUser(vc_from, member):
                    await removeUserVoice(vc_from)
                    return
                else:
                    if loadJson("config.json")["enable_nomic"]:
                        await changeNomicPerms("deny", vc_from, member)

        # If user joined vc
        else:
            if isUserVoice(vc_from):
                if isVoiceOfUser(vc_from, member):
                    await removeUserVoice(vc_from)
                else:
                    if loadJson("config.json")["enable_nomic"]:
                        await changeNomicPerms("deny", vc_from, member)
            if isUserVoice(vc_to):
                if loadJson("config.json")["enable_nomic"]:
                    await changeNomicPerms("allow", vc_to, member)
            if vc_to.id == guildConfGet(vc_to.guild, "channel"):
                if guildConfGet(vc_to.guild, "category") is not None:
                    voice_chan = await createUserVoice(
                        vc_to,
                        utils.get(
                            vc_to.guild.categories,
                            id=guildConfGet(vc_to.guild, "category"),
                        ),
                        member,
                    )
                    try:
                        await member.move_to(voice_chan)
                    except:
                        await removeUserVoice(voice_chan)
                else:
                    appendLog(
                        f"Category for guild {vc_to.guild} ({str(vc_to.guild.id)}) is not set",
                        guild=vc_to.guild,
                    )

    channel = SlashCommandGroup("channel", "Commands related to parent voice channel")

    @channel.command(
        name="set",
        description="Select the voice channel that will be parent to private ones",
    )
    @option("channel", description="Parent Voice Channel")
    async def channel_set(self, ctx: ApplicationContext, channel: VoiceChannel):
        config = loadJson("config.json")
        if ctx.guild is not None:
            guildConfSet(ctx.guild, "channel", channel.id)
            await ctx.respond(
                embed=makeEmbed(
                    title=getMsg("set_channel_title", ctx.guild),
                    description=getMsg("set_channel_description", ctx.guild).format(
                        channel.name
                    ),
                    color=strToColor(config["color_ok"]),
                )
            )
            if guildConfGet(ctx.guild, "category") is None:
                await ctx.respond(
                    embed=makeEmbed(
                        title=getMsg("hint_none_category_title", ctx.guild),
                        description=getMsg("hint_none_category_description", ctx.guild),
                        color=strToColor(config["color_warn"]),
                    )
                )
        else:
            await ctx.respond(
                embed=makeEmbed(
                    title=getMsg("dm_title", ctx.guild),
                    description=getMsg("dm_description", ctx.guild),
                    color=strToColor(config["color_error"]),
                )
            )

    @channel.command(
        name="reset", description="Reset the currently selected parent voice channel"
    )
    async def channel_reset(self, ctx: ApplicationContext):
        config = loadJson("config.json")
        if ctx.guild is not None:
            if guildConfGet(ctx.guild, "channel") is not None:
                guildConfReset(ctx.guild, "channel")
                await ctx.respond(
                    embed=makeEmbed(
                        title=getMsg("reset_channel_title", ctx.guild),
                        description=getMsg("reset_channel_description", ctx.guild),
                        color=strToColor(config["color_ok"]),
                    )
                )
            else:
                await ctx.respond(
                    embed=makeEmbed(
                        title=getMsg("hint_none_channel_title", ctx.guild),
                        description=getMsg("hint_none_channel_description", ctx.guild),
                        color=strToColor(config["color_warn"]),
                    )
                )
        else:
            await ctx.respond(
                embed=makeEmbed(
                    title=getMsg("dm_title", ctx.guild),
                    description=getMsg("dm_description", ctx.guild),
                    color=strToColor(config["color_error"]),
                )
            )


# =========================================================================================================================