forked from GitHub/dbot
137 lines
4.5 KiB
JavaScript
137 lines
4.5 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 needs to be somewhere else
|
|
this.isUpperCase = function(word) {
|
|
return _.all(word.split(''), function(c) {
|
|
return c == c.toUpperCase();
|
|
});
|
|
};
|
|
|
|
this.internalAPI = {
|
|
// I'm not a huge fan of this but it's better than all the code
|
|
// repetition
|
|
'highscore': function(key, property, callback) {
|
|
var pList = {},
|
|
pPointer = property.split('.');
|
|
|
|
this.db.scan(key, function(item) {
|
|
var id = item.id;
|
|
for(var i=0;i<pPointer.length;i++) {
|
|
if(_.has(item, pPointer[i])) {
|
|
item = item[pPointer[i]];
|
|
} else {
|
|
item = null; break;
|
|
}
|
|
}
|
|
|
|
if(item) {
|
|
pList[id] = item;
|
|
}
|
|
}, function() {
|
|
var pCounts = _.chain(pList)
|
|
.pairs()
|
|
.sortBy(function(p) { return p[1]; })
|
|
.reverse()
|
|
.first(10)
|
|
.value();
|
|
|
|
callback(pCounts);
|
|
});
|
|
}.bind(this),
|
|
|
|
'formatHighscore': function(string, pCounts) {
|
|
var output = string;
|
|
for(var i=0;i<pCounts.length;i++) {
|
|
output += pCounts[i][0] + " (" + pCounts[i][1] + "), ";
|
|
}
|
|
return output.slice(0, -2);
|
|
}
|
|
};
|
|
|
|
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(_.any(this.config.curses, function(curse) { return word.toLowerCase().indexOf(curse) != -1; })) {
|
|
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);
|
|
};
|