dbot/modules/nickserv/nickserv.js

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

var nickserv = function(dbot) {
    this.authStack = {};
    this.userStack = {};
    this.servStack = {};

    this.api = {
        'auth': function(server, nick, callback) {
            var nickserv = dbot.config.servers[server].nickserv,
                infoCommand = this.config.servers[server].infoCommand;

            if(!_.has(this.authStack, server)) this.authStack[server] = {};
            this.authStack[server][nick] = callback;
            dbot.say(server, nickserv, infoCommand + ' ' + nick + ' *');
        },

        'getUserServer': function(server, nick, callback) {
            if(!_.has(this.servStack, server)) this.servStack[server] = {};
            this.servStack[server][nick] = callback;
            dbot.instance.connections[server].send('WHOIS ' + nick);
            setTimeout(function() {
                if(_.has(this.servStack[server], nick)) {
                    callback(false); 
                }
            }.bind(this), 6000);
        },

        'getUserHost': function(server, nick, callback, skipFallback) {
            if(!_.has(this.userStack, server)) this.userStack[server] = {};
            this.userStack[server][nick] = callback;
            dbot.instance.connections[server].send('USERHOST ' + nick);
            setTimeout(function() {
                if(_.has(this.userStack[server], nick)) {
                    if (skipFallback) {
                        callback(false);
                    } else {
                        dbot.instance.connections[server].send('WHOWAS ' + nick + ' 1');
                        setTimeout(function() {
                            if(_.has(this.userStack[server], nick)) {
                                callback(false);
                            }
                        }.bind(this), 2000);
                    }
                }
            }.bind(this), 4000);
        }
    };
    this.api['auth'].external = true;
    this.api['auth'].extMap = [ 'server', 'nick', 'callback' ];

    this.commands = {
        '~auth': function(event) {
            var user = event.params[1] || event.user;
            this.api.auth(event.server, user, function(isAuthed, account) {
                if(isAuthed) {
                    if(user == account) {
                        event.reply(dbot.t('authed', { 'nick': user })); 
                    } else {
                        event.reply(dbot.t('authed_as', { 
                            'nick': user,
                            'account': account
                        })); 
                    }
                } else {
                    event.reply(dbot.t('not_authed', { 'nick': user })); 
                }
            });
        },

        '~hostmask': function(event) {
            var user = event.params[1] || event.user;
            this.api.getUserHost(event.server, user, function(host) {
                if(host) {
                    event.reply(dbot.t('hostmask', {
                        'nick': user,
                        'host': host
                    }));
                } else {
                    event.reply(dbot.t('no_hostmask', { 'nick': user }));
                }
            });
        },

        '~server': function(event) {
            var user = event.params[1] || event.user;
            this.api.getUserServer(event.server, user, function(server) {
                if(server) {
                    event.reply(user + ' is on ' + server);
                } else {
                    event.reply('We don\'t know who ' + user + ' is.');
                }
            });
        }
    };

    this.listener = function(event) {
        if(event.action == 'NOTICE') {
            var nickserv = dbot.config.servers[event.server].nickserv,
                statusRegex = this.config.servers[event.server].matcher,
                acceptableState = this.config.servers[event.server].acceptableState;

            if(event.user == nickserv) {
                var info = event.message.match(statusRegex);
                if(info && _.has(this.authStack, event.server)) {
                    if(info[3] == acceptableState) {
                        this.authStack[event.server][info[1]](true, info[2]);
                    } else {
                        this.authStack[event.server][info[1]](false, info[2]);
                    }
                }
            }
        } else if(event.action == '302') {
            var match = event.message.match(/(.+)=([^@]+)@(.+)$/);

            if(match && match[1]) match[1] = match[1].replace('\*', '');
            if(match && _.has(this.userStack, event.server) && _.has(this.userStack[event.server], match[1])) {
                var callback = this.userStack[event.server][match[1]];
                delete this.userStack[event.server][match[1]];
                callback(match[3].trim());
            }
        } else if(event.action == '314') {
            var params = event.message.split(' '),
                nick = params[1],
                host = params[3];

            if(_.has(this.userStack, event.server) && _.has(this.userStack[event.server], nick)) {
                var callback = this.userStack[event.server][nick];
                delete this.userStack[event.server][nick];
                callback(host);
            }
        } else if(event.action == '338') {
            var params = event.message.split(' '),
                user = event.args[1],
                host = event.args[2];

            if(_.has(this.userStack, event.server) && _.has(this.userStack[event.server], user)) {
                var callback = this.userStack[event.server][user];
                delete this.userStack[event.server][user];
                callback(host);
            }
        }
    }.bind(this);
    this.on = ['NOTICE', '302', '314', '338'];
};

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