From c8fa89ae63eddb932c7ffdfe0f12e7243b88494f Mon Sep 17 00:00:00 2001 From: Sammy Date: Sat, 10 Jun 2023 17:27:32 +0200 Subject: [PATCH] Add command to close the poll and update the event --- server/commands/closepoll.ts | 115 +++++++++++++++++++++ server/events/guildScheduledEventCreate.ts | 9 +- 2 files changed, 120 insertions(+), 4 deletions(-) create mode 100644 server/commands/closepoll.ts diff --git a/server/commands/closepoll.ts b/server/commands/closepoll.ts new file mode 100644 index 0000000..6363fdd --- /dev/null +++ b/server/commands/closepoll.ts @@ -0,0 +1,115 @@ +import { Command } from '../structures/command' +import { RunOptions } from '../types/commandTypes' +import { APIEmbed, Guild, GuildScheduledEvent, GuildScheduledEventEditOptions, GuildScheduledEventSetStatusArg, GuildScheduledEventStatus, Message, MessageEditOptions, TextChannel } from 'discord.js' +import { v4 as uuid } from 'uuid' +import { logger } from '../logger' +import { config } from '../configuration' +import { Emotes } from '../events/guildScheduledEventCreate' +import { EndOfLineState } from 'typescript' + +export default new Command({ + name: 'closepoll', + description: 'Aktuelle Umfrage für nächste Watchparty beenden und Gewinner in Event eintragen.', + options: [], + run: async (interaction: RunOptions) => { + const command = interaction.interaction + const requestId = uuid() + const guildId = command.guildId! + logger.info("Got command for closing poll!", { guildId: guildId, requestId }) + + const announcementChannel: TextChannel = (await command.guild?.channels.fetch()) + ?.filter(channel => channel!.id === config.bot.announcement_channel_id) + .map((value, _) => value)[0] //todo: needs to be done less sketchy + + interaction.interaction.followUp("Alles klar, beende die Umfrage :)") + + + const messages: Message[] = (await announcementChannel.messages.fetch()) //todo: fetch only pinned messages + .map((value, _) => value) + .filter(message => !message.cleanContent.includes("[Abstimmung beendet]") && message.cleanContent.includes("[Abstimmung]")) + .sort((a, b) => b.createdTimestamp - a.createdTimestamp) + + + const lastMessage: Message = messages[0] + + logger.debug(`Found messages: ${JSON.stringify(messages, null, 2)}`, { guildId: guildId, requestId }) + + logger.debug(`Last message: ${JSON.stringify(lastMessage, null, 2)}`, { guildId: guildId, requestId }) + + const votes = await (await getVotesByEmote(lastMessage, guildId, requestId)) + .sort((a, b) => b.count - a.count) + + logger.debug(`votes: ${JSON.stringify(votes, null, 2)}`, { guildId: guildId, requestId }) + + updateEvent(votes, command.guild!, guildId, requestId) + updateMessage(votes[0].movie, lastMessage, guildId, requestId) + + //lastMessage.unpin() //todo: uncomment when bot has permission to pin/unpin + } +}) + +async function updateMessage(movie: string, message: Message, guildId: string, requestId: string) { + const body = `[Abstimmung beendet] Gewonnen hat: ${movie}` + .concat(message.cleanContent.substring("[Abstimmung]".length)) + + const options: MessageEditOptions = { + content: body, + } + message.edit(options) + +} + +async function updateEvent(votes: Vote[], guild: Guild, guildId: string, requestId: string) { + logger.info(`Updating event with movie ${votes[0].movie}.`, { guildId: guildId, requestId }) + const voteEvents = (await guild.scheduledEvents.fetch()) + .map((value, _) => value) + .filter(event => event.name.toLowerCase().includes("voting offen")) + logger.debug(`Found events: ${JSON.stringify(voteEvents, null, 2)}`, { guildId: guildId, requestId }) + + if(!voteEvents || voteEvents.length <= 0) { + logger.error("Could not find vote event. Cancelling update!", { guildId: guildId, requestId }) + } + + const voteEvent: GuildScheduledEvent = voteEvents[0] + const options: GuildScheduledEventEditOptions> = { + name: votes[0].movie, + description: `!wp\nNummer 2: ${votes[1].movie} mit ${votes[1].count -1} Stimmen\nNummer 3: ${votes[2].movie} mit ${votes[2].count -1} Stimmen` + } + logger.debug(`Updating event: ${JSON.stringify(voteEvent, null, 2)}`, { guildId: guildId, requestId }) + voteEvent.edit(options) +} + +type Vote = { + emote: string, //todo habs nicht hinbekommen hier Emotes zu nutzen + count: number, + movie: string +} + +async function getVotesByEmote(message: Message, guildId: string, requestId: string): Promise { + const votes: Vote[] = [] + logger.debug(`Number of items in emotes: ${Object.values(Emotes).length}`) + for (let i = 0; i < Object.keys(Emotes).length / 2; i++) { + const emote = Emotes[i] + // for (const emote of Object.keys(Emotes)) { + logger.debug(`Getting reaction for emote ${emote}`, { guildId: guildId, requestId }) + const reaction = await message.reactions.resolve(emote) + logger.debug(`Reaction for emote ${emote}: ${JSON.stringify(reaction, null, 2)}`, { guildId: guildId, requestId }) + if (reaction) { + const vote: Vote = { emote: emote, count: reaction.count, movie: extractMovieFromMessageByEmote(message, emote, guildId, requestId) } + votes.push(vote) + } + } + return votes +} + +function extractMovieFromMessageByEmote(message: Message, emote: string, guildId: string, requestId: string): string { + const lines = message.cleanContent.split("\n") + const emoteLines = lines.filter(line => line.includes(emote)) + + if (!emoteLines) { + return "" + } + const movie = emoteLines[0].substring(emoteLines[0].indexOf(emote) + emote.length + 2) // plus colon and space + + return movie +} \ No newline at end of file diff --git a/server/events/guildScheduledEventCreate.ts b/server/events/guildScheduledEventCreate.ts index 1896691..312ccdd 100644 --- a/server/events/guildScheduledEventCreate.ts +++ b/server/events/guildScheduledEventCreate.ts @@ -8,7 +8,7 @@ import { config } from "../configuration"; export const name = 'guildScheduledEventCreate' -const emotes = ["1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6️⃣", "7️⃣", "8️⃣", "9️⃣", "🔟"] +export enum Emotes {"1️⃣", "2️⃣", "3️⃣", "4️⃣", "5️⃣", "6️⃣", "7️⃣", "8️⃣", "9️⃣", "🔟"} export async function execute(event: GuildScheduledEvent) { const requestId = uuid() @@ -28,17 +28,18 @@ export async function execute(event: GuildScheduledEvent) { const channel: TextChannel = (await event.guild?.channels.fetch())?.filter(channel => channel!.id === config.bot.announcement_channel_id).map((value, _) => value)[0] //todo: needs to be done less sketchy logger.debug(`Found channel ${JSON.stringify(channel, null, 2)}`) - let message = "Es gibt eine neue Abstimmung für die nächste Watchparty! Stimme hierunter für den nächsten Film ab!\n" + let message = "[Abstimmung]\nEs gibt eine neue Abstimmung für die nächste Watchparty! Stimme hierunter für den nächsten Film ab!\n" for (let i = 0; i < movies.length; i++) { - message = message.concat(emotes[i]).concat(": ").concat(movies[i].name!).concat("\n") + message = message.concat(Emotes[i]).concat(": ").concat(movies[i].name!).concat("\n") } const sentMessage: Message = await (await channel.fetch()).send(message) for (let i = 0; i < movies.length; i++) { - sentMessage.react(emotes[i]) + sentMessage.react(Emotes[i]) } + // sentMessage.pin() //todo: uncomment when bot has permission to pin messages. Also update closepoll.ts to only fetch pinned messages } } \ No newline at end of file