@agimon-ai/doompi-computer-use 0.0.1-alpha.2 → 0.0.1-alpha.21

package/dist/src/services/computerScriptRunner/index.cjs

@@ -0,0 +1,171 @@
+let node_child_process = require("node:child_process");
+let node_fs_promises = require("node:fs/promises");
+//#region src/services/computerScriptRunner/index.ts
+const DEFAULT_TIMEOUT_MS = 3e4;
+const DEFAULT_MAX_OUTPUT_BYTES = 262144;
+const WORKER_SOURCE = String.raw`
+import { readFile } from 'node:fs/promises';
+import { stripTypeScriptTypes } from 'node:module';
+
+let nextRequestId = 0;
+const pending = new Map();
+const call = (operation, payload) => new Promise((resolve, reject) => {
+  const id = String(++nextRequestId);
+  pending.set(id, { resolve, reject });
+  process.send({ type: 'program', id, operation, payload });
+});
+process.on('message', async (message) => {
+  if (message?.type === 'response') {
+    const request = pending.get(message.id);
+    if (!request) return;
+    pending.delete(message.id);
+    if (message.error) request.reject(new Error(message.error));
+    else request.resolve(message.value);
+    return;
+  }
+  if (message?.type !== 'start') return;
+  const controller = new AbortController();
+  process.once('SIGTERM', () => controller.abort());
+  const logger = Object.fromEntries(['debug', 'info', 'warn', 'error'].map((level) => [level, (value) => {
+    process.send({ type: 'log', level, message: String(value) });
+  }]));
+  try {
+    const source = await readFile(message.scriptPath, 'utf8');
+    const compiled = stripTypeScriptTypes(source, { mode: 'strip' });
+    const moduleUrl = 'data:text/javascript;base64,' + Buffer.from(compiled).toString('base64');
+    const loaded = await import(moduleUrl);
+    if (typeof loaded.run !== 'function') throw new Error('Computer script must export a named run function.');
+    const program = {
+      observe: () => call('observe'),
+      act: (action) => call('act', action),
+    };
+    const result = await loaded.run({ context: { program }, input: message.input, logger, signal: controller.signal });
+    const resultJson = result === undefined ? undefined : JSON.stringify(result);
+    process.send({ type: 'complete', resultJson });
+  } catch (error) {
+    process.send({ type: 'failed', error: error instanceof Error ? error.message : String(error) });
+  }
+});
+`;
+var ComputerScriptRunner = class {
+	client;
+	allowedScriptPaths;
+	timeoutMs;
+	maxOutputBytes;
+	constructor(options) {
+		this.client = options.client;
+		this.allowedScriptPaths = options.allowedScriptPaths;
+		this.timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+		this.maxOutputBytes = options.maxOutputBytes ?? DEFAULT_MAX_OUTPUT_BYTES;
+	}
+	async execute(scriptPath, input, signal) {
+		const resolvedPath = await (0, node_fs_promises.realpath)(scriptPath);
+		if (!(await Promise.all(this.allowedScriptPaths.map(async (candidate) => (0, node_fs_promises.realpath)(candidate).catch(() => "")))).includes(resolvedPath)) throw new Error("Computer script path is not explicitly allowed.");
+		if (!resolvedPath.endsWith(".ts")) throw new Error("Computer scripts must be TypeScript files.");
+		const serializedInput = JSON.stringify(input);
+		if (serializedInput === void 0) throw new Error("Computer script input must be JSON serializable.");
+		if (Buffer.byteLength(serializedInput) > this.maxOutputBytes) throw new Error("Computer script input exceeds the size limit.");
+		return new Promise((resolve, reject) => {
+			const child = (0, node_child_process.spawn)(process.execPath, [
+				"--input-type=module",
+				"--eval",
+				WORKER_SOURCE
+			], { stdio: [
+				"ignore",
+				"pipe",
+				"pipe",
+				"ipc"
+			] });
+			const controller = new AbortController();
+			const logs = [];
+			let outputBytes = 0;
+			let settled = false;
+			let operation = Promise.resolve();
+			const terminate = () => {
+				controller.abort();
+				child.kill("SIGTERM");
+				setTimeout(() => child.kill("SIGKILL"), 250).unref?.();
+			};
+			const finish = (error, value) => {
+				if (settled) return;
+				settled = true;
+				clearTimeout(timeout);
+				signal?.removeEventListener("abort", onAbort);
+				terminate();
+				if (error !== void 0) reject(error);
+				else resolve(value);
+			};
+			const addOutput = (value) => {
+				outputBytes += Buffer.byteLength(value);
+				if (outputBytes <= this.maxOutputBytes) return true;
+				finish(/* @__PURE__ */ new Error("Computer script output exceeded the size limit."));
+				return false;
+			};
+			const onAbort = () => finish(/* @__PURE__ */ new Error("Computer script execution was cancelled."));
+			const timeout = setTimeout(() => finish(/* @__PURE__ */ new Error("Computer script execution timed out.")), this.timeoutMs);
+			timeout.unref?.();
+			signal?.addEventListener("abort", onAbort, { once: true });
+			if (signal?.aborted === true) return onAbort();
+			child.stdout?.on("data", (chunk) => addOutput(chunk.toString("utf8")));
+			child.stderr?.on("data", (chunk) => addOutput(chunk.toString("utf8")));
+			child.on("error", (error) => finish(error));
+			child.on("exit", (code, exitSignal) => {
+				if (!settled) finish(/* @__PURE__ */ new Error(`Computer script worker exited before completion (${exitSignal ?? code ?? "unknown"}).`));
+			});
+			child.on("message", (rawMessage) => {
+				if (settled) return;
+				if (rawMessage.type === "log" && typeof rawMessage.message === "string") {
+					const line = `[${String(rawMessage.level)}] ${rawMessage.message}`;
+					if (addOutput(line)) logs.push(line);
+					return;
+				}
+				if (rawMessage.type === "program" && typeof rawMessage.id === "string" && (rawMessage.operation === "observe" || rawMessage.operation === "act")) {
+					operation = operation.then(async () => {
+						try {
+							const value = rawMessage.operation === "observe" ? await this.client.observe(controller.signal) : await this.client.act(rawMessage.payload, controller.signal);
+							child.send({
+								type: "response",
+								id: rawMessage.id,
+								value
+							});
+						} catch (error) {
+							child.send({
+								type: "response",
+								id: rawMessage.id,
+								error: error instanceof Error ? error.message : String(error)
+							});
+						}
+					});
+					return;
+				}
+				if (rawMessage.type === "failed" && typeof rawMessage.error === "string") {
+					finish(new Error(rawMessage.error));
+					return;
+				}
+				if (rawMessage.type === "complete") operation = operation.then(async () => {
+					const resultJson = rawMessage.resultJson;
+					if (resultJson !== void 0 && typeof resultJson !== "string") {
+						finish(/* @__PURE__ */ new Error("Computer script returned an invalid result."));
+						return;
+					}
+					if (typeof resultJson === "string" && !addOutput(resultJson)) return;
+					const observation = await this.client.observe(controller.signal);
+					finish(void 0, {
+						...typeof resultJson === "string" ? { result: JSON.parse(resultJson) } : {},
+						observation,
+						logs
+					});
+				}).catch((error) => finish(error instanceof Error ? error : new Error(String(error))));
+			});
+			child.send({
+				type: "start",
+				scriptPath: resolvedPath,
+				input: JSON.parse(serializedInput)
+			});
+		});
+	}
+};
+//#endregion
+exports.ComputerScriptRunner = ComputerScriptRunner;
+
+//# sourceMappingURL=index.cjs.map

