Merge pull request #180 from tjjfvi/suggestions

Use bot to manage #ts-discord-suggestions

Author: robertt

.env.example

@@ -19,3 +19,5 @@ GENERAL_HELP_CHANNEL=
 
 # Time in milliseconds before !helper can be run
 TIME_BEFORE_HELPER_PING=
+
+SUGGESTIONS_CHANNEL=

src/env.ts

@@ -32,3 +32,5 @@ export const BLOCKQUOTE_GREY = '#4f545c';
 export const timeBeforeHelperPing = parseInt(
 	process.env.TIME_BEFORE_HELPER_PING!,
 );
+
+export const suggestionsChannelId = process.env.SUGGESTIONS_CHANNEL!;

src/modules/etc.ts

@@ -5,15 +5,18 @@ import {
 	listener,
 	CommonInhibitors,
 } from 'cookiecord';
+import { ThreadAutoArchiveDuration } from 'discord-api-types';
 import {
 	Message,
 	MessageReaction,
 	GuildMember,
 	User,
 	ReactionEmoji,
 	TextChannel,
+	ThreadChannel,
 } from 'discord.js';
-import { MessageChannel } from 'worker_threads';
+import { MessageChannel, threadId } from 'worker_threads';
+import { suggestionsChannelId } from '../env';
 import {
 	clearMessageOwnership,
 	DELETE_EMOJI,
@@ -22,6 +25,8 @@ import {
 
 const emojiRegex = /<:\w+?:(\d+?)>|(\p{Emoji_Presentation})/gu;
 
+const defaultPollEmojis = ['✅', '❌', '🤷'];
+
 export class EtcModule extends Module {
 	constructor(client: CookiecordClient) {
 		super(client);
@@ -33,18 +38,67 @@ export class EtcModule extends Module {
 	}
 
 	@listener({ event: 'messageCreate' })
-	async onMessage(msg: Message) {
+	async onPoll(msg: Message) {
 		if (msg.author.bot || !msg.content.toLowerCase().startsWith('poll:'))
 			return;
 		let emojis = [
 			...new Set(
 				[...msg.content.matchAll(emojiRegex)].map(x => x[1] ?? x[2]),
 			),
 		];
-		if (!emojis.length) emojis = ['✅', '❌', '🤷'];
+		if (!emojis.length) emojis = defaultPollEmojis;
 		for (const emoji of emojis) await msg.react(emoji);
 	}
 
+	@listener({ event: 'messageCreate' })
+	async onSuggestion(msg: Message) {
+		if (msg.author.bot || msg.channelId !== suggestionsChannelId) return;
+		// First 50 characters of the first line of the content (without cutting off a word)
+		const title =
+			msg.content
+				.split('\n')[0]
+				.split(/(^.{0,50}\b)/)
+				.find(x => x) ?? 'Suggestion';
+		await msg.startThread({
+			name: title,
+			autoArchiveDuration: ThreadAutoArchiveDuration.OneDay,
+		});
+		for (let emoji of defaultPollEmojis) await msg.react(emoji);
+	}
+
+	@listener({ event: 'threadUpdate' })
+	async onSuggestionClose(thread: ThreadChannel) {
+		if (
+			thread.parentId !== suggestionsChannelId ||
+			!((await thread.fetch()) as ThreadChannel).archived
+		)
+			return;
+		const channel = thread.parent!;
+		let lastMessage = null;
+		let suggestion: Message;
+		while (!suggestion!) {
+			const msgs = await channel.messages.fetch({
+				before: lastMessage ?? undefined,
+				limit: 5,
+			});
+			suggestion = msgs.find(msg => msg.thread?.id === thread.id)!;
+			lastMessage = msgs.last()!.id as string;
+		}
+		const pollingResults = defaultPollEmojis.map(emoji => {
+			const reactions = suggestion.reactions.resolve(emoji);
+			// Subtract the bot's vote
+			const count = (reactions?.count ?? 0) - 1;
+			return [emoji, count] as const;
+		});
+		const pollingResultStr = pollingResults
+			.sort((a, b) => b[1] - a[1])
+			.map(([emoji, count], i) => `${count}  ${emoji}`)
+			.join('   ');
+		await suggestion.reply({
+			content: `Polling finished; result:  ${pollingResultStr}`,
+		});
+	}
+
 	@listener({ event: 'messageReactionAdd' })
 	async onReact(reaction: MessageReaction, member: GuildMember) {
 		if (reaction.partial) return;