104 lines
3.3 KiB
TypeScript
104 lines
3.3 KiB
TypeScript
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);
|
|
});
|
|
}
|