forked from GitHub/dbot
194 lines
6.9 KiB
JavaScript
194 lines
6.9 KiB
JavaScript
var _ = require('underscore')._,
|
|
uuid = require('node-uuid');
|
|
|
|
var quotes = function(dbot) {
|
|
dbot.sessionData.rmCache = [];
|
|
this.rmCache = dbot.sessionData.rmCache;
|
|
this.quotes = dbot.db.quoteArrs;
|
|
this.rmAllowed = true;
|
|
this.rmTimer;
|
|
|
|
this.internalAPI = {
|
|
// Parse quote interpolations
|
|
'interpolatedQuote': function(server, channel, username, key, quote, callback) {
|
|
var quoteRefs = quote.match(/~~([\d\w\s-]*)~~/g);
|
|
if(quoteRefs) {
|
|
var ref = quoteRefs[0].replace(/^~~/,'').replace(/~~$/,'').trim();
|
|
if(ref === '-nicks-') {
|
|
if(_.has(dbot.instance.connections[server].channels, channel)) {
|
|
var nicks =
|
|
_.keys(dbot.instance.connections[server].channels[channel].nicks);
|
|
var randomUser = nicks[_.random(0, nicks.length -1)];
|
|
quote = quote.replace('~~' + ref + '~~', randomUser);
|
|
this.internalAPI.interpolatedQuote(server, channel,
|
|
username, key, quote, callback);
|
|
}
|
|
} else if(ref === '-nick-') {
|
|
quote = quote.replace('~~' + ref + '~~', username);
|
|
this.internalAPI.interpolatedQuote(server, channel,
|
|
username, key, quote, callback);
|
|
} else {
|
|
this.api.getQuote(ref, function(interQuote) {
|
|
if(!interQuote) {
|
|
interQuote = '';
|
|
}
|
|
quote = quote.replace('~~' + ref + '~~', interQuote);
|
|
this.internalAPI.interpolatedQuote(server, channel,
|
|
username, key, quote, callback);
|
|
}.bind(this), true);
|
|
}
|
|
} else {
|
|
callback(quote);
|
|
}
|
|
}.bind(this),
|
|
|
|
'resetRemoveTimer': function(event, key, quote) {
|
|
this.rmAllowed = false;
|
|
setTimeout(function() {
|
|
this.rmAllowed = true;
|
|
}.bind(this), 5000);
|
|
|
|
this.rmCache.push({
|
|
'key': key,
|
|
'quote': quote
|
|
});
|
|
|
|
clearTimeout(this.rmTimer);
|
|
if(this.rmCache.length < this.config.rmLimit) {
|
|
this.rmTimer = setTimeout(function() {
|
|
this.rmCache.length = 0; // lol what
|
|
}.bind(this), 600000);
|
|
} else {
|
|
_.each(dbot.config.admins, function(admin) {
|
|
dbot.say(event.server, admin, dbot.t('rm_cache_limit'));
|
|
});
|
|
}
|
|
}.bind(this)
|
|
};
|
|
|
|
this.api = {
|
|
'addQuote': function(key, quote, user, callback) {
|
|
var key = key.toLowerCase().trim(),
|
|
newCount,
|
|
category = false;
|
|
|
|
this.db.search('quote_category', { 'name': key }, function(result) {
|
|
category = result;
|
|
}, function(err) {
|
|
if(!category) {
|
|
var id = uuid.v4();
|
|
category = {
|
|
'id': id,
|
|
'name': key,
|
|
'quotes': [],
|
|
'owner': user
|
|
};
|
|
}
|
|
|
|
if(_.include(category.quotes, quote)) {
|
|
callback(false);
|
|
} else {
|
|
newCount = category.quotes.push(quote);
|
|
this.db.save('quote_category', category.id, category, function(err) {
|
|
this.rmAllowed = true;
|
|
callback(newCount);
|
|
}.bind(this));
|
|
}
|
|
}.bind(this));
|
|
|
|
},
|
|
|
|
'getQuote': function(key, callback, removeRefs) {
|
|
this.api.getQuoteCategory(key, function(category) {
|
|
if(category) {
|
|
var quotes = category.quotes;
|
|
|
|
if(removeRefs) {
|
|
quotes = _.filter(quotes, function(q) {
|
|
console.log('DEBUG (if removeRefs): ' + removeRefs + quotes + key)
|
|
return !q.match('~~'+key+'~~');
|
|
});
|
|
}
|
|
|
|
var index = _.random(0, quotes.length - 1);
|
|
callback(quotes[index]);
|
|
} else {
|
|
callback(false);
|
|
}
|
|
});
|
|
},
|
|
|
|
'getInterpolatedQuote': function(server, channel, username, key, callback) {
|
|
console.log('DEBUG: ' + channel + ' ' + username + ' ' + key)
|
|
key = key.trim().toLowerCase(),
|
|
|
|
this.api.getQuote(key, function(quote) {
|
|
if(quote) {
|
|
this.internalAPI.interpolatedQuote(server, channel,
|
|
username, key, quote, callback);
|
|
} else {
|
|
console.log('DEBUG (else callback quote): ' + quote)
|
|
callback(quote);
|
|
}
|
|
}.bind(this));
|
|
},
|
|
|
|
'getQuoteCategory': function(key, callback) {
|
|
console.log('DEBUG: ' + key)
|
|
var category = false,
|
|
key = key.trim().toLowerCase();
|
|
|
|
this.db.search('quote_category', { 'name': key }, function(result) {
|
|
console.log('DEBUG: ' + key + ' ' + result)
|
|
category = result;
|
|
}, function(err) {
|
|
callback(category);
|
|
});
|
|
},
|
|
|
|
'getCategoryKeys': function(callback) {
|
|
var keys = [];
|
|
this.db.scan('quote_category', function(result) {
|
|
console.log('DEBUG: ' + result)
|
|
if(result) keys.push(result.name);
|
|
}, function(err) {
|
|
callback(keys);
|
|
});
|
|
}
|
|
};
|
|
|
|
this.api['getQuote'].external = true;
|
|
this.api['getQuote'].extMap = [ 'key', 'callback' ];
|
|
this.api['getQuoteCategory'].external = true;
|
|
this.api['getQuoteCategory'].extMap = [ 'key', 'callback' ];
|
|
|
|
this.listener = function(event) {
|
|
if(event.action == 'PRIVMSG') {
|
|
if(event.user == 'reality') {
|
|
var once = event.message.valMatch(/^I ([\d\w\s,'-]* once)/, 2);
|
|
} else {
|
|
var once = event.message.valMatch(/^reality ([\d\w\s,'-]* once)/, 2);
|
|
}
|
|
|
|
if(once) {
|
|
this.api.addQuote('realityonce', 'reality' + once[1], event.user, function(newCount) {
|
|
event.reply('\'reality ' + once[1] + '\' saved (' + newCount + ').');
|
|
});
|
|
}
|
|
} else if(event.action == 'JOIN') {
|
|
if(this.config.quotesOnJoin == true) {
|
|
this.api.getQuote(event.user, function(quote) {
|
|
if(quote) {
|
|
event.reply(event.user + ': ' + quote);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
}.bind(this);
|
|
this.on = ['PRIVMSG', 'JOIN'];
|
|
};
|
|
|
|
exports.fetch = function(dbot) {
|
|
return new quotes(dbot);
|
|
};
|