1 files changed, 78 insertions, 0 deletions
diff --git a/lib/common/DeleteButton.ts b/lib/common/DeleteButton.ts
new file mode 100644
index 0000000..340d07f
--- /dev/null
+++ b/lib/common/DeleteButton.ts
@@ -0,0 +1,78 @@
+import { PaginateEmojis, type CommandMessage, type SlashMessage } from '#lib';
+import { CommandUtil } from 'discord-akairo';
+import {
+	ActionRowBuilder,
+	ButtonBuilder,
+	ButtonStyle,
+	MessageComponentInteraction,
+	MessageEditOptions,
+	MessagePayload,
+	type MessageOptions
+} from 'discord.js';
+
+/**
+ * Sends a message with a button for the user to delete it.
+ */
+export class DeleteButton {
+	/**
+	 * @param message The message to respond to
+	 * @param messageOptions The send message options
+	 */
+	protected constructor(protected message: CommandMessage | SlashMessage, protected messageOptions: MessageOptions) {}
+
+	/**
+	 * Sends a message with a button for the user to delete it.
+	 */
+	protected async send() {
+		this.updateComponents();
+
+		const msg = await this.message.util.reply(this.messageOptions);
+
+		const collector = msg.createMessageComponentCollector({
+			filter: (interaction) => interaction.customId == 'paginate__stop' && interaction.message?.id == msg.id,
+			time: 300000
+		});
+
+		collector.on('collect', async (interaction: MessageComponentInteraction) => {
+			await interaction.deferUpdate().catch(() => undefined);
+			if (interaction.user.id == this.message.author.id || this.message.client.config.owners.includes(interaction.user.id)) {
+				if (msg.deletable && !CommandUtil.deletedMessages.has(msg.id)) await msg.delete();
+			}
+		});
+
+		collector.on('end', async () => {
+			this.updateComponents(true, true);
+			await msg.edit(<string | MessagePayload | MessageEditOptions>this.messageOptions).catch(() => undefined);
+		});
+	}
+
+	/**
+	 * Generates the components for the message
+	 * @param edit Whether or not the message is being edited
+	 * @param disable Whether or not to disable the buttons
+	 */
+	protected updateComponents(edit = false, disable = false): void {
+		this.messageOptions.components = [
+			new ActionRowBuilder<ButtonBuilder>().addComponents(
+				new ButtonBuilder({
+					style: ButtonStyle.Primary,
+					customId: 'paginate__stop',
+					emoji: PaginateEmojis.STOP,
+					disabled: disable
+				})
+			)
+		];
+		if (edit) {
+			this.messageOptions.reply = undefined;
+		}
+	}
+
+	/**
+	 * Sends a message with a button for the user to delete it.
+	 * @param message The message to respond to
+	 * @param options The send message options
+	 */
+	public static async send(message: CommandMessage | SlashMessage, options: Omit<MessageOptions, 'components'>) {
+		return new DeleteButton(message, options).send();
+	}
+}