@open-wxa/wxa-runner-core 0.2.3 → 0.2.4

package/dist/index.mjs

@@ -1,15 +1,15 @@
 import {
   __dirname
-} from "./chunk-22T3D5KS.mjs";
+} from "./chunk-NGXYLLYS.mjs";
 
 // src/server/DevServer.ts
-import { randomUUID } from "crypto";
+import { randomUUID as randomUUID2 } from "crypto";
 import fs4 from "fs";
 import http from "http";
 import path4 from "path";
 import express2 from "express";
-import { WebSocket, WebSocketServer } from "ws";
-import { findDeveloperTicket as findDeveloperTicket2, findRegionDataFile, resolveBaseLib, usesIsolatedContext } from "@open-wxa/wxa-runner-extractor";
+import { WebSocket as WebSocket2, WebSocketServer } from "ws";
+import { readRegionData, resolveBaseLib, usesIsolatedContext } from "@open-wxa/wxa-runner-extractor";
 
 // src/project/AppBuilder.ts
 import { compileWxml, compileWxss } from "@open-wxa/wxa-runner-compiler";
@@ -653,28 +653,164 @@ function planHotUpdate(files, units) {
   return plan;
 }
 
+// src/server/developerTicket.ts
+import { execFile } from "child_process";
+import { randomUUID } from "crypto";
+import { setTimeout as sleep } from "timers/promises";
+import { WebSocket } from "ws";
+import { findDeveloperTicket, findElectronDevtoolsCli, TICKET_ENV } from "@open-wxa/wxa-runner-extractor";
+var AUTO_PORT = 9420;
+var CLI_TIMEOUT = 6e4;
+var AUTOMATOR_TIMEOUT = 5e3;
+var WINDOW_TIMEOUT = 1e4;
+var WINDOW_RETRY_INTERVAL = 500;
+function developerTicketSource(projectPath, logger) {
+  const cli = findElectronDevtoolsCli();
+  if (!cli) return async () => findDeveloperTicket();
+  let automatorPort = null;
+  const startOnce = () => {
+    if (!automatorPort) {
+      const starting = startAutomation(cli, projectPath).then((port) => {
+        logger.log(`[wxa-runner] \u771F\u5B9E\u767B\u5F55\uFF1A\u5DF2\u5728\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u4E3A\u9879\u76EE\u5F00\u542F\u81EA\u52A8\u5316\u63A5\u53E3\uFF08\u7AEF\u53E3 ${port}\uFF09\uFF0Cticket \u4ECE\u5F00\u53D1\u8005\u5DE5\u5177\u5B9E\u65F6\u83B7\u53D6`);
+        return port;
+      });
+      starting.catch(() => {
+        if (automatorPort === starting) automatorPort = null;
+      });
+      automatorPort = starting;
+    }
+    return automatorPort;
+  };
+  return async () => {
+    if (process.env[TICKET_ENV]) return { newticket: process.env[TICKET_ENV] };
+    const port = startOnce();
+    try {
+      return await requestTicketOnceWindowReady(await port, projectPath);
+    } catch (error) {
+      if (!(error instanceof AutomatorGone)) throw error;
+      if (automatorPort === port) automatorPort = null;
+      return requestTicketOnceWindowReady(await startOnce(), projectPath);
+    }
+  };
+}
+var AutomatorGone = class extends Error {
+};
+var WindowNotFound = class extends Error {
+};
+async function requestTicketOnceWindowReady(port, projectPath) {
+  const deadline = Date.now() + WINDOW_TIMEOUT;
+  for (; ; ) {
+    try {
+      return await requestTicket(port);
+    } catch (error) {
+      if (!(error instanceof WindowNotFound)) throw error;
+      if (Date.now() >= deadline) throw new Error(`\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u627E\u4E0D\u5230\u9879\u76EE\u7A97\u53E3\uFF0C\u8BF7\u5728\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u6253\u5F00\u9879\u76EE ${projectPath}`);
+      await sleep(WINDOW_RETRY_INTERVAL);
+    }
+  }
+}
+async function startAutomation(cli, projectPath) {
+  const stdout = await runCli(cli, ["agent", "start", "--project", projectPath, "--auto-port", String(AUTO_PORT), "--lang", "zh"]);
+  let autoPort = NaN;
+  try {
+    autoPort = Number(JSON.parse(stdout.match(/^\{$[\s\S]*?^\}$/m)?.[0] ?? "").autoPort);
+  } catch {
+  }
+  if (Number.isInteger(autoPort) && autoPort > 0) return autoPort;
+  if (stdout.includes("\u670D\u52A1\u7AEF\u53E3\u5DF2\u5173\u95ED")) {
+    throw new Error("\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u670D\u52A1\u7AEF\u53E3\u5DF2\u5173\u95ED\uFF0C\u8BF7\u5728\u5F00\u53D1\u8005\u5DE5\u5177\u300C\u8BBE\u7F6E \u2192 \u5B89\u5168\u8BBE\u7F6E\u300D\u91CC\u5F00\u542F\u670D\u52A1\u7AEF\u53E3");
+  }
+  throw new Error(`\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u6CA1\u80FD\u4E3A\u9879\u76EE\u5F00\u542F\u81EA\u52A8\u5316\u63A5\u53E3\uFF1A${cliErrorMessage(stdout)}`);
+}
+function runCli(cli, args) {
+  const bat = cli.endsWith(".bat");
+  const [file, fileArgs] = bat ? [`"${cli}"`, args.map((arg) => `"${arg}"`)] : [cli, args];
+  return new Promise((resolve, reject) => {
+    const child = execFile(file, fileArgs, { shell: bat, timeout: CLI_TIMEOUT, windowsHide: true }, (error, stdout) => {
+      if (error?.killed) {
+        reject(new Error(`\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u547D\u4EE4\u884C ${CLI_TIMEOUT / 1e3}s \u5185\u6CA1\u6709\u5B8C\u6210\uFF0C\u8BF7\u770B\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u662F\u5426\u6709\u7B49\u5F85\u786E\u8BA4\u7684\u5F39\u7A97\uFF08\u5982\u4FE1\u4EFB\u9879\u76EE\u3001\u626B\u7801\u767B\u5F55\uFF09`));
+      } else if (error && !stdout) {
+        reject(new Error(`\u8FD0\u884C\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u547D\u4EE4\u884C\u5931\u8D25\uFF1A${error.message}`));
+      } else {
+        resolve(stdout);
+      }
+    });
+    child.stdin?.end();
+  });
+}
+function cliErrorMessage(stdout) {
+  const output = stdout.replace(/\x1b\[[0-9;]*m/g, "");
+  const start = output.indexOf("[error]");
+  const error = (start < 0 ? output : output.slice(start + "[error]".length)).trim();
+  const message = error.match(/message: (['"`])((?:\\.|(?!\1)[^\\])*?)(?:\\n|\1)/)?.[2];
+  return message ?? (error.replace(/\s+/g, " ").slice(0, 200) || "\u547D\u4EE4\u884C\u6CA1\u6709\u8F93\u51FA");
+}
+function requestTicket(port) {
+  return new Promise((resolve, reject) => {
+    const id = `wxa-runner-${randomUUID()}`;
+    const socket = new WebSocket(`ws://127.0.0.1:${port}`);
+    const settle = (finish) => {
+      clearTimeout(timer);
+      socket.removeAllListeners().on("error", () => void 0).terminate();
+      finish();
+    };
+    const timer = setTimeout(
+      () => settle(
+        () => reject(new AutomatorGone(`\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u81EA\u52A8\u5316\u63A5\u53E3 ${AUTOMATOR_TIMEOUT / 1e3}s \u5185\u6CA1\u6709\u54CD\u5E94\uFF0C\u8BF7\u786E\u8BA4\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u6253\u5F00\u4E86\u9879\u76EE\u7A97\u53E3`))
+      ),
+      AUTOMATOR_TIMEOUT
+    );
+    const gone = () => settle(() => reject(new AutomatorGone(`\u8FDE\u4E0D\u4E0A\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u81EA\u52A8\u5316\u63A5\u53E3\uFF08\u7AEF\u53E3 ${port}\uFF09\uFF0C\u8BF7\u786E\u8BA4\u5F00\u53D1\u8005\u5DE5\u5177\u5728\u8FD0\u884C`)));
+    socket.on("error", gone);
+    socket.on("close", gone);
+    socket.on("open", () => socket.send(JSON.stringify({ id, method: "Tool.getTicket", params: {} })));
+    socket.on("message", (data) => {
+      let message;
+      try {
+        message = JSON.parse(String(data));
+      } catch {
+        return;
+      }
+      if (message?.id !== id) return;
+      if (message.error) settle(() => reject(explainAutomatorError(String(message.error.message))));
+      else if (typeof message.result?.ticket !== "string") settle(() => reject(new Error("\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u6CA1\u6709\u8FD4\u56DE\u767B\u5F55\u7968\u636E")));
+      else settle(() => resolve({ newticket: message.result.ticket, expiredAt: message.result.expiredTime }));
+    });
+  });
+}
+function explainAutomatorError(message) {
+  if (message.includes("GUI project window is not found")) return new WindowNotFound(message);
+  if (message.includes("not allow to get")) {
+    return new Error("\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u4E0D\u5141\u8BB8\u83B7\u53D6\u767B\u5F55\u7968\u636E\uFF0C\u8BF7\u5728\u5F00\u53D1\u8005\u5DE5\u5177\u300C\u8BBE\u7F6E \u2192 \u5B89\u5168\u8BBE\u7F6E\u300D\u91CC\u6253\u5F00\u300C\u5141\u8BB8\u83B7\u53D6\u5DE5\u5177\u767B\u5F55\u7968\u636E\u300D");
+  }
+  if (message.includes("need to login")) return new Error("\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u672A\u767B\u5F55\uFF0C\u8BF7\u5148\u5728\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u767B\u5F55");
+  return new Error(`\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u8FD4\u56DE\u9519\u8BEF\uFF1A${message}`);
+}
+
 // src/server/realLogin.ts
 import express from "express";
-import { findDeveloperTicket, TICKET_ENV } from "@open-wxa/wxa-runner-extractor";
+import { findDeveloperTicket as findDeveloperTicket2, TICKET_ENV as TICKET_ENV2 } from "@open-wxa/wxa-runner-extractor";
 var DEV_LOGIC = "https://servicewechat.com/wxa-dev-logic";
 var CLIENT_VERSION = "1021810240";
 var UPSTREAM_TIMEOUT = 1e4;
 var LOOPBACK_ADDRESSES = ["127.0.0.1", "::1", "::ffff:127.0.0.1"];
 var LOOPBACK_HOSTS = ["localhost", "127.0.0.1", "[::1]"];
 var ERROR_HINTS = {
+  // 40001：开发者工具刷新过 ticket，旧的随即作废
+  40001: "\uFF08\u767B\u5F55\u6001\u5DF2\u5931\u6548\uFF0C\u8BF7\u5728\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u91CD\u65B0\u767B\u5F55\uFF09",
   40029: "\uFF08\u767B\u5F55\u6001\u5DF2\u5931\u6548\uFF0C\u8BF7\u5728\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u91CC\u91CD\u65B0\u767B\u5F55\uFF09",
   85023: "\uFF08\u5F53\u524D\u5FAE\u4FE1\u8D26\u53F7\u4E0D\u662F\u8BE5\u5C0F\u7A0B\u5E8F\u7684\u5F00\u53D1\u8005\uFF09"
 };
 function requireUsableTicket(ticket) {
-  if (!ticket) throw new Error(`\u672A\u627E\u5230\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u767B\u5F55\u6001\uFF0C\u8BF7\u767B\u5F55\u5F00\u53D1\u8005\u5DE5\u5177\uFF0C\u6216\u8BBE\u7F6E\u73AF\u5883\u53D8\u91CF ${TICKET_ENV}`);
+  if (!ticket) throw new Error(`\u672A\u627E\u5230\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u767B\u5F55\u6001\uFF0C\u8BF7\u767B\u5F55\u5F00\u53D1\u8005\u5DE5\u5177\uFF0C\u6216\u8BBE\u7F6E\u73AF\u5883\u53D8\u91CF ${TICKET_ENV2}`);
   if (ticket.expiredAt !== void 0 && ticket.expiredAt <= Date.now()) {
     throw new Error("\u5FAE\u4FE1\u5F00\u53D1\u8005\u5DE5\u5177\u7684\u767B\u5F55\u6001\u5DF2\u8FC7\u671F\uFF0C\u8BF7\u6253\u5F00\u5F00\u53D1\u8005\u5DE5\u5177\u5237\u65B0\u767B\u5F55\u6001");
   }
   return ticket.newticket;
 }
-function realLoginRouter(appid, findTicket = findDeveloperTicket) {
+function realLoginRouter(appid, findTicket = findDeveloperTicket2) {
   const callDevLogic = async (endpoint, body) => {
-    const query = new URLSearchParams({ appid, newticket: requireUsableTicket(findTicket()), clientversion: CLIENT_VERSION });
+    const query = new URLSearchParams({ appid, newticket: requireUsableTicket(await findTicket()), clientversion: CLIENT_VERSION });
     const response = await fetch(`${DEV_LOGIC}/${endpoint}?${query}`, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
@@ -853,13 +989,7 @@ var DevServer = class {
     if (!this.baseLib.compatible) {
       this.logger.warn(`[wxa-runner] \u57FA\u7840\u5E93 ${this.baseLib.version} \u9AD8\u4E8E\u5DF2\u9A8C\u8BC1\u7684\u7248\u672C\u8303\u56F4\uFF0C\u53EF\u80FD\u65E0\u6CD5\u6B63\u5E38\u8FD0\u884C`);
     }
-    if (options.realLogin) {
-      try {
-        requireUsableTicket(findDeveloperTicket2());
-      } catch (error) {
-        this.logger.warn(`[wxa-runner] \u771F\u5B9E\u767B\u5F55\u6682\u4E0D\u53EF\u7528\uFF1A${error.message}`);
-      }
-    }
+    this.ticketSource = options.realLogin ? developerTicketSource(this.project.projectPath, this.logger) : null;
     this.device = { ...DEFAULT_DEVICE, ...options.device };
     this.builder = new AppBuilder(this.project);
     this.setupFile = resolveSetupFile(options.setup, this.project.projectPath);
@@ -895,6 +1025,7 @@ ${error.message}`);
     const { port } = server.address();
     this.address = `http://${host}:${port}`;
     if (this.options.watch !== false) this.startWatching();
+    this.ticketSource?.().then(requireUsableTicket).catch((error) => this.logger.warn(`[wxa-runner] \u771F\u5B9E\u767B\u5F55\u6682\u4E0D\u53EF\u7528\uFF1A${error.message}`));
     return this.address;
   }
   async close() {
@@ -1005,18 +1136,18 @@ ${error.message}`);
         res.status(400).type("text").send(`\u975E\u6CD5\u7684\u6587\u4EF6\u7C7B\u578B ${type}`);
         return;
       }
-      const name = `${randomUUID()}.${type}`;
+      const name = `${randomUUID2()}.${type}`;
       this.tempFiles.set(name, Buffer.isBuffer(req.body) ? req.body : Buffer.alloc(0));
       res.json({ url: `${req.protocol}://${req.get("host")}/__wxa__/tmp/${name}` });
     });
     app.get("/__wxa__/region/:kind", (req, res) => {
       const kind = req.params.kind;
-      const file = kind === "districts" || kind === "streets" ? findRegionDataFile(kind) : null;
-      if (!file) {
+      const data = kind === "districts" || kind === "streets" ? readRegionData(kind) : null;
+      if (!data) {
         res.status(404).end();
         return;
       }
-      res.type("text/plain; charset=utf-8").sendFile(file, { maxAge: "1d" });
+      res.type("text/plain; charset=utf-8").set("Cache-Control", "public, max-age=86400").send(data);
     });
     app.get("/__wxa__/tmp/:name", (req, res) => {
       const data = this.tempFiles.get(req.params.name);
@@ -1029,7 +1160,7 @@ ${error.message}`);
     app.delete("/__wxa__/tmp/:name", (req, res) => {
       res.status(this.tempFiles.delete(req.params.name) ? 204 : 404).end();
     });
-    if (this.options.realLogin) app.use("/__wxa__/api", realLoginRouter(this.project.appid));
+    if (this.ticketSource) app.use("/__wxa__/api", realLoginRouter(this.project.appid, this.ticketSource));
     app.use("/__wxa__/webview", express2.static(root));
     app.use(express2.static(root));
     app.use((error, _req, res, _next) => {
@@ -1108,7 +1239,7 @@ ${message}`);
   broadcast(message) {
     const data = JSON.stringify(message);
     this.sockets?.clients.forEach((client) => {
-      if (client.readyState === WebSocket.OPEN) client.send(data);
+      if (client.readyState === WebSocket2.OPEN) client.send(data);
     });
   }
 };