overleaf/services/chat/app/js/Features/Messages/MessageManager.js

let MessageManager
const { db, ObjectId } = require('../../mongodb')
const metrics = require('@overleaf/metrics')
const logger = require('@overleaf/logger')

function createMessage(roomId, userId, content, timestamp, callback) {
  let newMessageOpts = {
    content,
    room_id: roomId,
    user_id: userId,
    timestamp,
  }
  newMessageOpts = _ensureIdsAreObjectIds(newMessageOpts)
  db.messages.insertOne(newMessageOpts, function (error, confirmation) {
    if (error) {
      return callback(error)
    }
    newMessageOpts._id = confirmation.insertedId
    callback(null, newMessageOpts)
  })
}

function getMessages(roomId, limit, before, callback) {
  let query = { room_id: roomId }
  if (before) {
    query.timestamp = { $lt: before }
  }
  query = _ensureIdsAreObjectIds(query)
  db.messages.find(query).sort({ timestamp: -1 }).limit(limit).toArray(callback)
}

function findAllMessagesInRooms(roomIds, callback) {
  db.messages
    .find({
      room_id: { $in: roomIds },
    })
    .toArray(callback)
}

function deleteAllMessagesInRoom(roomId, callback) {
  db.messages.deleteMany(
    {
      room_id: roomId,
    },
    callback
  )
}

function updateMessage(roomId, messageId, content, timestamp, callback) {
  const query = _ensureIdsAreObjectIds({
    _id: messageId,
    room_id: roomId,
  })
  db.messages.updateOne(
    query,
    {
      $set: {
        content,
        edited_at: timestamp,
      },
    },
    callback
  )
}

function deleteMessage(roomId, messageId, callback) {
  const query = _ensureIdsAreObjectIds({
    _id: messageId,
    room_id: roomId,
  })
  db.messages.deleteOne(query, callback)
}

function _ensureIdsAreObjectIds(query) {
  if (query.user_id && !(query.user_id instanceof ObjectId)) {
    query.user_id = ObjectId(query.user_id)
  }
  if (query.room_id && !(query.room_id instanceof ObjectId)) {
    query.room_id = ObjectId(query.room_id)
  }
  if (query._id && !(query._id instanceof ObjectId)) {
    query._id = ObjectId(query._id)
  }
  return query
}

module.exports = MessageManager = {
  createMessage,
  getMessages,
  findAllMessagesInRooms,
  deleteAllMessagesInRoom,
  updateMessage,
  deleteMessage,
}
;[
  'createMessage',
  'getMessages',
  'findAllMessagesInRooms',
  'updateMessage',
  'deleteMessage',
].map(method =>
  metrics.timeAsyncMethod(
    MessageManager,
    method,
    'mongo.MessageManager',
    logger
  )
)