'use strict'; var async = require('async'); var LRU = require('lru-cache'); var db = require('../database'); var pubsub = require('../pubsub'); module.exports = function (User) { User.blocks = { _cache: new LRU({ max: 100, length: function () { return 1; }, maxAge: 0, }), }; User.blocks.is = function (targetUid, uid, callback) { User.blocks.list(uid, function (err, blocks) { callback(err, blocks.includes(parseInt(targetUid, 10))); }); }; User.blocks.can = function (callerUid, blockerUid, blockeeUid, callback) { // Guests can't block if (blockerUid === 0 || blockeeUid === 0) { return setImmediate(callback, new Error('[[error:cannot-block-guest]]')); } else if (blockerUid === blockeeUid) { return setImmediate(callback, new Error('[[error:cannot-block-self]]')); } // Administrators and global moderators cannot be blocked // Only admins/mods can block users as another user async.waterfall([ function (next) { async.parallel({ isCallerAdminOrMod: function (next) { User.isAdminOrGlobalMod(callerUid, next); }, isBlockeeAdminOrMod: function (next) { User.isAdminOrGlobalMod(blockeeUid, next); }, }, next); }, function (results, next) { if (results.isBlockeeAdminOrMod) { return callback(new Error('[[error:cannot-block-privileged]]')); } if (parseInt(callerUid, 10) !== parseInt(blockerUid, 10) && !results.isCallerAdminOrMod) { return callback(new Error()); } next(); }, ], callback); }; User.blocks.list = function (uid, callback) { if (User.blocks._cache.has(parseInt(uid, 10))) { return setImmediate(callback, null, User.blocks._cache.get(parseInt(uid, 10))); } db.getSortedSetRange('uid:' + uid + ':blocked_uids', 0, -1, function (err, blocked) { if (err) { return callback(err); } blocked = blocked.map(uid => parseInt(uid, 10)).filter(Boolean); User.blocks._cache.set(parseInt(uid, 10), blocked); callback(null, blocked); }); }; pubsub.on('user:blocks:cache:del', function (uid) { User.blocks._cache.del(uid); }); User.blocks.add = function (targetUid, uid, callback) { async.waterfall([ async.apply(this.applyChecks, true, targetUid, uid), async.apply(db.sortedSetAdd.bind(db), 'uid:' + uid + ':blocked_uids', Date.now(), targetUid), async.apply(User.incrementUserFieldBy, uid, 'blocksCount', 1), function (_blank, next) { User.blocks._cache.del(parseInt(uid, 10)); pubsub.publish('user:blocks:cache:del', parseInt(uid, 10)); setImmediate(next); }, ], callback); }; User.blocks.remove = function (targetUid, uid, callback) { async.waterfall([ async.apply(this.applyChecks, false, targetUid, uid), async.apply(db.sortedSetRemove.bind(db), 'uid:' + uid + ':blocked_uids', targetUid), async.apply(User.decrementUserFieldBy, uid, 'blocksCount', 1), function (_blank, next) { User.blocks._cache.del(parseInt(uid, 10)); pubsub.publish('user:blocks:cache:del', parseInt(uid, 10)); setImmediate(next); }, ], callback); }; User.blocks.applyChecks = function (block, targetUid, uid, callback) { User.blocks.can(uid, uid, targetUid, function (err) { if (err) { return callback(err); } User.blocks.is(targetUid, uid, function (err, is) { callback(err || (is === block ? new Error('[[error:already-' + (block ? 'blocked' : 'unblocked') + ']]') : null)); }); }); }; User.blocks.filterUids = function (targetUid, uids, callback) { async.filter(uids, function (uid, next) { User.blocks.is(targetUid, uid, function (err, blocked) { next(err, !blocked); }); }, callback); }; User.blocks.filter = function (uid, property, set, callback) { // Given whatever is passed in, iterates through it, and removes entries made by blocked uids // property is optional if (Array.isArray(property) && typeof set === 'function' && !callback) { callback = set; set = property; property = 'uid'; } if (!Array.isArray(set) || !set.length || !set.every((item) => { if (!item) { return false; } const check = item.hasOwnProperty(property) ? item[property] : item; return ['number', 'string'].includes(typeof check); })) { return callback(null, set); } const isPlain = typeof set[0] !== 'object'; User.blocks.list(uid, function (err, blocked_uids) { if (err) { return callback(err); } set = set.filter(function (item) { return !blocked_uids.includes(parseInt(isPlain ? item : item[property], 10)); }); callback(null, set); }); }; };