@my-life-buddies/cli 0.11.0 → 0.13.0

package/dist/bin/preview.js

@@ -43,6 +43,7 @@ var PreviewController = class {
   #simulator;
   #startController;
   #startPromise;
+  #stopPromise;
   #messages = [];
   #simulatorState = Object.freeze({ messages: Object.freeze([]) });
   #lastActiveRunId;
@@ -140,7 +141,17 @@ var PreviewController = class {
     this.#startPromise = startPromise;
     return startPromise;
   }
-  async stop() {
+  stop() {
+    if (this.#stopPromise) return this.#stopPromise;
+    const stopping = this.#stop();
+    this.#stopPromise = stopping;
+    const clear = () => {
+      if (this.#stopPromise === stopping) this.#stopPromise = void 0;
+    };
+    void stopping.then(clear, clear);
+    return stopping;
+  }
+  async #stop() {
     if (this.#state.phase === "idle") return this.#state;
     if (this.#state.phase === "stopping") return this.#state;
     this.#setState({ phase: "stopping" });
@@ -1673,9 +1684,149 @@ var MlbSimulatorClient = class {
     return value;
   }
 };
+
+// apps/preview/src/bridge.ts
+import { hostname } from "node:os";
+import { randomBytes as randomBytes2 } from "node:crypto";
+import { setTimeout as delay } from "node:timers/promises";
+async function connectPreviewBridge(options) {
+  const connectionId = randomBytes2(32).toString("base64url");
+  const shutdown = new AbortController();
+  const signal = AbortSignal.any([options.signal, shutdown.signal]);
+  const running = /* @__PURE__ */ new Map();
+  const replies = /* @__PURE__ */ new Map();
+  const completed = /* @__PURE__ */ new Set();
+  let lastContact = Date.now();
+  let stopping;
+  let interrupted = false;
+  async function call(path, token, payload, requestSignal = signal) {
+    const response = await fetch(new URL(path, options.webUrl), {
+      method: "POST",
+      redirect: "error",
+      signal: AbortSignal.any([requestSignal, AbortSignal.timeout(4e3)]),
+      headers: { Authorization: `Bearer ${token}`, "Content-Type": "application/json" },
+      body: JSON.stringify(payload ?? {})
+    });
+    const value = await response.json();
+    if (!response.ok) throw Object.assign(new Error(value.error || "\u540E\u53F0\u8FDE\u63A5\u5931\u8D25"), { status: response.status });
+    return value;
+  }
+  const attach = async () => {
+    const result = await call("api/preview/connect", await options.token(), {
+      connectionId,
+      buddyId: options.buddyId,
+      projectRoot: options.controller.projectRoot,
+      machineName: hostname(),
+      platformOrigin: options.platformOrigin
+    });
+    lastContact = Date.now();
+    return result;
+  };
+  const bootstrap = await fetch(options.localServer.bootstrapUrl, { redirect: "manual", signal });
+  const cookie = bootstrap.headers.get("set-cookie")?.split(";", 1)[0];
+  if (bootstrap.status !== 303 || !cookie) throw new Error("\u672C\u5730\u8C03\u8BD5\u670D\u52A1\u521D\u59CB\u5316\u5931\u8D25");
+  const initial = await attach();
+  if (initial.reused) return { reused: true, completion: Promise.resolve(), async close() {
+  } };
+  const watchdog = setInterval(() => {
+    if (Date.now() - lastContact < 1e4 || stopping || interrupted) return;
+    interrupted = true;
+    stopping = options.controller.stop();
+    void stopping.catch(() => shutdown.abort());
+    options.onStatus?.("\u540E\u53F0\u8FDE\u63A5\u5DF2\u65AD\u5F00\uFF0C\u6B63\u5728\u505C\u6B62 DEV\uFF1B\u8FDE\u63A5\u6062\u590D\u540E\u53EF\u5728\u540E\u53F0\u91CD\u65B0\u542F\u52A8\u3002");
+  }, 500);
+  watchdog.unref();
+  async function execute(command) {
+    try {
+      if (!command.path.startsWith("/api/") || command.path === "/api/events" || command.path.includes("..")) throw new Error("\u8C03\u8BD5\u8DEF\u5F84\u65E0\u6548");
+      const target = new URL(command.path, options.localServer.origin);
+      if (target.origin !== options.localServer.origin.origin) throw new Error("\u8C03\u8BD5\u8DEF\u5F84\u65E0\u6548");
+      if ((await options.controller.project()).buddyId !== options.buddyId) throw new Error("\u5DE5\u7A0B buddyId \u5DF2\u6539\u53D8\uFF0C\u8BF7\u9000\u51FA CLI \u540E\u91CD\u65B0\u8FDE\u63A5\u5BF9\u5E94\u642D\u5B50");
+      signal.throwIfAborted();
+      if (interrupted) throw new Error("\u672C\u5730\u8FDE\u63A5\u5DF2\u4E2D\u65AD");
+      const response = await fetch(target, {
+        method: command.method,
+        redirect: "error",
+        signal: AbortSignal.any([signal, AbortSignal.timeout(28e3)]),
+        headers: { Cookie: cookie, ...command.body ? { "Content-Type": "application/json" } : {} },
+        ...command.body ? { body: command.body } : {}
+      });
+      const chunks = [];
+      let size = 0;
+      for await (const chunk of response.body ?? []) {
+        size += chunk.length;
+        if (size > 8 * 1024 * 1024) throw new Error("\u8C03\u8BD5\u54CD\u5E94\u8FC7\u5927");
+        chunks.push(Buffer.from(chunk));
+      }
+      replies.set(command.id, { id: command.id, status: response.status, contentType: response.headers.get("content-type") ?? "application/json", body: Buffer.concat(chunks).toString("base64") });
+    } catch {
+      replies.set(command.id, { id: command.id, status: 502, contentType: "application/json", body: Buffer.from(JSON.stringify({ error: { code: "PREVIEW_LOCAL_FAILED", message: "\u672C\u5730\u8C03\u8BD5\u8BF7\u6C42\u5931\u8D25\uFF0C\u8BF7\u68C0\u67E5\u7EC8\u7AEF\u53CA\u5DE5\u7A0B buddyId" } })).toString("base64") });
+    } finally {
+      running.delete(command.id);
+    }
+  }
+  const completion = (async () => {
+    try {
+      while (!signal.aborted) {
+        try {
+          if (stopping) {
+            await stopping;
+            stopping = void 0;
+          }
+          const result = await call("api/preview/poll", connectionId, { replies: [...replies.values()].slice(0, 1) });
+          lastContact = Date.now();
+          const recovered = interrupted;
+          if (stopping) {
+            await stopping;
+            stopping = void 0;
+          }
+          if (recovered) {
+            interrupted = false;
+            options.onStatus?.("\u5DF2\u91CD\u65B0\u8FDE\u63A5\u540E\u53F0\uFF0C\u8BF7\u5728\u5F00\u53D1\u8C03\u8BD5\u9875\u542F\u52A8 DEV\u3002");
+          }
+          for (const id of result.acknowledged ?? []) {
+            replies.delete(id);
+            completed.add(id);
+          }
+          while (completed.size > 256) completed.delete(completed.values().next().value);
+          for (const command of result.commands ?? []) {
+            if (recovered && !running.has(command.id) && !replies.has(command.id) && !completed.has(command.id)) {
+              replies.set(command.id, { id: command.id, status: 409, contentType: "application/json", body: Buffer.from(JSON.stringify({ error: { code: "PREVIEW_RECONNECTED", message: "\u672C\u5730\u8FDE\u63A5\u5DF2\u6062\u590D\uFF0C\u8BF7\u91CD\u65B0\u64CD\u4F5C" } })).toString("base64") });
+            }
+            if (!running.has(command.id) && !replies.has(command.id) && !completed.has(command.id) && running.size < 32) running.set(command.id, execute(command));
+          }
+        } catch (cause) {
+          if (signal.aborted) break;
+          const status = cause.status;
+          if (status === 410) {
+            await options.controller.stop();
+            if ((await attach()).reused) throw new Error("\u5F53\u524D\u5DE5\u7A0B\u5DF2\u6709\u5176\u4ED6 CLI \u8FDE\u63A5\uFF0C\u8BF7\u91CD\u65B0\u8FD0\u884C preview \u6253\u5F00\u540E\u53F0");
+          } else if (status && status >= 400 && status < 500 && status !== 429) throw cause;
+        }
+        await delay(replies.size ? 20 : 500, void 0, { signal }).catch(() => void 0);
+      }
+    } finally {
+      clearInterval(watchdog);
+    }
+  })();
+  void completion.catch(() => void 0);
+  return {
+    reused: false,
+    completion,
+    async close() {
+      shutdown.abort();
+      await completion.catch(() => void 0);
+      await Promise.allSettled(running.values());
+      await stopping;
+      await options.controller.stop();
+      await call("api/preview/disconnect", connectionId, {}, AbortSignal.timeout(2e3)).catch(() => void 0);
+    }
+  };
+}
 export {
   MlbSimulatorClient,
   PreviewController,
+  connectPreviewBridge,
   createBuddyPreviewUrl,
   startPreviewControlServer
 };