import { type BushMessage, type BushSlashMessage } from '#lib'; import { ActionRow, ButtonComponent, ButtonStyle, type MessageComponentInteraction, type MessageOptions } from 'discord.js'; /** * Sends a message with buttons for the user to confirm or cancel the action. */ export class ConfirmationPrompt { /** * Options for sending the message */ protected messageOptions: MessageOptions; /** * The message that triggered the command */ protected message: BushMessage | BushSlashMessage; /** * @param message The message to respond to * @param options The send message options */ protected constructor(message: BushMessage | BushSlashMessage, messageOptions: MessageOptions) { this.message = message; this.messageOptions = messageOptions; } /** * Sends a message with buttons for the user to confirm or cancel the action. */ protected async send(): Promise { this.messageOptions.components = [ new ActionRow().addComponents( new ButtonComponent({ style: ButtonStyle.Success, customId: 'confirmationPrompt_confirm', label: 'Yes' }), new ButtonComponent({ style: ButtonStyle.Danger, customId: 'confirmationPrompt_cancel', label: 'No' }) ) ]; const msg = await this.message.channel!.send(this.messageOptions); return await new Promise((resolve) => { let responded = false; const collector = msg.createMessageComponentCollector({ filter: (interaction) => interaction.message?.id == msg.id, time: 300_000 }); collector.on('collect', async (interaction: MessageComponentInteraction) => { await interaction.deferUpdate().catch(() => undefined); if (interaction.user.id == this.message.author.id || client.config.owners.includes(interaction.user.id)) { if (interaction.customId === 'confirmationPrompt_confirm') { responded = true; collector.stop(); resolve(true); } else if (interaction.customId === 'confirmationPrompt_cancel') { responded = true; collector.stop(); resolve(false); } } }); collector.on('end', async () => { await msg.delete().catch(() => undefined); if (!responded) resolve(false); }); }); } /** * Sends a message with buttons for the user to confirm or cancel the action. * @param message The message to respond to * @param options The send message options */ public static async send(message: BushMessage | BushSlashMessage, sendOptions: MessageOptions): Promise { return new ConfirmationPrompt(message, sendOptions).send(); } }