refactor: improve Feed service

Author: eug-vs

services/feed/feed.class.ts

@@ -1,23 +1,13 @@
-import _ from 'lodash';
 import { Application } from '@feathersjs/express';
 import { Params } from '@feathersjs/feathers';
 import { Poll } from 'which-types';
 
-import { PollSchema } from '../../models/polls/poll.schema';
-import PollModel from '../../models/polls/poll.model';
-
 
 export default class Feed {
   app!: Application;
 
   async find(params: Params): Promise<Poll[]> {
-    return this.app.service('polls')
-      .find(params)
-      .then( // Move new verified polls on top
-        (polls: Poll[]) => _.sortBy(polls, poll => poll.author.verified && !poll.userChoice)
-      ).then( // But all seen posts go down
-        (polls: Poll[]) => _.sortBy(polls, poll => !!poll.userChoice)
-      );
+    return this.app.service('polls').find(params);
   }
 
   setup (app: Application) {

services/feed/feed.hooks.ts

@@ -0,0 +1,28 @@
+import _ from 'lodash';
+import { HookContext } from '@feathersjs/feathers';
+
+
+const raiseNewVerifedPolls = async (context: HookContext): Promise<HookContext> => {
+  // Raise unseen verified polls to the very top
+  context.result = _.sortBy(
+    context.result,
+    poll => !(poll.author.verified && !poll.userChoice)
+  );
+  return context;
+};
+
+const lowerOldPolls = async (context: HookContext): Promise<HookContext> => {
+  // Move all seen polls down
+  context.result = _.sortBy(
+    context.result,
+    poll => !!poll.userChoice
+  );
+  return context;
+};
+
+export default {
+  after: {
+    find: [raiseNewVerifedPolls, lowerOldPolls]
+  }
+};
+

services/feed/feed.service.ts

@@ -1,7 +1,10 @@
 import { Application } from '@feathersjs/express';
 import Feed from './feed.class';
 
+import hooks from './feed.hooks';
+
 export default (app: Application): void => {
   app.use('/feed', new Feed());
+  app.service('feed').hooks(hooks);
 };