habitica/website/server/libs/inbox/index.js

import { mapInboxMessage, inboxModel } from '../../models/message';
import { getUserInfo, sendTxn as sendTxnEmail } from '../email'; // eslint-disable-line import/no-cycle
import { sendNotification as sendPushNotification } from '../pushNotifications';

export async function sentMessage (sender, receiver, message, translate) {
  const fakeSending = sender.flags.chatShadowMuted;
  const messageSent = await sender.sendMessage(receiver, {
    receiverMsg: message,
    fakeSending,
  });
  const senderName = getUserInfo(sender, ['name']).name;

  if (!fakeSending) {
    if (receiver.preferences.emailNotifications.newPM !== false) {
      sendTxnEmail(receiver, 'new-pm', [
        { name: 'SENDER', content: senderName },
      ]);
    }

    if (receiver.preferences.pushNotifications.newPM !== false && messageSent.unformattedText) {
      await sendPushNotification(
        receiver,
        {
          title: translate(
            'newPMNotificationTitle',
            { name: getUserInfo(sender, ['name']).name },
            receiver.preferences.language,
          ),
          message: messageSent.unformattedText,
          identifier: 'newPM',
          category: 'newPM',
          payload: { replyTo: sender._id, senderName, message: messageSent.unformattedText },
        },
      );
    }
  }

  return messageSent;
}
// Paginate per every 50
const PM_PER_PAGE = 50;
const MAX_PM_COUNT = 400;

const getUserInboxDefaultOptions = {
  asArray: true,
  page: undefined,
  conversation: null,
  mapProps: false,
};

export async function getUserInbox (user, optionParams = getUserInboxDefaultOptions) {
  // if not all properties are passed, fill the default values
  const options = { ...getUserInboxDefaultOptions, ...optionParams };

  const findObj = { ownerId: user._id };

  if (options.conversation) {
    findObj.uuid = options.conversation;
  }

  let query = inboxModel
    .find(findObj)
    .sort({ timestamp: -1 });

  if (typeof options.page !== 'undefined') {
    const page = Number(options.page);
    const skip = PM_PER_PAGE * page;
    if (skip >= MAX_PM_COUNT) {
      return options.asArray ? [] : {};
    }
    const remainingAllowed = MAX_PM_COUNT - skip;
    const limit = Math.min(PM_PER_PAGE, remainingAllowed);
    query = query
      .skip(skip)
      .limit(limit);
  } else {
    query = query.limit(MAX_PM_COUNT);
  }

  const messages = (await query.lean().exec()).map(msgObj => {
    delete msgObj.__v;
    if (options.mapProps) {
      mapInboxMessage(msgObj, user);
    }

    return msgObj;
  });

  if (options.asArray) {
    return messages;
  }
  const messagesObj = {};
  messages.forEach(msg => { messagesObj[msg._id] = msg; });

  return messagesObj;
}

export async function applyLikeToMessages (user, uniqueMessages) {
  const bulkWriteOperations = [];

  for (const message of uniqueMessages) {
    if (!message.likes) {
      message.likes = {};
    }

    message.likes[user._id] = !message.likes[user._id];

    bulkWriteOperations.push({
      updateOne: {
        filter: { _id: message._id },
        update: {
          $set: {
            likes: message.likes,
          },
        },
      },
    });
  }

  await inboxModel.bulkWrite(bulkWriteOperations, {});
}

export async function getInboxMessagesByUniqueId (uniqueMessageId) {
  return inboxModel
    .find({ uniqueMessageId })
    // prevents creating the proxies, no .save() and other stuff
    .lean()
    // since there can be only 2 messages maximum for this uniqueMessageId,
    // this might speed up the query
    .limit(2)
    .exec();
}

export async function getUserInboxMessage (user, messageId) {
  return inboxModel.findOne({ ownerId: user._id, _id: messageId }).exec();
}

export async function deleteMessage (user, messageId) {
  const message = await inboxModel.findOne({ _id: messageId, ownerId: user._id }).exec();
  if (!message) return false;
  await inboxModel.deleteOne({ _id: message._id, ownerId: user._id }).exec();

  return true;
}

export async function clearPMs (user) {
  user.inbox.newMessages = 0;

  await Promise.all([
    user.save(),
    inboxModel.deleteMany({ ownerId: user._id }).exec(),
  ]);
}