dbot/modules/atheme/atheme.js

/**
 * Module Name: atheme
 * Description: atheme mode references & retrieve channel flags
 */
var _ = require('underscore')._,
    async = require('async');

var atheme = function(dbot) {
    this.flagStack = {};
    this.hostStack = {};

    this.api = {
        'getChannelFlags': function(server, channel, callback) {
            if(!_.has(this.flagStack, server)) this.flagStack[server] = {};
            if(_.has(this.flagStack[server], channel)) { // Already an active flag call
                this.flagStack[server][channel].callbacks.push(callback);
            } else {
                this.flagStack[server][channel] = {
                    'flags': {},
                    'callbacks': [ callback ]
                };
            }

            dbot.say(server, 'chanserv', 'FLAGS ' + channel);
            setTimeout(function() { // Delete callback if no response
                if(_.has(this.flagStack[server], channel)) {
                    _.each(this.flagStack[server][channel].callbacks, function(callback) {
                        callback(true, null);
                    });
                    delete this.flagStack[server][channel];
                }
            }.bind(this), 20000);
        },

        'getVHosts': function(server, mask, callback) {
            if(!_.has(this.hostStack, server)) this.hostStack[server] = {};
            if(_.has(this.hostStack[server], mask)) { // Already an active host call
                this.hostStack[server][channel].callbacks.push(callback);
            } else {
                this.hostStack[server][mask] = {
                    'users': [],
                    'callbacks': [ callback ]
                };
            }

            dbot.say(server, 'hostserv', 'LISTVHOST ' + mask);
            setTimeout(function() { // Delete callback if no response
                if(_.has(this.hostStack[server], mask)) {
                    _.each(this.hostStack[server][mask].callbacks, function(callback) {
                        callback(true, null);
                    });
                    delete this.hostStack[server][mask];
                }
            }.bind(this), 2000);
        }
    };

    this.commands = {
        '~chanserv': function(event) {
            if(_.has(this.config.chanserv, event.input[1])) {
                event.reply('ChanServ flag ' + event.input[1] + ': ' + this.config.chanserv[event.input[1]]);
            } else {
                event.reply('I don\'t know anything about ' + event.input[1]);
            }
        },

        '~chanmode': function(event) {
            if(_.has(this.config.chanmodes, event.input[1])) {
                event.reply('Channel Mode ' + event.input[1] + ': ' + this.config.chanmodes[event.input[1]]);
            } else {
                event.reply('I don\'t know anything about ' + event.input[1]);
            }
        }
    };
    this.commands['~chanserv'].regex = [/^chanserv (\+.)/, 2];
    this.commands['~chanmode'].regex = [/^chanmode (\+.)/, 2];

    this.listener = function(event) {
        if(event.user === 'ChanServ') {
            var flags = event.params.match(/(\d+)\s+([^ ]+)\s+(\+\w+)\s+\((\#[\w\.]+)\)/),
                end = event.params.match(/end of \u0002(\#[\w\.]+)\u0002 flags listing/i);

            if(flags && _.has(this.flagStack[event.server], flags[4])) {
                this.flagStack[event.server][flags[4]].flags[flags[2]] = flags[3];
            } else if(end) {
                if(_.has(this.flagStack[event.server], end[1])) {
                    // Parse wildcard hostmasks to nicks
                    var allFlags = this.flagStack[event.server][end[1]].flags,
                        hostMasks = {};

                    _.each(allFlags, function(f, u) { // TODO: combine to one loop
                        if(u.indexOf('*!*@') !== -1) {
                            hostMasks[u] = f;
                            delete allFlags[u];
                        }
                    });

                    async.each(_.keys(hostMasks), function(hostMask, done) {
                        this.api.getVHosts(event.server, hostMask.split('@')[1], function(err, users) {
                            _.each(users, function(user) {
                                allFlags[user] = hostMasks[hostMask];
                            });
                            done();
                        });
                    }.bind(this), function() {
                    console.log('DONE');
                        _.each(this.flagStack[event.server][end[1]].callbacks, function(callback) {
                            callback(null, this.flagStack[event.server][end[1]].flags);
                        }.bind(this));
                        delete this.flagStack[event.server][end[1]];
                    }.bind(this));
                }
            }
        } else if(event.user === 'HostServ') {
            _.each(this.hostStack[event.server], function(el, mask) {
                if(event.params.match(mask)) {
                    var user = event.params.match(/- ([^ ]+)/),
                        end = event.params.match(/matches for pattern/);

                    if(user) {
                        this.hostStack[event.server][mask].users.push(user[1]);
                    } else if(end) {
                        _.each(this.hostStack[event.server][mask].callbacks, function(callback) {
                            callback(null, this.hostStack[event.server][mask].users);
                        }, this);
                        delete this.hostStack[event.server][mask];
                    }
                }
            }, this);
        }
    }.bind(this);
    this.on = 'NOTICE';
};

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