@modern-js/server 2.15.0 → 2.16.0

package/dist/esm/dev-tools/dev-middleware/socketServer.js

@@ -1,214 +1,228 @@
-function _classCallCheck(instance, Constructor) {
-    if (!(instance instanceof Constructor)) {
-        throw new TypeError("Cannot call a class as a function");
-    }
+function _class_call_check(instance, Constructor) {
+  if (!(instance instanceof Constructor)) {
+    throw new TypeError("Cannot call a class as a function");
+  }
 }
 function _defineProperties(target, props) {
-    for(var i = 0; i < props.length; i++){
-        var descriptor = props[i];
-        descriptor.enumerable = descriptor.enumerable || false;
-        descriptor.configurable = true;
-        if ("value" in descriptor) descriptor.writable = true;
-        Object.defineProperty(target, descriptor.key, descriptor);
-    }
+  for (var i = 0; i < props.length; i++) {
+    var descriptor = props[i];
+    descriptor.enumerable = descriptor.enumerable || false;
+    descriptor.configurable = true;
+    if ("value" in descriptor)
+      descriptor.writable = true;
+    Object.defineProperty(target, descriptor.key, descriptor);
+  }
 }
-function _createClass(Constructor, protoProps, staticProps) {
-    if (protoProps) _defineProperties(Constructor.prototype, protoProps);
-    if (staticProps) _defineProperties(Constructor, staticProps);
-    return Constructor;
+function _create_class(Constructor, protoProps, staticProps) {
+  if (protoProps)
+    _defineProperties(Constructor.prototype, protoProps);
+  if (staticProps)
+    _defineProperties(Constructor, staticProps);
+  return Constructor;
 }
-function _defineProperty(obj, key, value) {
-    if (key in obj) {
-        Object.defineProperty(obj, key, {
-            value: value,
-            enumerable: true,
-            configurable: true,
-            writable: true
-        });
-    } else {
-        obj[key] = value;
-    }
-    return obj;
+function _define_property(obj, key, value) {
+  if (key in obj) {
+    Object.defineProperty(obj, key, {
+      value,
+      enumerable: true,
+      configurable: true,
+      writable: true
+    });
+  } else {
+    obj[key] = value;
+  }
+  return obj;
 }
 import ws from "ws";
 import { logger } from "@modern-js/utils";
-var SocketServer = /*#__PURE__*/ function() {
-    "use strict";
-    function SocketServer(options) {
-        _classCallCheck(this, SocketServer);
-        _defineProperty(this, "wsServer", void 0);
-        _defineProperty(this, "sockets", []);
-        _defineProperty(this, "options", void 0);
-        _defineProperty(this, "app", void 0);
-        _defineProperty(this, "stats", void 0);
-        _defineProperty(this, "timer", null);
-        this.options = options;
-    }
-    _createClass(SocketServer, [
-        {
-            // create socket, install socket handler, bind socket event
-            key: "prepare",
-            value: function prepare(app) {
-                var _this = this;
-                var _this_options_client;
-                this.app = app;
-                this.wsServer = new ws.Server({
-                    noServer: true,
-                    path: (_this_options_client = this.options.client) === null || _this_options_client === void 0 ? void 0 : _this_options_client.path
-                });
-                this.app.on("upgrade", function(req, sock, head) {
-                    if (!_this.wsServer.shouldHandle(req)) {
-                        return;
-                    }
-                    _this.wsServer.handleUpgrade(req, sock, head, function(connection) {
-                        _this.wsServer.emit("connection", connection, req);
-                    });
-                });
-                this.wsServer.on("error", function(err) {
-                    logger.error(err);
-                });
-                this.timer = setInterval(function() {
-                    _this.wsServer.clients.forEach(function(socket) {
-                        var extWs = socket;
-                        if (!extWs.isAlive) {
-                            extWs.terminate();
-                        } else {
-                            extWs.isAlive = false;
-                            extWs.ping(function() {});
-                        }
-                    });
-                }, 3e4);
-                this.wsServer.on("connection", function(socket) {
-                    _this.onConnect(socket);
-                });
-            }
-        },
-        {
-            key: "updateStats",
-            value: function updateStats(stats) {
-                this.stats = stats;
-                this.sendStats();
-            }
-        },
-        {
-            // write message to each socket
-            key: "sockWrite",
-            value: function sockWrite(type, data) {
-                var _this = this;
-                this.sockets.forEach(function(socket) {
-                    _this.send(socket, JSON.stringify({
-                        type: type,
-                        data: data
-                    }));
-                });
-            }
-        },
-        {
-            key: "singleWrite",
-            value: function singleWrite(socket, type, data) {
-                this.send(socket, JSON.stringify({
-                    type: type,
-                    data: data
-                }));
-            }
-        },
-        {
-            key: "close",
-            value: function close() {
-                this.sockets.forEach(function(socket) {
-                    socket.close();
-                });
-                if (this.timer) {
-                    clearInterval(this.timer);
-                    this.timer = null;
-                }
-            }
-        },
-        {
-            key: "onConnect",
-            value: function onConnect(socket) {
-                var _this = this;
-                var connection = socket;
-                connection.isAlive = true;
-                connection.on("pong", function() {
-                    connection.isAlive = true;
-                });
-                if (!connection) {
-                    return;
-                }
-                this.sockets.push(connection);
-                connection.on("close", function() {
-                    var idx = _this.sockets.indexOf(connection);
-                    if (idx >= 0) {
-                        _this.sockets.splice(idx, 1);
-                    }
-                });
-                if (this.options.hot || this.options.hot === "only") {
-                    this.singleWrite(connection, "hot");
-                }
-                if (this.options.liveReload) {
-                    this.singleWrite(connection, "liveReload");
-                }
-                if (this.stats) {
-                    this.sendStats(true);
-                }
-            }
-        },
-        {
-            // get standard stats
-            key: "getStats",
-            value: function getStats() {
-                var curStats = this.stats;
-                if (!curStats) {
-                    return null;
-                }
-                var defaultStats = {
-                    all: false,
-                    hash: true,
-                    assets: true,
-                    warnings: true,
-                    errors: true,
-                    errorDetails: false
-                };
-                return curStats.toJson(defaultStats);
+var SocketServer = /* @__PURE__ */ function() {
+  "use strict";
+  function SocketServer2(options) {
+    _class_call_check(this, SocketServer2);
+    _define_property(this, "wsServer", void 0);
+    _define_property(this, "sockets", []);
+    _define_property(this, "options", void 0);
+    _define_property(this, "app", void 0);
+    _define_property(this, "stats", void 0);
+    _define_property(this, "timer", null);
+    this.options = options;
+  }
+  _create_class(SocketServer2, [
+    {
+      key: "prepare",
+      value: (
+        // create socket, install socket handler, bind socket event
+        function prepare(app) {
+          var _this = this;
+          var _this_options_client;
+          this.app = app;
+          this.wsServer = new ws.Server({
+            noServer: true,
+            path: (_this_options_client = this.options.client) === null || _this_options_client === void 0 ? void 0 : _this_options_client.path
+          });
+          this.app.on("upgrade", function(req, sock, head) {
+            if (!_this.wsServer.shouldHandle(req)) {
+              return;
             }
-        },
-        {
-            // determine what message should send by stats
-            key: "sendStats",
-            value: function sendStats() {
-                var force = arguments.length > 0 && arguments[0] !== void 0 ? arguments[0] : false;
-                var stats = this.getStats();
-                if (!stats) {
-                    return null;
-                }
-                var shouldEmit = !force && stats && (!stats.errors || stats.errors.length === 0) && stats.assets && stats.assets.every(function(asset) {
-                    return !asset.emitted;
+            _this.wsServer.handleUpgrade(req, sock, head, function(connection) {
+              _this.wsServer.emit("connection", connection, req);
+            });
+          });
+          this.wsServer.on("error", function(err) {
+            logger.error(err);
+          });
+          this.timer = setInterval(function() {
+            _this.wsServer.clients.forEach(function(socket) {
+              var extWs = socket;
+              if (!extWs.isAlive) {
+                extWs.terminate();
+              } else {
+                extWs.isAlive = false;
+                extWs.ping(function() {
                 });
-                if (shouldEmit) {
-                    return this.sockWrite("still-ok");
-                }
-                this.sockWrite("hash", stats.hash);
-                if (stats.errors && stats.errors.length > 0) {
-                    return this.sockWrite("errors", stats.errors);
-                } else if (stats.warnings && stats.warnings.length > 0) {
-                    return this.sockWrite("warnings", stats.warnings);
-                } else {
-                    return this.sockWrite("ok");
-                }
-            }
-        },
-        {
-            // send message to connecting socket
-            key: "send",
-            value: function send(connection, message) {
-                if (connection.readyState !== 1) {
-                    return;
-                }
-                connection.send(message);
-            }
+              }
+            });
+          }, 3e4);
+          this.wsServer.on("connection", function(socket) {
+            _this.onConnect(socket);
+          });
+        }
+      )
+    },
+    {
+      key: "updateStats",
+      value: function updateStats(stats) {
+        this.stats = stats;
+        this.sendStats();
+      }
+    },
+    {
+      key: "sockWrite",
+      value: (
+        // write message to each socket
+        function sockWrite(type, data) {
+          var _this = this;
+          this.sockets.forEach(function(socket) {
+            _this.send(socket, JSON.stringify({
+              type,
+              data
+            }));
+          });
+        }
+      )
+    },
+    {
+      key: "singleWrite",
+      value: function singleWrite(socket, type, data) {
+        this.send(socket, JSON.stringify({
+          type,
+          data
+        }));
+      }
+    },
+    {
+      key: "close",
+      value: function close() {
+        this.sockets.forEach(function(socket) {
+          socket.close();
+        });
+        if (this.timer) {
+          clearInterval(this.timer);
+          this.timer = null;
+        }
+      }
+    },
+    {
+      key: "onConnect",
+      value: function onConnect(socket) {
+        var _this = this;
+        var connection = socket;
+        connection.isAlive = true;
+        connection.on("pong", function() {
+          connection.isAlive = true;
+        });
+        if (!connection) {
+          return;
         }
-    ]);
-    return SocketServer;
+        this.sockets.push(connection);
+        connection.on("close", function() {
+          var idx = _this.sockets.indexOf(connection);
+          if (idx >= 0) {
+            _this.sockets.splice(idx, 1);
+          }
+        });
+        if (this.options.hot || this.options.hot === "only") {
+          this.singleWrite(connection, "hot");
+        }
+        if (this.options.liveReload) {
+          this.singleWrite(connection, "liveReload");
+        }
+        if (this.stats) {
+          this.sendStats(true);
+        }
+      }
+    },
+    {
+      key: "getStats",
+      value: (
+        // get standard stats
+        function getStats() {
+          var curStats = this.stats;
+          if (!curStats) {
+            return null;
+          }
+          var defaultStats = {
+            all: false,
+            hash: true,
+            assets: true,
+            warnings: true,
+            errors: true,
+            errorDetails: false
+          };
+          return curStats.toJson(defaultStats);
+        }
+      )
+    },
+    {
+      key: "sendStats",
+      value: (
+        // determine what message should send by stats
+        function sendStats() {
+          var force = arguments.length > 0 && arguments[0] !== void 0 ? arguments[0] : false;
+          var stats = this.getStats();
+          if (!stats) {
+            return null;
+          }
+          var shouldEmit = !force && stats && (!stats.errors || stats.errors.length === 0) && stats.assets && stats.assets.every(function(asset) {
+            return !asset.emitted;
+          });
+          if (shouldEmit) {
+            return this.sockWrite("still-ok");
+          }
+          this.sockWrite("hash", stats.hash);
+          if (stats.errors && stats.errors.length > 0) {
+            return this.sockWrite("errors", stats.errors);
+          } else if (stats.warnings && stats.warnings.length > 0) {
+            return this.sockWrite("warnings", stats.warnings);
+          } else {
+            return this.sockWrite("ok");
+          }
+        }
+      )
+    },
+    {
+      key: "send",
+      value: (
+        // send message to connecting socket
+        function send(connection, message) {
+          if (connection.readyState !== 1) {
+            return;
+          }
+          connection.send(message);
+        }
+      )
+    }
+  ]);
+  return SocketServer2;
 }();
 export { SocketServer as default };