Class MessageClass

api/server/models/Message.ts:147–528

}

class MessageClass extends mongoose.Model {
  public static async getList({ userId, chatId, batchNumberForMessages, limit }) {
    await this.checkPermission({ userId, chatId });

    const filter = { chatId, parentMessageId: null };

    const messages = await this.find(filter)
      .sort({ createdAt: -1, _id: 1 })
      .skip((batchNumberForMessages - 1) * limit)
      .limit(limit)
      .setOptions({ lean: true });

    // use for...of instead of forEach
    // forEach does not care about promises from await
    for (const message of messages) {
      const threadMessages = await this.find({
        parentMessageId: message._id.toString(),
      });

      message.countOfThreadMessages = threadMessages.length;
    }

    return messages;
  }

  public static async getListForThread({ userId, chatId, messageId }) {
    await this.checkPermission({ userId, chatId });

    const filter = { chatId, parentMessageId: messageId };

    return this.find(filter).sort({ createdAt: 1, _id: 1 }).setOptions({ lean: true });
  }

  // done: add parentMessageId
  public static async addOrEdit({ content, chatId, teamId, userId, id, files, parentMessageId }) {
    if (!content || !teamId) {
      throw new Error('Bad data');
    }

    const { isSubscriptionActiveForAccount, isTrialPeriodOverForAccount } =
      await User.getSubscriptionStatus(userId);

    if (isTrialPeriodOverForAccount && !isSubscriptionActiveForAccount) {
      throw new Error('This team is not subscribed to a paid plan and free trial period is over.');
    }

    await Chat.updateOne({ _id: chatId }, { lastUpdatedAt: new Date() });

    if (id) {
      const editedMessage = await this.edit({
        content,
        chatId,
        userId,
        id,
      });

      // no realtime update for editing message
      return { message: editedMessage, userIdsToNotify: [] };