Restruct

node_modules/websocket-driver/lib/websocket/driver/base.js

@@ -1,7 +1,6 @@
 'use strict';
 
-var Buffer  = require('safe-buffer').Buffer,
-    Emitter = require('events').EventEmitter,
+var Emitter = require('events').EventEmitter,
     util    = require('util'),
     streams = require('../streams'),
     Headers = require('./headers'),
@@ -26,15 +25,6 @@ var Base = function(request, url, options) {
 };
 util.inherits(Base, Emitter);
 
-Base.isWebSocket = function(request) {
-  var connection = request.headers.connection || '',
-      upgrade    = request.headers.upgrade || '';
-
-  return request.method === 'GET' &&
-         connection.toLowerCase().split(/ *, */).indexOf('upgrade') >= 0 &&
-         upgrade.toLowerCase() === 'websocket';
-};
-
 Base.validateOptions = function(options, validKeys) {
   for (var key in options) {
     if (validKeys.indexOf(key) < 0)
@@ -89,35 +79,13 @@ var instance = {
 
   start: function() {
     if (this.readyState !== 0) return false;
-
-    if (!Base.isWebSocket(this._request))
-      return this._failHandshake(new Error('Not a WebSocket request'));
-
-    var response;
-
-    try {
-      response = this._handshakeResponse();
-    } catch (error) {
-      return this._failHandshake(error);
-    }
-
+    var response = this._handshakeResponse();
+    if (!response) return false;
     this._write(response);
     if (this._stage !== -1) this._open();
     return true;
   },
 
-  _failHandshake: function(error) {
-    var headers = new Headers();
-    headers.set('Content-Type', 'text/plain');
-    headers.set('Content-Length', Buffer.byteLength(error.message, 'utf8'));
-
-    headers = ['HTTP/1.1 400 Bad Request', headers.toString(), error.message];
-    this._write(Buffer.from(headers.join('\r\n'), 'utf8'));
-    this._fail('protocol_error', error.message);
-
-    return false;
-  },
-
   text: function(message) {
     return this.frame(message);
   },
@@ -156,12 +124,6 @@ var instance = {
   _write: function(chunk) {
     var io = this.io;
     if (io.readable) io.emit('data', chunk);
-  },
-
-  _fail: function(type, message) {
-    this.readyState = 2;
-    this.emit('error', new Error(message));
-    this.close();
   }
 };
 
@@ -182,12 +144,4 @@ Base.MessageEvent = function(data) {
   this.data = data;
 };
 
-Base.PingEvent = function(data) {
-  this.data = data;
-};
-
-Base.PongEvent = function(data) {
-  this.data = data;
-};
-
 module.exports = Base;