talk/graph/loaders/comments.js

const DataLoader = require('dataloader');

const util = require('./util');

const ActionModel = require('../../models/action');
const CommentModel = require('../../models/comment');
const CommentsService = require('../../services/comments');

/**
 * Retrieves comments by an array of asset id's.
 * @param {Array} ids array of ids to lookup
 */
const genCommentsByAssetID = (context, ids) => CommentModel.find({
  asset_id: {
    $in: ids
  },
  parent_id: null,
  status: {
    $in: [null, 'ACCEPTED']
  }
}).then(util.arrayJoinBy(ids, 'asset_id'));

/**
 * Retrieves comments by an array of parent ids.
 * @param {Array} ids array of ids to lookup
 */
const genCommentsByParentID = (context, ids) => CommentModel.find({
  parent_id: {
    $in: ids
  },
  status: {
    $in: [null, 'ACCEPTED']
  }
}).then(util.arrayJoinBy(ids, 'parent_id'));

const getCommentsByStatusAndAssetID = (context, {status = null, asset_id = null}) => {

  // TODO: remove when we move the enum over to the uppercase.
  if (status) {
    status = status.toLowerCase();
  }

  return CommentsService.moderationQueue(status, asset_id);
};

const getCommentsByActionTypeAndAssetID = (context, {action_type, asset_id = null}) => {
  return ActionModel.find({
    action_type,
    item_type: 'COMMENTS'
  }).then((actions) => {
    let comments = CommentModel.find({
      id: {
        $in: actions.map((action) => action.item_id)
      }
    });

    if (asset_id) {
      comments = comments.where({asset_id});
    }

    return comments;
  });
};

const genCommentsByAuthorID = (context, authorIDs) => CommentModel.find({
  author_id: {
    $in: authorIDs
  }
}).then(util.arrayJoinBy(authorIDs, 'author_id'));

/**
 * Creates a set of loaders based on a GraphQL context.
 * @param  {Object} context the context of the GraphQL request
 * @return {Object}         object of loaders
 */
module.exports = (context) => ({
  Comments: {
    getByParentID: new DataLoader((ids) => genCommentsByParentID(context, ids)),
    getByAssetID: new DataLoader((ids) => genCommentsByAssetID(context, ids)),
    getByStatusAndAssetID: (query) => getCommentsByStatusAndAssetID(context, query),
    getByActionTypeAndAssetID: (query) => getCommentsByActionTypeAndAssetID(context, query),
    getByAuthorID: new DataLoader((authorIDs) => genCommentsByAuthorID(context, authorIDs))
  }
});