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

/* eslint-disable
    camelcase,
    handle-callback-err,
    max-len,
    no-unused-vars,
*/
// TODO: This file was created by bulk-decaffeinate.
// Fix any style issues and re-enable lint.
/*
 * decaffeinate suggestions:
 * DS102: Remove unnecessary code created because of implicit returns
 * DS207: Consider shorter variations of null checks
 * Full docs: https://github.com/decaffeinate/decaffeinate/blob/master/docs/suggestions.md
 */
let MessageManager
const { ObjectId, getCollection } = require('../../mongodb')
const metrics = require('@overleaf/metrics')
const logger = require('logger-sharelatex')

const messagesCollectionPromise = getCollection('messages')

module.exports = MessageManager = {
  createMessage(room_id, user_id, content, timestamp, callback) {
    if (callback == null) {
      callback = function (error, message) {}
    }
    let newMessageOpts = {
      content,
      room_id,
      user_id,
      timestamp
    }
    newMessageOpts = this._ensureIdsAreObjectIds(newMessageOpts)
    messagesCollectionPromise.then((messages) =>
      messages.insertOne(newMessageOpts, function (error, confirmation) {
        if (error) {
          return callback(error)
        }
        newMessageOpts._id = confirmation.insertedId
        callback(null, newMessageOpts)
      })
    )
  },

  getMessages(room_id, limit, before, callback) {
    if (callback == null) {
      callback = function (error, messages) {}
    }
    let query = { room_id }
    if (before != null) {
      query.timestamp = { $lt: before }
    }
    query = this._ensureIdsAreObjectIds(query)
    messagesCollectionPromise.then((messages) =>
      messages
        .find(query)
        .sort({ timestamp: -1 })
        .limit(limit)
        .toArray(callback)
    )
  },

  findAllMessagesInRooms(room_ids, callback) {
    if (callback == null) {
      callback = function (error, messages) {}
    }
    messagesCollectionPromise.then((messages) =>
      messages
        .find({
          room_id: { $in: room_ids }
        })
        .toArray(callback)
    )
  },

  deleteAllMessagesInRoom(room_id, callback) {
    if (callback == null) {
      callback = function (error) {}
    }
    messagesCollectionPromise.then((messages) =>
      messages.deleteMany(
        {
          room_id
        },
        callback
      )
    )
  },

  updateMessage(room_id, message_id, content, timestamp, callback) {
    if (callback == null) {
      callback = function (error, message) {}
    }
    const query = this._ensureIdsAreObjectIds({
      _id: message_id,
      room_id
    })
    messagesCollectionPromise.then((messages) =>
      messages.updateOne(
        query,
        {
          $set: {
            content,
            edited_at: timestamp
          }
        },
        callback
      )
    )
  },

  deleteMessage(room_id, message_id, callback) {
    if (callback == null) {
      callback = function (error) {}
    }
    const query = this._ensureIdsAreObjectIds({
      _id: message_id,
      room_id
    })
    messagesCollectionPromise.then((messages) =>
      messages.deleteOne(query, callback)
    )
  },

  _ensureIdsAreObjectIds(query) {
    if (query.user_id != null && !(query.user_id instanceof ObjectId)) {
      query.user_id = ObjectId(query.user_id)
    }
    if (query.room_id != null && !(query.room_id instanceof ObjectId)) {
      query.room_id = ObjectId(query.room_id)
    }
    if (query._id != null && !(query._id instanceof ObjectId)) {
      query._id = ObjectId(query._id)
    }
    return query
  }
}
;[
  'createMessage',
  'getMessages',
  'findAllMessagesInRooms',
  'updateMessage',
  'deleteMessage'
].map((method) =>
  metrics.timeAsyncMethod(
    MessageManager,
    method,
    'mongo.MessageManager',
    logger
  )
)