overleaf/services/chat/app/js/Features/Messages/MessageManager.js
Jakob Ackermann 4080784310 [misc] simplify mongodb collection access using a shared db construct
Resolve the getCollection Promises once and store the result in a shared
 `db` object which can get imported by all the call-sites.

The http server is starting only after a Promise of `waitForDb()`
 resolves. This covers the app code and the acceptance tests:
REF: 586706a9439c3591fc9613dc877f055096ca073a
REF: d026569d2eb4123e30c771a55a001b42d5ade72f
2020-09-10 10:08:30 +01:00

137 lines
3.2 KiB
JavaScript

/* 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 { db, ObjectId } = require('../../mongodb')
const metrics = require('@overleaf/metrics')
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)
db.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)
db.messages
.find(query)
.sort({ timestamp: -1 })
.limit(limit)
.toArray(callback)
},
findAllMessagesInRooms(room_ids, callback) {
if (callback == null) {
callback = function (error, messages) {}
}
db.messages
.find({
room_id: { $in: room_ids }
})
.toArray(callback)
},
deleteAllMessagesInRoom(room_id, callback) {
if (callback == null) {
callback = function (error) {}
}
db.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
})
db.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
})
db.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
)
)