refactor voteInfo to include event instead of eventid and startDate
This commit is contained in:
		@@ -1,10 +1,10 @@
 | 
			
		||||
import { format, isToday } from "date-fns";
 | 
			
		||||
import { utcToZonedTime } from "date-fns-tz"
 | 
			
		||||
import { GuildScheduledEvent } from "discord.js";
 | 
			
		||||
import { logger } from "../logger";
 | 
			
		||||
import de from "date-fns/locale/de";
 | 
			
		||||
import { Maybe } from "../interfaces";
 | 
			
		||||
 | 
			
		||||
export function createDateStringFromEvent(eventStartDate:Date, requestId: string, guildId?: string): string {
 | 
			
		||||
export function createDateStringFromEvent(eventStartDate: Maybe<Date>, requestId: string, guildId?: string): string {
 | 
			
		||||
	if (!eventStartDate) {
 | 
			
		||||
		logger.error("Event has no start. Cannot create dateString.", { guildId, requestId })
 | 
			
		||||
		return `"habe keinen Startzeitpunkt ermitteln können"`
 | 
			
		||||
 
 | 
			
		||||
@@ -21,8 +21,7 @@ export type Vote = {
 | 
			
		||||
}
 | 
			
		||||
export type VoteMessageInfo = {
 | 
			
		||||
	votes: Vote[],
 | 
			
		||||
	eventId: string,
 | 
			
		||||
	eventDate: Date
 | 
			
		||||
	event: GuildScheduledEvent,
 | 
			
		||||
}
 | 
			
		||||
export default class VoteController {
 | 
			
		||||
	private client: ExtendedClient
 | 
			
		||||
@@ -74,11 +73,8 @@ export default class VoteController {
 | 
			
		||||
		logger.debug(`${vote.movie} : ${vote.count} -> above: ${overOneVote}`)
 | 
			
		||||
		return overOneVote
 | 
			
		||||
	}
 | 
			
		||||
	public async handleReroll(voteMessage: VoteMessage, guildId: string, requestId: string) {
 | 
			
		||||
		// get the movies currently being voted on, their votes, the eventId and its date
 | 
			
		||||
		const voteInfo: VoteMessageInfo = await this.parseVoteInfoFromVoteMessage(voteMessage, requestId)
 | 
			
		||||
 | 
			
		||||
		let movies: string[] = Array()
 | 
			
		||||
	public async generateRerollMovieList(voteInfo: VoteMessageInfo, guildId: string, requestId: string) {
 | 
			
		||||
		if (config.bot.reroll_retains_top_picks) {
 | 
			
		||||
			const votedOnMovies = voteInfo.votes.filter(this.hasAtLeastOneVote).filter(x => x.emote !== NONE_OF_THAT)
 | 
			
		||||
			logger.info(`Found ${votedOnMovies.length} with votes`, { requestId, guildId })
 | 
			
		||||
@@ -86,40 +82,53 @@ export default class VoteController {
 | 
			
		||||
			logger.info(`Fetching ${newMovieCount} from jellyfin`)
 | 
			
		||||
			const newMovies: string[] = await this.yavinJellyfinHandler.getRandomMovieNames(newMovieCount, guildId, requestId)
 | 
			
		||||
			// merge
 | 
			
		||||
			movies = newMovies.concat(votedOnMovies.map(x => x.movie))
 | 
			
		||||
			return newMovies.concat(votedOnMovies.map(x => x.movie))
 | 
			
		||||
		} else {
 | 
			
		||||
			// get movies from jellyfin to fill the remaining slots
 | 
			
		||||
			const newMovieCount: number = config.bot.random_movie_count
 | 
			
		||||
			logger.info(`Fetching ${newMovieCount} from jellyfin`)
 | 
			
		||||
			movies = await this.yavinJellyfinHandler.getRandomMovieNames(newMovieCount, guildId, requestId)
 | 
			
		||||
			return await this.yavinJellyfinHandler.getRandomMovieNames(newMovieCount, guildId, requestId)
 | 
			
		||||
		}
 | 
			
		||||
		// create new message
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	public async handleReroll(voteMessage: VoteMessage, guildId: string, requestId: string) {
 | 
			
		||||
		// get the movies currently being voted on, their votes, the eventId and its date
 | 
			
		||||
		const voteInfo: VoteMessageInfo = await this.parseVoteInfoFromVoteMessage(voteMessage, requestId)
 | 
			
		||||
		if (!voteInfo.event.scheduledStartAt) {
 | 
			
		||||
			logger.info("Event does not have a start date, cancelling", { guildId: voteInfo.event.guildId, requestId })
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		let movies: string[] = await this.generateRerollMovieList(voteInfo, guildId, requestId)
 | 
			
		||||
 | 
			
		||||
		logger.info(`Creating new poll message with new movies: ${movies}`, { requestId, guildId })
 | 
			
		||||
		const messageText = this.createVoteMessageText(voteInfo.eventId, voteInfo.eventDate, movies, guildId, requestId)
 | 
			
		||||
		const announcementChannel = this.client.getAnnouncementChannelForGuild(guildId)
 | 
			
		||||
		if (!announcementChannel) {
 | 
			
		||||
			logger.error(`No announcementChannel found for ${guildId}, can't post poll`)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		try {
 | 
			
		||||
			logger.info(`Trying to remove old vote Message`, { requestId, guildId })
 | 
			
		||||
			this.removeMessage(voteMessage)
 | 
			
		||||
		} catch (err) {
 | 
			
		||||
			// TODO: integrate failure DM to media Admin to inform about inability to delete old message
 | 
			
		||||
			logger.error(`Error during removeMessage: ${err}`)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		const sentMessage = await this.sendVoteMessage(messageText, movies.length, announcementChannel)
 | 
			
		||||
		sentMessage.pin()
 | 
			
		||||
		logger.info(`Sent and pinned new poll message`, { requestId, guildId })
 | 
			
		||||
		const sentMessage = this.prepareAndSendVoteMessage({
 | 
			
		||||
			event: voteInfo.event,
 | 
			
		||||
			movies,
 | 
			
		||||
			announcementChannel,
 | 
			
		||||
			startDate: voteInfo.event.scheduledStartAt,
 | 
			
		||||
			pinAfterSending: true
 | 
			
		||||
		},
 | 
			
		||||
			guildId,
 | 
			
		||||
			requestId)
 | 
			
		||||
		logger.debug(`Sent reroll message: ${JSON.stringify(sentMessage)}`, { requestId, guildId })
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	private async fetchEventStartDateByEventId(guild: Guild, eventId: string, requestId: string): Promise<Maybe<Date>> {
 | 
			
		||||
	private async fetchEventByEventId(guild: Guild, eventId: string, requestId: string): Promise<Maybe<GuildScheduledEvent>> {
 | 
			
		||||
		const guildEvent: GuildScheduledEvent = await guild.scheduledEvents.fetch(eventId)
 | 
			
		||||
		if (!guildEvent) logger.error(`GuildScheduledEvent with id${eventId} could not be found`, { requestId, guildId: guild.id })
 | 
			
		||||
		if (guildEvent.scheduledStartAt)
 | 
			
		||||
			return guildEvent.scheduledStartAt
 | 
			
		||||
		return guildEvent
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	public async parseVoteInfoFromVoteMessage(message: VoteMessage, requestId: string): Promise<VoteMessageInfo> {
 | 
			
		||||
@@ -129,8 +138,7 @@ export default class VoteController {
 | 
			
		||||
		if (!message.guild)
 | 
			
		||||
			throw new Error(`Message ${message.id} not a guild message`)
 | 
			
		||||
 | 
			
		||||
		let eventStartDate: Maybe<Date> = await this.fetchEventStartDateByEventId(message.guild, parsedIds.eventId, requestId)
 | 
			
		||||
		if (!eventStartDate) eventStartDate = this.parseEventDateFromMessage(message.cleanContent, message.guild.id, requestId)
 | 
			
		||||
		const event: Maybe<GuildScheduledEvent> = await this.fetchEventByEventId(message.guild, parsedIds.eventId, requestId)
 | 
			
		||||
 | 
			
		||||
		let votes: Vote[] = []
 | 
			
		||||
		for (const line of lines) {
 | 
			
		||||
@@ -149,7 +157,7 @@ export default class VoteController {
 | 
			
		||||
				}
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		return <VoteMessageInfo>{ eventId: parsedIds.eventId, eventDate: eventStartDate, votes }
 | 
			
		||||
		return <VoteMessageInfo>{ event, votes }
 | 
			
		||||
	}
 | 
			
		||||
	public parseEventDateFromMessage(message: string, guildId: string, requestId: string): Date {
 | 
			
		||||
		logger.warn(`Falling back to RegEx parsing to get Event Date`, { guildId, requestId })
 | 
			
		||||
@@ -168,15 +176,15 @@ export default class VoteController {
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	public async prepareAndSendVoteMessage(inputInfo: prepareVoteMessageInput, guildId: string, requestId: string) {
 | 
			
		||||
		const messageText = this.createVoteMessageText(inputInfo.event.id, inputInfo.startDate, inputInfo.movies, guildId, requestId)
 | 
			
		||||
		const messageText = this.createVoteMessageText(inputInfo.event, inputInfo.movies, guildId, requestId)
 | 
			
		||||
		const sentMessage = await this.sendVoteMessage(messageText, inputInfo.movies.length, inputInfo.announcementChannel)
 | 
			
		||||
		if (inputInfo.pinAfterSending)
 | 
			
		||||
			sentMessage.pin()
 | 
			
		||||
		return sentMessage
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	public createVoteMessageText(eventId: string, eventStartDate: Date, movies: string[], guildId: string, requestId: string): string {
 | 
			
		||||
		let message = `[Abstimmung] für https://discord.com/events/${guildId}/${eventId} \n<@&${config.bot.announcement_role}> Es gibt eine neue Abstimmung für die nächste Watchparty ${createDateStringFromEvent(eventStartDate, guildId, requestId)}! Stimme hierunter für den nächsten Film ab!\n`
 | 
			
		||||
	public createVoteMessageText(event: GuildScheduledEvent, movies: string[], guildId: string, requestId: string): string {
 | 
			
		||||
		let message = `[Abstimmung] für https://discord.com/events/${guildId}/${event.id} \n<@&${config.bot.announcement_role}> Es gibt eine neue Abstimmung für die nächste Watchparty ${createDateStringFromEvent(event.scheduledStartAt, guildId, requestId)}! 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]).concat("\n")
 | 
			
		||||
@@ -186,6 +194,7 @@ export default class VoteController {
 | 
			
		||||
		return message
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// TODO: Refactor into separate message controller
 | 
			
		||||
	public async sendVoteMessage(messageText: string, movieCount: number, announcementChannel: TextChannel) {
 | 
			
		||||
 | 
			
		||||
		const options: MessageCreateOptions = {
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user