handle voting status

Author: ManAnRuck

src/graphql/resolvers/Vote.js

@@ -1,18 +1,42 @@
 /* eslint no-underscore-dangle: ["error", { "allow": ["_id"] }] */
 
+const statesVoting = ['Beschlussempfehlung liegt vor'];
+const statesCompleted = [
+  'Zurückgezogen',
+  'Erledigt durch Ablauf der Wahlperiode',
+  'Einbringung abgelehnt',
+  'Für erledigt erklärt',
+  'Bundesrat hat zugestimmt',
+  'Abgelehnt',
+  'Verkündet',
+  'Verabschiedet',
+  'Zusammengeführt mit... (siehe Vorgangsablauf)',
+];
+
 export default {
   Query: {
     votes: (parent, { procedure }, { VoteModel }) =>
       VoteModel.findOne({ procedure }).then(result => result.voteResults),
   },
 
   Mutation: {
-    vote: async (parent, { procedure, selection }, { VoteModel, user }) => {
+    vote: async (
+      parent,
+      { procedure: procedureId, selection },
+      { VoteModel, ProcedureModel, user },
+    ) => {
       // TODO check if procedure is votable
+      const procedure = await ProcedureModel.findById(procedureId);
+      let state;
+      if (statesVoting.some(s => s === procedure.currentStatus)) {
+        state = 'VOTING';
+      } else if (statesCompleted.some(s => s === procedure.currentStatus)) {
+        state = 'COMPLETED';
+      }
+
       let vote = await VoteModel.findOne({ procedure });
       if (!vote) {
-        console.log('### Create new Vote Instance');
-        vote = await VoteModel.create({ procedure });
+        vote = await VoteModel.create({ procedure, state });
       }
       const hasVoted = vote.users.some(uId => uId.equals(user._id));
       if (!hasVoted) {
@@ -31,11 +55,19 @@ export default {
           default:
             break;
         }
-        return VoteModel.findByIdAndUpdate(vote._id, voteUpdate, {
-          new: true,
-        }).then(result => result.voteResults);
+        await VoteModel.findByIdAndUpdate(vote._id, { ...voteUpdate, state });
       }
-      return vote.voteResults;
+      return VoteModel.aggregate([
+        { $match: { procedure: procedure._id } },
+        {
+          $group: {
+            _id: '$procedure',
+            yes: { $sum: '$voteResults.yes' },
+            no: { $sum: '$voteResults.no' },
+            abstination: { $sum: '$voteResults.abstination' },
+          },
+        },
+      ]).then(result => result[0] || { yes: null, no: null, abstination: null });
     },
   },
 };

src/models/Vote.js

@@ -6,14 +6,16 @@ const VoteSchema = new Schema({
     type: Schema.Types.ObjectId,
     ref: 'Procedure',
     required: true,
-    index: { unique: true },
   },
-  users: [{ type: Schema.Types.ObjectId, ref: 'User', required: true }],
+  state: { type: String, enum: ['VOTING', 'COMPLETED'], required: true },
+  users: [{ type: Schema.Types.ObjectId, ref: 'User' }],
   voteResults: {
     yes: { type: Number, default: 0 },
     no: { type: Number, default: 0 },
     abstination: { type: Number, default: 0 },
   },
 });
 
+VoteSchema.index({ procedure: 1, state: 1 }, { unique: true });
+
 export default mongoose.model('Vote', VoteSchema);