Georg/dbot
1
0
Fork 0
forked from GitHub/dbot
Code Issues Pull Requests Projects Releases Wiki Activity

all commands but ~qstats and ~rq in quotes use databank [#272]

Browse Source
This commit is contained in:
reality 2013-03-23 13:24:13 +00:00
parent 7bcb136e26
commit 71c0bdb760
1 changed files with 130 additions and 100 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

230
modules/quotes/commands.js
View File

@ -6,11 +6,30 @@ var _ = require('underscore')._,
var commands = function(dbot) { var commands = function(dbot) {
var quotes = dbot.db.quoteArrs; var quotes = dbot.db.quoteArrs;
var commands = { var commands = {
// Alternative syntax to ~q
/*** Quote Retrieval ***/
// Alternative ~q syntax
'~': function(event) { '~': function(event) {
commands['~q'].bind(this)(event); commands['~q'].bind(this)(event);
}, },
// Retrieve quote from a category in the database.
'~q': function(event) {
var name = event.input[1].trim().toLowerCase();
this.db.read('quote_category', name, function(err, category) {
if(!err) {
var quoteIndex = _.random(0, category.length - 1);
event.reply(name + ': ' + category[quoteIndex]);
} else if(err instanceof NoSuchThingError) {
event.reply(dbot.t('category_not_found', { 'category': name }));
}
});
},
/*** Quote Removal ***/
// Show number of quotes in removal cache
'~rmstatus': function(event) { '~rmstatus': function(event) {
var rmCacheCount = this.rmCache.length; var rmCacheCount = this.rmCache.length;
if(rmCacheCount < dbot.config.quotes.rmLimit) { if(rmCacheCount < dbot.config.quotes.rmLimit) {
@ -22,6 +41,7 @@ var commands = function(dbot) {
} }
}, },
// Confirm removal of quote cache
'~rmconfirm': function(event) { '~rmconfirm': function(event) {
var rmCacheCount = this.rmCache.length; var rmCacheCount = this.rmCache.length;
this.rmCache.length = 0; this.rmCache.length = 0;
@ -29,99 +49,46 @@ var commands = function(dbot) {
{ 'count': rmCacheCount })); { 'count': rmCacheCount }));
}, },
// Reinstate all quotes in removal cache
'~rmdeny': function(event) { '~rmdeny': function(event) {
var rmCache = this.rmCache; var rmCache = this.rmCache;
var rmCacheCount = rmCache.length; var rmCacheCount = rmCache.length;
for(var i=0;i<rmCacheCount;i++) {
if(!_.has(quotes, rmCache[i].key)) { _.each(rmCache, function(quote, index) {
quotes[rmCache[i].key] = []; this.db.append('quote_category', quote.key, quote.quote, function(err, length) {
} if(err) {
quotes[rmCache[i].key].push(rmCache[i].quote); // QQ
} }
rmCache.length = 0; });
});
rmCache.length = 0;
event.reply(dbot.t('quote_cache_reinstated', event.reply(dbot.t('quote_cache_reinstated',
{ 'count': rmCacheCount })); { 'count': rmCacheCount }));
}, },
// Retrieve quote from a category in the database. // Remove last quote from category
'~q': function(event) {
var name = event.input[1].trim().toLowerCase();
this.db.read('quote_category', name, function(err, category) {
if(!err) {
var quoteIndex = _.random(0, category.length - 1);
event.reply(key + ': ' + category[quoteIndex]);
} else if(err instanceof AlreadyExistsError) {
event.reply(dbot.t('category_not_found', { 'category': name }));
}
});
},
// Shows a list of the biggest categories
'~qstats': function(event) {
this.db.readAll('quote_category)
var qSizes = _.chain(quotes)
.pairs()
.sortBy(function(category) { return category[1].length })
.reverse()
.first(10)
.value();
var qString = dbot.t('large_categories');
for(var i=0;i<qSizes.length;i++) {
qString += qSizes[i][0] + " (" + qSizes[i][1].length + "), ";
}
event.reply(qString.slice(0, -2));
},
// Search a given category for some text.
// TODO fix
'~qsearch': function(event) {
var haystack = event.input[1].trim().toLowerCase();
var needle = event.input[2];
if(_.has(quotes, haystack)) {
var matches = _.filter(quotes[haystack], function(quote) {
return quote.indexOf(needle) != -1;
}, this);
if(matches.length == 0) {
event.reply(dbot.t('no_results'));
} else {
event.reply(dbot.t('search_results', {
'category': haystack,
'needle': needle,
'quote': matches[0],
'matches': matches.length
}));
}
} else {
event.reply(dbot.t('empty_category'));
}
},
'~rmlast': function(event) { '~rmlast': function(event) {
if(this.rmAllowed === true || _.include(dbot.config.admins, event.user)) { if(this.rmAllowed === true || _.include(dbot.config.admins, event.user)) {
var key = event.input[1].trim().toLowerCase(); var key = event.input[1].trim().toLowerCase();
if(_.has(quotes, key)) {
var quote = quotes[key].pop();
if(quotes[key].length == 0) {
delete quotes[key];
}
this.internalAPI.resetRemoveTimer(event, key, quote);
event.reply(dbot.t('removed_from', { this.db.slice('quote_category', key, -1, 1, function(err, removed) {
'quote': quote, if(!err) {
'category': key this.internalAPI.resetRemoveTimer(event, key, removed);
})); event.reply(dbot.t('removed_from', {
} else { 'quote': removed,
event.reply(dbot.t('no_quotes', {'category': q[1]})); 'category': key
} }));
} else if(err instanceof NoSuchThingError) {
event.reply(dbot.t('category_not_found', { 'category': key }));
}
}.bind(this));
} else { } else {
event.reply(dbot.t('rmlast_spam')); event.reply(dbot.t('rmlast_spam'));
} }
}, },
// Remove specific quote from category
'~rm': function(event) { '~rm': function(event) {
if(this.rmAllowed == true || _.include(dbot.config.admins, event.user)) { if(this.rmAllowed == true || _.include(dbot.config.admins, event.user)) {
var key = event.input[1].trim().toLowerCase(); var key = event.input[1].trim().toLowerCase();
@ -148,11 +115,68 @@ var commands = function(dbot) {
} }
}, },
/*** Quote Statistics and Searching ***/
// Shows a list of the biggest categories
'~qstats': function(event) {
var quoteSizes = {};
this.db.scan('quote_category', function(category) {
// TODO: get name?
quoteSizes[name] = category.length;
}.bind(this), function(err) {
if(err) {
// QQ
}
});
var qSizes = _.chain(quoteSizes)
.pairs()
.sortBy(function(category) { return category[1] })
.reverse()
.first(10)
.value();
var qString = dbot.t('large_categories');
for(var i=0;i<qSizes.length;i++) {
qString += qSizes[i][0] + " (" + qSizes[i][1].length + "), ";
}
event.reply(qString.slice(0, -2));
},
// Search a given category for some text.
// TODO fix
'~qsearch': function(event) {
var haystack = event.input[1].trim().toLowerCase();
var needle = event.input[2];
this.db.read('quote_category', haystack, function(err, category) {
if(!err) {
var matches = _.filter(category, function(quote) {
return quote.indexOf(needle) != -1;
}, this);
if(matches.length == 0) {
event.reply(dbot.t('no_results'));
} else {
event.reply(dbot.t('search_results', {
'category': haystack,
'needle': needle,
'quote': matches[0],
'matches': matches.length
}));
}
} else if(err == NoSuchThingError) {
event.reply(dbot.t('empty_category'));
}
}.bind(this));
},
'~qcount': function(event) { '~qcount': function(event) {
var input = event.message.valMatch(/^~qcount ([\d\w\s-]*)/, 2); var input = event.message.valMatch(/^~qcount ([\d\w\s-]*)/, 2);
if(input) { // Give quote count for named category if(input) { // Give quote count for named category
var key = input[1].trim().toLowerCase(); var key = input[1].trim().toLowerCase();
this.db.get('quote_category', key, function(err, category) { this.db.read('quote_category', key, function(err, category) {
if(!err) { if(!err) {
event.reply(dbot.t('quote_count', { event.reply(dbot.t('quote_count', {
'category': key, 'category': key,
@ -162,11 +186,15 @@ var commands = function(dbot) {
event.reply(dbot.t('category_not_found', { 'category': name })); event.reply(dbot.t('category_not_found', { 'category': name }));
} }
}.bind(this)); }.bind(this));
} else { // TODO: databankise total quote count } else {
var totalQuoteCount = _.reduce(quotes, function(memo, category) { var quoteCount = 0;
return memo + category.length; this.db.scan('quote_category', function(category) {
}, 0); quoteCount += category.length;
event.reply(dbot.t('total_quotes', { 'count': totalQuoteCount })); }.bind(this), function(err) {
if(!err) {
event.reply(dbot.t('total_quotes', { 'count': quoteCount }));
}
}.bind(this));
} }
}, },
@ -174,17 +202,17 @@ var commands = function(dbot) {
var key = event.input[1].toLowerCase(); var key = event.input[1].toLowerCase();
var quote = event.input[2]; var quote = event.input[2];
this.db.indexOf('quote_category', key, text, function(err, index) { this.db.indexOf('quote_category', key, quote, function(err, index) {
if(index == -1) { if(index == null || index == -1) {
this.db.append('quote_category', key, quote, function(err) { this.db.append('quote_category', key, quote, function(err, newCount) {
this.rmAllowed = true; this.rmAllowed = true;
dbot.api.event.emit('~qadd', { dbot.api.event.emit('~qadd', {
'key': key, 'key': key,
'text': text 'text': quote
}); });
event.reply(dbot.t('quote_saved', { event.reply(dbot.t('quote_saved', {
'category': key, 'category': key,
'count': 0 // TODO: Figure out a way to get the count in the shim 'count': newCount
})); }));
}.bind(this)); }.bind(this));
} else { } else {
@ -200,18 +228,20 @@ var commands = function(dbot) {
'~link': function(event) { '~link': function(event) {
var key = event.input[1].toLowerCase(); var key = event.input[1].toLowerCase();
if(_.has(quotes, key)) { this.db.read('quote_category', key, function(err, category) {
event.reply(dbot.t('quote_link', { if(!err) {
'category': key, event.reply(dbot.t('quote_link', {
'url': dbot.t('url', { 'category': key,
'host': dbot.config.web.webHost, 'url': dbot.t('url', {
'port': dbot.config.web.webPort, 'host': dbot.config.web.webHost,
'path': 'quotes/' + key 'port': dbot.config.web.webPort,
}) 'path': 'quotes/' + key
})); })
} else { }));
event.reply(dbot.t('category_not_found', { 'category': key })); } else if(err == NoSuchThingError) {
} event.reply(dbot.t('category_not_found', { 'category': key }));
}
});
}, },
}; };