dbot/modules/sstats/commands.js

var _ = require('underscore')._,
    async = require('async');

var commands = function(dbot) {
    var commands = {
        '~words': function(event) {
            var getWords = function(user) {
                this.api.getUserStats(user.id, function(uStats) {
                    if(uStats) {
                        var output = dbot.t('sstats_uwords', {
                            'user': user.primaryNick,
                            'words': uStats.words,
                            'curses': uStats.curses,
                            'capitals': uStats.capitals
                        });
                        if(event.rChannel && _.has(uStats.channels, event.rChannel.id)) {
                            ucStats = uStats.channels[event.rChannel.id];
                            output += dbot.t('sstats_ucwords', {
                                'channel': event.channel,
                                'words': ucStats.words,
                                'curses': ucStats.curses,
                                'capitals': ucStats.capitals
                            });
                        }
                        event.reply(output);
                    } else {
                        event.reply(dbot.t('sstats_noustats'));
                    }
                });
            }.bind(this);

            if(event.params[1]) {
                dbot.api.users.resolveUser(event.server, event.params[1], function(user) {
                    if(user) {
                        getWords(user);
                    } else {
                        event.reply(dbot.t('sstats_unknown_user'));
                    }
                });
            } else {
                getWords(event.rUser);
            }
        },

        '~lines': function(event) {
            var getLines = function(user) {
                this.api.getUserStats(user.id, function(uStats) {
                    if(uStats) {
                        var output = dbot.t('sstats_tlines', { 
                            'user': user.primaryNick,
                            'lines': uStats.lines 
                        });
                        if(event.rChannel && _.has(uStats.channels, event.rChannel.id)) {
                            output += dbot.t('sstats_uclines', { 
                                'channel': event.channel,
                                'lines': uStats.channels[event.rChannel.id].lines 
                            });
                        }
                        event.reply(output);
                    } else {
                        event.reply(dbot.t('sstats_noustats'));
                    }
                });       
            }.bind(this);

            if(event.params[1]) {
                dbot.api.users.resolveUser(event.server, event.params[1], function(user) {
                    if(user) {
                        getLines(user);
                    } else {
                        event.reply(dbot.t('sstats_unknown_user'));
                    }
                });
            } else {
                getLines(event.rUser);
            }
        },

        // TODO: too much repeated code between loudest and cloudest yo
        '~loudest': function(event) {
            var lines = {};
            this.db.scan('user_stats', function(uStats) {
                lines[uStats.id] = uStats.lines;   
            }, function() {
                var lCounts = _.chain(lines)
                    .pairs()
                    .sortBy(function(user) { return user[1]; })
                    .reverse()
                    .first(10)
                    .value();

                async.eachSeries(lCounts, function(lCount, next) {
                    dbot.api.users.getUser(lCount[0], function(user) {
                        lCount[0] = user.primaryNick;
                        next();
                    });
                }, function() {
                    var output = "Loudest users: ";
                    for(var i=0;i<lCounts.length;i++) {
                        output += lCounts[i][0] + " (" + lCounts[i][1] + "), ";
                    }
                    event.reply(output.slice(0, -2));
                });
            });
        },

        '~cloudest': function(event) {
            var lines = {};
            this.db.scan('user_stats', function(uStats) {
                if(_.has(uStats.channels, event.rChannel.id)) {
                    lines[uStats.id] = uStats.channels[event.rChannel.id].lines;   
                }
            }, function() {
                var lCounts = _.chain(lines)
                    .pairs()
                    .sortBy(function(user) { return user[1]; })
                    .reverse()
                    .first(10)
                    .value();

                async.eachSeries(lCounts, function(lCount, next) {
                    dbot.api.users.getUser(lCount[0], function(user) {
                        lCount[0] = user.primaryNick;
                        next();
                    });
                }, function() {
                    var output = "Loudest users in " + event.channel + ": ";
                    for(var i=0;i<lCounts.length;i++) {
                        output += lCounts[i][0] + " (" + lCounts[i][1] + "), ";
                    }
                    event.reply(output.slice(0, -2));
                });
            });
        },

        '~clines': function(event) {
            if(!event.cStats) return;
            event.reply(dbot.t('sstats_clines', { 
                'channel': event.channel, 
                'lines': event.cStats.lines 
            }));
        }
    };
    return commands;
};

exports.fetch = function(dbot) {
    return commands(dbot);
};