package/dist/src/services/computerScriptRunner/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.cjs","names":["realpath","spawn"],"sources":["../../../../src/services/computerScriptRunner/index.ts"],"sourcesContent":["import { spawn } from 'node:child_process';\nimport { realpath } from 'node:fs/promises';\n\nimport type { ComputerScriptExecutionResult, ComputerScriptRunnerOptions } from '../../types/computerScript';\nimport type { ComputerUseAction } from '../../types/computerUse';\nimport type { ComputerUseSessionClient } from '../sessionApiClient';\n\nconst DEFAULT_TIMEOUT_MS = 30_000;\nconst DEFAULT_MAX_OUTPUT_BYTES = 256 * 1024;\n\nconst WORKER_SOURCE = String.raw`\nimport { readFile } from 'node:fs/promises';\nimport { stripTypeScriptTypes } from 'node:module';\n\nlet nextRequestId = 0;\nconst pending = new Map();\nconst call = (operation, payload) => new Promise((resolve, reject) => {\n  const id = String(++nextRequestId);\n  pending.set(id, { resolve, reject });\n  process.send({ type: 'program', id, operation, payload });\n});\nprocess.on('message', async (message) => {\n  if (message?.type === 'response') {\n    const request = pending.get(message.id);\n    if (!request) return;\n    pending.delete(message.id);\n    if (message.error) request.reject(new Error(message.error));\n    else request.resolve(message.value);\n    return;\n  }\n  if (message?.type !== 'start') return;\n  const controller = new AbortController();\n  process.once('SIGTERM', () => controller.abort());\n  const logger = Object.fromEntries(['debug', 'info', 'warn', 'error'].map((level) => [level, (value) => {\n    process.send({ type: 'log', level, message: String(value) });\n  }]));\n  try {\n    const source = await readFile(message.scriptPath, 'utf8');\n    const compiled = stripTypeScriptTypes(source, { mode: 'strip' });\n    const moduleUrl = 'data:text/javascript;base64,' + Buffer.from(compiled).toString('base64');\n    const loaded = await import(moduleUrl);\n    if (typeof loaded.run !== 'function') throw new Error('Computer script must export a named run function.');\n    const program = {\n      observe: () => call('observe'),\n      act: (action) => call('act', action),\n    };\n    const result = await loaded.run({ context: { program }, input: message.input, logger, signal: controller.signal });\n    const resultJson = result === undefined ? undefined : JSON.stringify(result);\n    process.send({ type: 'complete', resultJson });\n  } catch (error) {\n    process.send({ type: 'failed', error: error instanceof Error ? error.message : String(error) });\n  }\n});\n`;\n\ninterface WorkerMessage {\n  readonly type?: unknown;\n  readonly id?: unknown;\n  readonly operation?: unknown;\n  readonly payload?: unknown;\n  readonly level?: unknown;\n  readonly message?: unknown;\n  readonly resultJson?: unknown;\n  readonly error?: unknown;\n}\n\nexport class ComputerScriptRunner {\n  private readonly client: ComputerUseSessionClient;\n  private readonly allowedScriptPaths: readonly string[];\n  private readonly timeoutMs: number;\n  private readonly maxOutputBytes: number;\n\n  public constructor(options: ComputerScriptRunnerOptions) {\n    this.client = options.client;\n    this.allowedScriptPaths = options.allowedScriptPaths;\n    this.timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;\n    this.maxOutputBytes = options.maxOutputBytes ?? DEFAULT_MAX_OUTPUT_BYTES;\n  }\n\n  public async execute(\n    scriptPath: string,\n    input: unknown,\n    signal?: AbortSignal,\n  ): Promise<ComputerScriptExecutionResult> {\n    const resolvedPath = await realpath(scriptPath);\n    const allowedPaths = await Promise.all(\n      this.allowedScriptPaths.map(async (candidate) => realpath(candidate).catch(() => '')),\n    );\n    if (!allowedPaths.includes(resolvedPath)) throw new Error('Computer script path is not explicitly allowed.');\n    if (!resolvedPath.endsWith('.ts')) throw new Error('Computer scripts must be TypeScript files.');\n\n    const serializedInput = JSON.stringify(input);\n    if (serializedInput === undefined) throw new Error('Computer script input must be JSON serializable.');\n    if (Buffer.byteLength(serializedInput) > this.maxOutputBytes)\n      throw new Error('Computer script input exceeds the size limit.');\n\n    return new Promise((resolve, reject) => {\n      const child = spawn(process.execPath, ['--input-type=module', '--eval', WORKER_SOURCE], {\n        stdio: ['ignore', 'pipe', 'pipe', 'ipc'],\n      });\n      const controller = new AbortController();\n      const logs: string[] = [];\n      let outputBytes = 0;\n      let settled = false;\n      let operation = Promise.resolve();\n\n      const terminate = (): void => {\n        controller.abort();\n        child.kill('SIGTERM');\n        const killTimer = setTimeout(() => child.kill('SIGKILL'), 250);\n        killTimer.unref?.();\n      };\n      const finish = (error?: Error, value?: ComputerScriptExecutionResult): void => {\n        if (settled) return;\n        settled = true;\n        clearTimeout(timeout);\n        signal?.removeEventListener('abort', onAbort);\n        terminate();\n        if (error !== undefined) reject(error);\n        else resolve(value as ComputerScriptExecutionResult);\n      };\n      const addOutput = (value: string): boolean => {\n        outputBytes += Buffer.byteLength(value);\n        if (outputBytes <= this.maxOutputBytes) return true;\n        finish(new Error('Computer script output exceeded the size limit.'));\n        return false;\n      };\n      const onAbort = () => finish(new Error('Computer script execution was cancelled.'));\n      const timeout = setTimeout(() => finish(new Error('Computer script execution timed out.')), this.timeoutMs);\n      timeout.unref?.();\n      signal?.addEventListener('abort', onAbort, { once: true });\n      if (signal?.aborted === true) return onAbort();\n\n      child.stdout?.on('data', (chunk: Buffer) => addOutput(chunk.toString('utf8')));\n      child.stderr?.on('data', (chunk: Buffer) => addOutput(chunk.toString('utf8')));\n      child.on('error', (error) => finish(error));\n      child.on('exit', (code, exitSignal) => {\n        if (!settled)\n          finish(new Error(`Computer script worker exited before completion (${exitSignal ?? code ?? 'unknown'}).`));\n      });\n      child.on('message', (rawMessage: WorkerMessage) => {\n        if (settled) return;\n        if (rawMessage.type === 'log' && typeof rawMessage.message === 'string') {\n          const line = `[${String(rawMessage.level)}] ${rawMessage.message}`;\n          if (addOutput(line)) logs.push(line);\n          return;\n        }\n        if (\n          rawMessage.type === 'program' &&\n          typeof rawMessage.id === 'string' &&\n          (rawMessage.operation === 'observe' || rawMessage.operation === 'act')\n        ) {\n          operation = operation.then(async () => {\n            try {\n              const value =\n                rawMessage.operation === 'observe'\n                  ? await this.client.observe(controller.signal)\n                  : await this.client.act(rawMessage.payload as ComputerUseAction, controller.signal);\n              child.send({ type: 'response', id: rawMessage.id, value });\n            } catch (error) {\n              child.send({\n                type: 'response',\n                id: rawMessage.id,\n                error: error instanceof Error ? error.message : String(error),\n              });\n            }\n          });\n          return;\n        }\n        if (rawMessage.type === 'failed' && typeof rawMessage.error === 'string') {\n          finish(new Error(rawMessage.error));\n          return;\n        }\n        if (rawMessage.type === 'complete') {\n          operation = operation\n            .then(async () => {\n              const resultJson = rawMessage.resultJson;\n              if (resultJson !== undefined && typeof resultJson !== 'string') {\n                finish(new Error('Computer script returned an invalid result.'));\n                return;\n              }\n              if (typeof resultJson === 'string' && !addOutput(resultJson)) return;\n              const observation = await this.client.observe(controller.signal);\n              finish(undefined, {\n                ...(typeof resultJson === 'string' ? { result: JSON.parse(resultJson) as unknown } : {}),\n                observation,\n                logs,\n              });\n            })\n            .catch((error: unknown) => finish(error instanceof Error ? error : new Error(String(error))));\n        }\n      });\n      child.send({ type: 'start', scriptPath: resolvedPath, input: JSON.parse(serializedInput) as unknown });\n    });\n  }\n}\n"],"mappings":";;;AAOA,MAAM,qBAAqB;AAC3B,MAAM,2BAA2B;AAEjC,MAAM,gBAAgB,OAAO,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDhC,IAAa,uBAAb,MAAkC;CAChC;CACA;CACA;CACA;CAEA,YAAmB,SAAsC;EACvD,KAAK,SAAS,QAAQ;EACtB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,YAAY,QAAQ,aAAa;EACtC,KAAK,iBAAiB,QAAQ,kBAAkB;CAClD;CAEA,MAAa,QACX,YACA,OACA,QACwC;EACxC,MAAM,eAAe,OAAA,GAAMA,iBAAAA,SAAAA,CAAS,UAAU;EAI9C,IAAI,EAAC,MAHsB,QAAQ,IACjC,KAAK,mBAAmB,IAAI,OAAO,eAAA,GAAcA,iBAAAA,SAAAA,CAAS,SAAS,CAAC,CAAC,YAAY,EAAE,CAAC,CACtF,EAAA,CACkB,SAAS,YAAY,GAAG,MAAM,IAAI,MAAM,iDAAiD;EAC3G,IAAI,CAAC,aAAa,SAAS,KAAK,GAAG,MAAM,IAAI,MAAM,4CAA4C;EAE/F,MAAM,kBAAkB,KAAK,UAAU,KAAK;EAC5C,IAAI,oBAAoB,KAAA,GAAW,MAAM,IAAI,MAAM,kDAAkD;EACrG,IAAI,OAAO,WAAW,eAAe,IAAI,KAAK,gBAC5C,MAAM,IAAI,MAAM,+CAA+C;EAEjE,OAAO,IAAI,SAAS,SAAS,WAAW;GACtC,MAAM,SAAA,GAAQC,mBAAAA,MAAAA,CAAM,QAAQ,UAAU;IAAC;IAAuB;IAAU;GAAa,GAAG,EACtF,OAAO;IAAC;IAAU;IAAQ;IAAQ;GAAK,EACzC,CAAC;GACD,MAAM,aAAa,IAAI,gBAAgB;GACvC,MAAM,OAAiB,CAAC;GACxB,IAAI,cAAc;GAClB,IAAI,UAAU;GACd,IAAI,YAAY,QAAQ,QAAQ;GAEhC,MAAM,kBAAwB;IAC5B,WAAW,MAAM;IACjB,MAAM,KAAK,SAAS;IAEpB,iBADmC,MAAM,KAAK,SAAS,GAAG,GAClD,CAAC,CAAC,QAAQ;GACpB;GACA,MAAM,UAAU,OAAe,UAAgD;IAC7E,IAAI,SAAS;IACb,UAAU;IACV,aAAa,OAAO;IACpB,QAAQ,oBAAoB,SAAS,OAAO;IAC5C,UAAU;IACV,IAAI,UAAU,KAAA,GAAW,OAAO,KAAK;SAChC,QAAQ,KAAsC;GACrD;GACA,MAAM,aAAa,UAA2B;IAC5C,eAAe,OAAO,WAAW,KAAK;IACtC,IAAI,eAAe,KAAK,gBAAgB,OAAO;IAC/C,uBAAO,IAAI,MAAM,iDAAiD,CAAC;IACnE,OAAO;GACT;GACA,MAAM,gBAAgB,uBAAO,IAAI,MAAM,0CAA0C,CAAC;GAClF,MAAM,UAAU,iBAAiB,uBAAO,IAAI,MAAM,sCAAsC,CAAC,GAAG,KAAK,SAAS;GAC1G,QAAQ,QAAQ;GAChB,QAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;GACzD,IAAI,QAAQ,YAAY,MAAM,OAAO,QAAQ;GAE7C,MAAM,QAAQ,GAAG,SAAS,UAAkB,UAAU,MAAM,SAAS,MAAM,CAAC,CAAC;GAC7E,MAAM,QAAQ,GAAG,SAAS,UAAkB,UAAU,MAAM,SAAS,MAAM,CAAC,CAAC;GAC7E,MAAM,GAAG,UAAU,UAAU,OAAO,KAAK,CAAC;GAC1C,MAAM,GAAG,SAAS,MAAM,eAAe;IACrC,IAAI,CAAC,SACH,uBAAO,IAAI,MAAM,oDAAoD,cAAc,QAAQ,UAAU,GAAG,CAAC;GAC7G,CAAC;GACD,MAAM,GAAG,YAAY,eAA8B;IACjD,IAAI,SAAS;IACb,IAAI,WAAW,SAAS,SAAS,OAAO,WAAW,YAAY,UAAU;KACvE,MAAM,OAAO,IAAI,OAAO,WAAW,KAAK,EAAE,IAAI,WAAW;KACzD,IAAI,UAAU,IAAI,GAAG,KAAK,KAAK,IAAI;KACnC;IACF;IACA,IACE,WAAW,SAAS,aACpB,OAAO,WAAW,OAAO,aACxB,WAAW,cAAc,aAAa,WAAW,cAAc,QAChE;KACA,YAAY,UAAU,KAAK,YAAY;MACrC,IAAI;OACF,MAAM,QACJ,WAAW,cAAc,YACrB,MAAM,KAAK,OAAO,QAAQ,WAAW,MAAM,IAC3C,MAAM,KAAK,OAAO,IAAI,WAAW,SAA8B,WAAW,MAAM;OACtF,MAAM,KAAK;QAAE,MAAM;QAAY,IAAI,WAAW;QAAI;OAAM,CAAC;MAC3D,SAAS,OAAO;OACd,MAAM,KAAK;QACT,MAAM;QACN,IAAI,WAAW;QACf,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;OAC9D,CAAC;MACH;KACF,CAAC;KACD;IACF;IACA,IAAI,WAAW,SAAS,YAAY,OAAO,WAAW,UAAU,UAAU;KACxE,OAAO,IAAI,MAAM,WAAW,KAAK,CAAC;KAClC;IACF;IACA,IAAI,WAAW,SAAS,YACtB,YAAY,UACT,KAAK,YAAY;KAChB,MAAM,aAAa,WAAW;KAC9B,IAAI,eAAe,KAAA,KAAa,OAAO,eAAe,UAAU;MAC9D,uBAAO,IAAI,MAAM,6CAA6C,CAAC;MAC/D;KACF;KACA,IAAI,OAAO,eAAe,YAAY,CAAC,UAAU,UAAU,GAAG;KAC9D,MAAM,cAAc,MAAM,KAAK,OAAO,QAAQ,WAAW,MAAM;KAC/D,OAAO,KAAA,GAAW;MAChB,GAAI,OAAO,eAAe,WAAW,EAAE,QAAQ,KAAK,MAAM,UAAU,EAAa,IAAI,CAAC;MACtF;MACA;KACF,CAAC;IACH,CAAC,CAAC,CACD,OAAO,UAAmB,OAAO,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAAC,CAAC;GAElG,CAAC;GACD,MAAM,KAAK;IAAE,MAAM;IAAS,YAAY;IAAc,OAAO,KAAK,MAAM,eAAe;GAAa,CAAC;EACvG,CAAC;CACH;AACF"}

