@atlaspack/utils 2.14.5-canary.21 → 2.14.5-canary.210

package/lib/http-server.js

@@ -0,0 +1,85 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.createHTTPServer = createHTTPServer;
+function _http() {
+  const data = _interopRequireDefault(require("http"));
+  _http = function () {
+    return data;
+  };
+  return data;
+}
+function _https() {
+  const data = _interopRequireDefault(require("https"));
+  _https = function () {
+    return data;
+  };
+  return data;
+}
+function _nullthrows() {
+  const data = _interopRequireDefault(require("nullthrows"));
+  _nullthrows = function () {
+    return data;
+  };
+  return data;
+}
+var _ = require("./");
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
+// Creates either an http or https server with an awaitable dispose
+// that closes any connections
+async function createHTTPServer(options) {
+  let server;
+  if (!options.https) {
+    server = _http().default.createServer(options.listener);
+  } else if (options.https === true) {
+    let {
+      cert,
+      key
+    } = await (0, _.generateCertificate)(options.outputFS, options.cacheDir, options.host);
+    server = _https().default.createServer({
+      cert,
+      key
+    }, options.listener);
+  } else {
+    let {
+      cert,
+      key
+    } = await (0, _.getCertificate)(options.inputFS, options.https);
+    server = _https().default.createServer({
+      cert,
+      key
+    }, options.listener);
+  }
+
+  // HTTPServer#close only stops accepting new connections, and does not close existing ones.
+  // Before closing, destroy any active connections through their sockets. Additionally, remove sockets when they close:
+  // https://stackoverflow.com/questions/18874689/force-close-all-connections-in-a-node-js-http-server
+  // https://stackoverflow.com/questions/14626636/how-do-i-shutdown-a-node-js-https-server-immediately/14636625#14636625
+  let sockets = new Set();
+  server.on('connection', socket => {
+    (0, _nullthrows().default)(sockets).add(socket);
+    socket.on('close', () => {
+      (0, _nullthrows().default)(sockets).delete(socket);
+    });
+  });
+  return {
+    server,
+    stop() {
+      return new Promise((resolve, reject) => {
+        for (let socket of (0, _nullthrows().default)(sockets)) {
+          socket.destroy();
+        }
+        sockets = new Set();
+        server.close(err => {
+          if (err != null) {
+            reject(err);
+            return;
+          }
+          resolve();
+        });
+      });
+    }
+  };
+}