diff --git a/src/api/routes/channels/#channel_id/polls/#poll_id/answers/@me.ts b/src/api/routes/channels/#channel_id/polls/#poll_id/answers/@me.ts index 95660c308..a3594c8ee 100644 --- a/src/api/routes/channels/#channel_id/polls/#poll_id/answers/@me.ts +++ b/src/api/routes/channels/#channel_id/polls/#poll_id/answers/@me.ts @@ -20,7 +20,7 @@ import { Request, Response, Router } from "express"; import { route } from "@spacebar/api/util/handlers/route"; import { PollAnswerCount, PollUserAnswersSchema } from "@spacebar/schemas"; import { Message } from "#database"; -import { DiscordApiErrors, ErrorList, FieldError, makeObjectErrorContent } from "#util"; +import { DiscordApiErrors, emitEvent, ErrorList, FieldError, makeObjectErrorContent, MessagePollVoteAddEvent, MessagePollVoteRemoveEvent } from "#util"; const router: Router = Router({ mergeParams: true }); @@ -44,6 +44,9 @@ router.put("/", route({ requestBody: "PollUserAnswersSchema", permission: "VIEW_ throw new FieldError(50035, "Invalid form body", errors); } + const channel_id = message.channel_id!; + const guild_id = message.guild_id; + const allAnswerCounts = message.poll.results.answer_counts as unknown as (Omit & { voters: string[] })[]; for (const answer_id of payload.answer_ids) { @@ -57,12 +60,36 @@ router.put("/", route({ requestBody: "PollUserAnswersSchema", permission: "VIEW_ if (!answerCount.voters.includes(req.user_id)) { answerCount.voters.push(req.user_id); answerCount.count = answerCount.voters.length; + + await emitEvent({ + event: "MESSAGE_POLL_VOTE_ADD", + channel_id, + data: { + answer_id: Number(answerCount.id), + channel_id: channel_id, + message_id: poll_id, + user_id: req.user_id, + guild_id, + }, + } satisfies MessagePollVoteAddEvent); } } for (const answerCount of allAnswerCounts.filter((a) => !payload.answer_ids.includes(a.id))) { answerCount.voters = answerCount.voters.filter((voter) => voter != req.user_id); answerCount.count = answerCount.voters.length; + + await emitEvent({ + event: "MESSAGE_POLL_VOTE_REMOVE", + channel_id, + data: { + answer_id: Number(answerCount.id), + channel_id, + message_id: poll_id, + user_id: req.user_id, + guild_id, + }, + } satisfies MessagePollVoteRemoveEvent); } await message.save(); diff --git a/src/util/interfaces/Event.ts b/src/util/interfaces/Event.ts index 22eec64ff..e211044be 100644 --- a/src/util/interfaces/Event.ts +++ b/src/util/interfaces/Event.ts @@ -366,6 +366,29 @@ export interface MessageDeleteBulkEvent extends Event { guild_id?: string; }; } + +export interface MessagePollVoteAddEvent extends Event { + event: "MESSAGE_POLL_VOTE_ADD"; + data: { + answer_id: number; + channel_id: string; + guild_id?: string; + message_id: string; + user_id: string; + }; +} + +export interface MessagePollVoteRemoveEvent extends Event { + event: "MESSAGE_POLL_VOTE_REMOVE"; + data: { + answer_id: number; + channel_id: string; + guild_id?: string; + message_id: string; + user_id: string; + }; +} + export const enum ReactionType { normal = 0, burst = 1, @@ -685,6 +708,8 @@ export type EventData = | MessageUpdateEvent | MessageDeleteEvent | MessageDeleteBulkEvent + | MessagePollVoteAddEvent + | MessagePollVoteRemoveEvent | MessageReactionAddEvent | MessageReactionRemoveEvent | MessageReactionRemoveAllEvent @@ -746,6 +771,8 @@ export enum EVENTEnum { MessageUpdate = "MESSAGE_UPDATE", MessageDelete = "MESSAGE_DELETE", MessageDeleteBulk = "MESSAGE_DELETE_BULK", + MessagePollVoteAdd = "MESSAGE_POLL_VOTE_ADD", + MessageePollVoteRemove = "MESSAGE_POLL_VOTE_REMOVE", MessageReactionAdd = "MESSAGE_REACTION_ADD", MessageReactionRemove = "MESSAGE_REACTION_REMOVE", MessageReactionRemoveAll = "MESSAGE_REACTION_REMOVE_ALL", @@ -804,6 +831,8 @@ export type EVENT = | "MESSAGE_UPDATE" | "MESSAGE_DELETE" | "MESSAGE_DELETE_BULK" + | "MESSAGE_POLL_VOTE_ADD" + | "MESSAGE_POLL_VOTE_REMOVE" | "MESSAGE_REACTION_ADD" // TODO: add a new event: bulk add reaction: // | "MESSAGE_REACTION_BULK_ADD"