源码快照 @ 2026-09-02 (source-only, 剔除非源码素材)

packages/world/src/hold.ts

@@ -0,0 +1,103 @@
+import { chmod, mkdir, readFile, unlink, writeFile } from "node:fs/promises";
+import { basename, dirname, extname, join, resolve } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const REPO_ROOT = fileURLToPath(new URL("../../..", import.meta.url));
+const SAFE_WORLD_NAME = /^[A-Za-z0-9][A-Za-z0-9._-]{0,127}$/;
+
+export interface ScriptWorldSnapshot {
+  version: 1;
+  kind: "script";
+  name: string;
+  createdAt: string;
+  pid: number;
+  sourcePath: string;
+  outputs: Record<string, string>;
+}
+
+export interface HoldOptions {
+  name?: string;
+  outputs?: Record<string, string>;
+  snapshotDir?: string;
+}
+
+function recordedPid(text: string): number | undefined {
+  const value: unknown = JSON.parse(text);
+  if (typeof value !== "object" || value === null || !("pid" in value)) return undefined;
+  return typeof value.pid === "number" && Number.isInteger(value.pid) && value.pid > 0
+    ? value.pid
+    : undefined;
+}
+
+function isAlive(pid: number): boolean {
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+async function aliveSnapshotPid(path: string): Promise<number | undefined> {
+  try {
+    const pid = recordedPid(await readFile(path, "utf8"));
+    return pid !== undefined && isAlive(pid) ? pid : undefined;
+  } catch {
+    return undefined;
+  }
+}
+
+/** Keep a script world alive until SIGINT or SIGTERM, then release its resources. */
+export async function hold(options: HoldOptions = {}): Promise<void> {
+  const sourcePath = process.argv[1];
+  const name = options.name ?? basename(sourcePath, extname(sourcePath));
+  if (!SAFE_WORLD_NAME.test(name)) {
+    throw new Error("World names must use only letters, numbers, dots, underscores, and hyphens.");
+  }
+
+  const snapshotDirectory = resolve(
+    options.snapshotDir
+      ?? process.env.OPENWORK_WORLD_SNAPSHOT_DIR
+      ?? join(REPO_ROOT, "evals", "results", ".worlds", "scripts"),
+  );
+  const snapshotPath = join(snapshotDirectory, `${name}.json`);
+  const existingPid = await aliveSnapshotPid(snapshotPath);
+  if (existingPid !== undefined) {
+    throw new Error(
+      `Script world ${JSON.stringify(name)} is already running (pid ${existingPid}); run \`pnpm world down ${name}\` first.`,
+    );
+  }
+
+  const outputs = options.outputs ?? {};
+  const snapshot: ScriptWorldSnapshot = {
+    version: 1,
+    kind: "script",
+    name,
+    createdAt: new Date().toISOString(),
+    pid: process.pid,
+    sourcePath,
+    outputs,
+  };
+  await mkdir(dirname(snapshotPath), { recursive: true });
+  await writeFile(snapshotPath, `${JSON.stringify(snapshot, null, 2)}\n`, { encoding: "utf8", mode: 0o600 });
+  await chmod(snapshotPath, 0o600);
+
+  for (const [key, value] of Object.entries(outputs)) console.log(`${key}  ${value}`);
+  console.log(`World ${JSON.stringify(name)} is up. Ctrl-C (or pnpm world down ${name}) tears it down.`);
+
+  await new Promise<void>((done) => {
+    let stopping = false;
+    const stop = (): void => {
+      if (stopping) return;
+      stopping = true;
+      process.off("SIGINT", stop);
+      process.off("SIGTERM", stop);
+      void unlink(snapshotPath).then(done, (error: unknown) => {
+        console.error(`Could not remove script world snapshot ${snapshotPath}: ${error instanceof Error ? error.message : String(error)}`);
+        done();
+      });
+    };
+    process.once("SIGINT", stop);
+    process.once("SIGTERM", stop);
+  });
+}