routes and model fixes

SaranshDawra · Dec 13, 2020 · 2d21f94 · 2d21f94
1 parent 5a57062
commit 2d21f94
Show file tree

Hide file tree

Showing 5 changed files with 75 additions and 34 deletions.
diff --git a/models/poll.js b/models/poll.js
@@ -1,40 +1,59 @@
 const { ObjectId } = require("bson");
 const mongoose = require("mongoose");
 
-const pollSchema = new mongoose.Schema({
-    question: {
-        type: String,
-        required: true,
-    },
-    optionA: {
-        type: String,
-        required: true,
-    },
-    votesA: {
-        type: Array,
-        required: true,
-    },
-    optionB: {
-        type: String,
-        required: true,
-    },
-    votesB: {
-        type: Array,
-        required: true,
-    },
-    userId: {
-        type: ObjectId,
-        required: true,
-    },
-    votes: {
-        type: Array,
-        required: true,
-    },
-    isHalted: {
-        type: Boolean,
-        default: false,
-    },
-});
+const pollSchema = new mongoose.Schema(
+    {
+        question: {
+            type: String,
+            required: true,
+        },
+        optionA: {
+            type: String,
+            required: true,
+        },
+        votesA: {
+            type: Array,
+            required: true,
+        },
+        optionB: {
+            type: String,
+            required: true,
+        },
+        votesB: {
+            type: Array,
+            required: true,
+        },
+        optionC: {
+            type: String,
+            required: true,
+        },
+        votesC: {
+            type: Array,
+            required: true,
+        },
+        optionD: {
+            type: String,
+            required: true,
+        },
+        votesD: {
+            type: Array,
+            required: true,
+        },
+        userId: {
+            type: ObjectId,
+            required: true,
+        },
+        votes: {
+            type: Array,
+            required: true,
+        },
+        isHalted: {
+            type: Boolean,
+            default: false,
+        },
+    },
+    { timestamps: true }
+);
 
 const Poll = mongoose.model("Poll", pollSchema);
 

diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -25,6 +25,7 @@
     "bcryptjs": "^2.4.3",
     "chart.js": "^2.9.4",
     "concurrently": "^5.3.0",
+    "cors": "^2.8.5",
     "dotenv": "^8.2.0",
     "express": "^4.17.1",
     "express-validator": "^6.7.0",

diff --git a/routes/polls.js b/routes/polls.js
@@ -36,6 +36,8 @@ router.post("/", auth, async (req, res) => {
         question: req.body.question,
         optionA: req.body.optionA,
         optionB: req.body.optionB,
+        optionC: req.body.optionC,
+        optionD: req.body.optionD,
         votes: req.body.votes,
         userId: req.body.userId,
     });
@@ -70,6 +72,14 @@ router.put("/:qid", auth, async (req, res) => {
             poll.votesB.push(req.body.userId);
         }
 
+        if(req.body.option === 2) {
+            poll.votesC.push(req.body.userId);
+        }
+
+        if(req.body.option === 3) {
+            poll.votesD.push(req.body.userId);
+        }
+
         poll = await poll.save();
 
         return res.send(poll);

diff --git a/server.js b/server.js
@@ -1,12 +1,14 @@
 const express = require("express");
 const mongoose = require("mongoose");
+const cors = require("cors");
 const path = require("path");
 const users = require("./routes/users");
 const polls = require("./routes/polls");
 require("dotenv").config();
 const app = express();
 
 app.use(express.json());
+app.use(cors());
 
 const URL = process.env.MONGODB_URI;