import discord
from discord.ext import commands as cmds
from discord import app_commands as appcmds

from meta import LionBot, LionCog, LionContext
from meta.logger import log_wrap
from utils.lib import utc_now

from . import logger
from ..data import VoiceLogData, VoiceLogGuild
from .lib import ThreadedWebhook


class VoiceLogCog(LionCog):
    def __init__(self, bot: LionBot):
        self.bot = bot
        self.data = bot.db.load_registry(VoiceLogData())

    async def cog_load(self):
        await self.data.init()

    @LionCog.listener("on_voice_state_update")
    @log_wrap(action="Voice Log Update")
    async def voicelog_update(
        self,
        member: discord.Member,
        before: discord.VoiceState,
        after: discord.VoiceState,
    ):
        if member.bot:
            return

        after_channel = after.channel
        before_channel = before.channel
        if after_channel == before.channel:
            return

        row = await VoiceLogGuild.fetch(member.guild.id)
        if row is None or row.webhook_url is None:
            return

        hook = ThreadedWebhook.from_url(row.webhook_url, client=self.bot)

        embed = discord.Embed(timestamp=utc_now())

        if before_channel is None:
            embed.title = "User Joined Voice Channel"
            embed.description = f"{member.mention} joined {after_channel.mention}"
        elif after_channel is None:
            embed.title = "User Left Voice Channel"
            embed.description = f"{member.mention} left {before_channel.mention}"
        else:
            embed.title = "User Switched Voice Channel"
            embed.description = f"{member.mention} moved from {before_channel.mention} to {after_channel.mention}"

        await hook.send(embed=embed)

    @cmds.hybrid_group(
        name="voicelog", description="Base command group for the voice logging system"
    )
    @appcmds.default_permissions(manage_guild=True)
    async def voicelog_group(self, ctx: LionContext): ...

    @voicelog_group.command(
        name="enable",
        description="Enable voice activity logging and set the webhook url to use.",
    )
    async def voicelog_enable(self, ctx: LionContext, webhook_url: str):
        if not ctx.guild:
            return
        if not ctx.author.guild_permissions.manage_guild:
            return

        webhook = ThreadedWebhook.from_url(webhook_url, client=self.bot)
        try:
            embed = discord.Embed(
                title="Testing",
                description="Testing logging webhook, feel free to delete.",
            )
            await webhook.send(embed=embed, wait=True)
            existing = await VoiceLogGuild.fetch(ctx.guild.id)
            if existing:
                await existing.update(webhook_url=webhook_url)
            else:
                await VoiceLogGuild.create(
                    guildid=ctx.guild.id, webhook_url=webhook_url
                )

            await ctx.reply("Voice logging enabled!")
        except discord.HTTPException:
            await ctx.error_reply("Could not post to the given webhook!")

    @voicelog_group.command(
        name="disable", description="Disable voice activity logging in this server."
    )
    async def voicelog_disable(self, ctx: LionContext):
        if not ctx.guild:
            return
        if not ctx.author.guild_permissions.manage_guild:
            return

        await self.data.voicelog_guilds.update_where(guildid=ctx.guild.id).set(
            webhook_url=None
        )

        await ctx.reply("Voice Activity logging disabled.")