add vote process to server

Author: ManAnRuck

.eslintrc.js

@@ -1,4 +1,4 @@
 module.exports = {
-    extends: 'airbnb-base',
-  };
-  
+  extends: "airbnb-base",
+  rules: { "newline-per-chained-call": [2] }
+};

src/graphql/resolvers/Vote.js

@@ -0,0 +1,40 @@
+/* eslint no-underscore-dangle: ["error", { "allow": ["_id"] }] */
+
+export default {
+  Query: {
+    votes: (parent, { procedure }, { VoteModel }) =>
+      VoteModel.findOne({ procedure }).then(result => result.voteResults),
+  },
+
+  Mutation: {
+    vote: async (parent, { procedure, selection }, { VoteModel, user }) => {
+      let vote = await VoteModel.findOne({ procedure });
+      if (!vote) {
+        console.log('### Create new Vote Instance');
+        vote = await VoteModel.create({ procedure });
+      }
+      const hasVoted = vote.users.some(uId => uId.equals(user._id));
+      if (!hasVoted) {
+        const voteUpdate = { $push: { users: user } };
+        switch (selection) {
+          case 'YES':
+            voteUpdate.$inc = { 'voteResults.yes': 1 };
+            break;
+          case 'NO':
+            voteUpdate.$inc = { 'voteResults.no': 1 };
+            break;
+          case 'ABSTINATION':
+            voteUpdate.$inc = { 'voteResults.abstination': 1 };
+            break;
+
+          default:
+            break;
+        }
+        return VoteModel.findByIdAndUpdate(vote._id, voteUpdate, {
+          new: true,
+        }).then(result => result.voteResults);
+      }
+      return vote.voteResults;
+    },
+  },
+};

src/graphql/schemas/Procedure.js

@@ -5,13 +5,6 @@ enum ProcedureType {
   HOT
 }
 
-type VoteResult {
-  yes: Int
-  no: Int
-  abstination: Int
-  notVote: Int
-}
-
 type Procedure {
   _id: ID!
   title: String!

src/graphql/schemas/Vote.js

@@ -0,0 +1,29 @@
+export default `
+
+  enum VoteSelection {
+    YES
+    NO
+    ABSTINATION
+  }
+
+  type VoteResult {
+    yes: Int
+    no: Int
+    abstination: Int
+    notVote: Int
+  }
+
+  type Vote {
+    _id: ID!
+    voted: Boolean
+    voteResults: VoteResult
+  }
+  
+  type Mutation {
+    vote(procedure: ID!, selection: VoteSelection!): VoteResult
+  }
+  
+  type Query {
+    votes(procedure: ID!): VoteResult
+  }
+  `;

src/index.js

@@ -21,6 +21,7 @@ import auth from './express/auth';
 import ProcedureModel from './models/Procedure';
 import UserModel from './models/User';
 import ActivityModel from './models/Activity';
+import VoteModel from './models/Vote';
 
 const app = express();
 
@@ -60,6 +61,7 @@ app.use(constants.GRAPHQL_PATH, (req, res, next) => {
       ProcedureModel,
       UserModel,
       ActivityModel,
+      VoteModel,
     },
     tracing: true,
     cacheControl: true,

src/models/Vote.js

@@ -2,7 +2,12 @@
 import mongoose, { Schema } from 'mongoose';
 
 const VoteSchema = new Schema({
-  procedure: { type: Schema.Types.ObjectId, ref: 'Procedure', required: true },
+  procedure: {
+    type: Schema.Types.ObjectId,
+    ref: 'Procedure',
+    required: true,
+    index: { unique: true },
+  },
   users: [{ type: Schema.Types.ObjectId, ref: 'User', required: true }],
   voteResults: {
     yes: { type: Number, default: 0 },