@codeworksh/aikit 0.2.1 → 0.3.0

package/agent/codemode/drivers/drivers.d.mts

@@ -0,0 +1,7 @@
+import { t as CodeMode } from "../../../codemode-z3ZIw-5q.mjs";
+
+//#region src/agent/codemode/drivers/quickjs-wasi-driver.d.ts
+declare function createQuickJSWasiDriver(): CodeMode.Driver;
+//#endregion
+export { createQuickJSWasiDriver };
+//# sourceMappingURL=drivers.d.mts.map

package/agent/codemode/drivers/drivers.mjs

@@ -0,0 +1,122 @@
+import { QuickJS } from "quickjs-wasi";
+//#region src/agent/codemode/drivers/quickjs-wasi-driver.ts
+const DEFAULT_TIMEOUT_MS = 3e4;
+const DEFAULT_MEMORY_LIMIT_MB = 128;
+function withMessageInStack(message, stack) {
+	if (!stack) return void 0;
+	return stack.includes(message) ? stack : `${message}\n${stack}`;
+}
+function parseErrorLine(stack) {
+	if (typeof stack !== "string") return void 0;
+	const lineMatch = stack.match(/:(\d+)(?::\d+)?\)?$/m);
+	return lineMatch ? Number(lineMatch[1]) : void 0;
+}
+function normalizeExecutionError(error) {
+	if (error instanceof Error) return {
+		name: error.name,
+		message: error.message,
+		stack: withMessageInStack(error.message, error.stack),
+		line: parseErrorLine(error.stack)
+	};
+	if (typeof error === "object" && error !== null) {
+		const candidate = error;
+		const message = typeof candidate.message === "string" ? candidate.message : JSON.stringify(candidate) || "Unknown Execution Error";
+		return {
+			name: typeof candidate.name === "string" ? candidate.name : void 0,
+			message,
+			stack: typeof candidate.stack === "string" ? withMessageInStack(message, candidate.stack) : void 0,
+			line: typeof candidate.line === "number" ? candidate.line : parseErrorLine(candidate.stack)
+		};
+	}
+	return {
+		name: "Error",
+		message: String(error)
+	};
+}
+function toMegabytes(value) {
+	return (value ?? DEFAULT_MEMORY_LIMIT_MB) * 1024 * 1024;
+}
+function createConsole(vm, logs) {
+	const consoleObject = vm.newObject();
+	for (const [name, prefix] of [
+		["log", ""],
+		["info", "INFO: "],
+		["warn", "WARN: "],
+		["error", "ERROR: "]
+	]) {
+		const fn = vm.newFunction(name, (...args) => {
+			const message = args.map((arg) => String(vm.dump(arg))).join(" ");
+			logs.push(`${prefix}${message}`);
+			return vm.undefined;
+		});
+		consoleObject.setProp(name, fn);
+	}
+	vm.setProp(vm.global, "console", consoleObject);
+}
+function registerBindings(vm, bindings, signal) {
+	let callCounter = 0;
+	for (const [name, binding] of Object.entries(bindings)) {
+		const fn = vm.newFunction(name, (...args) => {
+			const input = args[0] ? vm.dump(args[0]) : void 0;
+			const result = Promise.resolve(binding.execute(`${name}:${++callCounter}`, input, signal));
+			return vm.hostToHandle(result);
+		});
+		vm.setProp(vm.global, name, fn);
+	}
+}
+function createQuickJSWasiDriver() {
+	return { async createContext(config) {
+		const timeout = config.timeout ?? DEFAULT_TIMEOUT_MS;
+		const deadline = Date.now() + timeout;
+		const logs = [];
+		const vm = await QuickJS.create({
+			memoryLimit: toMegabytes(config.memoryLimit),
+			interruptHandler: () => Boolean(config.signal?.aborted) || Date.now() > deadline
+		});
+		createConsole(vm, logs);
+		registerBindings(vm, config.bindings, config.signal);
+		return {
+			async execute(code) {
+				let resultHandle;
+				try {
+					resultHandle = vm.evalCode(code, "script.js");
+					vm.executePendingJobs();
+					const settled = await vm.resolvePromise(resultHandle);
+					if ("error" in settled) try {
+						return {
+							success: false,
+							error: normalizeExecutionError(vm.dump(settled.error)),
+							logs: [...logs]
+						};
+					} finally {
+						settled.error.dispose();
+					}
+					try {
+						return {
+							success: true,
+							value: vm.dump(settled.value),
+							logs: [...logs]
+						};
+					} finally {
+						settled.value.dispose();
+					}
+				} catch (error) {
+					return {
+						success: false,
+						error: normalizeExecutionError(error),
+						logs: [...logs]
+					};
+				} finally {
+					resultHandle?.dispose();
+				}
+			},
+			async dispose() {
+				vm.dispose();
+			}
+		};
+	} };
+}
+//#endregion
+export { createQuickJSWasiDriver };
+
+//# sourceMappingURL=drivers.mjs.map