package/dist/src/services/computerScriptRunner/index.d.cts

@@ -0,0 +1,12 @@
+import { ComputerScriptExecutionResult, ComputerScriptRunnerOptions } from "../../types/computerScript.cjs";
+//#region src/services/computerScriptRunner/index.d.ts
+export declare class ComputerScriptRunner {
+  private readonly client;
+  private readonly allowedScriptPaths;
+  private readonly timeoutMs;
+  private readonly maxOutputBytes;
+  constructor(options: ComputerScriptRunnerOptions);
+  execute(scriptPath: string, input: unknown, signal?: AbortSignal): Promise<ComputerScriptExecutionResult>;
+}
+//#endregion
+//# sourceMappingURL=index.d.cts.map

package/dist/src/services/computerScriptRunner/index.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.cts","names":[],"sources":["../../../../src/services/computerScriptRunner/index.ts"],"mappings":";;qBAkEa;mBACM;mBACA;mBACA;mBACA;EAEjB,YAAmB,SAAS;EAOf,QACX,oBACA,gBACA,SAAS,cACR,QAAQ"}

package/dist/src/services/computerScriptRunner/index.d.mts

@@ -0,0 +1,12 @@
+import { ComputerScriptExecutionResult, ComputerScriptRunnerOptions } from "../../types/computerScript.mjs";
+//#region src/services/computerScriptRunner/index.d.ts
+export declare class ComputerScriptRunner {
+  private readonly client;
+  private readonly allowedScriptPaths;
+  private readonly timeoutMs;
+  private readonly maxOutputBytes;
+  constructor(options: ComputerScriptRunnerOptions);
+  execute(scriptPath: string, input: unknown, signal?: AbortSignal): Promise<ComputerScriptExecutionResult>;
+}
+//#endregion
+//# sourceMappingURL=index.d.mts.map

