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, results are returned in reverse
 * chronological order.
 * @param {Array} ids array of ids to lookup
 */
const genCommentsByAssetID = (context, ids) => {
  return CommentModel.find({
    asset_id: {
      $in: ids
    },
    parent_id: null,
    status: {
      $in: [null, 'ACCEPTED']
    }
  })
  .sort({created_at: -1})
  .then(util.arrayJoinBy(ids, 'asset_id'));
};

/**
 * Retrieves comments by an array of parent ids, results are returned in
 * chronological order.
 * @param {Array} ids array of ids to lookup
 */
const genCommentsByParentID = (context, ids) => {
  return CommentModel.find({
    parent_id: {
      $in: ids
    },
    status: {
      $in: [null, 'ACCEPTED']
    }
  })
  .sort({created_at: 1})
  .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)
      }
    }).sort({created_at: 1});

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

    return comments;
  });
};

const genCommentsByAuthorID = (context, authorIDs) => {
  return CommentModel.find({
    author_id: {
      $in: authorIDs
    }
  })
  .sort({created_at: -1})
  .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))
  }
});