unplugin-https-reverse-proxy 0.0.12 → 0.0.13

package/dist/index.cjs

@@ -3,8 +3,22 @@ var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -44,6 +58,15 @@ var import_consola = require("consola");
 var consola = (0, import_consola.createConsola)({
   level: 3
 }).withTag("reverse-proxy");
+function once(fn) {
+  let called = false;
+  return (...args) => {
+    if (called)
+      return;
+    called = true;
+    return fn(...args);
+  };
+}
 
 // src/caddy/index.ts
 var import_node_os = require("os");
@@ -53,7 +76,7 @@ var import_node_child_process = require("child_process");
 var import_got_cjs = require("got-cjs");
 var import_http_proxy_agent = require("http-proxy-agent");
 var import_https_proxy_agent = require("https-proxy-agent");
-var hostile = __toESM(require("hostile"), 1);
+var import_hosts_so_easy = __toESM(require("hosts-so-easy"), 1);
 var import_kill_port = __toESM(require("kill-port"), 1);
 
 // src/caddy/constants.ts
@@ -175,6 +198,8 @@ var supportList = [
 
 // src/caddy/utils.ts
 var import_node_net = require("net");
+var import_ora = __toESM(require("ora"), 1);
+var spinner;
 function tryPort(port) {
   return new Promise((resolve, reject) => {
     try {
@@ -193,8 +218,41 @@ function tryPort(port) {
     }
   });
 }
+function getNumber(number) {
+  if (number <= 0)
+    return "0%";
+  else if (number < 100)
+    return `${number.toFixed(2)}%`;
+  else
+    return "100%";
+}
+function logProgress(percent) {
+  if (!spinner) {
+    consola.log("");
+    spinner = (0, import_ora.default)({
+      text: "Start Download Caddy: 0%",
+      indent: 2
+    });
+    spinner.start();
+  } else {
+    spinner.text = `Download Caddy: ${getNumber(Number(percent) * 100)}`;
+  }
+}
+function logProgressOver() {
+  if (!spinner) {
+    spinner = (0, import_ora.default)({
+      text: "Start Download Caddy: 100%",
+      indent: 2
+    });
+    spinner.start();
+  } else {
+    spinner.succeed();
+    consola.log("");
+  }
+}
 
 // src/caddy/index.ts
+var Hosts = import_hosts_so_easy.default.default || import_hosts_so_easy.default;
 async function download() {
   if (await testCaddy())
     return caddyPath;
@@ -218,7 +276,10 @@ async function download() {
         http: httpAgent,
         https: httpsAgent
       }
+    }).on("downloadProgress", (progress) => {
+      logProgress(progress.percent);
     }).pipe(file).on("finish", () => {
+      logProgressOver();
       if (import_node_process2.default.platform === "win32")
         return resolve(caddyPath);
       (0, import_node_fs.chmodSync)(caddyPath, 493);
@@ -233,7 +294,7 @@ function testCaddy() {
     if (!(0, import_node_fs.existsSync)(caddyPath))
       return resolve(false);
     (0, import_node_fs.chmodSync)(caddyPath, 493);
-    const child = import_node_process2.default.platform === "win32" ? (0, import_node_child_process.spawn)(caddyPath, []) : (0, import_node_child_process.spawn)("sudo", [caddyPath]);
+    const child = import_node_process2.default.platform === "win32" ? (0, import_node_child_process.spawn)(caddyPath, []) : (0, import_node_child_process.spawn)("sudo", ["-E", caddyPath]);
     child.on("close", () => {
       return resolve(false);
     });
@@ -248,6 +309,8 @@ function testCaddy() {
 var CaddyInstant = class {
   constructor() {
     this.inited = false;
+    this.stoped = false;
+    this.host = new Hosts();
     testCaddy().then(() => {
       this.inited = true;
     });
@@ -256,52 +319,46 @@ var CaddyInstant = class {
     await download();
     this.inited = true;
   }
-  async updateHost(ip, host) {
+  updateHost(ip, host) {
     ["localhost", "0.0.0.0"].includes(ip) && (ip = "127.0.0.1");
+    this.host.add(ip, host);
     return new Promise((resolve, reject) => {
-      hostile.set(ip, host, (err) => {
-        if (err) {
-          return reject(err);
-        } else {
-          if (import_node_process2.default.platform === "win32") {
-            (0, import_node_child_process.spawn)("ipconfig", ["/flushdns"]).on("error", (err2) => {
-              return reject(err2);
-            }).on("close", () => {
-              resolve(true);
-            });
-          } else {
-            (0, import_node_child_process.spawn)("sudo", ["killall", "-HUP", "mDNSResponder"]).on("error", (err2) => {
-              return reject(err2);
-            }).on("close", () => {
-              resolve(true);
-            });
-          }
-        }
-      });
+      if (import_node_process2.default.platform === "win32") {
+        (0, import_node_child_process.spawn)("ipconfig", ["/flushdns"]).on("error", (err) => {
+          consola.error(err);
+          return resolve(false);
+        }).on("close", () => {
+          resolve(true);
+        });
+      } else {
+        (0, import_node_child_process.spawn)("sudo", ["-E", "killall", "-HUP", "mDNSResponder"]).on("error", (err) => {
+          consola.error(err);
+          return resolve(false);
+        }).on("close", () => {
+          resolve(true);
+        });
+      }
     });
   }
-  async restoreHost(ip, host) {
+  restoreHost(ip, host) {
     ["localhost", "0.0.0.0"].includes(ip) && (ip = "127.0.0.1");
+    this.host.remove(ip, host);
     return new Promise((resolve, reject) => {
-      hostile.remove(ip, host, (err) => {
-        if (err) {
-          return reject(err);
-        } else {
-          if (import_node_process2.default.platform === "win32") {
-            (0, import_node_child_process.spawn)("ipconfig", ["/flushdns"]).on("error", (err2) => {
-              return reject(err2);
-            }).on("close", () => {
-              resolve(true);
-            });
-          } else {
-            (0, import_node_child_process.spawn)("sudo", ["killall", "-HUP", "mDNSResponder"]).on("error", (err2) => {
-              return reject(err2);
-            }).on("close", () => {
-              resolve(true);
-            });
-          }
-        }
-      });
+      if (import_node_process2.default.platform === "win32") {
+        (0, import_node_child_process.spawn)("ipconfig", ["/flushdns"]).on("error", (err) => {
+          consola.error(err);
+          return resolve(false);
+        }).on("close", () => {
+          resolve(true);
+        });
+      } else {
+        (0, import_node_child_process.spawn)("sudo", ["-E", "killall", "-HUP", "mDNSResponder"]).on("error", (err) => {
+          consola.error(err);
+          return resolve(false);
+        }).on("close", () => {
+          resolve(true);
+        });
+      }
     });
   }
   /**
@@ -318,14 +375,15 @@ var CaddyInstant = class {
     } = options || {};
     if (!this.inited)
       await this.init();
-    if (!await this.updateHost(source.split(":")[0], target.split(":")[0]))
-      throw new Error("update host failed");
     if (await tryPort(443))
       await (0, import_kill_port.default)(443, "tcp");
     if (import_node_process2.default.platform === "win32" && await tryPort(80))
       await (0, import_kill_port.default)(80, "tcp");
+    if (!await this.updateHost(source.split(":")[0], target.split(":")[0]))
+      throw new Error("update host failed");
+    consola.success("update host success\n");
     return new Promise((resolve, reject) => {
-      const child = import_node_process2.default.platform !== "win32" ? (0, import_node_child_process.spawn)("sudo", [caddyPath, "reverse-proxy", "--from", `${target.split(":")[0]}`, "--to", `${source}`, "--internal-certs"]) : (0, import_node_child_process.spawn)(caddyPath, ["reverse-proxy", "--from", `${target.split(":")[0]}`, "--to", `${source}`, "--internal-certs"]);
+      const child = import_node_process2.default.platform !== "win32" ? (0, import_node_child_process.spawn)("sudo", ["-E", caddyPath, "reverse-proxy", "--from", `${target.split(":")[0]}`, "--to", `${source}`, "--internal-certs"]) : (0, import_node_child_process.spawn)(caddyPath, ["reverse-proxy", "--from", `${target.split(":")[0]}`, "--to", `${source}`, "--internal-certs"]);
       child.on("error", (err) => {
         return reject(err);
       });
@@ -341,14 +399,21 @@ var CaddyInstant = class {
       });
       child.stdout.on("data", (_data) => {
         consola.info(_data.toString());
-        resolve(async () => {
-          if (!restore)
+        resolve(async (callback) => {
+          if (!restore || this.stoped)
             return;
-          if (await this.restoreHost(source.split(":")[0], target.split(":")[0]))
-            consola.success("restore host success\n");
-          else
+          try {
+            if (await this.restoreHost(source.split(":")[0], target.split(":")[0]))
+              consola.success("restore host success\n");
+            else
+              consola.fail("restore host failed\n");
+          } catch (e) {
+            consola.error(e);
             consola.fail("restore host failed\n");
+          }
+          this.stoped = true;
           child.kill();
+          callback && callback();
         });
       });
     });
@@ -358,12 +423,53 @@ var CaddyInstant = class {
 // src/index.ts
 var config;
 var caddy;
+var _stop;
+var _servce;
+function registerExit(clear) {
+  import_node_process3.default.on("beforeExit", async (_code) => {
+    await clear();
+  });
+  import_node_process3.default.on("uncaughtException", async (err) => {
+    console.error("An uncaught error occurred!", err.stack);
+    await clear();
+    import_node_process3.default.nextTick(() => {
+      import_node_process3.default.exit(1);
+    });
+  });
+  import_node_process3.default.on("unhandledRejection", async (reason, promise) => {
+    console.error("Unhandled Rejection at:", promise, "reason:", reason);
+    await clear();
+    import_node_process3.default.nextTick(() => {
+      import_node_process3.default.exit(1);
+    });
+  });
+  import_node_process3.default.once("SIGINT", async () => {
+    await clear();
+    import_node_process3.default.emit("SIGINT");
+  });
+  import_node_process3.default.once("SIGTERM", async () => {
+    await clear();
+    import_node_process3.default.emit("SIGTERM");
+  });
+}
 var unpluginFactory = (options) => ({
   name: "unplugin-https-reverse-proxy",
-  enforce: "post",
+  enforce: "pre",
   vite: {
     configResolved(_config) {
       config = _config;
+      registerExit(async () => {
+        var _a;
+        try {
+          _servce && _servce.close();
+          (_a = _servce.httpServer) == null ? void 0 : _a.close(async () => {
+            _stop && await _stop();
+          });
+          _stop && await _stop();
+        } catch (e) {
+          consola.error(e);
+        }
+      });
     },
     configureServer(server) {
       const {
@@ -376,6 +482,7 @@ var unpluginFactory = (options) => ({
         consola.fail("please provide target");
         return;
       }
+      _servce = server;
       const _printUrls = server.printUrls;
       server.printUrls = () => {
         var _a;
@@ -388,18 +495,16 @@ var unpluginFactory = (options) => ({
         }
         const base = server.config.base || "/";
         try {
-          const caddy2 = new CaddyInstant();
-          caddy2.run(source, target, {
+          if (caddy)
+            return;
+          caddy = new CaddyInstant();
+          caddy.run(source, target, {
             base,
             showCaddyLog: options.showCaddyLog
           }).then((stop) => {
             const colorUrl = (url2) => import_picocolors.default.green(url2.replace(/:(\d+)\//, (_, port) => `:${import_picocolors.default.bold(port)}/`));
             consola.success(`  ${import_picocolors.default.green("\u279C")}  ${import_picocolors.default.bold("run caddy reverse proxy success")}: ${colorUrl(`https://${target}${base}`)}`);
-            import_node_process3.default.on("SIGINT", async () => {
-              await stop();
-              server.close();
-              import_node_process3.default.exit();
-            });
+            !_stop && (_stop = once(stop));
           }).catch((e) => {
             throw e;
           });
@@ -407,6 +512,9 @@ var unpluginFactory = (options) => ({
           consola.error(e);
         }
       };
+    },
+    async buildEnd() {
+      _stop && await _stop();
     }
   },
   webpack(compiler) {
@@ -421,23 +529,39 @@ var unpluginFactory = (options) => ({
       return;
     }
     try {
-      const devServer = compiler.options.devServer || import_node_process3.default.VUE_CLI_SERVICE.projectOptions.devServer;
-      const source = `${(devServer == null ? void 0 : devServer.host) || "127.0.0.1"}:${(devServer == null ? void 0 : devServer.port) || "8080"}`;
+      registerExit(async () => {
+        try {
+          _stop && await _stop();
+        } catch (e) {
+          consola.error(e);
+        }
+      });
+      const devServer = __spreadValues(__spreadValues({
+        host: "127.0.0.1",
+        port: 8080
+      }, compiler.options.devServer), import_node_process3.default.VUE_CLI_SERVICE.projectOptions.devServer);
+      const source = `${devServer.host}:${devServer.port}`;
       if (caddy)
         return;
+      const _close = compiler.close.bind(compiler);
+      compiler.close = async (callback) => {
+        try {
+          _stop && await _stop(() => {
+            _close && _close(callback);
+          });
+        } catch (e) {
+          consola.error(e);
+          import_node_process3.default.exit(1);
+        }
+      };
       caddy = new CaddyInstant();
       caddy.run(source, target, {
         showCaddyLog: options.showCaddyLog
       }).then((stop) => {
         const colorUrl = (url) => import_picocolors.default.green(url.replace(/:(\d+)\//, (_, port) => `:${import_picocolors.default.bold(port)}/`));
-        compiler.hooks.done.tap("unplugin-https-reverse-proxy", (stats) => {
-          if (stats.hasErrors())
-            return;
+        compiler.hooks.done.tap("unplugin-https-reverse-proxy", () => {
           consola.success(`  ${import_picocolors.default.green("\u279C")}  ${import_picocolors.default.bold("run caddy reverse proxy success")}: ${colorUrl(`https://${target}`)}`);
-        });
-        import_node_process3.default.on("SIGINT", async () => {
-          await stop();
-          import_node_process3.default.exit();
+          !_stop && (_stop = once(stop));
         });
       }).catch((e) => {
         throw e;

package/dist/index.js

@@ -2,8 +2,8 @@ import {
   src_default,
   unplugin,
   unpluginFactory
-} from "./chunk-WUSTJHAZ.js";
-import "./chunk-3HKPYCSE.js";
+} from "./chunk-DH6VMKTP.js";
+import "./chunk-CQC5SSJR.js";
 export {
   src_default as default,
   unplugin,