"use strict"; var winston = require('winston'); module.exports = function(db, module) { var helpers = module.helpers.mongo; module.searchIndex = function(key, data, id, callback) { callback = callback || function() {}; id = parseInt(id, 10); if (!id) { return callback(); } var setData = { id: id }; for(var field in data) { if (data.hasOwnProperty(field) && data[field]) { setData[field] = data[field].toString(); } } db.collection('search' + key).update({id: id}, {$set: setData}, {upsert:true, w: 1}, function(err) { if (err) { winston.error('Error indexing ' + err.message); } callback(err); }); }; module.search = function(key, data, limit, callback) { var searchQuery = {}; if (data.content) { searchQuery.$text = {$search: data.content}; } if (Array.isArray(data.cid) && data.cid.length) { data.cid = data.cid.filter(Boolean); if (data.cid.length > 1) { searchQuery.cid = {$in: data.cid.map(String)}; } else if (data.cid[0]) { searchQuery.cid = data.cid[0].toString(); } } if (Array.isArray(data.uid) && data.uid.length) { data.uid = data.uid.filter(Boolean); if (data.uid.length > 1) { searchQuery.uid = {$in: data.uid.map(String)}; } else if (data.uid[0]) { searchQuery.uid = data.uid[0].toString(); } } db.collection('search' + key).find(searchQuery, {limit: limit, fields:{_id: 0, id: 1}}).toArray(function(err, results) { if (err) { return callback(err); } if (!results || !results.length) { return callback(null, []); } var data = results.map(function(item) { return item.id; }); callback(null, data); }); }; module.searchRemove = function(key, id, callback) { callback = callback || helpers.noop; id = parseInt(id, 10); if (!id) { return callback(); } db.collection('search' + key).remove({id: id}, function(err, res) { callback(err); }); }; module.flushdb = function(callback) { callback = callback || helpers.noop; db.dropDatabase(callback); }; module.exists = function(key, callback) { if (!key) { return callback(); } db.collection('objects').findOne({_key: key}, function(err, item) { callback(err, item !== undefined && item !== null); }); }; module.delete = function(key, callback) { callback = callback || helpers.noop; if (!key) { return callback(); } db.collection('objects').remove({_key: key}, function(err, res) { callback(err); }); }; module.deleteAll = function(keys, callback) { callback = callback || helpers.noop; if (!Array.isArray(keys) || !keys.length) { return callback(); } db.collection('objects').remove({_key: {$in: keys}}, function(err, res) { callback(err); }); }; module.get = function(key, callback) { if (!key) { return callback(); } module.getObjectField(key, 'value', callback); }; module.set = function(key, value, callback) { callback = callback || helpers.noop; if (!key) { return callback(); } var data = {value: value}; module.setObject(key, data, callback); }; module.increment = function(key, callback) { callback = callback || helpers.noop; if (!key) { return callback(); } db.collection('objects').findAndModify({_key: key}, {}, {$inc: {value: 1}}, {new: true, upsert: true}, function(err, result) { callback(err, result && result.value ? result.value.value : null); }); }; module.rename = function(oldKey, newKey, callback) { callback = callback || helpers.noop; db.collection('objects').update({_key: oldKey}, {$set:{_key: newKey}}, {multi: true}, function(err, res) { callback(err); }); }; module.expire = function(key, seconds, callback) { module.expireAt(key, Math.round(Date.now() / 1000) + seconds, callback); }; module.expireAt = function(key, timestamp, callback) { module.setObjectField(key, 'expireAt', new Date(timestamp * 1000), callback); }; module.pexpire = function(key, ms, callback) { module.pexpireAt(key, Date.now() + parseInt(ms, 10), callback); }; module.pexpireAt = function(key, timestamp, callback) { module.setObjectField(key, 'expireAt', new Date(timestamp), callback); }; };