from naff import Extension, listen, slash_command, slash_option, context_menu, CommandTypes, events, InteractionContext, OptionTypes, Client
from naff.models.discord.enums import ButtonStyles, Permissions
from naff.models.discord.embed import Embed, EmbedAuthor, EmbedField, EmbedFooter
from naff.models.discord.components import ActionRow, Button

import re
import logging

class QuoteExtension(Extension):
    def __init__(self, client: Client):
        self.client = client

    @slash_command(name="quote", description="Quote a message", dm_permission=False)
    @slash_option(name="url", description="Message URL", required=True, opt_type=OptionTypes.STRING)
    async def quote_command(self, ctx: InteractionContext, url: str):

        url_match = re.match(r"https?://discord(?:app)?\.com/channels/(\d+)/(\d+)/(\d+)", url)
        if url_match is None:
            ctx.ephemeral = True
            await ctx.send("Invalid URL")
            return
        
        guild_id, channel_id, message_id = url_match.groups()
        
        chan = await self.client.fetch_channel(channel_id)
        msg = await chan.fetch_message(message_id)

        if chan is None or msg is None:
            await ctx.send("Failed to get message", ephemeral=True)
            print(chan)
            print(msg)
            return

        if not (ctx.author.channel_permissions(chan) & (Permissions.VIEW_CHANNEL | Permissions.READ_MESSAGE_HISTORY)) \
            == (Permissions.VIEW_CHANNEL | Permissions.READ_MESSAGE_HISTORY):

            ctx.ephemeral = True
            await ctx.send("You don't have permission to view this message", ephemeral=True)
            return

        author_embed = EmbedAuthor(name=msg.author.display_name, icon_url=msg.author.display_avatar.as_url())
        embed = Embed(
            #title=msg.author.display_name, 
            author=author_embed,
            description=msg.content, 
            color=0x00ff00, 
            timestamp=msg.created_at, 
            url=url)
        if len(msg.attachments) == 1 and msg.attachments[0].width is not None:
            embed.set_image(url=msg.attachments[0].url)
        elif len(msg.attachments) > 1:
            field = EmbedField(name="**Attachments**", value="\n".join([f"• [{a.filename}]({a.url})" for a in msg.attachments]))
            embed.fields.append(field)

        components: list[ActionRow] = [
            ActionRow(
                Button(
                    style=ButtonStyles.LINK,
                    emoji="🔗",
                    url=url,
                )
            )
        ]

        await ctx.send(embed=embed, components=components)


def setup(bot):
    QuoteExtension(bot)
    logging.info("Quote extension loaded")