package/agent/codemode/drivers/drivers.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"drivers.mjs","names":[],"sources":["../../../../../src/agent/codemode/drivers/quickjs-wasi-driver.ts"],"sourcesContent":["import { QuickJS } from \"quickjs-wasi\";\nimport type { CodeMode } from \"../codemode\";\n\nconst DEFAULT_TIMEOUT_MS = 30_000;\nconst DEFAULT_MEMORY_LIMIT_MB = 128;\n\nfunction withMessageInStack(message: string, stack: string | undefined): string | undefined {\n\tif (!stack) return undefined;\n\treturn stack.includes(message) ? stack : `${message}\\n${stack}`;\n}\n\nfunction parseErrorLine(stack: unknown): number | undefined {\n\tif (typeof stack !== \"string\") return undefined;\n\tconst lineMatch = stack.match(/:(\\d+)(?::\\d+)?\\)?$/m);\n\treturn lineMatch ? Number(lineMatch[1]) : undefined;\n}\n\nfunction normalizeExecutionError(error: unknown): CodeMode.NormalizedError {\n\tif (error instanceof Error) {\n\t\treturn {\n\t\t\tname: error.name,\n\t\t\tmessage: error.message,\n\t\t\tstack: withMessageInStack(error.message, error.stack),\n\t\t\tline: parseErrorLine(error.stack),\n\t\t};\n\t}\n\n\tif (typeof error === \"object\" && error !== null) {\n\t\tconst candidate = error as Record<string, unknown>;\n\t\tconst message =\n\t\t\ttypeof candidate.message === \"string\"\n\t\t\t\t? candidate.message\n\t\t\t\t: JSON.stringify(candidate) || \"Unknown Execution Error\";\n\t\treturn {\n\t\t\tname: typeof candidate.name === \"string\" ? candidate.name : undefined,\n\t\t\tmessage,\n\t\t\tstack: typeof candidate.stack === \"string\" ? withMessageInStack(message, candidate.stack) : undefined,\n\t\t\tline: typeof candidate.line === \"number\" ? candidate.line : parseErrorLine(candidate.stack),\n\t\t};\n\t}\n\n\treturn {\n\t\tname: \"Error\",\n\t\tmessage: String(error),\n\t};\n}\n\nfunction toMegabytes(value: number | undefined): number {\n\treturn (value ?? DEFAULT_MEMORY_LIMIT_MB) * 1024 * 1024;\n}\n\nfunction createConsole(vm: QuickJS, logs: string[]) {\n\tconst consoleObject = vm.newObject();\n\tconst methods = [\n\t\t[\"log\", \"\"],\n\t\t[\"info\", \"INFO: \"],\n\t\t[\"warn\", \"WARN: \"],\n\t\t[\"error\", \"ERROR: \"],\n\t] as const;\n\n\tfor (const [name, prefix] of methods) {\n\t\tconst fn = vm.newFunction(name, (...args) => {\n\t\t\tconst message = args.map((arg) => String(vm.dump(arg))).join(\" \");\n\t\t\tlogs.push(`${prefix}${message}`);\n\t\t\treturn vm.undefined;\n\t\t});\n\t\tconsoleObject.setProp(name, fn);\n\t}\n\n\tvm.setProp(vm.global, \"console\", consoleObject);\n}\n\nfunction registerBindings(vm: QuickJS, bindings: Record<string, CodeMode.ToolBinding>, signal?: AbortSignal) {\n\tlet callCounter = 0;\n\n\tfor (const [name, binding] of Object.entries(bindings)) {\n\t\tconst fn = vm.newFunction(name, (...args) => {\n\t\t\tconst input = args[0] ? vm.dump(args[0]) : undefined;\n\t\t\tconst result = Promise.resolve(binding.execute(`${name}:${++callCounter}`, input, signal));\n\t\t\treturn vm.hostToHandle(result);\n\t\t});\n\t\tvm.setProp(vm.global, name, fn);\n\t}\n}\n\nexport function createQuickJSWasiDriver(): CodeMode.Driver {\n\treturn {\n\t\tasync createContext(config: CodeMode.DriverContextConfig): Promise<CodeMode.Context> {\n\t\t\tconst timeout = config.timeout ?? DEFAULT_TIMEOUT_MS;\n\t\t\tconst deadline = Date.now() + timeout;\n\t\t\tconst logs: string[] = [];\n\t\t\t// create VM for execution sandbox\n\t\t\tconst vm = await QuickJS.create({\n\t\t\t\tmemoryLimit: toMegabytes(config.memoryLimit),\n\t\t\t\tinterruptHandler: () => Boolean(config.signal?.aborted) || Date.now() > deadline,\n\t\t\t});\n\n\t\t\t// create console log bindings\n\t\t\tcreateConsole(vm, logs);\n\t\t\t// register tool fn bindings\n\t\t\tregisterBindings(vm, config.bindings, config.signal);\n\n\t\t\treturn {\n\t\t\t\tasync execute<T = unknown>(code: string) {\n\t\t\t\t\tlet resultHandle: ReturnType<typeof vm.evalCode> | undefined;\n\n\t\t\t\t\ttry {\n\t\t\t\t\t\tresultHandle = vm.evalCode(code, \"script.js\");\n\t\t\t\t\t\tvm.executePendingJobs();\n\n\t\t\t\t\t\tconst settled = await vm.resolvePromise(resultHandle);\n\t\t\t\t\t\tif (\"error\" in settled) {\n\t\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\t\treturn {\n\t\t\t\t\t\t\t\t\tsuccess: false,\n\t\t\t\t\t\t\t\t\terror: normalizeExecutionError(vm.dump(settled.error)),\n\t\t\t\t\t\t\t\t\tlogs: [...logs],\n\t\t\t\t\t\t\t\t};\n\t\t\t\t\t\t\t} finally {\n\t\t\t\t\t\t\t\tsettled.error.dispose();\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t}\n\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\treturn {\n\t\t\t\t\t\t\t\tsuccess: true,\n\t\t\t\t\t\t\t\tvalue: vm.dump(settled.value) as T,\n\t\t\t\t\t\t\t\tlogs: [...logs],\n\t\t\t\t\t\t\t};\n\t\t\t\t\t\t} finally {\n\t\t\t\t\t\t\tsettled.value.dispose();\n\t\t\t\t\t\t}\n\t\t\t\t\t} catch (error) {\n\t\t\t\t\t\treturn {\n\t\t\t\t\t\t\tsuccess: false,\n\t\t\t\t\t\t\terror: normalizeExecutionError(error),\n\t\t\t\t\t\t\tlogs: [...logs],\n\t\t\t\t\t\t};\n\t\t\t\t\t} finally {\n\t\t\t\t\t\tresultHandle?.dispose();\n\t\t\t\t\t}\n\t\t\t\t},\n\t\t\t\tasync dispose() {\n\t\t\t\t\tvm.dispose();\n\t\t\t\t},\n\t\t\t};\n\t\t},\n\t};\n}\n"],"mappings":";;AAGA,MAAM,qBAAqB;AAC3B,MAAM,0BAA0B;AAEhC,SAAS,mBAAmB,SAAiB,OAA+C;AAC3F,KAAI,CAAC,MAAO,QAAO,KAAA;AACnB,QAAO,MAAM,SAAS,QAAQ,GAAG,QAAQ,GAAG,QAAQ,IAAI;;AAGzD,SAAS,eAAe,OAAoC;AAC3D,KAAI,OAAO,UAAU,SAAU,QAAO,KAAA;CACtC,MAAM,YAAY,MAAM,MAAM,uBAAuB;AACrD,QAAO,YAAY,OAAO,UAAU,GAAG,GAAG,KAAA;;AAG3C,SAAS,wBAAwB,OAA0C;AAC1E,KAAI,iBAAiB,MACpB,QAAO;EACN,MAAM,MAAM;EACZ,SAAS,MAAM;EACf,OAAO,mBAAmB,MAAM,SAAS,MAAM,MAAM;EACrD,MAAM,eAAe,MAAM,MAAM;EACjC;AAGF,KAAI,OAAO,UAAU,YAAY,UAAU,MAAM;EAChD,MAAM,YAAY;EAClB,MAAM,UACL,OAAO,UAAU,YAAY,WAC1B,UAAU,UACV,KAAK,UAAU,UAAU,IAAI;AACjC,SAAO;GACN,MAAM,OAAO,UAAU,SAAS,WAAW,UAAU,OAAO,KAAA;GAC5D;GACA,OAAO,OAAO,UAAU,UAAU,WAAW,mBAAmB,SAAS,UAAU,MAAM,GAAG,KAAA;GAC5F,MAAM,OAAO,UAAU,SAAS,WAAW,UAAU,OAAO,eAAe,UAAU,MAAM;GAC3F;;AAGF,QAAO;EACN,MAAM;EACN,SAAS,OAAO,MAAM;EACtB;;AAGF,SAAS,YAAY,OAAmC;AACvD,SAAQ,SAAS,2BAA2B,OAAO;;AAGpD,SAAS,cAAc,IAAa,MAAgB;CACnD,MAAM,gBAAgB,GAAG,WAAW;AAQpC,MAAK,MAAM,CAAC,MAAM,WAPF;EACf,CAAC,OAAO,GAAG;EACX,CAAC,QAAQ,SAAS;EAClB,CAAC,QAAQ,SAAS;EAClB,CAAC,SAAS,UAAU;EACpB,EAEqC;EACrC,MAAM,KAAK,GAAG,YAAY,OAAO,GAAG,SAAS;GAC5C,MAAM,UAAU,KAAK,KAAK,QAAQ,OAAO,GAAG,KAAK,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;AACjE,QAAK,KAAK,GAAG,SAAS,UAAU;AAChC,UAAO,GAAG;IACT;AACF,gBAAc,QAAQ,MAAM,GAAG;;AAGhC,IAAG,QAAQ,GAAG,QAAQ,WAAW,cAAc;;AAGhD,SAAS,iBAAiB,IAAa,UAAgD,QAAsB;CAC5G,IAAI,cAAc;AAElB,MAAK,MAAM,CAAC,MAAM,YAAY,OAAO,QAAQ,SAAS,EAAE;EACvD,MAAM,KAAK,GAAG,YAAY,OAAO,GAAG,SAAS;GAC5C,MAAM,QAAQ,KAAK,KAAK,GAAG,KAAK,KAAK,GAAG,GAAG,KAAA;GAC3C,MAAM,SAAS,QAAQ,QAAQ,QAAQ,QAAQ,GAAG,KAAK,GAAG,EAAE,eAAe,OAAO,OAAO,CAAC;AAC1F,UAAO,GAAG,aAAa,OAAO;IAC7B;AACF,KAAG,QAAQ,GAAG,QAAQ,MAAM,GAAG;;;AAIjC,SAAgB,0BAA2C;AAC1D,QAAO,EACN,MAAM,cAAc,QAAiE;EACpF,MAAM,UAAU,OAAO,WAAW;EAClC,MAAM,WAAW,KAAK,KAAK,GAAG;EAC9B,MAAM,OAAiB,EAAE;EAEzB,MAAM,KAAK,MAAM,QAAQ,OAAO;GAC/B,aAAa,YAAY,OAAO,YAAY;GAC5C,wBAAwB,QAAQ,OAAO,QAAQ,QAAQ,IAAI,KAAK,KAAK,GAAG;GACxE,CAAC;AAGF,gBAAc,IAAI,KAAK;AAEvB,mBAAiB,IAAI,OAAO,UAAU,OAAO,OAAO;AAEpD,SAAO;GACN,MAAM,QAAqB,MAAc;IACxC,IAAI;AAEJ,QAAI;AACH,oBAAe,GAAG,SAAS,MAAM,YAAY;AAC7C,QAAG,oBAAoB;KAEvB,MAAM,UAAU,MAAM,GAAG,eAAe,aAAa;AACrD,SAAI,WAAW,QACd,KAAI;AACH,aAAO;OACN,SAAS;OACT,OAAO,wBAAwB,GAAG,KAAK,QAAQ,MAAM,CAAC;OACtD,MAAM,CAAC,GAAG,KAAK;OACf;eACQ;AACT,cAAQ,MAAM,SAAS;;AAIzB,SAAI;AACH,aAAO;OACN,SAAS;OACT,OAAO,GAAG,KAAK,QAAQ,MAAM;OAC7B,MAAM,CAAC,GAAG,KAAK;OACf;eACQ;AACT,cAAQ,MAAM,SAAS;;aAEhB,OAAO;AACf,YAAO;MACN,SAAS;MACT,OAAO,wBAAwB,MAAM;MACrC,MAAM,CAAC,GAAG,KAAK;MACf;cACQ;AACT,mBAAc,SAAS;;;GAGzB,MAAM,UAAU;AACf,OAAG,SAAS;;GAEb;IAEF"}