diff --git a/app.js b/app.js index 05ba01c22f..075a029fbe 100644 --- a/app.js +++ b/app.js @@ -87,7 +87,8 @@ SocketIO = require('socket.io').listen(global.server, { log: false, transports: ['websocket', 'xhr-polling', 'jsonp-polling', 'flashsocket']}), websockets = require('./src/websockets.js'), posts = require('./src/posts.js'), - plugins = require('./src/plugins'); // Don't remove this - plugins initializes itself + plugins = require('./src/plugins'), // Don't remove this - plugins initializes itself + Notifications = require('./src/notifications'); websockets.init(SocketIO); @@ -106,6 +107,8 @@ ]); templates.ready(webserver.init); + + Notifications.init(); }); } else if (nconf.get('upgrade')) { diff --git a/package.json b/package.json index 89d3e9b69c..4890e1acd5 100644 --- a/package.json +++ b/package.json @@ -44,7 +44,8 @@ "nodebb-plugin-mentions": "~0.1.13", "nodebb-plugin-markdown": "~0.1.7", "nodebb-theme-vanilla": "designcreateplay/nodebb-theme-vanilla", - "nodebb-theme-cerulean": "0.0.3" + "nodebb-theme-cerulean": "0.0.3", + "cron": "~1.0.1" }, "optionalDependencies": { "hiredis": "~0.1.15" diff --git a/public/src/forum/footer.js b/public/src/forum/footer.js index 45cc71312c..fa2240a93c 100644 --- a/public/src/forum/footer.js +++ b/public/src/forum/footer.js @@ -101,6 +101,7 @@ numUnread = data.unread.length, x; notifList.innerHTML = ''; + console.log(data); if ((data.read.length + data.unread.length) > 0) { for (x = 0; x < numUnread; x++) { notifEl.setAttribute('data-nid', data.unread[x].nid); @@ -115,15 +116,16 @@ notifFrag.appendChild(notifEl.cloneNode(true)); } } else { + notifEl.className = 'no-notifs'; notifEl.innerHTML = 'You have no notifications'; - notifFrag.appendChild(notifEl); + notifFrag.appendChild(notifEl.cloneNode(true)); } // Add dedicated link to /notifications notifEl.removeAttribute('data-nid'); notifEl.className = 'pagelink'; notifEl.innerHTML = 'See all Notifications'; - notifFrag.appendChild(notifEl); + notifFrag.appendChild(notifEl.cloneNode(true)); notifList.appendChild(notifFrag); diff --git a/public/templates/header.tpl b/public/templates/header.tpl index a88a02339e..adceb8c17c 100644 --- a/public/templates/header.tpl +++ b/public/templates/header.tpl @@ -24,9 +24,6 @@ "forum": '../forum' } }); - requirejs.onError = function(err) { - console.log(err); - } diff --git a/src/notifications.js b/src/notifications.js index d6b2791028..f6d4f03e06 100644 --- a/src/notifications.js +++ b/src/notifications.js @@ -1,16 +1,30 @@ var RDB = require('./redis.js'), async = require('async'), utils = require('../public/src/utils.js'), + winston = require('winston'), + cron = require('cron').CronJob, notifications = { + init: function() { + if (process.env.NODE_ENV === 'development') { + winston.info('[notifications.init] Registering jobs.'); + } + + new cron('0 0 * * *', notifications.prune, null, true); + }, get: function(nid, uid, callback) { RDB.multi() .hmget('notifications:' + nid, 'text', 'score', 'path', 'datetime', 'uniqueId') .zrank('uid:' + uid + ':notifications:read', nid) + .exists('notifications:' + nid) .exec(function(err, results) { - var notification = results[0] + var notification = results[0], readIdx = results[1]; + if (!results[2]) { + return callback(null); + } + callback({ nid: nid, text: notification[0], @@ -30,16 +44,32 @@ var RDB = require('./redis.js'), * the new one put in its place. */ RDB.incr('notifications:next_nid', function(err, nid) { + RDB.sadd('notifications', nid); RDB.hmset('notifications:' + nid, { text: text || '', path: path || null, datetime: Date.now(), uniqueId: uniqueId || utils.generateUUID() }, function(err, status) { - if (!err) callback(nid); + if (!err) { + callback(nid); + } }); }); }, + destroy: function(nid) { + var multi = RDB.multi(); + + multi.del('notifications:' + nid); + multi.srem('notifications', nid); + + multi.exec(function(err) { + if (err) { + winston.error('Problem deleting expired notifications. Stack follows.'); + winston.error(err.stack); + } + }); + }, push: function(nid, uids, callback) { if (!Array.isArray(uids)) uids = [uids]; @@ -48,12 +78,14 @@ var RDB = require('./redis.js'), notifications.get(nid, null, function(notif_data) { for (x = 0; x < numUids; x++) { - if (parseInt(uids[x]) > 0) { + if (parseInt(uids[x], 10) > 0) { (function(uid) { notifications.remove_by_uniqueId(notif_data.uniqueId, uid, function() { RDB.zadd('uid:' + uid + ':notifications:unread', notif_data.datetime, nid); global.io.sockets.in('uid_' + uid).emit('event:new_notification'); - if (callback) callback(true); + if (callback) { + callback(true); + } }); })(uids[x]); } @@ -67,13 +99,18 @@ var RDB = require('./redis.js'), if (nids && nids.length > 0) { async.each(nids, function(nid, next) { notifications.get(nid, uid, function(nid_info) { - if (nid_info.uniqueId === uniqueId) RDB.zrem('uid:' + uid + ':notifications:unread', nid); + if (nid_info.uniqueId === uniqueId) { + RDB.zrem('uid:' + uid + ':notifications:unread', nid); + } + next(); }); }, function(err) { next(); }); - } else next(); + } else { + next(); + } }); }, function(next) { @@ -81,17 +118,24 @@ var RDB = require('./redis.js'), if (nids && nids.length > 0) { async.each(nids, function(nid, next) { notifications.get(nid, uid, function(nid_info) { - if (nid_info.uniqueId === uniqueId) RDB.zrem('uid:' + uid + ':notifications:read', nid); + if (nid_info.uniqueId === uniqueId) { + RDB.zrem('uid:' + uid + ':notifications:read', nid); + } + next(); }); }, function(err) { next(); }); - } else next(); + } else { + next(); + } }); } ], function(err) { - if (!err) callback(true); + if (!err) { + callback(true); + } }); }, mark_read: function(nid, uid, callback) { @@ -99,38 +143,126 @@ var RDB = require('./redis.js'), notifications.get(nid, uid, function(notif_data) { RDB.zrem('uid:' + uid + ':notifications:unread', nid); RDB.zadd('uid:' + uid + ':notifications:read', notif_data.datetime, nid); - if (callback) callback(); + if (callback) { + callback(); + } }); } }, mark_read_multiple: function(nids, uid, callback) { - if (!Array.isArray(nids) && parseInt(nids, 10) > 0) nids = [nids]; + if (!Array.isArray(nids) && parseInt(nids, 10) > 0) { + nids = [nids]; + } async.each(nids, function(nid, next) { notifications.mark_read(nid, uid, function(err) { - if (!err) next(null); + if (!err) { + next(null); + } }); }, function(err) { - if (callback) callback(err); + if (callback) { + callback(err); + } }); }, mark_all_read: function(uid, callback) { RDB.zrange('uid:' + uid + ':notifications:unread', 0, 10, function(err, nids) { - if (err) return callback(err); + if (err) { + return callback(err); + } if (nids.length > 0) { notifications.mark_read_multiple(nids, uid, function(err) { callback(err); }); - } else callback(); + } else { + callback(); + } }); + }, + prune: function(cutoff) { + if (process.env.NODE_ENV === 'development') { + winston.info('[notifications.prune] Removing expired notifications from the database.'); + } + + var today = new Date(), + numPruned = 0; + + if (!cutoff) { + cutoff = new Date(today.getFullYear(), today.getMonth(), today.getDate() - 7); + } + + var cutoffTime = cutoff.getTime(); + + async.parallel({ + "inboxes": function(next) { + RDB.keys('uid:*:notifications:unread', next); + }, + "nids": function(next) { + RDB.smembers('notifications', function(err, nids) { + async.filter(nids, function(nid, next) { + RDB.hget('notifications:' + nid, 'datetime', function(err, datetime) { + if (parseInt(datetime, 10) < cutoffTime) { + next(true); + } else { + next(false); + } + }); + }, function(expiredNids) { + next(null, expiredNids); + }); + }); + } + }, function(err, results) { + if (!err) { + var numInboxes = results.inboxes.length, + x; + + async.eachSeries(results.nids, function(nid, next) { + var multi = RDB.multi(); + + for(x=0;x 0) { async.eachSeries(nids, function(nid, next) { notifications.get(nid, uid, function(notif_data) { - unread.push(notif_data); + // If the notification could not be found, silently drop it + if (notif_data) { + unread.push(notif_data); + } else { + RDB.zrem('uid:' + uid + ':notifications:unread', nid); + } + next(); }); }, function(err) { @@ -925,7 +931,13 @@ var utils = require('./../public/src/utils.js'), if (nids && nids.length > 0) { async.eachSeries(nids, function(nid, next) { notifications.get(nid, uid, function(notif_data) { - read.push(notif_data); + // If the notification could not be found, silently drop it + if (notif_data) { + read.push(notif_data); + } else { + RDB.zrem('uid:' + uid + ':notifications:read', nid); + } + next(); }); }, function(err) { diff --git a/src/webserver.js b/src/webserver.js index e216c6ebfa..b660f34372 100644 --- a/src/webserver.js +++ b/src/webserver.js @@ -656,6 +656,10 @@ var express = require('express'), }); }); + // Debug routes + if (process.env.NODE_ENV === 'development') { + require('./routes/debug')(app); + } var custom_routes = { 'routes': [],