dandri/server
1
0
Fork 0
mirror of https://github.com/spacebarchat/server.git synced 2026-08-28 13:54:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: add poll utility

Browse Source
This commit is contained in:
CyberL1
2026-06-26 16:42:33 +02:00
committed by Rory&
parent 5454898b84
commit 73d1ae9e59
5 changed files with 145 additions and 276 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/api/util/handlers/Message.ts
+3 -94
View File
@@ -37,7 +37,6 @@ import {
makeObjectErrorContent,
MessageCreateEvent,
MessageFlags,
pendingPolls,
Permissions,
ROLE_MENTION,
Snowflake,
@@ -59,12 +58,12 @@ import {
MessageCreateSchema,
MessageReferenceType,
MessageType,
PollAnswerCount,
Reaction,
ReadStateType,
UnfurledMediaItem,
v1CompTypes,
} from "@spacebar/schemas";
import { addPendingPoll } from "../utility/polls";
const allow_empty = false;
// TODO: check webhook, application, system author, stickers
@@ -497,97 +496,7 @@ export async function handleMessage(opts: MessageOptions): Promise<Message> {
if (opts.poll?.duration) {
message.poll.expiry = new Date(Date.now() + opts.poll.duration * 3600000);
const pollTimeout = setTimeout(async () => {
if (!message.poll?.results) {
return;
}
const allAnswerCounts = message.poll.results.answer_counts as unknown as (Omit<PollAnswerCount, "me_voted"> & { voters: string[] })[];
const totalVotes = allAnswerCounts.map((a) => a.voters).length;
const winningAnswerCounts = allAnswerCounts.filter((a) => (a.count * totalVotes) / 100);
const pollResultsMessage = {
type: MessageType.POLL_RESULT,
channel_id: message.channel_id,
author_id: message.author_id,
message_reference: {
type: MessageReferenceType.DEFAULT,
message_id: message.id,
channel_id: message.channel_id,
},
embeds: [
{
type: EmbedType.poll_result,
id: message.id,
fields: [
{
name: "poll_question_text",
value: message.poll.question.text!,
},
{
name: "total_votes",
value: totalVotes.toString(),
},
],
},
],
};
if (winningAnswerCounts) {
const winningAnswer = message.poll.answers.find((a) => a.answer_id === Number(winningAnswerCounts[0]?.id))!;
if (winningAnswerCounts.length === 0) {
pollResultsMessage.embeds[0].fields.push({
name: "victor_answer_votes",
value: "0",
});
} else if (winningAnswerCounts.length === 1) {
pollResultsMessage.embeds[0].fields.push(
{
name: "victor_answer_votes",
value: winningAnswerCounts[0].count.toString(),
},
{
name: "victor_answer_id",
value: winningAnswerCounts[0].id,
},
{
name: "victor_answer_text",
value: winningAnswer.poll_media.text!,
},
);
} else if (winningAnswerCounts.length > 1) {
pollResultsMessage.embeds[0].fields.push({
name: "victor_answer_votes",
value: winningAnswerCounts[0].count.toString(),
});
}
if (winningAnswer?.poll_media.emoji) {
pollResultsMessage.embeds[0].fields.push(
{
name: "victor_answer_emoji_id",
value: winningAnswer.poll_media.emoji.id!.toString()!,
},
{
name: "victor_answer_emoji_name",
value: winningAnswer.poll_media.emoji.name!,
},
{
name: "victor_answer_emoji_animated",
value: `${winningAnswer.poll_media.emoji.animated}`,
},
);
}
}
await sendMessage(pollResultsMessage);
pendingPolls.delete(message.id);
}, opts.poll.duration * 3600000);
pendingPolls.set(message.id, { timeout: pollTimeout });
addPendingPoll(message, opts.poll.duration * 3600000);
}
if (opts.poll?.answers) {
@@ -689,7 +598,7 @@ export async function sendMessage(opts: MessageOptions) {
}
type MessageOptionAttachment = MessageCreateAttachment | MessageCreateCloudAttachment | Attachment;
interface MessageOptions extends MessageCreateSchema {
export interface MessageOptions extends MessageCreateSchema {
id?: string;
type?: MessageType;
pinned?: boolean;
src/api/util/index.ts
+1
View File
@@ -23,3 +23,4 @@ export * from "./handlers/route";
export * from "./handlers/Voice";
export * from "./utility/captcha";
export * from "./utility/EmbedHandlers";
export * from "./utility/polls";
src/api/util/utility/polls.ts
+136
View File
@@ -0,0 +1,136 @@
/*
Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
Copyright (C) 2026 Spacebar and Spacebar Contributors
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published
by the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
import { EmbedType, MessageReferenceType, MessageType, PollAnswerCount } from "@spacebar/schemas";
import { pendingPolls } from "@spacebar/util";
import { MessageOptions, sendMessage } from "../handlers/Message";
import { Message } from "@spacebar/database";
export async function generatePollResultsMessage(options: MessageOptions): Promise<MessageOptions> {
const message = await Message.create({
...options,
message_reference: options.message_reference ?? undefined,
poll: options.poll,
// sticker_items: stickers,
// guild_id: options.guild_id,
channel_id: options.channel_id,
attachments: [],
embeds: options.embeds || [],
reactions: [],
type: MessageType.POLL_RESULT,
mentions: [],
components: [],
});
if (!message.poll?.results) {
return {};
}
const allAnswerCounts = message.poll.results.answer_counts as unknown as (Omit<PollAnswerCount, "me_voted"> & { voters: string[] })[];
const totalVotes = allAnswerCounts.map((a) => a.voters).length;
const winningAnswerCounts = allAnswerCounts.filter((a) => (a.count * totalVotes) / 100);
const pollResultsMessage = {
type: MessageType.POLL_RESULT,
channel_id: message.channel_id,
author_id: message.author_id,
message_reference: {
type: MessageReferenceType.DEFAULT,
message_id: message.id,
channel_id: message.channel_id,
},
embeds: [
{
type: EmbedType.poll_result,
id: message.id,
fields: [
{
name: "poll_question_text",
value: message.poll.question.text!,
},
{
name: "total_votes",
value: totalVotes.toString(),
},
],
},
],
};
if (winningAnswerCounts) {
const winningAnswer = message.poll.answers.find((a) => a.answer_id === Number(winningAnswerCounts[0]?.id))!;
if (winningAnswerCounts.length === 0) {
pollResultsMessage.embeds[0].fields.push({
name: "victor_answer_votes",
value: "0",
});
} else if (winningAnswerCounts.length === 1) {
pollResultsMessage.embeds[0].fields.push(
{
name: "victor_answer_votes",
value: winningAnswerCounts[0].count.toString(),
},
{
name: "victor_answer_id",
value: winningAnswerCounts[0].id,
},
{
name: "victor_answer_text",
value: winningAnswer.poll_media.text!,
},
);
} else if (winningAnswerCounts.length > 1) {
pollResultsMessage.embeds[0].fields.push({
name: "victor_answer_votes",
value: winningAnswerCounts[0].count.toString(),
});
}
if (winningAnswer?.poll_media.emoji) {
pollResultsMessage.embeds[0].fields.push(
{
name: "victor_answer_emoji_id",
value: winningAnswer.poll_media.emoji.id!.toString()!,
},
{
name: "victor_answer_emoji_name",
value: winningAnswer.poll_media.emoji.name!,
},
{
name: "victor_answer_emoji_animated",
value: `${winningAnswer.poll_media.emoji.animated}`,
},
);
}
}
return pollResultsMessage;
}
export async function addPendingPoll(message: Message, timeoutTime: number) {
pendingPolls.set(message.id, {
timeout: setTimeout(async () => {
const pollResultsMessage = await generatePollResultsMessage(message);
await sendMessage(pollResultsMessage);
pendingPolls.delete(message.id);
}, timeoutTime),
});
}