feat: Support for removing playground links

Author: Gerrit0

src/modules/playground.ts

@@ -12,6 +12,7 @@ import {
 	findCodeblockFromChannel,
 	PLAYGROUND_REGEX,
 } from '../util/findCodeblockFromChannel';
+import { addMessageOwnership, sendWithMessageOwnership } from '../util/send';
 
 export class PlaygroundModule extends Module {
 	constructor(client: CookiecordClient) {
@@ -34,15 +35,16 @@ export class PlaygroundModule extends Module {
 				true,
 			);
 			if (!code)
-				return await msg.channel.send(
+				return sendWithMessageOwnership(
+					msg,
 					":warning: couldn't find a codeblock!",
 				);
 		}
 		const embed = new MessageEmbed()
 			.setURL(PLAYGROUND_BASE + compressToEncodedURIComponent(code))
 			.setTitle('View in Playground')
 			.setColor(TS_BLUE);
-		await msg.channel.send({ embed });
+		await sendWithMessageOwnership(msg, { embed });
 	}
 
 	@listener({ event: 'message' })
@@ -56,15 +58,16 @@ export class PlaygroundModule extends Module {
 			.setURL(exec[0]);
 		if (exec[0] == msg.content) {
 			// Message only contained the link
+			await sendWithMessageOwnership(msg, { embed });
 			msg.delete();
-			await msg.channel.send({ embed });
 		} else {
 			// Message also contained other characters
 			const botMsg = await msg.channel.send(
 				`${msg.author} Here's a shortened URL of your playground link! You can remove the full link from your message.`,
 				{ embed },
 			);
 			this.editedLongLink.set(msg.id, botMsg);
+			addMessageOwnership(botMsg, msg.author);
 		}
 	}
 	@listener({ event: 'messageUpdate' })

src/util/send.ts

@@ -1,27 +1,30 @@
-import { Message, MessageEmbed, GuildMember } from 'discord.js';
+import { Message, MessageEmbed, User } from 'discord.js';
 
 const MAX_TRACKED_MESSAGES = 1000;
 
-const messageIdToMemberId = new Map<string, string>();
+const messageToUserId = new Map<string, string>();
 
 export async function sendWithMessageOwnership(
 	message: Message,
-	toSend: string | MessageEmbed,
+	toSend: string | { embed: MessageEmbed },
 ) {
 	const sent = await message.channel.send(toSend);
-	messageIdToMemberId.set(sent.id, message.author.id);
+	addMessageOwnership(sent, message.author);
+}
 
+export function addMessageOwnership(message: Message, user: User) {
+	messageToUserId.set(message.id, user.id);
 	// Without this memory grows unboundedly... very slowly, but better to avoid the issue.
-	if (messageIdToMemberId.size > MAX_TRACKED_MESSAGES) {
+	if (messageToUserId.size > MAX_TRACKED_MESSAGES) {
 		// Keys returns an iterable in insertion order, so we remove the oldest message from the map.
-		messageIdToMemberId.delete(messageIdToMemberId.keys().next().value);
+		messageToUserId.delete(messageToUserId.keys().next().value);
 	}
 }
 
-export function ownsBotMessage(message: Message, member: GuildMember) {
-	return messageIdToMemberId.get(message.id) === member.id;
+export function ownsBotMessage(message: Message, user: User) {
+	return messageToUserId.get(message.id) === user.id;
 }
 
 export function clearMessageOwnership(message: Message) {
-	messageIdToMemberId.delete(message.id);
+	messageToUserId.delete(message.id);
 }