package/dist/src/services/computerScriptRunner/index.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.mts","names":[],"sources":["../../../../src/services/computerScriptRunner/index.ts"],"mappings":";;qBAkEa;mBACM;mBACA;mBACA;mBACA;EAEjB,YAAmB,SAAS;EAOf,QACX,oBACA,gBACA,SAAS,cACR,QAAQ"}

package/dist/src/services/computerScriptRunner/index.mjs

@@ -0,0 +1,171 @@
+import { spawn } from "node:child_process";
+import { realpath } from "node:fs/promises";
+//#region src/services/computerScriptRunner/index.ts
+const DEFAULT_TIMEOUT_MS = 3e4;
+const DEFAULT_MAX_OUTPUT_BYTES = 262144;
+const WORKER_SOURCE = String.raw`
+import { readFile } from 'node:fs/promises';
+import { stripTypeScriptTypes } from 'node:module';
+
+let nextRequestId = 0;
+const pending = new Map();
+const call = (operation, payload) => new Promise((resolve, reject) => {
+  const id = String(++nextRequestId);
+  pending.set(id, { resolve, reject });
+  process.send({ type: 'program', id, operation, payload });
+});
+process.on('message', async (message) => {
+  if (message?.type === 'response') {
+    const request = pending.get(message.id);
+    if (!request) return;
+    pending.delete(message.id);
+    if (message.error) request.reject(new Error(message.error));
+    else request.resolve(message.value);
+    return;
+  }
+  if (message?.type !== 'start') return;
+  const controller = new AbortController();
+  process.once('SIGTERM', () => controller.abort());
+  const logger = Object.fromEntries(['debug', 'info', 'warn', 'error'].map((level) => [level, (value) => {
+    process.send({ type: 'log', level, message: String(value) });
+  }]));
+  try {
+    const source = await readFile(message.scriptPath, 'utf8');
+    const compiled = stripTypeScriptTypes(source, { mode: 'strip' });
+    const moduleUrl = 'data:text/javascript;base64,' + Buffer.from(compiled).toString('base64');
+    const loaded = await import(moduleUrl);
+    if (typeof loaded.run !== 'function') throw new Error('Computer script must export a named run function.');
+    const program = {
+      observe: () => call('observe'),
+      act: (action) => call('act', action),
+    };
+    const result = await loaded.run({ context: { program }, input: message.input, logger, signal: controller.signal });
+    const resultJson = result === undefined ? undefined : JSON.stringify(result);
+    process.send({ type: 'complete', resultJson });
+  } catch (error) {
+    process.send({ type: 'failed', error: error instanceof Error ? error.message : String(error) });
+  }
+});
+`;
+var ComputerScriptRunner = class {
+	client;
+	allowedScriptPaths;
+	timeoutMs;
+	maxOutputBytes;
+	constructor(options) {
+		this.client = options.client;
+		this.allowedScriptPaths = options.allowedScriptPaths;
+		this.timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+		this.maxOutputBytes = options.maxOutputBytes ?? DEFAULT_MAX_OUTPUT_BYTES;
+	}
+	async execute(scriptPath, input, signal) {
+		const resolvedPath = await realpath(scriptPath);
+		if (!(await Promise.all(this.allowedScriptPaths.map(async (candidate) => realpath(candidate).catch(() => "")))).includes(resolvedPath)) throw new Error("Computer script path is not explicitly allowed.");
+		if (!resolvedPath.endsWith(".ts")) throw new Error("Computer scripts must be TypeScript files.");
+		const serializedInput = JSON.stringify(input);
+		if (serializedInput === void 0) throw new Error("Computer script input must be JSON serializable.");
+		if (Buffer.byteLength(serializedInput) > this.maxOutputBytes) throw new Error("Computer script input exceeds the size limit.");
+		return new Promise((resolve, reject) => {
+			const child = spawn(process.execPath, [
+				"--input-type=module",
+				"--eval",
+				WORKER_SOURCE
+			], { stdio: [
+				"ignore",
+				"pipe",
+				"pipe",
+				"ipc"
+			] });
+			const controller = new AbortController();
+			const logs = [];
+			let outputBytes = 0;
+			let settled = false;
+			let operation = Promise.resolve();
+			const terminate = () => {
+				controller.abort();
+				child.kill("SIGTERM");
+				setTimeout(() => child.kill("SIGKILL"), 250).unref?.();
+			};
+			const finish = (error, value) => {
+				if (settled) return;
+				settled = true;
+				clearTimeout(timeout);
+				signal?.removeEventListener("abort", onAbort);
+				terminate();
+				if (error !== void 0) reject(error);
+				else resolve(value);
+			};
+			const addOutput = (value) => {
+				outputBytes += Buffer.byteLength(value);
+				if (outputBytes <= this.maxOutputBytes) return true;
+				finish(/* @__PURE__ */ new Error("Computer script output exceeded the size limit."));
+				return false;
+			};
+			const onAbort = () => finish(/* @__PURE__ */ new Error("Computer script execution was cancelled."));
+			const timeout = setTimeout(() => finish(/* @__PURE__ */ new Error("Computer script execution timed out.")), this.timeoutMs);
+			timeout.unref?.();
+			signal?.addEventListener("abort", onAbort, { once: true });
+			if (signal?.aborted === true) return onAbort();
+			child.stdout?.on("data", (chunk) => addOutput(chunk.toString("utf8")));
+			child.stderr?.on("data", (chunk) => addOutput(chunk.toString("utf8")));
+			child.on("error", (error) => finish(error));
+			child.on("exit", (code, exitSignal) => {
+				if (!settled) finish(/* @__PURE__ */ new Error(`Computer script worker exited before completion (${exitSignal ?? code ?? "unknown"}).`));
+			});
+			child.on("message", (rawMessage) => {
+				if (settled) return;
+				if (rawMessage.type === "log" && typeof rawMessage.message === "string") {
+					const line = `[${String(rawMessage.level)}] ${rawMessage.message}`;
+					if (addOutput(line)) logs.push(line);
+					return;
+				}
+				if (rawMessage.type === "program" && typeof rawMessage.id === "string" && (rawMessage.operation === "observe" || rawMessage.operation === "act")) {
+					operation = operation.then(async () => {
+						try {
+							const value = rawMessage.operation === "observe" ? await this.client.observe(controller.signal) : await this.client.act(rawMessage.payload, controller.signal);
+							child.send({
+								type: "response",
+								id: rawMessage.id,
+								value
+							});
+						} catch (error) {
+							child.send({
+								type: "response",
+								id: rawMessage.id,
+								error: error instanceof Error ? error.message : String(error)
+							});
+						}
+					});
+					return;
+				}
+				if (rawMessage.type === "failed" && typeof rawMessage.error === "string") {
+					finish(new Error(rawMessage.error));
+					return;
+				}
+				if (rawMessage.type === "complete") operation = operation.then(async () => {
+					const resultJson = rawMessage.resultJson;
+					if (resultJson !== void 0 && typeof resultJson !== "string") {
+						finish(/* @__PURE__ */ new Error("Computer script returned an invalid result."));
+						return;
+					}
+					if (typeof resultJson === "string" && !addOutput(resultJson)) return;
+					const observation = await this.client.observe(controller.signal);
+					finish(void 0, {
+						...typeof resultJson === "string" ? { result: JSON.parse(resultJson) } : {},
+						observation,
+						logs
+					});
+				}).catch((error) => finish(error instanceof Error ? error : new Error(String(error))));
+			});
+			child.send({
+				type: "start",
+				scriptPath: resolvedPath,
+				input: JSON.parse(serializedInput)
+			});
+		});
+	}
+};
+//#endregion
+export { ComputerScriptRunner };
+
+//# sourceMappingURL=index.mjs.map

