forked from GitHub/dbot
93 lines
3.1 KiB
JavaScript
93 lines
3.1 KiB
JavaScript
/**
|
|
* Module Name: sstats
|
|
* Description: Simple Stats, in the absence of good ones.
|
|
*/
|
|
var _ = require('underscore')._;
|
|
|
|
var sstats = function(dbot) {
|
|
if(!_.has(dbot.db, 'ssinception')) dbot.db.ssinception = new Date().getTime();
|
|
|
|
this.isUpperCase = function(word) {
|
|
return _.all(word.split(''), function(c) {
|
|
return c == c.toUpperCase();
|
|
});
|
|
};
|
|
|
|
this.listener = function(event) {
|
|
event.cStats.lines++;
|
|
event.uStats.lines++;
|
|
|
|
var words = event.message.split(' '),
|
|
wCount = words.length,
|
|
capitals = 0,
|
|
curses = 0;
|
|
_.each(words, function(word) {
|
|
if(this.isUpperCase(word)) capitals++;
|
|
if(_.include(this.config.curses, word)) curses++;
|
|
}, this);
|
|
|
|
event.uStats.words += wCount;
|
|
event.uStats.capitals += capitals;
|
|
event.uStats.curses += curses;
|
|
|
|
event.cStats.words += wCount;
|
|
event.cStats.capitals += capitals;
|
|
event.cStats.curses += curses;
|
|
|
|
if(!_.has(event.uStats.channels, event.rChannel.id)) {
|
|
event.uStats.channels[event.rChannel.id] = {
|
|
'lines': 1,
|
|
'words': wCount,
|
|
'capitals': capitals,
|
|
'curses': curses
|
|
};
|
|
} else {
|
|
event.uStats.channels[event.rChannel.id].lines++;
|
|
event.uStats.channels[event.rChannel.id].words += wCount;
|
|
event.uStats.channels[event.rChannel.id].capitals += capitals;
|
|
event.uStats.channels[event.rChannel.id].curses += curses;
|
|
}
|
|
this.db.save('channel_stats', event.cStats.id, event.cStats, function() {});
|
|
this.db.save('user_stats', event.uStats.id, event.uStats, function() {});
|
|
}.bind(this);
|
|
this.on = 'PRIVMSG';
|
|
|
|
this.onLoad = function() {
|
|
// Preload user stats
|
|
dbot.instance.addPreEmitHook(function(event, callback) {
|
|
if(!event.rUser) return callback();
|
|
this.api.getUserStats(event.rUser.id, function(uStats) {
|
|
if(uStats) {
|
|
event.uStats = uStats;
|
|
callback();
|
|
} else {
|
|
this.api.createUserStats(event.rUser.id, function(uStats) {
|
|
event.uStats = uStats;
|
|
callback();
|
|
});
|
|
}
|
|
}.bind(this));
|
|
}.bind(this));
|
|
|
|
// Preload channel stats
|
|
dbot.instance.addPreEmitHook(function(event, callback) {
|
|
if(!event.rChannel) return callback();
|
|
this.api.getChannelStats(event.rChannel.id, function(cStats) {
|
|
if(cStats) {
|
|
event.cStats = cStats;
|
|
callback();
|
|
} else {
|
|
this.api.createChannelStats(event.rChannel.id, function(cStats) {
|
|
event.cStats = cStats;
|
|
callback();
|
|
});
|
|
}
|
|
}.bind(this));
|
|
}.bind(this));
|
|
}.bind(this);
|
|
};
|
|
|
|
exports.fetch = function(dbot) {
|
|
return new sstats(dbot);
|
|
};
|