feat: add voting route

2026-07-04 15:21:47 +00:00 · 2026-06-24 23:46:23 +02:00
parent 305d1d87b2
commit 0efdb5c35a
5 changed files with 78 additions and 0 deletions
@@ -0,0 +1,72 @@
+/*
+  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 { 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";
+
+const router: Router = Router({ mergeParams: true });
+
+router.put("/", route({ requestBody: "PollUserAnswersSchema", permission: "VIEW_CHANNEL" }), async (req: Request, res: Response) => {
+    const payload = req.body as PollUserAnswersSchema;
+    const { poll_id } = req.params as { [key: string]: string };
+
+    const message = await Message.findOne({ where: { id: poll_id } });
+
+    if (!message || !message.poll || !message.poll.results) {
+        throw DiscordApiErrors.UNKNOWN_MESSAGE;
+    }
+
+    if (new Date() > new Date(message.poll.expiry)) {
+        throw DiscordApiErrors.POLL_EXPIRED;
+    }
+
+    if (!message.poll.allow_multiselect && payload.answer_ids.length > 1) {
+        const errors: ErrorList = {};
+        errors["answer_ids"] = makeObjectErrorContent("CANNOT_ADD_MULTIPLE_POLL_ANSWERS", "Multiple votes are not allowed for this poll.");
+        throw new FieldError(50035, "Invalid form body", errors);
+    }
+
+    const allAnswerCounts = message.poll.results.answer_counts as unknown as (Omit<PollAnswerCount, "me_voted"> & { voters: string[] })[];
+
+    for (const answer_id of payload.answer_ids) {
+        let answerCount = allAnswerCounts.find((a) => a.id === answer_id);
+
+        if (!answerCount) {
+            allAnswerCounts.push({ id: answer_id, count: 0, voters: [] });
+            answerCount = allAnswerCounts.find((a) => a.id === answer_id)!;
+        }
+
+        if (!answerCount.voters.includes(req.user_id)) {
+            answerCount.voters.push(req.user_id);
+            answerCount.count = answerCount.voters.length;
+        }
+    }
+
+    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 message.save();
+    res.send();
+});
+
+export default router;