var utils = require('./../public/src/utils.js'), RDB = require('./redis.js'), crypto = require('crypto'), emailjs = require('emailjs'), meta = require('./meta.js'), emailjsServer = emailjs.server.connect(meta.config.mailer), bcrypt = require('bcrypt'), marked = require('marked'), notifications = require('./notifications.js'), topics = require('./topics.js'), async = require('async'); (function(User) { User.create = function(username, password, email, callback) { var userslug = utils.slugify(username); username = username.trim(); if (email !== undefined) email = email.trim(); async.parallel([ function(next) { if (email !== undefined) next(!utils.isEmailValid(email) ? new Error('Invalid Email!') : null); else next(); }, function(next) { next(!utils.isUserNameValid(username) ? new Error('Invalid Username!') : null); }, function(next) { if (password !== undefined) next(!utils.isPasswordValid(password) ? new Error('Invalid Password!') : null); else next(); }, function(next) { User.exists(userslug, function(exists) { next(exists ? new Error('Username taken!') : null); }); }, function(next) { if (email !== undefined) { User.isEmailAvailable(email, function(available) { next(!available ? new Error('Email taken!') : null); }); } else next(); } ], function(err, results) { if (err) return callback(err, null); RDB.incr('global:next_user_id', function(err, uid) { RDB.handle(err); var gravatar = User.createGravatarURLFromEmail(email); var timestamp = Date.now(); RDB.hmset('user:'+uid, { 'uid': uid, 'username' : username, 'userslug' : userslug, 'fullname': '', 'location':'', 'birthday':'', 'website':'', 'email' : email || '', 'signature':'', 'joindate' : timestamp, 'picture': gravatar, 'gravatarpicture' : gravatar, 'uploadedpicture': '', 'profileviews': 0, 'reputation': 0, 'postcount': 0, 'lastposttime': 0, 'administrator': (uid == 1) ? 1 : 0, 'banned': 0, 'showemail': 0 }); RDB.set('username:' + username + ':uid', uid); RDB.set('userslug:'+ userslug +':uid', uid); if (email !== undefined) { RDB.set('email:' + email +':uid', uid); User.sendConfirmationEmail(email); } RDB.incr('usercount', function(err, count) { RDB.handle(err); io.sockets.emit('user.count', {count: count}); }); RDB.zadd('users:joindate', timestamp, uid); RDB.zadd('users:postcount', 0, uid); RDB.zadd('users:reputation', 0, uid); io.sockets.emit('user.latest', {userslug: userslug, username: username}); if (password !== undefined) { User.hashPassword(password, function(hash) { User.setUserField(uid, 'password', hash); }); } callback(null, uid); }); }); }; User.delete = function(uid, callback) { RDB.exists('user:'+uid, function(err, exists) { if(exists === 1) { console.log('deleting uid ' + uid); User.getUserData(uid, function(data) { RDB.del('username:' + data['username'] + ':uid'); RDB.del('email:' + data['email'] +':uid'); RDB.del('userslug:'+ data['userslug'] +':uid'); RDB.del('user:' + uid); RDB.del('followers:' + uid); RDB.del('following:' + uid); RDB.zrem('users:joindate', uid); RDB.zrem('users:postcount', uid); RDB.zrem('users:reputation', uid); callback(true); }); } else { callback(false); } }); } User.ban = function(uid, callback) { User.setUserField(uid, 'banned', 1, callback); } User.unban = function(uid, callback) { User.setUserField(uid, 'banned', 0, callback); } User.getUserField = function(uid, field, callback) { RDB.hget('user:' + uid, field, callback); } User.getUserFields = function(uid, fields, callback) { RDB.hmgetObject('user:' + uid, fields, callback); } User.getMultipleUserFields = function(uids, fields, callback) { if(uids.length === 0) { callback([]); return; } var returnData = []; uuids = uids.filter(function(value, index, self) { return self.indexOf(value) === index; }); function iterator(uid, callback) { User.getUserFields(uid, fields, function(err, userData) { if(err) return callback(err); returnData.push(userData); callback(null); }); } async.eachSeries(uuids, iterator, function(err) { if(!err) { callback(returnData); } else { console.log(err); callback(null); } }); } User.getUserData = function(uid, callback) { RDB.hgetall('user:' + uid, function(err, data) { if(err === null) { if(data) { if(data['password']) delete data['password']; } callback(data); } else { console.log(err); } }); } User.filterBannedUsers = function(users) { return users.filter(function(user) { return (!user.banned || user.banned === '0'); }); } User.updateProfile = function(uid, data, callback) { var fields = ['email', 'fullname', 'website', 'location', 'birthday', 'signature']; var returnData = {success:false}; function isSignatureValid(next) { if(data['signature'] !== undefined && data['signature'].length > 150) { next({error:'Signature can\'t be longer than 150 characters!'}, false); } else { next(null, true); } } function isEmailAvailable(next) { if(!data['email']) { return next(null, true); } User.getUserField(uid, 'email', function(err, email) { if(email !== data['email']) { User.isEmailAvailable(data['email'], function(available) { if(!available) { next({error:'Email not available!'}, false); } else { next(null, true); } }); } else { next(null, true); } }); } async.series([isSignatureValid, isEmailAvailable], function(err, results) { if(err) { console.log(err); callback(returnData); } else { async.each(fields, updateField, function(err) { if(err) { console.log(err); callback(returnData); } else { returnData.success = true; callback(returnData); } }); } }); function updateField(field, callback) { if(data[field] !== undefined) { if(field === 'email') { var gravatarpicture = User.createGravatarURLFromEmail(data[field]); User.setUserField(uid, 'gravatarpicture', gravatarpicture); User.getUserFields(uid, ['email', 'picture', 'uploadedpicture'], function(err, userData) { if(err) return callback(err); RDB.del('email:' + userData['email'] + ':uid'); RDB.set('email:' + data['email'] + ':uid', uid); User.setUserField(uid, field, data[field]); if(userData.picture !== userData.uploadedpicture) { returnData.picture = gravatarpicture; User.setUserField(uid, 'picture', gravatarpicture); } returnData.gravatarpicture = gravatarpicture; callback(null); }); return; } else if(field === 'signature') { data[field] = utils.strip_tags(data[field]); } User.setUserField(uid, field, data[field]); callback(null); } else { callback(null); } } } User.isEmailAvailable = function(email, callback) { RDB.exists('email:' + email + ':uid' , function(err, exists) { if(!err) { callback(exists !== 1); return; } else { console.log(err); callback(false); } }); } User.changePassword = function(uid, data, callback) { if(!utils.isPasswordValid(data.newPassword)) { callback({err:'Invalid password!'}); return; } User.getUserField(uid, 'password', function(err, user_password) { bcrypt.compare(data.currentPassword, user_password, function(err, res) { if(err) { console.log(err); callback({err:'bcrpyt compare error!'}); return; } if (res) { User.hashPassword(data.newPassword, function(hash) { User.setUserField(uid, 'password', hash); callback({err:null}); }); } else { callback({err:'Your current password is not correct!'}); } }); }); } User.setUserField = function(uid, field, value, callback) { RDB.hset('user:' + uid, field, value, callback); } User.setUserFields = function(uid, data) { RDB.hmset('user:' + uid, data); } User.incrementUserFieldBy = function(uid, field, value, callback) { RDB.hincrby('user:' + uid, field, value, callback); } User.decrementUserFieldBy = function(uid, field, value, callback) { RDB.hincrby('user:' + uid, field, -value, callback); } User.getUsers = function(set, start, stop, callback) { var data = []; RDB.zrevrange(set, start, stop, function(err, uids) { if(err) { return callback(err, null); } function iterator(uid, callback) { User.getUserData(uid, function(userData) { if(userData) { data.push(userData); } callback(null); }); } async.eachSeries(uids, iterator, function(err) { callback(err, data); }); }); } User.createGravatarURLFromEmail = function(email) { var options = { size: '128', default: 'identicon', rating: 'pg' }; if (!email) { email = ''; options.forcedefault = 'y'; } return require('gravatar').url(email, options, https=nconf.get('https')); } User.hashPassword = function(password, callback) { if(!password) { callback(password); return; } bcrypt.genSalt(nconf.get('bcrypt_rounds'), function(err, salt) { bcrypt.hash(password, salt, function(err, hash) { callback(hash); }); }); } User.search = function(username, callback) { if(!username) { callback([]); return; } RDB.keys('username:*'+ username + '*:uid', function(err, keys) { if(!err) { if(keys && keys.length) { RDB.mget(keys, function(err, uids) { User.getDataForUsers(uids, function(userdata) { callback(userdata); }); }); } else { callback([]); } } else { console.log(err); } }); } User.onNewPostMade = function(uid, tid, pid, timestamp) { User.addPostIdToUser(uid, pid); User.incrementUserFieldBy(uid, 'postcount', 1, function(err, newpostcount) { RDB.zadd('users:postcount', newpostcount, uid); }); User.setUserField(uid, 'lastposttime', timestamp); User.sendPostNotificationToFollowers(uid, tid, pid); } User.addPostIdToUser = function(uid, pid) { RDB.lpush('uid:' + uid + ':posts', pid); } User.addTopicIdToUser = function(uid, tid) { RDB.lpush('uid:' + uid + ':topics', tid); } User.getPostIds = function(uid, start, end, callback) { RDB.lrange('uid:' + uid + ':posts', start, end, function(err, pids) { if(!err) { if(pids && pids.length) callback(pids); else callback([]); } else { console.log(err); callback([]); } }); } User.sendConfirmationEmail = function (email) { if (meta.config['email:host'] && meta.config['email:port'] && meta.config['email:from']) { var confirm_code = utils.generateUUID(), confirm_link = nconf.get('url') + 'confirm/' + confirm_code, confirm_email = global.templates['emails/header'] + global.templates['emails/email_confirm'].parse({'CONFIRM_LINK': confirm_link}) + global.templates['emails/footer'], confirm_email_plaintext = global.templates['emails/email_confirm_plaintext'].parse({ 'CONFIRM_LINK': confirm_link }); // Email confirmation code var expiry_time = 60*60*2, // Expire after 2 hours email_key = 'email:' + email + ':confirm', confirm_key = 'confirm:' + confirm_code + ':email'; RDB.set(email_key, confirm_code); RDB.expire(email_key, expiry_time); RDB.set(confirm_key, email); RDB.expire(confirm_key, expiry_time); // Send intro email w/ confirm code var message = emailjs.message.create({ text: confirm_email_plaintext, from: meta.config.mailer.from, to: email, subject: '[NodeBB] Registration Email Verification', attachment: [ { data: confirm_email, alternative: true } ] }); emailjsServer.send(message, function(err, success) { if (err) { console.log(err); } }); } } User.follow = function(uid, followid, callback) { RDB.sadd('following:' + uid, followid, function(err, data) { if(!err) { RDB.sadd('followers:' + followid, uid, function(err, data) { if(!err) { callback(true); } else { console.log(err); callback(false); } }); } else { console.log(err); callback(false); } }); } User.unfollow = function(uid, unfollowid, callback) { RDB.srem('following:' + uid, unfollowid, function(err, data){ if(!err) { RDB.srem('followers:' + unfollowid, uid, function(err, data){ callback(data); }); } else { console.log(err); } }); } User.getFollowing = function(uid, callback) { RDB.smembers('following:' + uid, function(err, userIds) { if(!err) { User.getDataForUsers(userIds, callback); } else { console.log(err); } }); } User.getFollowers = function(uid, callback) { RDB.smembers('followers:' + uid, function(err, userIds) { if(!err) { User.getDataForUsers(userIds, callback); } else { console.log(err); } }); } User.getFollowingCount = function(uid, callback) { RDB.smembers('following:' + uid, function(err, userIds) { if(!err) { callback(userIds.length); } else { console.log(err); } }); } User.getFollowerCount = function(uid, callback) { RDB.smembers('followers:' + uid, function(err, userIds) { // @note why are error-handling styles being mixed? // either go with not-error-dosomething-else-dosomethingelse, or // go with if-error-dosomething-return // also why is console.log(err) being used when below we're using RDB.handle()? if(!err) { callback(userIds.length); } else { console.log(err); } }); } User.getDataForUsers = function(uids, callback) { var returnData = []; if(!uids || !Array.isArray(uids) || uids.length === 0) { callback(returnData); return; } function iterator(uid, callback) { User.getUserData(uid, function(userData) { returnData.push(userData); callback(null); }); } async.eachSeries(uids, iterator, function(err) { callback(returnData); }); } User.sendPostNotificationToFollowers = function(uid, tid, pid) { User.getUserField(uid, 'username', function(err, username) { RDB.smembers('followers:' + uid, function(err, followers) { topics.getTopicField(tid, 'slug', function(err, slug) { var message = username + ' made a new post'; notifications.create(message, 5, nconf.get('url') + 'topic/' + slug + '#' + pid, 'notification_'+ Date.now(), function(nid) { notifications.push(nid, followers); }); }); }); }); } User.isFollowing = function(uid, theirid, callback) { RDB.sismember('following:' + uid, theirid, function(err, data) { if(!err) { callback(data === 1); } else { console.log(err); } }); } User.exists = function(userslug, callback) { User.get_uid_by_userslug(userslug, function(exists) { callback(!!exists); }); }; User.count = function(socket) { RDB.get('usercount', function(err, count) { RDB.handle(err); socket.emit('user.count', { count: count ? count : 0 }); }); }; User.latest = function(socket) { RDB.zrevrange('users:joindate', 0, 0, function(err, uid) { RDB.handle(err); User.getUserFields(uid, ['username', 'userslug'], function(err, userData) { if(!err && userData) socket.emit('user.latest', {userslug: userData.userslug, username: userData.username}); }); }); } User.get_uid_by_username = function(username, callback) { RDB.get('username:' + username + ':uid', function(err, data) { if (err) { RDB.handle(err); } callback(data); }); }; User.get_uid_by_userslug = function(userslug, callback) { RDB.get('userslug:' + userslug + ':uid', function(err, data) { if (err) { RDB.handle(err); } callback(data); }); }; User.get_usernames_by_uids = function(uids, callback) { var usernames = []; if (!Array.isArray(uids)) { return callback([]); } function iterator(uid, callback) { User.getUserField(uid, 'username', function(err, username) { usernames.push(username); callback(null); }); } async.eachSeries(uids, iterator, function(err) { callback(usernames); }); } User.get_userslugs_by_uids = function(uids, callback) { var userslugs = []; if (!Array.isArray(uids)) { return callback([]); } function iterator(uid, callback) { User.getUserField(uid, 'userslug', function(err, userslug) { userslugs.push(userslug); callback(null); }); } async.eachSeries(uids, iterator, function(err) { callback(userslugs); }); } User.get_uid_by_email = function(email, callback) { RDB.get('email:' + email + ':uid', function(err, data) { if (err) { RDB.handle(err); } callback(data); }); }; User.get_uid_by_session = function(session, callback) { RDB.get('sess:' + session + ':uid', function(err, data) { if (err) { RDB.handle(err); } callback(data); }); }; User.get_uid_by_twitter_id = function(twid, callback) { RDB.hget('twid:uid', twid, function(err, uid) { if (err) { RDB.handle(err); } callback(uid); }); } User.get_uid_by_google_id = function(gplusid, callback) { RDB.hget('gplusid:uid', gplusid, function(err, uid) { if (err) { RDB.handle(err); } callback(uid); }); } User.get_uid_by_fbid = function(fbid, callback) { RDB.hget('fbid:uid', fbid, function(err, uid) { if (err) { RDB.handle(err); } callback(uid); }); } User.session_ping = function(sessionID, uid) { // Start, replace, or extend a session RDB.get('sess:' + sessionID, function(err, session) { if (err) { RDB.handle(err); } var expiry = 60*60*24*14, // Login valid for two weeks sess_key = 'sess:' + sessionID + ':uid', uid_key = 'uid:' + uid + ':session'; RDB.set(sess_key, uid); RDB.expire(sess_key, expiry); RDB.set(uid_key, sessionID); RDB.expire(uid_key, expiry); }); } User.isModerator = function(uid, cid, callback) { RDB.sismember('cid:' + cid + ':moderators', uid, function(err, exists) { RDB.handle(err); callback(!!exists); }); } User.isAdministrator = function(uid, callback) { RDB.sismember('administrators', uid, function(err, exists) { RDB.handle(err); callback(!!exists); }); } User.makeAdministrator = function(uid, callback) { RDB.sadd('administrators', uid, function(err, data){ if(!err) { User.setUserField(uid, 'administrator', 1); } if(callback) { // @todo address why we're only sending back a boolean in the callback and not an error if it occurred callback(err === null); } }); } User.removeAdministrator = function(uid, callback) { RDB.srem('administrators', uid, function(err, data){ if(!err) { User.setUserField(uid, 'administrator', 0); } if(callback) { // @todo address why we're only sending back a boolean in the callback and not an error if it occurred callback(err === null); } }); } User.reset = { validate: function(socket, code, callback) { if (typeof callback !== 'function') { callback = null; } RDB.get('reset:' + code + ':uid', function(err, uid) { if (err) { RDB.handle(err); } if (uid !== null) { RDB.get('reset:' + code + ':expiry', function(err, expiry) { if (err) { RDB.handle(err); } if (expiry >= +Date.now()/1000|0) { if (!callback) { socket.emit('user:reset.valid', { valid: true }); } else { callback(true); } } else { // Expired, delete from db RDB.del('reset:' + code + ':uid'); RDB.del('reset:' + code + ':expiry'); if (!callback) { socket.emit('user:reset.valid', { valid: false }); } else { callback(false); } } }); } else { if (!callback) { socket.emit('user:reset.valid', { valid: false }); } else { callback(false); } } }); }, send: function(socket, email) { User.get_uid_by_email(email, function(uid) { if (uid !== null) { // Generate a new reset code var reset_code = utils.generateUUID(); RDB.set('reset:' + reset_code + ':uid', uid); RDB.set('reset:' + reset_code + ':expiry', (60*60)+new Date()/1000|0); // Active for one hour var reset_link = nconf.get('url') + 'reset/' + reset_code, reset_email = global.templates['emails/reset'].parse({'RESET_LINK': reset_link}), reset_email_plaintext = global.templates['emails/reset_plaintext'].parse({ 'RESET_LINK': reset_link }); var message = emailjs.message.create({ text: reset_email_plaintext, from: meta.config.mailer?meta.config.mailer.from:'localhost@example.org', to: email, subject: 'Password Reset Requested', attachment: [ { data: reset_email, alternative: true } ] }); emailjsServer.send(message, function(err, success) { if (err === null) { socket.emit('user.send_reset', { status: "ok", message: "code-sent", email: email }); } else { socket.emit('user.send_reset', { status: "error", message: "send-failed" }); // @todo handle error properly throw new Error(err); } }); } else { socket.emit('user.send_reset', { status: "error", message: "invalid-email", email: email }); } }); }, commit: function(socket, code, password) { this.validate(socket, code, function(validated) { if (validated) { RDB.get('reset:' + code + ':uid', function(err, uid) { if (err) { RDB.handle(err); } User.hashPassword(password, function(hash) { User.setUserField(uid, 'password', hash); }); RDB.del('reset:' + code + ':uid'); RDB.del('reset:' + code + ':expiry'); socket.emit('user:reset.commit', { status: 'ok' }); }); } }); } } User.email = { exists: function(socket, email, callback) { User.get_uid_by_email(email, function(exists) { exists = !!exists; if (typeof callback !== 'function') { socket.emit('user.email.exists', { exists: exists }); } else { callback(exists); } }); }, confirm: function(code, callback) { RDB.get('confirm:' + code + ':email', function(err, email) { if (err) { RDB.handle(err); } if (email !== null) { RDB.set('email:' + email + ':confirm', true); RDB.del('confirm:' + code + ':email'); callback({ status: 'ok' }); } else { callback({ status: 'not_ok' }); } }); } }; User.active = { get_record : function(socket) { RDB.mget(['global:active_user_record', 'global:active_user_record_date'], function(err, data) { RDB.handle(err); socket.emit('api:user.active.get_record', { record: data[0], timestamp: data[1] }); }); }, get: function(callback) { function user_record(total) { RDB.get('global:active_user_record', function(err, record) { RDB.handle(err); if (total > record) { RDB.set('global:active_user_record', total); RDB.set('global:active_user_record_date', Date.now()); } }); } RDB.keys('active:*', function(err, active) { RDB.handle(err); var returnObj = { users: 0, anon: 0, uids: [] }, keys = []; if (active.length > 0) { for(var a in active) { keys.push('sess:' + active[a].split(':')[1] + ':uid'); } RDB.mget(keys, function(err, uids) { RDB.handle(err); for(var u in uids) { if (uids[u] !== null) { if (returnObj.uids.indexOf(uids[u]) === -1) { returnObj.users++; returnObj.uids.push(uids[u]); } } else { returnObj.anon++; } } user_record(returnObj.anon + returnObj.users); if (callback === undefined) { io.sockets.emit('api:user.active.get', returnObj) } else { callback(returnObj); } }); } else { io.sockets.emit('api:user.active.get', returnObj) } }); }, register: function(sessionID) { // Active state persists for 10 minutes var active_session = 'active:' + sessionID; RDB.set(active_session, ''); RDB.expire(active_session, 60*10) this.get(); } } User.notifications = { get: function(uid, callback) { var maxNotifs = 15; async.parallel({ unread: function(next) { RDB.zrevrangebyscore('uid:' + uid + ':notifications:unread', 10, 0, function(err, nids) { // @todo handle err var unread = []; // Cap the number of notifications returned if (nids.length > maxNotifs) nids.length = maxNotifs; if (nids && nids.length > 0) { async.eachSeries(nids, function(nid, next) { notifications.get(nid, function(notif_data) { unread.push(notif_data); next(); }); }, function(err) { next(null, unread); }); } else { next(null, unread); } }); }, read: function(next) { RDB.zrevrangebyscore('uid:' + uid + ':notifications:read', 10, 0, function(err, nids) { // @todo handle err var read = []; // Cap the number of notifications returned if (nids.length > maxNotifs) nids.length = maxNotifs; if (nids && nids.length > 0) { async.eachSeries(nids, function(nid, next) { notifications.get(nid, function(notif_data) { read.push(notif_data); next(); }); }, function(err) { next(null, read); }); } else { next(null, read); } }); } }, function(err, notifications) { // While maintaining score sorting, sort by time var readCount = notifications.read.length, unreadCount = notifications.unread.length; notifications.read.sort(function(a, b) { if (a.score === b.score) { return (a.datetime - b.datetime) > 0 ? -1 : 1; } }); notifications.unread.sort(function(a, b) { if (a.score === b.score) { return (a.datetime - b.datetime) > 0 ? -1 : 1; } }); // Limit the number of notifications to `maxNotifs`, prioritising unread notifications if (notifications.read.length + notifications.unread.length > maxNotifs) { notifications.read.length = maxNotifs - notifications.unread.length; } callback(notifications); }); }, getUnreadCount: function(uid, callback) { RDB.zcount('uid:' + uid + ':notifications:unread', 0, 10, callback); } } }(exports));