'use strict'; var db = require('./database'), async = require('async'), user = require('./user'), plugins = require('./plugins'), meta = require('./meta'); (function(Messaging) { function sortUids(fromuid, touid) { return [fromuid, touid].sort(); } Messaging.addMessage = function(fromuid, touid, content, callback) { var uids = sortUids(fromuid, touid); db.incrObjectField('global', 'nextMid', function(err, mid) { if (err) { return callback(err); } var timestamp = Date.now(); var message = { content: content, timestamp: timestamp, fromuid: fromuid, touid: touid }; plugins.fireHook('filter:messaging.save', message, function(err, message) { if (err) { return callback(err); } /* TODO: Here I'd check the score of the previous message in this chat's sorted set. but unfortunately, chats are stored as a list. Once nodebb/nodebb#1902 is resolved, I can finish it up here. */ db.setObject('message:' + mid, message, function(err) { if (err) { return callback(err); } db.sortedSetAdd('messages:uid:' + uids[0] + ':to:' + uids[1], timestamp, mid); db.listPrepend('messages:recent:' + fromuid, message.content, function(err) { if (err) { return callback(err); } // Truncate recent chats list back down to 10 (should use LTRIM, see nodebb/nodebb#1901) db.getListRange('messages:recent:' + fromuid, 0, -1, function(err, list) { if (list.length > 10) { db.listRemoveLast('messages:recent:' + uids[0]); } }); }); Messaging.updateChatTime(fromuid, touid); Messaging.updateChatTime(touid, fromuid); async.parallel([ function(next) { Messaging.markRead(fromuid, touid, next); }, function(next) { Messaging.markUnread(touid, fromuid, next); } ], function(err, results) { if (err) { return callback(err); } getMessages([mid], fromuid, touid, true, function(err, messages) { Messaging.isNewSet(fromuid, touid, mid, function(err, isNewSet) { messages[0].newSet = isNewSet; callback(err, messages ? messages[0] : null); }); }); }); }); }); }); }; Messaging.getMessages = function(fromuid, touid, isNew, callback) { var uids = sortUids(fromuid, touid); db.getSortedSetRange('messages:uid:' + uids[0] + ':to:' + uids[1], -((meta.config.chatMessagesToDisplay || 50) - 1), -1, function(err, mids) { if (err) { return callback(err); } if (!mids || !mids.length) { return callback(null, []); } getMessages(mids, fromuid, touid, isNew, callback); }); }; function getMessages(mids, fromuid, touid, isNew, callback) { user.getMultipleUserFields([fromuid, touid], ['uid', 'username', 'userslug', 'picture'], function(err, userData) { if(err) { return callback(err); } var keys = mids.map(function(mid) { return 'message:' + mid; }); async.waterfall([ async.apply(db.getObjects, keys), function(messages, next) { async.map(messages, function(message, next) { var self = parseInt(message.fromuid, 10) === parseInt(fromuid, 10); message.fromUser = self ? userData[0] : userData[1]; message.toUser = self ? userData[1] : userData[0]; message.timestampISO = new Date(parseInt(message.timestamp, 10)).toISOString(); message.self = self ? 1 : 0; message.newSet = false; Messaging.parse(message.content, message.fromuid, fromuid, userData[1], userData[0], isNew, function(result) { message.content = result; next(null, message); }); }, next); }, function(messages, next) { // Add a spacer in between messages with time gaps between them messages = messages.map(function(message, index) { // Compare timestamps with the previous message, and check if a spacer needs to be added if (index > 0 && parseInt(message.timestamp, 10) > parseInt(messages[index-1].timestamp, 10) + (1000*60*5)) { // If it's been 5 minutes, this is a new set of messages message.newSet = true; } return message; }); next(undefined, messages); } ], callback); }); } Messaging.parse = function (message, fromuid, myuid, toUserData, myUserData, isNew, callback) { plugins.fireHook('filter:post.parse', message, function(err, parsed) { if (err) { return callback(message); } var messageData = { message: message, parsed: parsed, fromuid: fromuid, myuid: myuid, toUserData: toUserData, myUserData: myUserData, isNew: isNew, parsedMessage: parsed }; plugins.fireHook('filter:messaging.parse', messageData, function(err, messageData) { callback(messageData.parsedMessage); }); }); }; Messaging.isNewSet = function(fromuid, touid, mid, callback) { var uids = sortUids(fromuid, touid), setKey = 'messages:uid:' + uids[0] + ':to:' + uids[1]; async.waterfall([ async.apply(db.sortedSetRank, setKey, mid), function(index, next) { if (index > 0) { db.getSortedSetRange(setKey, index-1, index, next); } else { next(null, true); } }, function(mids, next) { db.getObjects(['message:' + mids[0], 'message:' + mids[1]], next); }, function(messages, next) { if (typeof messages !== 'boolean') { next(null, parseInt(messages[1].timestamp, 10) > parseInt(messages[0].timestamp, 10) + (1000*60*5)); } else { next(null, messages); } } ], callback); }; Messaging.updateChatTime = function(uid, toUid, callback) { callback = callback || function() {}; db.sortedSetAdd('uid:' + uid + ':chats', Date.now(), toUid, callback); }; Messaging.getRecentChats = function(uid, start, end, callback) { db.getSortedSetRevRange('uid:' + uid + ':chats', start, end, function(err, uids) { if(err) { return callback(err); } async.parallel({ unreadUids: async.apply(db.isSortedSetMembers, 'uid:' + uid + ':chats:unread', uids), users: async.apply(user.getMultipleUserFields, uids, ['username', 'picture', 'uid']) }, function(err, results) { if (err) { return callback(err); } var users = results.users; for (var i=0; i