From 2b78c17304aab898fd860d15ff452593da78ea0b Mon Sep 17 00:00:00 2001
From: Gabriel Tofvesson <contact@w1zzrd.dev>
Date: Sat, 22 Oct 2022 00:39:39 +0200
Subject: [PATCH] Implemented voting

---
 functions/src/index.ts  | 110 +++++++++++++++++++++++++++++++++++++---
 functions/tsconfig.json |   1 +
 2 files changed, 104 insertions(+), 7 deletions(-)

diff --git a/functions/src/index.ts b/functions/src/index.ts
index 557c347..4f6095d 100644
--- a/functions/src/index.ts
+++ b/functions/src/index.ts
@@ -1,6 +1,6 @@
 import * as functions from "firebase-functions";
-import { initializeApp, applicationDefault, cert } from "firebase-admin/app";
-import { getFirebase } from "firebase-admin/firestore";
+import { initializeApp } from "firebase-admin/app";
+import { DocumentReference, DocumentSnapshot, getFirestore, QuerySnapshot } from "firebase-admin/firestore";
 import express, { Request, Response } from "express";
 
 type VoteEntry = {
@@ -10,20 +10,116 @@ type VoteEntry = {
 
 type Vote = {
   prompt: string
-  entries: VoteEntry[]
+  options: number
 }
 
-
 initializeApp();
 
 
-const db = getFirebase();
+const db = getFirestore();
 const app = express();
 
-app.post("/vote", (req: Request, res: Response) => {
-  
+const getVote = async (id: string) =>
+  db.collection("votes").doc(id).get() as Promise<DocumentSnapshot<Vote>>
+const getVoteEntry = async (vote: DocumentReference<Vote>, username: string) =>
+  vote.collection("entries").where("username", "==", username).get() as Promise<QuerySnapshot<VoteEntry>>
+const makeVoteEntry = async (vote: DocumentReference<Vote>, username: string, voteIndex: number) =>
+  vote.collection("entries").add({ username, voteIndex }) as Promise<DocumentReference<VoteEntry>>
+
+app.post("/", async (req: Request, res: Response) => {
+  const voteId = req.body.voteId as string;
+  const voter = req.body.voter as string;
+  const voteIndex = parseInt(req.body.voteIndex as string);
+
+  if (!voteId) {
+    res.status(400).send("Missing voteId");
+    return;
+  }
+
+  if (!voter) {
+    res.status(400).send("Missing voter");
+    return;
+  }
+
+  const vote = await getVote(voteId);
+  if (!vote.exists) {
+    res.status(404).json({ error: "Vote not found" });
+    return;
+  }
+
+  if (Number.isNaN(voteIndex) || voteIndex < 0 || voteIndex >= vote.data()!.options) {
+    res.status(400).json({ error: "Invalid vote index" });
+    return;
+  }
+
+  const entry = await getVoteEntry(vote.ref, voter);
+  if (entry.empty) {
+    await makeVoteEntry(vote.ref, voter, voteIndex);
+    res.json({ success: true });
+    return;
+  }
+
+  entry.docs[0].ref.update({ voteIndex });
+  res.json({ success: true });
 });
 
+app.get("/", async (req: Request, res: Response) => {
+  const voteId = req.body.voteId as string;
+  if (!voteId) {
+    res.status(400).send("Missing voteId");
+    return;
+  }
+
+  const vote = await getVote(voteId);
+  if (!vote.exists) {
+    res.status(404).json({ error: "Vote not found" });
+    return;
+  }
+
+  res.json(vote.data());
+});
+
+app.get("/entries", async (req: Request, res: Response) => {
+  const voteId = req.body.voteId as string;
+  if (!voteId) {
+    res.status(400).send("Missing voteId");
+    return;
+  }
+
+  const vote = await getVote(voteId);
+  if (!vote.exists) {
+    res.status(404).json({ error: "Vote not found" });
+    return;
+  }
+
+  const entries = await vote.ref.collection("entries").get();
+  res.json(entries.docs.map(d => d.data()));
+});
+
+app.get("/count", async (req: Request, res: Response) => {
+  const voteId = req.body.voteId as string;
+
+  if (!voteId) {
+    res.status(400).send("Missing voteId");
+    return;
+  }
+
+  const vote = await getVote(voteId);
+  if (!vote.exists) {
+    res.status(404).json({ error: "Vote not found" });
+    return;
+  }
+
+  const collect = new Array<Promise<QuerySnapshot<VoteEntry>>>();
+  for (let i = 0; i < vote.data()!.options; i++) {
+    collect.push(vote.ref.collection("entries").where("voteIndex", "==", i).get() as Promise<QuerySnapshot<VoteEntry>>)
+  }
+
+  res.json((await Promise.all(collect)).map((query) => query.size));
+});
+
+export const vote = functions.https.onRequest(app)
+
 // // Start writing Firebase Functions
 // // https://firebase.google.com/docs/functions/typescript
 //
diff --git a/functions/tsconfig.json b/functions/tsconfig.json
index 7ce05d0..95a973b 100644
--- a/functions/tsconfig.json
+++ b/functions/tsconfig.json
@@ -3,6 +3,7 @@
     "module": "commonjs",
     "noImplicitReturns": true,
     "noUnusedLocals": true,
+    "esModuleInterop": true,
     "outDir": "lib",
     "sourceMap": true,
     "strict": true,