0.2.0 - Mid migration

web/node_modules/webpack-dev-server/lib/servers/BaseServer.js

@@ -0,0 +1,9 @@
+'use strict';
+
+// base class that users should extend if they are making their own
+// server implementation
+module.exports = class BaseServer {
+  constructor(server) {
+    this.server = server;
+  }
+};

web/node_modules/webpack-dev-server/lib/servers/SockJSServer.js

@@ -0,0 +1,74 @@
+'use strict';
+
+/* eslint-disable
+  class-methods-use-this,
+  func-names
+*/
+const sockjs = require('sockjs');
+const BaseServer = require('./BaseServer');
+
+// Workaround for sockjs@~0.3.19
+// sockjs will remove Origin header, however Origin header is required for checking host.
+// See https://github.com/webpack/webpack-dev-server/issues/1604 for more information
+{
+  const SockjsSession = require('sockjs/lib/transport').Session;
+  const decorateConnection = SockjsSession.prototype.decorateConnection;
+  SockjsSession.prototype.decorateConnection = function(req) {
+    decorateConnection.call(this, req);
+    const connection = this.connection;
+    if (
+      connection.headers &&
+      !('origin' in connection.headers) &&
+      'origin' in req.headers
+    ) {
+      connection.headers.origin = req.headers.origin;
+    }
+  };
+}
+
+module.exports = class SockJSServer extends BaseServer {
+  // options has: error (function), debug (function), server (http/s server), path (string)
+  constructor(server) {
+    super(server);
+    this.socket = sockjs.createServer({
+      // Use provided up-to-date sockjs-client
+      sockjs_url: '/__webpack_dev_server__/sockjs.bundle.js',
+      // Limit useless logs
+      log: (severity, line) => {
+        if (severity === 'error') {
+          this.server.log.error(line);
+        } else {
+          this.server.log.debug(line);
+        }
+      },
+    });
+
+    this.socket.installHandlers(this.server.listeningApp, {
+      prefix: this.server.sockPath,
+    });
+  }
+
+  send(connection, message) {
+    // prevent cases where the server is trying to send data while connection is closing
+    if (connection.readyState !== 1) {
+      return;
+    }
+
+    connection.write(message);
+  }
+
+  close(connection) {
+    connection.close();
+  }
+
+  // f should be passed the resulting connection and the connection headers
+  onConnection(f) {
+    this.socket.on('connection', (connection) => {
+      f(connection, connection ? connection.headers : null);
+    });
+  }
+
+  onConnectionClose(connection, f) {
+    connection.on('close', f);
+  }
+};

web/node_modules/webpack-dev-server/lib/servers/WebsocketServer.js

@@ -0,0 +1,72 @@
+'use strict';
+
+/* eslint-disable
+  class-methods-use-this
+*/
+const ws = require('ws');
+const BaseServer = require('./BaseServer');
+
+module.exports = class WebsocketServer extends BaseServer {
+  constructor(server) {
+    super(server);
+    this.wsServer = new ws.Server({
+      noServer: true,
+      path: this.server.sockPath,
+    });
+
+    this.server.listeningApp.on('upgrade', (req, sock, head) => {
+      if (!this.wsServer.shouldHandle(req)) {
+        return;
+      }
+
+      this.wsServer.handleUpgrade(req, sock, head, (connection) => {
+        this.wsServer.emit('connection', connection, req);
+      });
+    });
+
+    this.wsServer.on('error', (err) => {
+      this.server.log.error(err.message);
+    });
+
+    const noop = () => {};
+
+    setInterval(() => {
+      this.wsServer.clients.forEach((socket) => {
+        if (socket.isAlive === false) {
+          return socket.terminate();
+        }
+
+        socket.isAlive = false;
+        socket.ping(noop);
+      });
+    }, this.server.heartbeatInterval);
+  }
+
+  send(connection, message) {
+    // prevent cases where the server is trying to send data while connection is closing
+    if (connection.readyState !== 1) {
+      return;
+    }
+
+    connection.send(message);
+  }
+
+  close(connection) {
+    connection.close();
+  }
+
+  // f should be passed the resulting connection and the connection headers
+  onConnection(f) {
+    this.wsServer.on('connection', (connection, req) => {
+      connection.isAlive = true;
+      connection.on('pong', () => {
+        connection.isAlive = true;
+      });
+      f(connection, req.headers);
+    });
+  }
+
+  onConnectionClose(connection, f) {
+    connection.on('close', f);
+  }
+};