diff --git a/src/api/Server.ts b/src/api/Server.ts
index adc2429b..c22fcf2b 100644
--- a/src/api/Server.ts
+++ b/src/api/Server.ts
@@ -21,14 +21,15 @@ import { Request, Response, Router } from "express";
import morgan from "morgan";
import { Server, ServerOptions } from "lambert-server/Server";
import { red } from "picocolors";
-import { getDatabase, initDatabase } from "@spacebar/database";
-import { Config, ConnectionConfig, ConnectionLoader, Email, JSONReplacer, WebAuthn, initEvent, registerRoutes, getRevInfoOrFail } from "@spacebar/util";
+import { getDatabase, initDatabase, Message } from "@spacebar/database";
+import { Config, ConnectionConfig, ConnectionLoader, Email, JSONReplacer, WebAuthn, initEvent, registerRoutes, getRevInfoOrFail, pendingPolls } from "@spacebar/util";
import { ProcessLifecycle } from "../util/util/ProcessLifecycle";
import { Monitoring } from "../util/monitoring/Monitoring";
import { BcryptWorkerPool } from "../util/util/workers/bcrypt/BcryptWorkerPool";
import { Authentication, CORS, ImageProxy, BodyParser, ErrorHandler, initRateLimits, initTranslation } from "./middlewares";
import { initInstance } from "./util/handlers/Instance";
-import { route } from "./util";
+import { route, sendMessage } from "./util";
+import { EmbedType, MessageReferenceType, MessageType, PollAnswerCount } from "#schemas";
const ASSETS_FOLDER = path.join(__dirname, "..", "..", "assets");
const PUBLIC_ASSETS_FOLDER = path.join(ASSETS_FOLDER, "public");
@@ -189,6 +190,110 @@ export class SpacebarServer extends Server {
});
});
+ // Pickup non-expired polls
+ const nonExpiredPolls = await Message.createQueryBuilder("message").where("message.poll->>'expiry' > :now", { now: new Date().toISOString() }).getMany();
+ console.log(nonExpiredPolls);
+
+ for (const message of nonExpiredPolls) {
+ if (!message.poll) {
+ return;
+ }
+
+ pendingPolls.set(message.id, {
+ timeout: 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);
+ },
+ new Date(message.poll.expiry).getTime() - Date.now(),
+ ),
+ });
+ }
+
function isReady(req: Request, res: Response) {
if (!getDatabase()) return res.sendStatus(503);
return res.sendStatus(200);
|
