yaffs.net Git - yaffs-website/blob - node_modules/websocket-driver/lib/websocket/driver/base.js

   1 'use strict';
   2
   3 var Emitter = require('events').EventEmitter,
   4     util    = require('util'),
   5     streams = require('../streams'),
   6     Headers = require('./headers'),
   7     Reader  = require('./stream_reader');
   8
   9 var Base = function(request, url, options) {
  10   Emitter.call(this);
  11   Base.validateOptions(options || {}, ['maxLength', 'masking', 'requireMasking', 'protocols']);
  12
  13   this._request   = request;
  14   this._reader    = new Reader();
  15   this._options   = options || {};
  16   this._maxLength = this._options.maxLength || this.MAX_LENGTH;
  17   this._headers   = new Headers();
  18   this.__queue    = [];
  19   this.readyState = 0;
  20   this.url        = url;
  21
  22   this.io = new streams.IO(this);
  23   this.messages = new streams.Messages(this);
  24   this._bindEventListeners();
  25 };
  26 util.inherits(Base, Emitter);
  27
  28 Base.validateOptions = function(options, validKeys) {
  29   for (var key in options) {
  30     if (validKeys.indexOf(key) < 0)
  31       throw new Error('Unrecognized option: ' + key);
  32   }
  33 };
  34
  35 var instance = {
  36   // This is 64MB, small enough for an average VPS to handle without
  37   // crashing from process out of memory
  38   MAX_LENGTH: 0x3ffffff,
  39
  40   STATES: ['connecting', 'open', 'closing', 'closed'],
  41
  42   _bindEventListeners: function() {
  43     var self = this;
  44
  45     // Protocol errors are informational and do not have to be handled
  46     this.messages.on('error', function() {});
  47
  48     this.on('message', function(event) {
  49       var messages = self.messages;
  50       if (messages.readable) messages.emit('data', event.data);
  51     });
  52
  53     this.on('error', function(error) {
  54       var messages = self.messages;
  55       if (messages.readable) messages.emit('error', error);
  56     });
  57
  58     this.on('close', function() {
  59       var messages = self.messages;
  60       if (!messages.readable) return;
  61       messages.readable = messages.writable = false;
  62       messages.emit('end');
  63     });
  64   },
  65
  66   getState: function() {
  67     return this.STATES[this.readyState] || null;
  68   },
  69
  70   addExtension: function(extension) {
  71     return false;
  72   },
  73
  74   setHeader: function(name, value) {
  75     if (this.readyState > 0) return false;
  76     this._headers.set(name, value);
  77     return true;
  78   },
  79
  80   start: function() {
  81     if (this.readyState !== 0) return false;
  82     var response = this._handshakeResponse();
  83     if (!response) return false;
  84     this._write(response);
  85     if (this._stage !== -1) this._open();
  86     return true;
  87   },
  88
  89   text: function(message) {
  90     return this.frame(message);
  91   },
  92
  93   binary: function(message) {
  94     return false;
  95   },
  96
  97   ping: function() {
  98     return false;
  99   },
 100
 101   pong: function() {
 102       return false;
 103   },
 104
 105   close: function(reason, code) {
 106     if (this.readyState !== 1) return false;
 107     this.readyState = 3;
 108     this.emit('close', new Base.CloseEvent(null, null));
 109     return true;
 110   },
 111
 112   _open: function() {
 113     this.readyState = 1;
 114     this.__queue.forEach(function(args) { this.frame.apply(this, args) }, this);
 115     this.__queue = [];
 116     this.emit('open', new Base.OpenEvent());
 117   },
 118
 119   _queue: function(message) {
 120     this.__queue.push(message);
 121     return true;
 122   },
 123
 124   _write: function(chunk) {
 125     var io = this.io;
 126     if (io.readable) io.emit('data', chunk);
 127   }
 128 };
 129
 130 for (var key in instance)
 131   Base.prototype[key] = instance[key];
 132
 133
 134 Base.ConnectEvent = function() {};
 135
 136 Base.OpenEvent = function() {};
 137
 138 Base.CloseEvent = function(code, reason) {
 139   this.code   = code;
 140   this.reason = reason;
 141 };
 142
 143 Base.MessageEvent = function(data) {
 144   this.data = data;
 145 };
 146
 147 module.exports = Base;