'use strict'; var async = require('async'), winston = require('winston'), cron = require('cron').CronJob, nconf = require('nconf'), S = require('string'), _ = require('underscore'), db = require('./database'), utils = require('../public/src/utils'), events = require('./events'), User = require('./user'), groups = require('./groups'), meta = require('./meta'), plugins = require('./plugins'); (function(Notifications) { Notifications.init = function() { winston.verbose('[notifications.init] Registering jobs.'); new cron('*/30 * * * *', Notifications.prune, null, true); }; Notifications.get = function(nid, callback) { Notifications.getMultiple([nid], function(err, notifications) { callback(err, Array.isArray(notifications) && notifications.length ? notifications[0] : null); }); }; Notifications.getMultiple = function(nids, callback) { var keys = nids.map(function(nid) { return 'notifications:' + nid; }); db.getObjects(keys, function(err, notifications) { if (err) { return callback(err); } if (!Array.isArray(notifications) || !notifications.length) { return callback(null, []); } async.map(notifications, function(notification, next) { if (!notification) { return next(null, null); } if (notification.bodyShort) { notification.bodyShort = S(notification.bodyShort).escapeHTML().s; } if (notification.bodyLong) { notification.bodyLong = S(notification.bodyLong).escapeHTML().s; } if (notification.from && !notification.image) { User.getUserFields(notification.from, ['username', 'userslug', 'picture'], function(err, userData) { if (err) { return next(err); } notification.image = userData.picture; notification.user = userData; next(null, notification); }); return; } else if (notification.image) { switch(notification.image) { case 'brand:logo': notification.image = meta.config['brand:logo'] || nconf.get('relative_path') + '/logo.png'; break; } return next(null, notification); } else { notification.image = meta.config['brand:logo'] || nconf.get('relative_path') + '/logo.png'; return next(null, notification); } }, callback); }); }; Notifications.create = function(data, callback) { if (!data.nid) { return callback(new Error('no-notification-id')); } data.importance = data.importance || 5; db.getObject('notifications:' + data.nid, function(err, oldNotification) { if (err) { return callback(err); } if (oldNotification) { if (parseInt(oldNotification.pid, 10) === parseInt(data.pid, 10) && parseInt(oldNotification.importance, 10) > parseInt(data.importance, 10)) { return callback(); } } var now = Date.now(); data.datetime = now; async.parallel([ function(next) { db.sortedSetAdd('notifications', now, data.nid, next); }, function(next) { db.setObject('notifications:' + data.nid, data, next); } ], function(err) { callback(err, data); }); }); }; Notifications.push = function(notification, uids, callback) { callback = callback || function() {}; if (!notification.nid) { return callback(); } if (!Array.isArray(uids)) { uids = [uids]; } uids = uids.filter(function(uid) { return parseInt(uid, 10); }); if (!uids.length) { return callback(); } var done = false; var start = 0; var batchSize = 50; setTimeout(function() { async.whilst( function() { return !done; }, function(next) { var currentUids = uids.slice(start, start + batchSize); if (!currentUids.length) { done = true; return next(); } pushToUids(currentUids, notification, function(err) { if (err) { return next(err); } start = start + batchSize; setTimeout(next, 1000); }); }, function(err) { if (err) { winston.error(err.stack); } } ); }, 1000); callback(); }; function pushToUids(uids, notification, callback) { var unreadKeys = []; var readKeys = []; uids.forEach(function(uid) { unreadKeys.push('uid:' + uid + ':notifications:unread'); readKeys.push('uid:' + uid + ':notifications:read'); }); var oneWeekAgo = Date.now() - 604800000; async.series([ function(next) { db.sortedSetsAdd(unreadKeys, notification.datetime, notification.nid, next); }, function(next) { db.sortedSetsRemove(readKeys, notification.nid, next); }, function(next) { db.sortedSetsRemoveRangeByScore(unreadKeys, 0, oneWeekAgo, next); }, function(next) { db.sortedSetsRemoveRangeByScore(readKeys, 0, oneWeekAgo, next); } ], function(err) { if (err) { return callback(err); } plugins.fireHook('action:notification.pushed', {notification: notification, uids: uids}); var websockets = require('./socket.io'); if (websockets.server) { for(var i=0; i