jellyfin-discord-bot/server/commands/closepoll.ts

import { Guild, GuildScheduledEvent, GuildScheduledEventEditOptions, GuildScheduledEventSetStatusArg, GuildScheduledEventStatus, Message, MessageEditOptions, TextChannel } from 'discord.js'
import { v4 as uuid } from 'uuid'
import { config } from '../configuration'
import { Emotes } from '../events/guildScheduledEventCreate'
import { logger } from '../logger'
import { Command } from '../structures/command'
import { RunOptions } from '../types/commandTypes'
import { client } from '../..'

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, requestId })
        if (!command.guild) {
            logger.error("No guild found in interaction. Cancelling closing request", { guildId, requestId })
            command.followUp("Es gab leider ein Problem. Ich konnte deine Anfrage nicht bearbeiten :(")
            return
        }

        command.followUp("Alles klar, beende die Umfrage :)")
        closePoll(command.guild, requestId)
    }
})

export async function closePoll(guild: Guild, requestId: string) {
    const guildId = guild.id
    logger.info("stopping poll", { guildId, requestId })

    const announcementChannel: TextChannel = client.getAnnouncementChannelForGuild(guildId)

    const messages: Message<true>[] = (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)

    if (!messages || messages.length <= 0) {
        logger.info("Could not find any vote messages. Cancelling pollClose", { guildId, requestId })
        return
    }

    const lastMessage: Message<true> = messages[0]

    logger.debug(`Found messages: ${JSON.stringify(messages, null, 2)}`, { guildId, requestId })

    logger.debug(`Last message: ${JSON.stringify(lastMessage, null, 2)}`, { 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, requestId })

    updateEvent(votes, 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,
    }
    logger.info("Updating message.", { guildId, requestId })
    message.edit(options)

}

async function updateEvent(votes: Vote[], guild: Guild, guildId: string, requestId: string) {
    logger.info(`Updating event with movie ${votes[0].movie}.`, { 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, requestId })

    if (!voteEvents || voteEvents.length <= 0) {
        logger.error("Could not find vote event. Cancelling update!", { guildId, requestId })
        return
    }

    const voteEvent: GuildScheduledEvent<GuildScheduledEventStatus> = voteEvents[0]
    const options: GuildScheduledEventEditOptions<GuildScheduledEventStatus.Scheduled, GuildScheduledEventSetStatusArg<GuildScheduledEventStatus.Scheduled>> = {
        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, requestId })
    logger.info("Updating event.", { 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<Vote[]> {
    const votes: Vote[] = []
    logger.debug(`Number of items in emotes: ${Object.values(Emotes).length}`, { guildId, requestId })
    for (let i = 0; i < Object.keys(Emotes).length / 2; i++) {
        const emote = Emotes[i]
        logger.debug(`Getting reaction for emote ${emote}`, { guildId, requestId })
        const reaction = await message.reactions.resolve(emote)
        logger.debug(`Reaction for emote ${emote}: ${JSON.stringify(reaction, null, 2)}`, { 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
}