socket.js

/**
 * Module dependencies.
 */

var EventEmitter = require('events');

/**
 * Module exports.
 */

module.exports = Socket;

/**
 * Abtract Socket class.
 *
 * @param {http.Request} the node.js http Request
 * @api private
 */

function Socket (req) {
  this.req = req;
  this.socket = req.socket;
  this.readyState = 'opening';

  var self = this;
  this.socket
    .on('error', function (e) {
      self.emit('error', e);
      self.destroy();
    })
    .on('end', function () {
      self.close();
    })
    .on('close', function () {
      self.onClose();
    })
    .on('timeout', function() {
      self.destroy();
    })

  this.onOpen();
}

/**
 * Inherits from EventEmitter.
 */

Socket.prototype.__proto__ = EventEmitter.prototype;

/**
 * Request that originated the connection.
 *
 * @api public
 */

Socket.prototype.req;

/**
 * Stream that originated the connection.
 *
 * @api public
 */

Socket.prototype.socket;

/**
 * Called upon socket close.
 *
 * @return {Socket} for chaining.
 * @api private
 */

Socket.prototype.onClose = function () {
  if ('closed' != this.readyState) {
    this.socket.destroy();
    this.readyState = 'closed';
    this.emit('close');
  }

  return this;
};

/**
 * Handles a message.
 *
 * @param {String} message
 * @return {Socket} for chaining
 * @api public
 */

Socket.prototype.onMessage = function (msg) {
  this.emit('message', msg);
  this.emit('data', msg);
  return this;
};

/**
 * Writes to the socket.
 *
 * @return {Socket} for chaining
 * @api public
 */

Socket.prototype.send = function (data) {
  this.write(data);
  return this;
};

/**
 * Closes the connection.
 *
 * @api public
 */

Socket.prototype.close = Socket.prototype.end = function () {
  if ('open' == this.readyState) {
    this.socket.end();
    var self = this;
    process.nextTick(function () {
      // wait for 'end' packet from client
      self.onClose();
    });
  } else {
    this.onClose();
  }
  return this;
};

/**
 * Destroys the connection.
 *
 * @api public
 */

Socket.prototype.destroy = function () {
  if ('closed' != this.readyState) {
    this.socket.destroy();
    this.onClose();
  }
  return this;
};