Merge pull request #3 from eug-vs/hooks

Expand authorId field in polls

Author: eug-vs

app.ts

@@ -2,6 +2,7 @@ import feathers from '@feathersjs/feathers';
 import express from '@feathersjs/express';
 import socketio from '@feathersjs/socketio';
 import '@feathersjs/transport-commons';
+import cors from 'cors';
 
 import services from './services';
 
@@ -11,9 +12,11 @@ const app = express(feathers());
 app.use(express.json());
 app.use(express.urlencoded({ extended: true }));
 app.use(express.static(__dirname));
+app.use(express.errorHandler());
+app.use(cors());
+
 app.configure(express.rest());
 app.configure(socketio());
-app.use(express.errorHandler());
 app.configure(services);
 
 

index.ts

@@ -1,8 +1,8 @@
 import app from './app';
 import mongoose from 'mongoose';
+import Promise from 'bluebird';
 
-
-mongoose.Promise = global.Promise;
+mongoose.Promise = Promise;
 
 mongoose.connect('mongodb://localhost:27017/which', { useNewUrlParser: true });
 

models/polls/poll.model.ts

@@ -1,5 +1,5 @@
 import { Model, model } from "mongoose"
-import { Poll, PollSchema } from './poll.schema';
+import { PollSchema, pollSchema } from './poll.schema';
 
-export default model<Poll, Model<Poll>>("Poll", PollSchema);
+export default model<PollSchema, Model<PollSchema>>("Poll", pollSchema);
 

models/polls/poll.schema.ts

@@ -1,26 +1,30 @@
 import { Document, Schema, Types } from 'mongoose';
-import { User } from '../users/user.schema';
+import { User } from '../users/user.schema'
 
 interface ImageData {
   url: string;
   votes: number;
 }
 
-export interface Poll extends Document {
-  authorId: string;
+export interface Poll {
+  author: User;
   contents: {
     left: ImageData;
     right: ImageData;
   };
 }
 
+export interface PollSchema extends Document, Omit<Poll, 'author'> {
+  authorId: string;
+}
+
 
 const imageDataSchema = {
   url: String,
   votes: Number
 }
 
-export const PollSchema = new Schema({
+export const pollSchema = new Schema({
   contents: {
     left: imageDataSchema,
     right: imageDataSchema

models/users/user.model.ts

@@ -1,5 +1,5 @@
 import { Model, model } from "mongoose"
-import { User, UserSchema } from './user.schema';
+import { UserSchema, userSchema } from './user.schema';
 
-export default model<User, Model<User>>("User", UserSchema);
+export default model<UserSchema, Model<UserSchema>>("User", userSchema);
 

models/users/user.schema.ts

@@ -1,12 +1,14 @@
 import { Document, Schema } from "mongoose"
 
-export interface User extends Document {
+export interface User {
   name: string;
   avatarUrl?: string;
   age?: number;
 }
 
-export const UserSchema = new Schema({
+export interface UserSchema extends Document, User {}
+
+export const userSchema = new Schema({
   name: String,
   avatarUrl: {
     type: String,

package-lock.json

@@ -1,7 +1,7 @@
 {
-  "name": "@oneflow/which-api",
+  "name": "which-api",
   "version": "1.0.0",
-  "main": "index.js",
+  "main": "index.ts",
   "scripts": {
     "test": "echo \"Error: no test specified\" && exit 1"
   },
@@ -13,8 +13,9 @@
     "@feathersjs/feathers": "^4.5.3",
     "@feathersjs/socketio": "^4.5.4",
     "@feathersjs/transport-commons": "^4.5.3",
-    "@types/mongoose": "^5.7.23",
+    "cors": "^2.8.5",
     "feathers-mongoose": "^8.3.0",
+    "lodash": "^4.17.15",
     "mongoose": "^5.9.18"
   },
   "repository": {
@@ -25,5 +26,11 @@
     "url": "https://github.com/eug-vs/which-api/issues"
   },
   "homepage": "https://github.com/eug-vs/which-api#readme",
-  "description": ""
+  "description": "",
+  "devDependencies": {
+    "@types/bluebird": "^3.5.32",
+    "@types/cors": "^2.8.6",
+    "@types/lodash": "^4.14.155",
+    "@types/mongoose": "^5.7.23"
+  }
 }

package.json

@@ -0,0 +1,40 @@
+import { HookContext } from '@feathersjs/feathers';
+import bluebird from 'bluebird';
+import _ from 'lodash';
+
+import { Poll, PollSchema } from '../../models/polls/poll.schema';
+import { User } from '../../models/users/user.schema';
+import UserModel from '../../models/users/user.model';
+
+
+const expandAuthor = async (poll: PollSchema): Promise<Poll | null> => {
+  return UserModel.findById(poll.authorId)
+    .lean<User>()
+    .exec()
+    .then((author: User | null): Poll | null => {
+      return author && _.merge(_.omit(poll, 'authorId'), { author });
+    })
+    .catch(err => {
+      console.error(err);
+      return err;
+    });
+};
+
+const expandAuthorHook = async (context: HookContext): Promise<HookContext> => {
+  context.result = await expandAuthor(context.result);
+  return context;
+};
+
+const expandAuthorManyHook = async (context: HookContext): Promise<HookContext> => {
+  const polls = await bluebird.map(context.result, (poll: any) => expandAuthor(poll));
+  context.result = _.compact(polls);
+  return context;
+};
+
+
+export default {
+  after: {
+    get: [expandAuthorHook],
+    find: [expandAuthorManyHook]
+  }
+}

services/polls/polls.hooks.ts

@@ -1,10 +1,12 @@
 import { Application } from '@feathersjs/express';
 import Model from '../../models/polls/poll.model';
 import service from 'feathers-mongoose';
+import hooks from './polls.hooks'
 
 const PollService = service({ Model });
 
 export default (app: Application): void => {
   app.use('/polls', PollService);
+  app.service('polls').hooks(hooks);
 };