3
0
mirror of https://github.com/reality/dbot.git synced 2024-12-29 14:12:40 +01:00
dbot/modules/quotes/quotes.js
2014-09-08 10:41:09 +00:00

189 lines
6.6 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) {
console.log(quote);
var quoteRefs = quote.match(/~~([\d\w\s-]*)~~/g);
if(quoteRefs) {
var ref = this.internalAPI.cleanRef(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 || ref == key) {
interQuote = '';
}
quote = quote.replace('~~' + ref + '~~', interQuote);
this.internalAPI.interpolatedQuote(server, channel,
username, key, quote, callback);
}.bind(this));
}
} else {
callback(quote);
}
}.bind(this),
'cleanRef': function(key) {
key = key.toLowerCase();
while(key.slice(-1) == '_') {
key = key.substring(0, key.length-1);
}
return key;
},
'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) {
this.api.getQuoteCategory(key, function(category) {
if(category) {
var quotes = category.quotes;
var index = _.random(0, quotes.length - 1);
callback(quotes[index]);
} else {
callback(false);
}
});
},
'getInterpolatedQuote': function(server, channel, username, key, callback) {
console.log(key);
key = key.trim().toLowerCase(),
this.api.getQuote(key, function(quote) {
if(quote) {
this.internalAPI.interpolatedQuote(server, channel,
username, key, quote, callback);
} else {
callback(quote);
}
}.bind(this));
},
'getQuoteCategory': function(key, callback) {
var category = false,
key = key.trim().toLowerCase();
this.db.search('quote_category', { 'name': key }, function(result) {
category = result;
}, function(err) {
callback(category);
});
},
'getCategoryKeys': function(callback) {
var keys = [];
this.db.scan('quote_category', function(result) {
if(result) keys.push(result.name);
}, function(err) {
callback(keys);
});
}
};
this.api['getQuote'].external = true;
this.api['getQuote'].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);
};