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

/*
 * 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 mongojs = require("../../mongojs");
const { db } = mongojs;
const { ObjectId } = mongojs;
const async = require("async");
const metrics = require('metrics-sharelatex');
const logger = require('logger-sharelatex');

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);
		return db.messages.save(newMessageOpts, callback);
	},

	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);
		const cursor = db.messages.find(query).sort({ timestamp: -1 }).limit(limit);
		return cursor.toArray(callback);
	},
	
	findAllMessagesInRooms(room_ids, callback) {
		if (callback == null) { callback = function(error, messages) {}; }
		return db.messages.find({
			room_id: { $in: room_ids }
		}, callback);
	},

	deleteAllMessagesInRoom(room_id, callback) {
		if (callback == null) { callback = function(error) {}; }
		return db.messages.remove({
			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
		});
		return db.messages.update(query, {
			$set: {
				content,
				edited_at: timestamp
			}
		}, function(error) {
			if (error != null) { return callback(error); }
			return callback();
		});
	},

	deleteMessage(room_id, message_id, callback) {
		if (callback == null) { callback = function(error) {}; }
		const query = this._ensureIdsAreObjectIds({
			_id: message_id,
			room_id
		});
		return db.messages.remove(query, function(error) {
			if (error != null) { return callback(error); }
			return 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));