From e99b51895afd532a529744396ecae87d47c68503 Mon Sep 17 00:00:00 2001
From: eug-vs <eug-vs@keemail.me>
Date: Sun, 21 Jun 2020 14:29:59 +0300
Subject: feat: change hooks and install which-types

---
 hooks/convertPoll.ts | 45 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 hooks/convertPoll.ts

(limited to 'hooks/convertPoll.ts')

diff --git a/hooks/convertPoll.ts b/hooks/convertPoll.ts
new file mode 100644
index 0000000..62ddea1
--- /dev/null
+++ b/hooks/convertPoll.ts
@@ -0,0 +1,45 @@
+import { HookContext } from '@feathersjs/feathers';
+import bluebird from 'bluebird';
+import _ from 'lodash';
+import { Poll, User } from 'which-types';
+
+import { PollSchema } from '../models/polls/poll.schema';
+import UserModel from '../models/users/user.model';
+
+const convertPoll = 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,
+          contents: {
+            left: {
+              votes: poll.contents.left.votes.length
+            },
+            right: {
+              votes: poll.contents.right.votes.length
+            }
+          }
+        }
+      );
+    })
+    .catch(err => {
+      console.error(err);
+      return err;
+    });
+};
+
+export const convertPollHook = async (context: HookContext): Promise<HookContext> => {
+  context.result = await convertPoll(context.result);
+  return context;
+};
+
+export const convertPollManyHook = async (context: HookContext): Promise<HookContext> => {
+  const polls = await bluebird.map(context.result, (poll: PollSchema) => convertPoll(poll));
+  context.result = _.compact(polls);
+  return context;
+};
+
-- 
cgit v1.2.3