Merge pull request #8 from which-ecosystem/votes

Votes

Author: eug-vs

app.ts

@@ -1,7 +1,7 @@
 import feathers from '@feathersjs/feathers';
 import express from '@feathersjs/express';
 import socketio from '@feathersjs/socketio';
-import configuration from '@feathersjs/configuration'
+import configuration from '@feathersjs/configuration';
 import '@feathersjs/transport-commons';
 import cors from 'cors';
 

config/default.json

@@ -8,7 +8,7 @@
       "local"
     ],
     "local": {
-      "usernameField": "name",
+      "usernameField": "username",
       "passwordField": "password"
     }
   }

hooks/convertPoll.ts

@@ -0,0 +1,43 @@
+import { HookContext } from '@feathersjs/feathers';
+import { Types } from 'mongoose';
+import bluebird from 'bluebird';
+import _ from 'lodash';
+import { Poll, User, Vote } from 'which-types';
+
+import { PollSchema } from '../models/polls/poll.schema';
+import VoteModel from '../models/votes/vote.model';
+
+
+export default async (context: HookContext): Promise<HookContext> => {
+  const { app, result, params: { user } } = context;
+
+  const convert = async (poll: PollSchema): Promise<Poll | null> => {
+    const author = await app.service('users').get(poll.authorId);
+
+    const contents = await VoteModel.aggregate([
+      { $match: { pollId: Types.ObjectId(poll._id) } },
+      { $group: { _id: '$which', total: { $sum: 1 } } }
+    ]).then(groups => groups.reduce(
+      (acc, group) =>  _.set(acc, group._id + '.votes', group.total),
+      { left: { votes: 0 }, right: { votes: 0 } }
+    ));
+
+    const userChoice = await VoteModel.findOne(
+      { pollId: poll._id, userId: user?._id }
+    ).then(vote => vote?.which);
+
+    return _.merge(
+      _.omit(poll, ['authorId']),
+      { author, contents, userChoice }
+    );
+  };
+
+  if (Array.isArray(result)) {
+    const polls = await bluebird.map(result, (poll: PollSchema) => convert(poll));
+    context.result = _.compact(polls);
+  } else {
+    context.result = await convert(result);
+  }
+  return context;
+};
+

hooks/expandAuthor.ts

@@ -0,0 +1,7 @@
+import { HookContext } from '@feathersjs/feathers';
+
+export default async (context: HookContext): Promise<HookContext> => {
+  if (!context.params.user) throw new Error('This endpoint requires auth!');
+  return context;
+};
+

hooks/requireAuth.ts

@@ -0,0 +1,8 @@
+import { HookContext } from '@feathersjs/feathers';
+import { authenticate } from '@feathersjs/authentication';
+
+
+export default async (context: HookContext): Promise<HookContext> => {
+  return authenticate('jwt')(context).catch(() => context);
+};
+

hooks/tryAuthenticate.ts

@@ -1,5 +1,20 @@
 import { Model, model } from 'mongoose';
 import { PollSchema, pollSchema } from './poll.schema';
+import { Types } from 'mongoose';
+import { Which } from 'which-types';
+
+pollSchema.methods.vote = function(userId: string, which: Which): PollSchema {
+  const participants: Types.ObjectId[] = ['left', 'right'].reduce((acc, option) => {
+    const { votes } = this.contents[option];
+    return acc.concat(votes);
+  }, []);
+
+  if (!participants.some(user => user.equals(userId))) {
+    this.contents[which].votes.push(userId);
+  }
+
+  return this.save();
+}
 
 export default model<PollSchema, Model<PollSchema>>('Poll', pollSchema);
 

models/polls/poll.model.ts

@@ -1,27 +1,21 @@
 import { Document, Schema, Types } from 'mongoose';
-import { User } from '../users/user.schema';
 
-export interface ImageData {
+export interface ImageDataSchema {
   url: string;
-  votes: number;
 }
 
-export interface Poll {
-  author: User;
+export interface PollSchema extends Document {
   contents: {
-    left: ImageData;
-    right: ImageData;
+    left: ImageDataSchema;
+    right: ImageDataSchema;
   };
-}
-
-export interface PollSchema extends Document, Omit<Poll, 'author'> {
+  createdAt: Date;
   authorId: string;
+  vote: (userId: string, which: 'left' | 'right') => PollSchema;
 }
 
-
-const imageDataSchema = {
+export const imageDataSchema = {
   url: String,
-  votes: Number
 };
 
 export const pollSchema = new Schema({
@@ -33,5 +27,5 @@ export const pollSchema = new Schema({
     type: Types.ObjectId,
     ref: 'User'
   }
-});
+}, { timestamps: true });
 

models/polls/poll.schema.ts

@@ -1,24 +1,21 @@
 import { Document, Schema } from 'mongoose';
+import { User } from 'which-types';
 
-export interface User {
-  name: string;
-  avatarUrl?: string;
-  age?: number;
-}
-
-export interface UserSchema extends Document, User {
+export interface UserSchema extends Document, Omit<User, '_id'> {
   password: string;
 }
 
 export const userSchema = new Schema({
-  name: String,
+  username: String,
   password: String,
+  email: String,
   avatarUrl: {
     type: String,
     required: false
   },
   age: {
-    type: Number
+    type: Number,
+    required: false
   }
-});
+}, { timestamps: true });
 

models/users/user.schema.ts

@@ -0,0 +1,7 @@
+import { Model, model } from 'mongoose';
+import { VoteSchema, voteSchema } from './vote.schema';
+
+voteSchema.index({ pollId: 1, userId: 1 }, { unique: true }); // Unique together
+
+export default model<VoteSchema, Model<VoteSchema>>('Vote', voteSchema);
+