irc-socket.js

/**
 *
 * IRC Socket
 *
 * Socket that connects to an IRC network and emits each line from the server.
 *
 * Send messages to server with .raw(String) method.
 */

var net = require('net');
var tls = require('tls');
var events = require('events');
var util = require('util');

 // var log = function (input, msg) {
 //     var date = new Date();
 //     console.log(Date().toString() + "|" + (input ? "<-" : "->") + "|" + msg);
 // };

var create = function (prototype, properties) {
    if (typeof properties == 'object') {
        var props = {};
        Object.keys(properties).forEach(function (key) {
            props[key] = { value: properties[key] };
        });
    }

    return Object.create(prototype, props);
};

var Socket = module.exports = function Socket (network, NetSocket) {
    NetSocket = NetSocket || net.Socket;

    var socket = create(Socket.prototype);
    socket.port = network.port || 6667;
    socket.netname = network.server;
    socket.secure = network.secure || false;
    socket.capab = network.capab || false;
    socket.password = network.password || null;
    socket.network = network;
    socket.impl = new NetSocket();
    socket.connected = false;

    socket._setupEvents = function () {
        var onData = function onData (data) {
            lines = data.split('\r\n');

            if (onData.buffer) {
                lines[0] = onData.buffer + lines[0];
                onData.buffer = null;
            }

            if (lines[lines.length - 1] !== "") {
                onData.buffer = lines.pop();
            }

            lines
            .filter(function (line) { return line !== ''; })
            .filter(function (line) {
                if (line.slice(0, 4) === 'PING') {
                    socket.raw(['PONG', line.slice(line.indexOf(':'))]);
                    return false;
                }

                return true;
            })
            .forEach(function (line) {
                //log(true, line);
                socket.emit('data', line);
            });
        };

        onData.buffer = null;

        void function readyEvent () {
            var emitWhenReady = function (data) {
                if (Socket.isReady(data)) {
                    socket.emit('ready');
                }
            };

            socket.impl.on('data', emitWhenReady);
            socket.on('ready', function remove () {
                socket.impl.removeListener('data', emitWhenReady);
            });
        }();

        void function connectEvent () {
            var emitEvent = (socket.secure) ? 'secureConnect' : 'connect';
            var emitWhenConnected = function () {
                socket.connected = true;

                if (socket.capab) {
                    socket.raw(["CAP", "LS"]);
                }

                if (socket.password !== null) {
                    socket.raw(["PASS", socket.password]);
                }

                socket.raw(["NICK", socket.network.nick]);
                socket.raw(["USER", socket.network.user || "user", "8 * :" + socket.network.realname]);
            };

            socket.impl.on(emitEvent, emitWhenConnected);
        }();

        socket.impl.on('error', function () {
            socket.connected = false;
            socket.emit('error');
        });

        socket.impl.on('close', function () {
            socket.connected = false;
            socket.emit('close');
        });

        socket.impl.on('data', onData);
        socket.impl.setEncoding('utf-8');
        socket.impl.setNoDelay();
    };

    socket._setupEvents();

    return socket;
};

Socket.isReady = function (data) {
    // We are 'ready' when we get a 001 message.
    return data.split('\r\n')
    .filter(function (line) { return line !== ''; })
    .some(function (line) { return line.split(' ')[1] === '001'; });
};

Socket.prototype = create(events.EventEmitter.prototype, {
    connect : function () {
        if (this.isConnected()) {
            return;
        }

        if (this.secure) {
            this.impl = tls.connect(this.port, this.netname, {rejectUnauthorized: false});
            this._setupEvents();
            // set rejectUnauthorized because most IRC servers don't have certified certificates anyway.
        } else {
            this.impl.connect(this.port, this.netname);
        }
    },

    end : function () {
        if (!this.isConnected()) {
            return;
        }

        this.impl.end();
    },

    raw : function (message) {
        if (!this.connected) {
            return;
        }

        if (util.isArray(message)) {
            message = message.join(" ");
        }

        if (message.indexOf('\n') !== -1) {
            throw new Error('Newline detected in message. Use multiple raws instead.');
        }

        //log(false, message);
        this.impl.write(message + '\r\n', 'utf-8');
    },

    isConnected : function () {
        return this.connected;
    },

    getRealName : function () {
        return this._realname;
    }
});