package/dist/src/services/computerScriptRunner/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","names":[],"sources":["../../../../src/services/computerScriptRunner/index.ts"],"sourcesContent":["import { spawn } from 'node:child_process';\nimport { realpath } from 'node:fs/promises';\n\nimport type { ComputerScriptExecutionResult, ComputerScriptRunnerOptions } from '../../types/computerScript';\nimport type { ComputerUseAction } from '../../types/computerUse';\nimport type { ComputerUseSessionClient } from '../sessionApiClient';\n\nconst DEFAULT_TIMEOUT_MS = 30_000;\nconst DEFAULT_MAX_OUTPUT_BYTES = 256 * 1024;\n\nconst WORKER_SOURCE = String.raw`\nimport { readFile } from 'node:fs/promises';\nimport { stripTypeScriptTypes } from 'node:module';\n\nlet nextRequestId = 0;\nconst pending = new Map();\nconst call = (operation, payload) => new Promise((resolve, reject) => {\n  const id = String(++nextRequestId);\n  pending.set(id, { resolve, reject });\n  process.send({ type: 'program', id, operation, payload });\n});\nprocess.on('message', async (message) => {\n  if (message?.type === 'response') {\n    const request = pending.get(message.id);\n    if (!request) return;\n    pending.delete(message.id);\n    if (message.error) request.reject(new Error(message.error));\n    else request.resolve(message.value);\n    return;\n  }\n  if (message?.type !== 'start') return;\n  const controller = new AbortController();\n  process.once('SIGTERM', () => controller.abort());\n  const logger = Object.fromEntries(['debug', 'info', 'warn', 'error'].map((level) => [level, (value) => {\n    process.send({ type: 'log', level, message: String(value) });\n  }]));\n  try {\n    const source = await readFile(message.scriptPath, 'utf8');\n    const compiled = stripTypeScriptTypes(source, { mode: 'strip' });\n    const moduleUrl = 'data:text/javascript;base64,' + Buffer.from(compiled).toString('base64');\n    const loaded = await import(moduleUrl);\n    if (typeof loaded.run !== 'function') throw new Error('Computer script must export a named run function.');\n    const program = {\n      observe: () => call('observe'),\n      act: (action) => call('act', action),\n    };\n    const result = await loaded.run({ context: { program }, input: message.input, logger, signal: controller.signal });\n    const resultJson = result === undefined ? undefined : JSON.stringify(result);\n    process.send({ type: 'complete', resultJson });\n  } catch (error) {\n    process.send({ type: 'failed', error: error instanceof Error ? error.message : String(error) });\n  }\n});\n`;\n\ninterface WorkerMessage {\n  readonly type?: unknown;\n  readonly id?: unknown;\n  readonly operation?: unknown;\n  readonly payload?: unknown;\n  readonly level?: unknown;\n  readonly message?: unknown;\n  readonly resultJson?: unknown;\n  readonly error?: unknown;\n}\n\nexport class ComputerScriptRunner {\n  private readonly client: ComputerUseSessionClient;\n  private readonly allowedScriptPaths: readonly string[];\n  private readonly timeoutMs: number;\n  private readonly maxOutputBytes: number;\n\n  public constructor(options: ComputerScriptRunnerOptions) {\n    this.client = options.client;\n    this.allowedScriptPaths = options.allowedScriptPaths;\n    this.timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;\n    this.maxOutputBytes = options.maxOutputBytes ?? DEFAULT_MAX_OUTPUT_BYTES;\n  }\n\n  public async execute(\n    scriptPath: string,\n    input: unknown,\n    signal?: AbortSignal,\n  ): Promise<ComputerScriptExecutionResult> {\n    const resolvedPath = await realpath(scriptPath);\n    const allowedPaths = await Promise.all(\n      this.allowedScriptPaths.map(async (candidate) => realpath(candidate).catch(() => '')),\n    );\n    if (!allowedPaths.includes(resolvedPath)) throw new Error('Computer script path is not explicitly allowed.');\n    if (!resolvedPath.endsWith('.ts')) throw new Error('Computer scripts must be TypeScript files.');\n\n    const serializedInput = JSON.stringify(input);\n    if (serializedInput === undefined) throw new Error('Computer script input must be JSON serializable.');\n    if (Buffer.byteLength(serializedInput) > this.maxOutputBytes)\n      throw new Error('Computer script input exceeds the size limit.');\n\n    return new Promise((resolve, reject) => {\n      const child = spawn(process.execPath, ['--input-type=module', '--eval', WORKER_SOURCE], {\n        stdio: ['ignore', 'pipe', 'pipe', 'ipc'],\n      });\n      const controller = new AbortController();\n      const logs: string[] = [];\n      let outputBytes = 0;\n      let settled = false;\n      let operation = Promise.resolve();\n\n      const terminate = (): void => {\n        controller.abort();\n        child.kill('SIGTERM');\n        const killTimer = setTimeout(() => child.kill('SIGKILL'), 250);\n        killTimer.unref?.();\n      };\n      const finish = (error?: Error, value?: ComputerScriptExecutionResult): void => {\n        if (settled) return;\n        settled = true;\n        clearTimeout(timeout);\n        signal?.removeEventListener('abort', onAbort);\n        terminate();\n        if (error !== undefined) reject(error);\n        else resolve(value as ComputerScriptExecutionResult);\n      };\n      const addOutput = (value: string): boolean => {\n        outputBytes += Buffer.byteLength(value);\n        if (outputBytes <= this.maxOutputBytes) return true;\n        finish(new Error('Computer script output exceeded the size limit.'));\n        return false;\n      };\n      const onAbort = () => finish(new Error('Computer script execution was cancelled.'));\n      const timeout = setTimeout(() => finish(new Error('Computer script execution timed out.')), this.timeoutMs);\n      timeout.unref?.();\n      signal?.addEventListener('abort', onAbort, { once: true });\n      if (signal?.aborted === true) return onAbort();\n\n      child.stdout?.on('data', (chunk: Buffer) => addOutput(chunk.toString('utf8')));\n      child.stderr?.on('data', (chunk: Buffer) => addOutput(chunk.toString('utf8')));\n      child.on('error', (error) => finish(error));\n      child.on('exit', (code, exitSignal) => {\n        if (!settled)\n          finish(new Error(`Computer script worker exited before completion (${exitSignal ?? code ?? 'unknown'}).`));\n      });\n      child.on('message', (rawMessage: WorkerMessage) => {\n        if (settled) return;\n        if (rawMessage.type === 'log' && typeof rawMessage.message === 'string') {\n          const line = `[${String(rawMessage.level)}] ${rawMessage.message}`;\n          if (addOutput(line)) logs.push(line);\n          return;\n        }\n        if (\n          rawMessage.type === 'program' &&\n          typeof rawMessage.id === 'string' &&\n          (rawMessage.operation === 'observe' || rawMessage.operation === 'act')\n        ) {\n          operation = operation.then(async () => {\n            try {\n              const value =\n                rawMessage.operation === 'observe'\n                  ? await this.client.observe(controller.signal)\n                  : await this.client.act(rawMessage.payload as ComputerUseAction, controller.signal);\n              child.send({ type: 'response', id: rawMessage.id, value });\n            } catch (error) {\n              child.send({\n                type: 'response',\n                id: rawMessage.id,\n                error: error instanceof Error ? error.message : String(error),\n              });\n            }\n          });\n          return;\n        }\n        if (rawMessage.type === 'failed' && typeof rawMessage.error === 'string') {\n          finish(new Error(rawMessage.error));\n          return;\n        }\n        if (rawMessage.type === 'complete') {\n          operation = operation\n            .then(async () => {\n              const resultJson = rawMessage.resultJson;\n              if (resultJson !== undefined && typeof resultJson !== 'string') {\n                finish(new Error('Computer script returned an invalid result.'));\n                return;\n              }\n              if (typeof resultJson === 'string' && !addOutput(resultJson)) return;\n              const observation = await this.client.observe(controller.signal);\n              finish(undefined, {\n                ...(typeof resultJson === 'string' ? { result: JSON.parse(resultJson) as unknown } : {}),\n                observation,\n                logs,\n              });\n            })\n            .catch((error: unknown) => finish(error instanceof Error ? error : new Error(String(error))));\n        }\n      });\n      child.send({ type: 'start', scriptPath: resolvedPath, input: JSON.parse(serializedInput) as unknown });\n    });\n  }\n}\n"],"mappings":";;;AAOA,MAAM,qBAAqB;AAC3B,MAAM,2BAA2B;AAEjC,MAAM,gBAAgB,OAAO,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAwDhC,IAAa,uBAAb,MAAkC;CAChC;CACA;CACA;CACA;CAEA,YAAmB,SAAsC;EACvD,KAAK,SAAS,QAAQ;EACtB,KAAK,qBAAqB,QAAQ;EAClC,KAAK,YAAY,QAAQ,aAAa;EACtC,KAAK,iBAAiB,QAAQ,kBAAkB;CAClD;CAEA,MAAa,QACX,YACA,OACA,QACwC;EACxC,MAAM,eAAe,MAAM,SAAS,UAAU;EAI9C,IAAI,EAAC,MAHsB,QAAQ,IACjC,KAAK,mBAAmB,IAAI,OAAO,cAAc,SAAS,SAAS,CAAC,CAAC,YAAY,EAAE,CAAC,CACtF,EAAA,CACkB,SAAS,YAAY,GAAG,MAAM,IAAI,MAAM,iDAAiD;EAC3G,IAAI,CAAC,aAAa,SAAS,KAAK,GAAG,MAAM,IAAI,MAAM,4CAA4C;EAE/F,MAAM,kBAAkB,KAAK,UAAU,KAAK;EAC5C,IAAI,oBAAoB,KAAA,GAAW,MAAM,IAAI,MAAM,kDAAkD;EACrG,IAAI,OAAO,WAAW,eAAe,IAAI,KAAK,gBAC5C,MAAM,IAAI,MAAM,+CAA+C;EAEjE,OAAO,IAAI,SAAS,SAAS,WAAW;GACtC,MAAM,QAAQ,MAAM,QAAQ,UAAU;IAAC;IAAuB;IAAU;GAAa,GAAG,EACtF,OAAO;IAAC;IAAU;IAAQ;IAAQ;GAAK,EACzC,CAAC;GACD,MAAM,aAAa,IAAI,gBAAgB;GACvC,MAAM,OAAiB,CAAC;GACxB,IAAI,cAAc;GAClB,IAAI,UAAU;GACd,IAAI,YAAY,QAAQ,QAAQ;GAEhC,MAAM,kBAAwB;IAC5B,WAAW,MAAM;IACjB,MAAM,KAAK,SAAS;IAEpB,iBADmC,MAAM,KAAK,SAAS,GAAG,GAClD,CAAC,CAAC,QAAQ;GACpB;GACA,MAAM,UAAU,OAAe,UAAgD;IAC7E,IAAI,SAAS;IACb,UAAU;IACV,aAAa,OAAO;IACpB,QAAQ,oBAAoB,SAAS,OAAO;IAC5C,UAAU;IACV,IAAI,UAAU,KAAA,GAAW,OAAO,KAAK;SAChC,QAAQ,KAAsC;GACrD;GACA,MAAM,aAAa,UAA2B;IAC5C,eAAe,OAAO,WAAW,KAAK;IACtC,IAAI,eAAe,KAAK,gBAAgB,OAAO;IAC/C,uBAAO,IAAI,MAAM,iDAAiD,CAAC;IACnE,OAAO;GACT;GACA,MAAM,gBAAgB,uBAAO,IAAI,MAAM,0CAA0C,CAAC;GAClF,MAAM,UAAU,iBAAiB,uBAAO,IAAI,MAAM,sCAAsC,CAAC,GAAG,KAAK,SAAS;GAC1G,QAAQ,QAAQ;GAChB,QAAQ,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;GACzD,IAAI,QAAQ,YAAY,MAAM,OAAO,QAAQ;GAE7C,MAAM,QAAQ,GAAG,SAAS,UAAkB,UAAU,MAAM,SAAS,MAAM,CAAC,CAAC;GAC7E,MAAM,QAAQ,GAAG,SAAS,UAAkB,UAAU,MAAM,SAAS,MAAM,CAAC,CAAC;GAC7E,MAAM,GAAG,UAAU,UAAU,OAAO,KAAK,CAAC;GAC1C,MAAM,GAAG,SAAS,MAAM,eAAe;IACrC,IAAI,CAAC,SACH,uBAAO,IAAI,MAAM,oDAAoD,cAAc,QAAQ,UAAU,GAAG,CAAC;GAC7G,CAAC;GACD,MAAM,GAAG,YAAY,eAA8B;IACjD,IAAI,SAAS;IACb,IAAI,WAAW,SAAS,SAAS,OAAO,WAAW,YAAY,UAAU;KACvE,MAAM,OAAO,IAAI,OAAO,WAAW,KAAK,EAAE,IAAI,WAAW;KACzD,IAAI,UAAU,IAAI,GAAG,KAAK,KAAK,IAAI;KACnC;IACF;IACA,IACE,WAAW,SAAS,aACpB,OAAO,WAAW,OAAO,aACxB,WAAW,cAAc,aAAa,WAAW,cAAc,QAChE;KACA,YAAY,UAAU,KAAK,YAAY;MACrC,IAAI;OACF,MAAM,QACJ,WAAW,cAAc,YACrB,MAAM,KAAK,OAAO,QAAQ,WAAW,MAAM,IAC3C,MAAM,KAAK,OAAO,IAAI,WAAW,SAA8B,WAAW,MAAM;OACtF,MAAM,KAAK;QAAE,MAAM;QAAY,IAAI,WAAW;QAAI;OAAM,CAAC;MAC3D,SAAS,OAAO;OACd,MAAM,KAAK;QACT,MAAM;QACN,IAAI,WAAW;QACf,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;OAC9D,CAAC;MACH;KACF,CAAC;KACD;IACF;IACA,IAAI,WAAW,SAAS,YAAY,OAAO,WAAW,UAAU,UAAU;KACxE,OAAO,IAAI,MAAM,WAAW,KAAK,CAAC;KAClC;IACF;IACA,IAAI,WAAW,SAAS,YACtB,YAAY,UACT,KAAK,YAAY;KAChB,MAAM,aAAa,WAAW;KAC9B,IAAI,eAAe,KAAA,KAAa,OAAO,eAAe,UAAU;MAC9D,uBAAO,IAAI,MAAM,6CAA6C,CAAC;MAC/D;KACF;KACA,IAAI,OAAO,eAAe,YAAY,CAAC,UAAU,UAAU,GAAG;KAC9D,MAAM,cAAc,MAAM,KAAK,OAAO,QAAQ,WAAW,MAAM;KAC/D,OAAO,KAAA,GAAW;MAChB,GAAI,OAAO,eAAe,WAAW,EAAE,QAAQ,KAAK,MAAM,UAAU,EAAa,IAAI,CAAC;MACtF;MACA;KACF,CAAC;IACH,CAAC,CAAC,CACD,OAAO,UAAmB,OAAO,iBAAiB,QAAQ,QAAQ,IAAI,MAAM,OAAO,KAAK,CAAC,CAAC,CAAC;GAElG,CAAC;GACD,MAAM,KAAK;IAAE,MAAM;IAAS,YAAY;IAAc,OAAO,KAAK,MAAM,eAAe;GAAa,CAAC;EACvG,CAAC;CACH;AACF"}