From 709416b125f7099b34686ea8c1d35cf7328a19e8 Mon Sep 17 00:00:00 2001
From: eug-vs <eug-vs@keemail.me>
Date: Wed, 10 Jun 2020 18:43:09 +0300
Subject: feat: add expandAuthor hook to polls

---
 services/polls/polls.hooks.ts | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 services/polls/polls.hooks.ts

(limited to 'services/polls/polls.hooks.ts')

diff --git a/services/polls/polls.hooks.ts b/services/polls/polls.hooks.ts
new file mode 100644
index 0000000..ff6a83f
--- /dev/null
+++ b/services/polls/polls.hooks.ts
@@ -0,0 +1,38 @@
+import { HookContext } from '@feathersjs/feathers';
+import bluebird from 'bluebird';
+import _ from 'lodash';
+
+import { PollSchema } from '../../models/polls/poll.schema';
+import { UserSchema } from '../../models/users/user.schema';
+import UserModel from '../../models/users/user.model';
+
+
+interface Poll extends Omit<PollSchema, 'authorId'> {
+  author: UserSchema;
+}
+
+const expandAuthor = async (poll: PollSchema): Promise<Poll | null> => {
+  return UserModel.findById(poll.authorId).then((author: UserSchema | null): Poll | null => {
+    if (author) return _.merge(_.omit(poll, 'authorId'), { author });
+    return null;
+  });
+};
+
+const expandAuthorHook = async (context: HookContext): Promise<HookContext> => {
+  context.result = await expandAuthor(context.result);
+  return context;
+};
+
+const expandAuthorManyHook = async (context: HookContext): Promise<HookContext> => {
+  context.result = await bluebird.map(context.result, (poll: any) => expandAuthor(poll));
+  console.log(context.result);
+  return context;
+};
+
+
+export default {
+  after: {
+    get: [expandAuthorHook],
+    find: [expandAuthorManyHook]
+  }
+}
-- 
cgit v1.2.3