73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
|
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")
|
||
|
|
|
||
|
|
|