openvibe 0.63.2 → 0.63.3

package/dist/core/companion/scheduler/workers/cpu-worker.js

@@ -0,0 +1,134 @@
+import { spawn } from "child_process";
+import { cpus, totalmem } from "os";
+import { BaseWorker } from "./base-worker.js";
+export class CPUWorker extends BaseWorker {
+    id;
+    type = "cpu";
+    capabilities;
+    constructor(id) {
+        super();
+        this.id = id || `cpu_${Date.now()}_${Math.random().toString(36).substr(2, 6)}`;
+        this.capabilities = {
+            cpuCores: cpus().length,
+            memoryMb: Math.floor(totalmem() / (1024 * 1024)),
+            hasGpu: false,
+            supportsTypes: ["cpu_intensive", "code", "script", "bash"],
+        };
+    }
+    async execute(task, signal) {
+        this.markBusy(task.id);
+        const startTime = Date.now();
+        try {
+            let result;
+            if (task.type === "bash" || task.type === "script") {
+                result = await this.executeBash(task, signal);
+            }
+            else if (task.type === "code") {
+                result = await this.executeCode(task, signal);
+            }
+            else {
+                result = await this.executeCPUTask(task, signal);
+            }
+            const executionTime = Date.now() - startTime;
+            this.updateStats(executionTime, result.success);
+            return result;
+        }
+        catch (error) {
+            const executionTime = Date.now() - startTime;
+            this.updateStats(executionTime, false);
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "EXECUTION_ERROR",
+            };
+        }
+        finally {
+            this.markIdle();
+        }
+    }
+    executeBash(task, signal) {
+        return new Promise((resolve) => {
+            const stdout = [];
+            const stderr = [];
+            const payload = task.payload;
+            const child = spawn(payload.command || "bash", payload.args || [], {
+                cwd: payload.workingDir,
+                env: { ...process.env, ...payload.env },
+                stdio: ["ignore", "pipe", "pipe"],
+            });
+            const timeout = task.requirements?.maxExecutionTimeMs || 300000;
+            const timeoutId = setTimeout(() => {
+                child.kill("SIGTERM");
+            }, timeout);
+            const onAbort = () => {
+                child.kill("SIGTERM");
+            };
+            signal?.addEventListener("abort", onAbort);
+            child.stdout?.on("data", (data) => stdout.push(data));
+            child.stderr?.on("data", (data) => stderr.push(data));
+            child.on("close", (code) => {
+                clearTimeout(timeoutId);
+                signal?.removeEventListener("abort", onAbort);
+                resolve({
+                    success: code === 0,
+                    exitCode: code || 0,
+                    stdout: Buffer.concat(stdout).toString(),
+                    stderr: Buffer.concat(stderr).toString(),
+                });
+            });
+            child.on("error", (error) => {
+                clearTimeout(timeoutId);
+                signal?.removeEventListener("abort", onAbort);
+                resolve({
+                    success: false,
+                    stdout: Buffer.concat(stdout).toString(),
+                    stderr: Buffer.concat(stderr).toString(),
+                    error: error.message,
+                    errorCode: "SPAWN_ERROR",
+                });
+            });
+        });
+    }
+    async executeCode(task, signal) {
+        const payload = task.payload;
+        const language = payload.language?.toLowerCase() || "python";
+        let command;
+        let args;
+        switch (language) {
+            case "python":
+            case "py":
+                command = "python3";
+                args = ["-c", payload.code || ""];
+                break;
+            case "javascript":
+            case "js":
+            case "node":
+                command = "node";
+                args = ["-e", payload.code || ""];
+                break;
+            case "typescript":
+            case "ts":
+                command = "npx";
+                args = ["tsx", "-e", payload.code || ""];
+                break;
+            default:
+                return {
+                    success: false,
+                    error: `Unsupported language: ${language}`,
+                    errorCode: "UNSUPPORTED_LANGUAGE",
+                };
+        }
+        return this.executeBash({
+            ...task,
+            payload: { ...payload, command, args },
+        }, signal);
+    }
+    async executeCPUTask(task, signal) {
+        return this.executeBash({
+            ...task,
+            type: "bash",
+            payload: task.payload,
+        }, signal);
+    }
+}
+//# sourceMappingURL=cpu-worker.js.map

package/dist/core/companion/scheduler/workers/cpu-worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cpu-worker.js","sourceRoot":"","sources":["../../../../../src/core/companion/scheduler/workers/cpu-worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,eAAe,CAAC;AACtC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,EAAE,UAAU,EAAgE,MAAM,kBAAkB,CAAC;AAE5G,MAAM,OAAO,SAAU,SAAQ,UAAU;IAC/B,EAAE,CAAS;IACX,IAAI,GAAG,KAAK,CAAC;IACb,YAAY,CAAqB;IAE1C,YAAY,EAAW,EAAE;QACxB,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAC/E,IAAI,CAAC,YAAY,GAAG;YACnB,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM;YACvB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;YAChD,MAAM,EAAE,KAAK;YACb,aAAa,EAAE,CAAC,eAAe,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,CAAC;SAC1D,CAAC;IAAA,CACF;IAED,KAAK,CAAC,OAAO,CAAC,IAAmB,EAAE,MAAoB,EAAuB;QAC7E,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACJ,IAAI,MAAkB,CAAC;YAEvB,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACpD,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC/C,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACjC,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAC/C,CAAC;iBAAM,CAAC;gBACP,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAClD,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YAChD,OAAO,MAAM,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;YACvC,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;gBAC7D,SAAS,EAAE,iBAAiB;aAC5B,CAAC;QACH,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjB,CAAC;IAAA,CACD;IAEO,WAAW,CAAC,IAAmB,EAAE,MAAoB,EAAuB;QACnF,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;YAE7B,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE,EAAE;gBAClE,GAAG,EAAE,OAAO,CAAC,UAAU;gBACvB,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE;gBACvC,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;aACjC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,kBAAkB,IAAI,MAAM,CAAC;YAChE,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC;gBAClC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAAA,CACtB,EAAE,OAAO,CAAC,CAAC;YAEZ,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC;gBACrB,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAAA,CACtB,CAAC;YAEF,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAE3C,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC9D,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAE9D,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC3B,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAE9C,OAAO,CAAC;oBACP,OAAO,EAAE,IAAI,KAAK,CAAC;oBACnB,QAAQ,EAAE,IAAI,IAAI,CAAC;oBACnB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;oBACxC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;iBACxC,CAAC,CAAC;YAAA,CACH,CAAC,CAAC;YAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC5B,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAE9C,OAAO,CAAC;oBACP,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;oBACxC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;oBACxC,KAAK,EAAE,KAAK,CAAC,OAAO;oBACpB,SAAS,EAAE,aAAa;iBACxB,CAAC,CAAC;YAAA,CACH,CAAC,CAAC;QAAA,CACH,CAAC,CAAC;IAAA,CACH;IAEO,KAAK,CAAC,WAAW,CAAC,IAAmB,EAAE,MAAoB,EAAuB;QACzF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,EAAE,WAAW,EAAE,IAAI,QAAQ,CAAC;QAE7D,IAAI,OAAe,CAAC;QACpB,IAAI,IAAc,CAAC;QAEnB,QAAQ,QAAQ,EAAE,CAAC;YAClB,KAAK,QAAQ,CAAC;YACd,KAAK,IAAI;gBACR,OAAO,GAAG,SAAS,CAAC;gBACpB,IAAI,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBAClC,MAAM;YACP,KAAK,YAAY,CAAC;YAClB,KAAK,IAAI,CAAC;YACV,KAAK,MAAM;gBACV,OAAO,GAAG,MAAM,CAAC;gBACjB,IAAI,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBAClC,MAAM;YACP,KAAK,YAAY,CAAC;YAClB,KAAK,IAAI;gBACR,OAAO,GAAG,KAAK,CAAC;gBAChB,IAAI,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBACzC,MAAM;YACP;gBACC,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,yBAAyB,QAAQ,EAAE;oBAC1C,SAAS,EAAE,sBAAsB;iBACjC,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,WAAW,CACtB;YACC,GAAG,IAAI;YACP,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE;SACtC,EACD,MAAM,CACN,CAAC;IAAA,CACF;IAEO,KAAK,CAAC,cAAc,CAAC,IAAmB,EAAE,MAAoB,EAAuB;QAC5F,OAAO,IAAI,CAAC,WAAW,CACtB;YACC,GAAG,IAAI;YACP,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,IAAI,CAAC,OAAO;SACrB,EACD,MAAM,CACN,CAAC;IAAA,CACF;CACD","sourcesContent":["import { spawn } from \"child_process\";\nimport { cpus, totalmem } from \"os\";\nimport { BaseWorker, type ScheduledTask, type TaskResult, type WorkerCapabilities } from \"./base-worker.js\";\n\nexport class CPUWorker extends BaseWorker {\n\treadonly id: string;\n\treadonly type = \"cpu\";\n\treadonly capabilities: WorkerCapabilities;\n\n\tconstructor(id?: string) {\n\t\tsuper();\n\t\tthis.id = id || `cpu_${Date.now()}_${Math.random().toString(36).substr(2, 6)}`;\n\t\tthis.capabilities = {\n\t\t\tcpuCores: cpus().length,\n\t\t\tmemoryMb: Math.floor(totalmem() / (1024 * 1024)),\n\t\t\thasGpu: false,\n\t\t\tsupportsTypes: [\"cpu_intensive\", \"code\", \"script\", \"bash\"],\n\t\t};\n\t}\n\n\tasync execute(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tthis.markBusy(task.id);\n\t\tconst startTime = Date.now();\n\n\t\ttry {\n\t\t\tlet result: TaskResult;\n\n\t\t\tif (task.type === \"bash\" || task.type === \"script\") {\n\t\t\t\tresult = await this.executeBash(task, signal);\n\t\t\t} else if (task.type === \"code\") {\n\t\t\t\tresult = await this.executeCode(task, signal);\n\t\t\t} else {\n\t\t\t\tresult = await this.executeCPUTask(task, signal);\n\t\t\t}\n\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, result.success);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, false);\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"EXECUTION_ERROR\",\n\t\t\t};\n\t\t} finally {\n\t\t\tthis.markIdle();\n\t\t}\n\t}\n\n\tprivate executeBash(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\treturn new Promise((resolve) => {\n\t\t\tconst stdout: Buffer[] = [];\n\t\t\tconst stderr: Buffer[] = [];\n\t\t\tconst payload = task.payload;\n\n\t\t\tconst child = spawn(payload.command || \"bash\", payload.args || [], {\n\t\t\t\tcwd: payload.workingDir,\n\t\t\t\tenv: { ...process.env, ...payload.env },\n\t\t\t\tstdio: [\"ignore\", \"pipe\", \"pipe\"],\n\t\t\t});\n\n\t\t\tconst timeout = task.requirements?.maxExecutionTimeMs || 300000;\n\t\t\tconst timeoutId = setTimeout(() => {\n\t\t\t\tchild.kill(\"SIGTERM\");\n\t\t\t}, timeout);\n\n\t\t\tconst onAbort = () => {\n\t\t\t\tchild.kill(\"SIGTERM\");\n\t\t\t};\n\n\t\t\tsignal?.addEventListener(\"abort\", onAbort);\n\n\t\t\tchild.stdout?.on(\"data\", (data: Buffer) => stdout.push(data));\n\t\t\tchild.stderr?.on(\"data\", (data: Buffer) => stderr.push(data));\n\n\t\t\tchild.on(\"close\", (code) => {\n\t\t\t\tclearTimeout(timeoutId);\n\t\t\t\tsignal?.removeEventListener(\"abort\", onAbort);\n\n\t\t\t\tresolve({\n\t\t\t\t\tsuccess: code === 0,\n\t\t\t\t\texitCode: code || 0,\n\t\t\t\t\tstdout: Buffer.concat(stdout).toString(),\n\t\t\t\t\tstderr: Buffer.concat(stderr).toString(),\n\t\t\t\t});\n\t\t\t});\n\n\t\t\tchild.on(\"error\", (error) => {\n\t\t\t\tclearTimeout(timeoutId);\n\t\t\t\tsignal?.removeEventListener(\"abort\", onAbort);\n\n\t\t\t\tresolve({\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\tstdout: Buffer.concat(stdout).toString(),\n\t\t\t\t\tstderr: Buffer.concat(stderr).toString(),\n\t\t\t\t\terror: error.message,\n\t\t\t\t\terrorCode: \"SPAWN_ERROR\",\n\t\t\t\t});\n\t\t\t});\n\t\t});\n\t}\n\n\tprivate async executeCode(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tconst payload = task.payload;\n\t\tconst language = payload.language?.toLowerCase() || \"python\";\n\n\t\tlet command: string;\n\t\tlet args: string[];\n\n\t\tswitch (language) {\n\t\t\tcase \"python\":\n\t\t\tcase \"py\":\n\t\t\t\tcommand = \"python3\";\n\t\t\t\targs = [\"-c\", payload.code || \"\"];\n\t\t\t\tbreak;\n\t\t\tcase \"javascript\":\n\t\t\tcase \"js\":\n\t\t\tcase \"node\":\n\t\t\t\tcommand = \"node\";\n\t\t\t\targs = [\"-e\", payload.code || \"\"];\n\t\t\t\tbreak;\n\t\t\tcase \"typescript\":\n\t\t\tcase \"ts\":\n\t\t\t\tcommand = \"npx\";\n\t\t\t\targs = [\"tsx\", \"-e\", payload.code || \"\"];\n\t\t\t\tbreak;\n\t\t\tdefault:\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: `Unsupported language: ${language}`,\n\t\t\t\t\terrorCode: \"UNSUPPORTED_LANGUAGE\",\n\t\t\t\t};\n\t\t}\n\n\t\treturn this.executeBash(\n\t\t\t{\n\t\t\t\t...task,\n\t\t\t\tpayload: { ...payload, command, args },\n\t\t\t},\n\t\t\tsignal,\n\t\t);\n\t}\n\n\tprivate async executeCPUTask(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\treturn this.executeBash(\n\t\t\t{\n\t\t\t\t...task,\n\t\t\t\ttype: \"bash\",\n\t\t\t\tpayload: task.payload,\n\t\t\t},\n\t\t\tsignal,\n\t\t);\n\t}\n}\n"]}

package/dist/core/companion/scheduler/workers/gpu-worker.d.ts

@@ -0,0 +1,16 @@
+import { BaseWorker, type ScheduledTask, type TaskResult, type WorkerCapabilities } from "./base-worker.js";
+export declare class GPUWorker extends BaseWorker {
+    readonly id: string;
+    readonly type = "gpu";
+    readonly capabilities: WorkerCapabilities;
+    private devices;
+    constructor(id?: string);
+    private detectGPUs;
+    private buildCapabilities;
+    execute(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult>;
+    private executeGPUTask;
+    private selectBestDevice;
+    private executeGPUCode;
+    refreshGPUStatus(): void;
+}
+//# sourceMappingURL=gpu-worker.d.ts.map

package/dist/core/companion/scheduler/workers/gpu-worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpu-worker.d.ts","sourceRoot":"","sources":["../../../../../src/core/companion/scheduler/workers/gpu-worker.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,KAAK,aAAa,EAAE,KAAK,UAAU,EAAE,KAAK,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAU5G,qBAAa,SAAU,SAAQ,UAAU;IACxC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,SAAS;IACtB,QAAQ,CAAC,YAAY,EAAE,kBAAkB,CAAC;IAC1C,OAAO,CAAC,OAAO,CAAmB;IAElC,YAAY,EAAE,CAAC,EAAE,MAAM,EAKtB;IAED,OAAO,CAAC,UAAU;IA8ClB,OAAO,CAAC,iBAAiB;IAcnB,OAAO,CAAC,IAAI,EAAE,aAAa,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,CA4B5E;YAEa,cAAc;IA0B5B,OAAO,CAAC,gBAAgB;YAgBV,cAAc;IA6C5B,gBAAgB,IAAI,IAAI,CAGvB;CACD","sourcesContent":["import { execSync } from \"child_process\";\nimport { cpus, totalmem } from \"os\";\nimport { BaseWorker, type ScheduledTask, type TaskResult, type WorkerCapabilities } from \"./base-worker.js\";\n\ninterface GPUDevice {\n\tindex: number;\n\tname: string;\n\tmemoryTotalMb: number;\n\tmemoryFreeMb: number;\n\tutilization: number;\n}\n\nexport class GPUWorker extends BaseWorker {\n\treadonly id: string;\n\treadonly type = \"gpu\";\n\treadonly capabilities: WorkerCapabilities;\n\tprivate devices: GPUDevice[] = [];\n\n\tconstructor(id?: string) {\n\t\tsuper();\n\t\tthis.id = id || `gpu_${Date.now()}_${Math.random().toString(36).substr(2, 6)}`;\n\t\tthis.devices = this.detectGPUs();\n\t\tthis.capabilities = this.buildCapabilities();\n\t}\n\n\tprivate detectGPUs(): GPUDevice[] {\n\t\tconst devices: GPUDevice[] = [];\n\n\t\tif (process.platform === \"win32\") {\n\t\t\ttry {\n\t\t\t\tconst output = execSync(\n\t\t\t\t\t\"nvidia-smi --query-gpu=index,name,memory.total,memory.free,utilization.gpu --format=csv,noheader,nounits\",\n\t\t\t\t\t{ encoding: \"utf-8\", timeout: 5000 },\n\t\t\t\t);\n\n\t\t\t\tconst lines = output.trim().split(\"\\n\");\n\t\t\t\tfor (const line of lines) {\n\t\t\t\t\tconst [index, name, memoryTotal, memoryFree, utilization] = line.split(\",\").map((s) => s.trim());\n\t\t\t\t\tdevices.push({\n\t\t\t\t\t\tindex: parseInt(index, 10),\n\t\t\t\t\t\tname,\n\t\t\t\t\t\tmemoryTotalMb: parseInt(memoryTotal, 10),\n\t\t\t\t\t\tmemoryFreeMb: parseInt(memoryFree, 10),\n\t\t\t\t\t\tutilization: parseInt(utilization, 10),\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t} catch {}\n\t\t} else {\n\t\t\ttry {\n\t\t\t\tconst output = execSync(\n\t\t\t\t\t\"nvidia-smi --query-gpu=index,name,memory.total,memory.free,utilization.gpu --format=csv,noheader,nounits\",\n\t\t\t\t\t{ encoding: \"utf-8\", timeout: 5000 },\n\t\t\t\t);\n\n\t\t\t\tconst lines = output.trim().split(\"\\n\");\n\t\t\t\tfor (const line of lines) {\n\t\t\t\t\tconst [index, name, memoryTotal, memoryFree, utilization] = line.split(\",\").map((s) => s.trim());\n\t\t\t\t\tdevices.push({\n\t\t\t\t\t\tindex: parseInt(index, 10),\n\t\t\t\t\t\tname,\n\t\t\t\t\t\tmemoryTotalMb: parseInt(memoryTotal, 10),\n\t\t\t\t\t\tmemoryFreeMb: parseInt(memoryFree, 10),\n\t\t\t\t\t\tutilization: parseInt(utilization, 10),\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t} catch {}\n\t\t}\n\n\t\treturn devices;\n\t}\n\n\tprivate buildCapabilities(): WorkerCapabilities {\n\t\tconst hasGpu = this.devices.length > 0;\n\t\tconst totalGpuMemory = this.devices.reduce((sum, d) => sum + d.memoryTotalMb, 0);\n\n\t\treturn {\n\t\t\tcpuCores: cpus().length,\n\t\t\tmemoryMb: Math.floor(totalmem() / (1024 * 1024)),\n\t\t\thasGpu,\n\t\t\tgpuMemoryMb: totalGpuMemory > 0 ? totalGpuMemory : undefined,\n\t\t\tgpuName: this.devices[0]?.name,\n\t\t\tsupportsTypes: hasGpu ? [\"render\", \"cpu_intensive\", \"code\", \"script\"] : [\"cpu_intensive\", \"code\", \"script\"],\n\t\t};\n\t}\n\n\tasync execute(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tthis.markBusy(task.id);\n\t\tconst startTime = Date.now();\n\n\t\ttry {\n\t\t\tif (!this.capabilities.hasGpu) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: \"No GPU available\",\n\t\t\t\t\terrorCode: \"NO_GPU\",\n\t\t\t\t};\n\t\t\t}\n\n\t\t\tconst result = await this.executeGPUTask(task, signal);\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, result.success);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, false);\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"GPU_EXECUTION_ERROR\",\n\t\t\t};\n\t\t} finally {\n\t\t\tthis.markIdle();\n\t\t}\n\t}\n\n\tprivate async executeGPUTask(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tconst device = this.selectBestDevice();\n\t\tif (!device) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: \"No suitable GPU device available\",\n\t\t\t\terrorCode: \"NO_GPU_DEVICE\",\n\t\t\t};\n\t\t}\n\n\t\tconst env: Record<string, string> = {\n\t\t\t...task.payload.env,\n\t\t\tCUDA_VISIBLE_DEVICES: device.index.toString(),\n\t\t};\n\n\t\tif (task.payload.code) {\n\t\t\treturn this.executeGPUCode(task, env, signal);\n\t\t}\n\n\t\treturn {\n\t\t\tsuccess: false,\n\t\t\terror: \"GPU task requires code payload\",\n\t\t\terrorCode: \"INVALID_PAYLOAD\",\n\t\t};\n\t}\n\n\tprivate selectBestDevice(): GPUDevice | null {\n\t\tif (this.devices.length === 0) {\n\t\t\treturn null;\n\t\t}\n\n\t\tthis.devices = this.detectGPUs();\n\n\t\tconst sorted = [...this.devices].sort((a, b) => {\n\t\t\tconst freeDiff = b.memoryFreeMb - a.memoryFreeMb;\n\t\t\tif (freeDiff !== 0) return freeDiff;\n\t\t\treturn a.utilization - b.utilization;\n\t\t});\n\n\t\treturn sorted[0];\n\t}\n\n\tprivate async executeGPUCode(\n\t\ttask: ScheduledTask,\n\t\tenv: Record<string, string>,\n\t\tsignal?: AbortSignal,\n\t): Promise<TaskResult> {\n\t\tconst { spawn } = await import(\"child_process\");\n\n\t\treturn new Promise((resolve) => {\n\t\t\tconst stdout: Buffer[] = [];\n\t\t\tconst stderr: Buffer[] = [];\n\n\t\t\tconst child = spawn(\"python3\", [\"-c\", task.payload.code || \"\"], {\n\t\t\t\tenv: { ...process.env, ...env },\n\t\t\t\tcwd: task.payload.workingDir,\n\t\t\t});\n\n\t\t\tconst timeout = task.requirements?.maxExecutionTimeMs || 600000;\n\t\t\tconst timeoutId = setTimeout(() => child.kill(\"SIGTERM\"), timeout);\n\n\t\t\tsignal?.addEventListener(\"abort\", () => child.kill(\"SIGTERM\"));\n\n\t\t\tchild.stdout?.on(\"data\", (data: Buffer) => stdout.push(data));\n\t\t\tchild.stderr?.on(\"data\", (data: Buffer) => stderr.push(data));\n\n\t\t\tchild.on(\"close\", (code) => {\n\t\t\t\tclearTimeout(timeoutId);\n\t\t\t\tresolve({\n\t\t\t\t\tsuccess: code === 0,\n\t\t\t\t\texitCode: code || 0,\n\t\t\t\t\tstdout: Buffer.concat(stdout).toString(),\n\t\t\t\t\tstderr: Buffer.concat(stderr).toString(),\n\t\t\t\t});\n\t\t\t});\n\n\t\t\tchild.on(\"error\", (error) => {\n\t\t\t\tclearTimeout(timeoutId);\n\t\t\t\tresolve({\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t\terrorCode: \"GPU_CODE_ERROR\",\n\t\t\t\t});\n\t\t\t});\n\t\t});\n\t}\n\n\trefreshGPUStatus(): void {\n\t\tthis.devices = this.detectGPUs();\n\t\tthis.capabilities.hasGpu = this.devices.length > 0;\n\t}\n}\n"]}

package/dist/core/companion/scheduler/workers/gpu-worker.js

@@ -0,0 +1,167 @@
+import { execSync } from "child_process";
+import { cpus, totalmem } from "os";
+import { BaseWorker } from "./base-worker.js";
+export class GPUWorker extends BaseWorker {
+    id;
+    type = "gpu";
+    capabilities;
+    devices = [];
+    constructor(id) {
+        super();
+        this.id = id || `gpu_${Date.now()}_${Math.random().toString(36).substr(2, 6)}`;
+        this.devices = this.detectGPUs();
+        this.capabilities = this.buildCapabilities();
+    }
+    detectGPUs() {
+        const devices = [];
+        if (process.platform === "win32") {
+            try {
+                const output = execSync("nvidia-smi --query-gpu=index,name,memory.total,memory.free,utilization.gpu --format=csv,noheader,nounits", { encoding: "utf-8", timeout: 5000 });
+                const lines = output.trim().split("\n");
+                for (const line of lines) {
+                    const [index, name, memoryTotal, memoryFree, utilization] = line.split(",").map((s) => s.trim());
+                    devices.push({
+                        index: parseInt(index, 10),
+                        name,
+                        memoryTotalMb: parseInt(memoryTotal, 10),
+                        memoryFreeMb: parseInt(memoryFree, 10),
+                        utilization: parseInt(utilization, 10),
+                    });
+                }
+            }
+            catch { }
+        }
+        else {
+            try {
+                const output = execSync("nvidia-smi --query-gpu=index,name,memory.total,memory.free,utilization.gpu --format=csv,noheader,nounits", { encoding: "utf-8", timeout: 5000 });
+                const lines = output.trim().split("\n");
+                for (const line of lines) {
+                    const [index, name, memoryTotal, memoryFree, utilization] = line.split(",").map((s) => s.trim());
+                    devices.push({
+                        index: parseInt(index, 10),
+                        name,
+                        memoryTotalMb: parseInt(memoryTotal, 10),
+                        memoryFreeMb: parseInt(memoryFree, 10),
+                        utilization: parseInt(utilization, 10),
+                    });
+                }
+            }
+            catch { }
+        }
+        return devices;
+    }
+    buildCapabilities() {
+        const hasGpu = this.devices.length > 0;
+        const totalGpuMemory = this.devices.reduce((sum, d) => sum + d.memoryTotalMb, 0);
+        return {
+            cpuCores: cpus().length,
+            memoryMb: Math.floor(totalmem() / (1024 * 1024)),
+            hasGpu,
+            gpuMemoryMb: totalGpuMemory > 0 ? totalGpuMemory : undefined,
+            gpuName: this.devices[0]?.name,
+            supportsTypes: hasGpu ? ["render", "cpu_intensive", "code", "script"] : ["cpu_intensive", "code", "script"],
+        };
+    }
+    async execute(task, signal) {
+        this.markBusy(task.id);
+        const startTime = Date.now();
+        try {
+            if (!this.capabilities.hasGpu) {
+                return {
+                    success: false,
+                    error: "No GPU available",
+                    errorCode: "NO_GPU",
+                };
+            }
+            const result = await this.executeGPUTask(task, signal);
+            const executionTime = Date.now() - startTime;
+            this.updateStats(executionTime, result.success);
+            return result;
+        }
+        catch (error) {
+            const executionTime = Date.now() - startTime;
+            this.updateStats(executionTime, false);
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "GPU_EXECUTION_ERROR",
+            };
+        }
+        finally {
+            this.markIdle();
+        }
+    }
+    async executeGPUTask(task, signal) {
+        const device = this.selectBestDevice();
+        if (!device) {
+            return {
+                success: false,
+                error: "No suitable GPU device available",
+                errorCode: "NO_GPU_DEVICE",
+            };
+        }
+        const env = {
+            ...task.payload.env,
+            CUDA_VISIBLE_DEVICES: device.index.toString(),
+        };
+        if (task.payload.code) {
+            return this.executeGPUCode(task, env, signal);
+        }
+        return {
+            success: false,
+            error: "GPU task requires code payload",
+            errorCode: "INVALID_PAYLOAD",
+        };
+    }
+    selectBestDevice() {
+        if (this.devices.length === 0) {
+            return null;
+        }
+        this.devices = this.detectGPUs();
+        const sorted = [...this.devices].sort((a, b) => {
+            const freeDiff = b.memoryFreeMb - a.memoryFreeMb;
+            if (freeDiff !== 0)
+                return freeDiff;
+            return a.utilization - b.utilization;
+        });
+        return sorted[0];
+    }
+    async executeGPUCode(task, env, signal) {
+        const { spawn } = await import("child_process");
+        return new Promise((resolve) => {
+            const stdout = [];
+            const stderr = [];
+            const child = spawn("python3", ["-c", task.payload.code || ""], {
+                env: { ...process.env, ...env },
+                cwd: task.payload.workingDir,
+            });
+            const timeout = task.requirements?.maxExecutionTimeMs || 600000;
+            const timeoutId = setTimeout(() => child.kill("SIGTERM"), timeout);
+            signal?.addEventListener("abort", () => child.kill("SIGTERM"));
+            child.stdout?.on("data", (data) => stdout.push(data));
+            child.stderr?.on("data", (data) => stderr.push(data));
+            child.on("close", (code) => {
+                clearTimeout(timeoutId);
+                resolve({
+                    success: code === 0,
+                    exitCode: code || 0,
+                    stdout: Buffer.concat(stdout).toString(),
+                    stderr: Buffer.concat(stderr).toString(),
+                });
+            });
+            child.on("error", (error) => {
+                clearTimeout(timeoutId);
+                resolve({
+                    success: false,
+                    error: error.message,
+                    errorCode: "GPU_CODE_ERROR",
+                });
+            });
+        });
+    }
+    refreshGPUStatus() {
+        this.devices = this.detectGPUs();
+        this.capabilities.hasGpu = this.devices.length > 0;
+    }
+}
+//# sourceMappingURL=gpu-worker.js.map

package/dist/core/companion/scheduler/workers/gpu-worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpu-worker.js","sourceRoot":"","sources":["../../../../../src/core/companion/scheduler/workers/gpu-worker.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAC;AACpC,OAAO,EAAE,UAAU,EAAgE,MAAM,kBAAkB,CAAC;AAU5G,MAAM,OAAO,SAAU,SAAQ,UAAU;IAC/B,EAAE,CAAS;IACX,IAAI,GAAG,KAAK,CAAC;IACb,YAAY,CAAqB;IAClC,OAAO,GAAgB,EAAE,CAAC;IAElC,YAAY,EAAW,EAAE;QACxB,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;QAC/E,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAAA,CAC7C;IAEO,UAAU,GAAgB;QACjC,MAAM,OAAO,GAAgB,EAAE,CAAC;QAEhC,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;YAClC,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,QAAQ,CACtB,0GAA0G,EAC1G,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CACpC,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACxC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBAC1B,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBACjG,OAAO,CAAC,IAAI,CAAC;wBACZ,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC;wBAC1B,IAAI;wBACJ,aAAa,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;wBACxC,YAAY,EAAE,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC;wBACtC,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;qBACtC,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACX,CAAC;aAAM,CAAC;YACP,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,QAAQ,CACtB,0GAA0G,EAC1G,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CACpC,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACxC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBAC1B,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,UAAU,EAAE,WAAW,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;oBACjG,OAAO,CAAC,IAAI,CAAC;wBACZ,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,EAAE,CAAC;wBAC1B,IAAI;wBACJ,aAAa,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;wBACxC,YAAY,EAAE,QAAQ,CAAC,UAAU,EAAE,EAAE,CAAC;wBACtC,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,EAAE,CAAC;qBACtC,CAAC,CAAC;gBACJ,CAAC;YACF,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QACX,CAAC;QAED,OAAO,OAAO,CAAC;IAAA,CACf;IAEO,iBAAiB,GAAuB;QAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;QACvC,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;QAEjF,OAAO;YACN,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM;YACvB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;YAChD,MAAM;YACN,WAAW,EAAE,cAAc,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,SAAS;YAC5D,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,IAAI;YAC9B,aAAa,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,eAAe,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,EAAE,MAAM,EAAE,QAAQ,CAAC;SAC3G,CAAC;IAAA,CACF;IAED,KAAK,CAAC,OAAO,CAAC,IAAmB,EAAE,MAAoB,EAAuB;QAC7E,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACJ,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC;gBAC/B,OAAO;oBACN,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,kBAAkB;oBACzB,SAAS,EAAE,QAAQ;iBACnB,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACvD,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YAChD,OAAO,MAAM,CAAC;QACf,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAC7C,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;YACvC,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;gBAC7D,SAAS,EAAE,qBAAqB;aAChC,CAAC;QACH,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,QAAQ,EAAE,CAAC;QACjB,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,cAAc,CAAC,IAAmB,EAAE,MAAoB,EAAuB;QAC5F,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACvC,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,OAAO;gBACN,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,kCAAkC;gBACzC,SAAS,EAAE,eAAe;aAC1B,CAAC;QACH,CAAC;QAED,MAAM,GAAG,GAA2B;YACnC,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG;YACnB,oBAAoB,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE;SAC7C,CAAC;QAEF,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACvB,OAAO,IAAI,CAAC,cAAc,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QAC/C,CAAC;QAED,OAAO;YACN,OAAO,EAAE,KAAK;YACd,KAAK,EAAE,gCAAgC;YACvC,SAAS,EAAE,iBAAiB;SAC5B,CAAC;IAAA,CACF;IAEO,gBAAgB,GAAqB;QAC5C,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO,IAAI,CAAC;QACb,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAEjC,MAAM,MAAM,GAAG,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/C,MAAM,QAAQ,GAAG,CAAC,CAAC,YAAY,GAAG,CAAC,CAAC,YAAY,CAAC;YACjD,IAAI,QAAQ,KAAK,CAAC;gBAAE,OAAO,QAAQ,CAAC;YACpC,OAAO,CAAC,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;QAAA,CACrC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC;IAAA,CACjB;IAEO,KAAK,CAAC,cAAc,CAC3B,IAAmB,EACnB,GAA2B,EAC3B,MAAoB,EACE;QACtB,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,eAAe,CAAC,CAAC;QAEhD,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;YAC/B,MAAM,MAAM,GAAa,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAa,EAAE,CAAC;YAE5B,MAAM,KAAK,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,EAAE;gBAC/D,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,GAAG,GAAG,EAAE;gBAC/B,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,kBAAkB,IAAI,MAAM,CAAC;YAChE,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;YAEnE,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;YAE/D,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC9D,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,IAAY,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAE9D,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC3B,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,OAAO,CAAC;oBACP,OAAO,EAAE,IAAI,KAAK,CAAC;oBACnB,QAAQ,EAAE,IAAI,IAAI,CAAC;oBACnB,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;oBACxC,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE;iBACxC,CAAC,CAAC;YAAA,CACH,CAAC,CAAC;YAEH,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC5B,YAAY,CAAC,SAAS,CAAC,CAAC;gBACxB,OAAO,CAAC;oBACP,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,CAAC,OAAO;oBACpB,SAAS,EAAE,gBAAgB;iBAC3B,CAAC,CAAC;YAAA,CACH,CAAC,CAAC;QAAA,CACH,CAAC,CAAC;IAAA,CACH;IAED,gBAAgB,GAAS;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QACjC,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC;IAAA,CACnD;CACD","sourcesContent":["import { execSync } from \"child_process\";\nimport { cpus, totalmem } from \"os\";\nimport { BaseWorker, type ScheduledTask, type TaskResult, type WorkerCapabilities } from \"./base-worker.js\";\n\ninterface GPUDevice {\n\tindex: number;\n\tname: string;\n\tmemoryTotalMb: number;\n\tmemoryFreeMb: number;\n\tutilization: number;\n}\n\nexport class GPUWorker extends BaseWorker {\n\treadonly id: string;\n\treadonly type = \"gpu\";\n\treadonly capabilities: WorkerCapabilities;\n\tprivate devices: GPUDevice[] = [];\n\n\tconstructor(id?: string) {\n\t\tsuper();\n\t\tthis.id = id || `gpu_${Date.now()}_${Math.random().toString(36).substr(2, 6)}`;\n\t\tthis.devices = this.detectGPUs();\n\t\tthis.capabilities = this.buildCapabilities();\n\t}\n\n\tprivate detectGPUs(): GPUDevice[] {\n\t\tconst devices: GPUDevice[] = [];\n\n\t\tif (process.platform === \"win32\") {\n\t\t\ttry {\n\t\t\t\tconst output = execSync(\n\t\t\t\t\t\"nvidia-smi --query-gpu=index,name,memory.total,memory.free,utilization.gpu --format=csv,noheader,nounits\",\n\t\t\t\t\t{ encoding: \"utf-8\", timeout: 5000 },\n\t\t\t\t);\n\n\t\t\t\tconst lines = output.trim().split(\"\\n\");\n\t\t\t\tfor (const line of lines) {\n\t\t\t\t\tconst [index, name, memoryTotal, memoryFree, utilization] = line.split(\",\").map((s) => s.trim());\n\t\t\t\t\tdevices.push({\n\t\t\t\t\t\tindex: parseInt(index, 10),\n\t\t\t\t\t\tname,\n\t\t\t\t\t\tmemoryTotalMb: parseInt(memoryTotal, 10),\n\t\t\t\t\t\tmemoryFreeMb: parseInt(memoryFree, 10),\n\t\t\t\t\t\tutilization: parseInt(utilization, 10),\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t} catch {}\n\t\t} else {\n\t\t\ttry {\n\t\t\t\tconst output = execSync(\n\t\t\t\t\t\"nvidia-smi --query-gpu=index,name,memory.total,memory.free,utilization.gpu --format=csv,noheader,nounits\",\n\t\t\t\t\t{ encoding: \"utf-8\", timeout: 5000 },\n\t\t\t\t);\n\n\t\t\t\tconst lines = output.trim().split(\"\\n\");\n\t\t\t\tfor (const line of lines) {\n\t\t\t\t\tconst [index, name, memoryTotal, memoryFree, utilization] = line.split(\",\").map((s) => s.trim());\n\t\t\t\t\tdevices.push({\n\t\t\t\t\t\tindex: parseInt(index, 10),\n\t\t\t\t\t\tname,\n\t\t\t\t\t\tmemoryTotalMb: parseInt(memoryTotal, 10),\n\t\t\t\t\t\tmemoryFreeMb: parseInt(memoryFree, 10),\n\t\t\t\t\t\tutilization: parseInt(utilization, 10),\n\t\t\t\t\t});\n\t\t\t\t}\n\t\t\t} catch {}\n\t\t}\n\n\t\treturn devices;\n\t}\n\n\tprivate buildCapabilities(): WorkerCapabilities {\n\t\tconst hasGpu = this.devices.length > 0;\n\t\tconst totalGpuMemory = this.devices.reduce((sum, d) => sum + d.memoryTotalMb, 0);\n\n\t\treturn {\n\t\t\tcpuCores: cpus().length,\n\t\t\tmemoryMb: Math.floor(totalmem() / (1024 * 1024)),\n\t\t\thasGpu,\n\t\t\tgpuMemoryMb: totalGpuMemory > 0 ? totalGpuMemory : undefined,\n\t\t\tgpuName: this.devices[0]?.name,\n\t\t\tsupportsTypes: hasGpu ? [\"render\", \"cpu_intensive\", \"code\", \"script\"] : [\"cpu_intensive\", \"code\", \"script\"],\n\t\t};\n\t}\n\n\tasync execute(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tthis.markBusy(task.id);\n\t\tconst startTime = Date.now();\n\n\t\ttry {\n\t\t\tif (!this.capabilities.hasGpu) {\n\t\t\t\treturn {\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: \"No GPU available\",\n\t\t\t\t\terrorCode: \"NO_GPU\",\n\t\t\t\t};\n\t\t\t}\n\n\t\t\tconst result = await this.executeGPUTask(task, signal);\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, result.success);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, false);\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"GPU_EXECUTION_ERROR\",\n\t\t\t};\n\t\t} finally {\n\t\t\tthis.markIdle();\n\t\t}\n\t}\n\n\tprivate async executeGPUTask(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tconst device = this.selectBestDevice();\n\t\tif (!device) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: \"No suitable GPU device available\",\n\t\t\t\terrorCode: \"NO_GPU_DEVICE\",\n\t\t\t};\n\t\t}\n\n\t\tconst env: Record<string, string> = {\n\t\t\t...task.payload.env,\n\t\t\tCUDA_VISIBLE_DEVICES: device.index.toString(),\n\t\t};\n\n\t\tif (task.payload.code) {\n\t\t\treturn this.executeGPUCode(task, env, signal);\n\t\t}\n\n\t\treturn {\n\t\t\tsuccess: false,\n\t\t\terror: \"GPU task requires code payload\",\n\t\t\terrorCode: \"INVALID_PAYLOAD\",\n\t\t};\n\t}\n\n\tprivate selectBestDevice(): GPUDevice | null {\n\t\tif (this.devices.length === 0) {\n\t\t\treturn null;\n\t\t}\n\n\t\tthis.devices = this.detectGPUs();\n\n\t\tconst sorted = [...this.devices].sort((a, b) => {\n\t\t\tconst freeDiff = b.memoryFreeMb - a.memoryFreeMb;\n\t\t\tif (freeDiff !== 0) return freeDiff;\n\t\t\treturn a.utilization - b.utilization;\n\t\t});\n\n\t\treturn sorted[0];\n\t}\n\n\tprivate async executeGPUCode(\n\t\ttask: ScheduledTask,\n\t\tenv: Record<string, string>,\n\t\tsignal?: AbortSignal,\n\t): Promise<TaskResult> {\n\t\tconst { spawn } = await import(\"child_process\");\n\n\t\treturn new Promise((resolve) => {\n\t\t\tconst stdout: Buffer[] = [];\n\t\t\tconst stderr: Buffer[] = [];\n\n\t\t\tconst child = spawn(\"python3\", [\"-c\", task.payload.code || \"\"], {\n\t\t\t\tenv: { ...process.env, ...env },\n\t\t\t\tcwd: task.payload.workingDir,\n\t\t\t});\n\n\t\t\tconst timeout = task.requirements?.maxExecutionTimeMs || 600000;\n\t\t\tconst timeoutId = setTimeout(() => child.kill(\"SIGTERM\"), timeout);\n\n\t\t\tsignal?.addEventListener(\"abort\", () => child.kill(\"SIGTERM\"));\n\n\t\t\tchild.stdout?.on(\"data\", (data: Buffer) => stdout.push(data));\n\t\t\tchild.stderr?.on(\"data\", (data: Buffer) => stderr.push(data));\n\n\t\t\tchild.on(\"close\", (code) => {\n\t\t\t\tclearTimeout(timeoutId);\n\t\t\t\tresolve({\n\t\t\t\t\tsuccess: code === 0,\n\t\t\t\t\texitCode: code || 0,\n\t\t\t\t\tstdout: Buffer.concat(stdout).toString(),\n\t\t\t\t\tstderr: Buffer.concat(stderr).toString(),\n\t\t\t\t});\n\t\t\t});\n\n\t\t\tchild.on(\"error\", (error) => {\n\t\t\t\tclearTimeout(timeoutId);\n\t\t\t\tresolve({\n\t\t\t\t\tsuccess: false,\n\t\t\t\t\terror: error.message,\n\t\t\t\t\terrorCode: \"GPU_CODE_ERROR\",\n\t\t\t\t});\n\t\t\t});\n\t\t});\n\t}\n\n\trefreshGPUStatus(): void {\n\t\tthis.devices = this.detectGPUs();\n\t\tthis.capabilities.hasGpu = this.devices.length > 0;\n\t}\n}\n"]}

package/dist/core/companion/scheduler/workers/index.d.ts

@@ -0,0 +1,6 @@
+export { BaseWorker, type ScheduledTask, type TaskPayload, type TaskPriority, type TaskRequirements, type TaskResult, type TaskStatus, type TaskType, type WorkerCapabilities, type WorkerInfo, type WorkerStats, } from "./base-worker.js";
+export { CPUWorker } from "./cpu-worker.js";
+export { GPUWorker } from "./gpu-worker.js";
+export { IOWorker } from "./io-worker.js";
+export { NetworkWorker } from "./network-worker.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/core/companion/scheduler/workers/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../../src/core/companion/scheduler/workers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,UAAU,EACV,KAAK,aAAa,EAClB,KAAK,WAAW,EAChB,KAAK,YAAY,EACjB,KAAK,gBAAgB,EACrB,KAAK,UAAU,EACf,KAAK,UAAU,EACf,KAAK,QAAQ,EACb,KAAK,kBAAkB,EACvB,KAAK,UAAU,EACf,KAAK,WAAW,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC","sourcesContent":["export {\n\tBaseWorker,\n\ttype ScheduledTask,\n\ttype TaskPayload,\n\ttype TaskPriority,\n\ttype TaskRequirements,\n\ttype TaskResult,\n\ttype TaskStatus,\n\ttype TaskType,\n\ttype WorkerCapabilities,\n\ttype WorkerInfo,\n\ttype WorkerStats,\n} from \"./base-worker.js\";\nexport { CPUWorker } from \"./cpu-worker.js\";\nexport { GPUWorker } from \"./gpu-worker.js\";\nexport { IOWorker } from \"./io-worker.js\";\nexport { NetworkWorker } from \"./network-worker.js\";\n"]}

package/dist/core/companion/scheduler/workers/index.js

@@ -0,0 +1,6 @@
+export { BaseWorker, } from "./base-worker.js";
+export { CPUWorker } from "./cpu-worker.js";
+export { GPUWorker } from "./gpu-worker.js";
+export { IOWorker } from "./io-worker.js";
+export { NetworkWorker } from "./network-worker.js";
+//# sourceMappingURL=index.js.map

package/dist/core/companion/scheduler/workers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/core/companion/scheduler/workers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,UAAU,GAWV,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC","sourcesContent":["export {\n\tBaseWorker,\n\ttype ScheduledTask,\n\ttype TaskPayload,\n\ttype TaskPriority,\n\ttype TaskRequirements,\n\ttype TaskResult,\n\ttype TaskStatus,\n\ttype TaskType,\n\ttype WorkerCapabilities,\n\ttype WorkerInfo,\n\ttype WorkerStats,\n} from \"./base-worker.js\";\nexport { CPUWorker } from \"./cpu-worker.js\";\nexport { GPUWorker } from \"./gpu-worker.js\";\nexport { IOWorker } from \"./io-worker.js\";\nexport { NetworkWorker } from \"./network-worker.js\";\n"]}

package/dist/core/companion/scheduler/workers/io-worker.d.ts

@@ -0,0 +1,17 @@
+import { BaseWorker, type ScheduledTask, type TaskResult, type WorkerCapabilities } from "./base-worker.js";
+export declare class IOWorker extends BaseWorker {
+    readonly id: string;
+    readonly type = "io";
+    readonly capabilities: WorkerCapabilities;
+    constructor(id?: string);
+    execute(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult>;
+    private executeIOTask;
+    private handleRead;
+    private handleWrite;
+    private handleList;
+    private listDirectory;
+    private handleMkdir;
+    private handleDelete;
+    private handleCopy;
+}
+//# sourceMappingURL=io-worker.d.ts.map

package/dist/core/companion/scheduler/workers/io-worker.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"io-worker.d.ts","sourceRoot":"","sources":["../../../../../src/core/companion/scheduler/workers/io-worker.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,UAAU,EAAE,KAAK,aAAa,EAAE,KAAK,UAAU,EAAE,KAAK,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAE5G,qBAAa,QAAS,SAAQ,UAAU;IACvC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,QAAQ;IACrB,QAAQ,CAAC,YAAY,EAAE,kBAAkB,CAAC;IAE1C,YAAY,EAAE,CAAC,EAAE,MAAM,EAStB;IAEK,OAAO,CAAC,IAAI,EAAE,aAAa,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,CAoB5E;YAEa,aAAa;YA0Bb,UAAU;YAkBV,WAAW;YAqBX,UAAU;YAgBV,aAAa;YA+Bb,WAAW;YAkBX,YAAY;YAoBZ,UAAU;CAqBxB","sourcesContent":["import { mkdir, readdir, readFile, rm, stat, writeFile } from \"fs/promises\";\nimport { cpus, totalmem } from \"os\";\nimport { dirname, join } from \"path\";\nimport { BaseWorker, type ScheduledTask, type TaskResult, type WorkerCapabilities } from \"./base-worker.js\";\n\nexport class IOWorker extends BaseWorker {\n\treadonly id: string;\n\treadonly type = \"io\";\n\treadonly capabilities: WorkerCapabilities;\n\n\tconstructor(id?: string) {\n\t\tsuper();\n\t\tthis.id = id || `io_${Date.now()}_${Math.random().toString(36).substr(2, 6)}`;\n\t\tthis.capabilities = {\n\t\t\tcpuCores: cpus().length,\n\t\t\tmemoryMb: Math.floor(totalmem() / (1024 * 1024)),\n\t\t\thasGpu: false,\n\t\t\tsupportsTypes: [\"io_heavy\", \"script\"],\n\t\t};\n\t}\n\n\tasync execute(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tthis.markBusy(task.id);\n\t\tconst startTime = Date.now();\n\n\t\ttry {\n\t\t\tconst result = await this.executeIOTask(task, signal);\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, result.success);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tconst executionTime = Date.now() - startTime;\n\t\t\tthis.updateStats(executionTime, false);\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"IO_ERROR\",\n\t\t\t};\n\t\t} finally {\n\t\t\tthis.markIdle();\n\t\t}\n\t}\n\n\tprivate async executeIOTask(task: ScheduledTask, signal?: AbortSignal): Promise<TaskResult> {\n\t\tconst payload = task.payload as Record<string, unknown>;\n\t\tconst operation = (payload.data as { operation?: string })?.operation;\n\n\t\tif (signal?.aborted) {\n\t\t\treturn { success: false, error: \"Task cancelled\", errorCode: \"CANCELLED\" };\n\t\t}\n\n\t\tswitch (operation) {\n\t\t\tcase \"read\":\n\t\t\t\treturn this.handleRead(payload);\n\t\t\tcase \"write\":\n\t\t\t\treturn this.handleWrite(payload);\n\t\t\tcase \"list\":\n\t\t\t\treturn this.handleList(payload);\n\t\t\tcase \"mkdir\":\n\t\t\t\treturn this.handleMkdir(payload);\n\t\t\tcase \"delete\":\n\t\t\t\treturn this.handleDelete(payload);\n\t\t\tcase \"copy\":\n\t\t\t\treturn this.handleCopy(payload);\n\t\t\tdefault:\n\t\t\t\treturn { success: false, error: `Unknown IO operation: ${operation}`, errorCode: \"UNKNOWN_OPERATION\" };\n\t\t}\n\t}\n\n\tprivate async handleRead(payload: Record<string, unknown>): Promise<TaskResult> {\n\t\tconst path = payload.path as string;\n\t\tif (!path) {\n\t\t\treturn { success: false, error: \"Path is required\", errorCode: \"MISSING_PATH\" };\n\t\t}\n\n\t\ttry {\n\t\t\tconst content = await readFile(path, \"utf-8\");\n\t\t\treturn { success: true, output: content };\n\t\t} catch (error) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"READ_ERROR\",\n\t\t\t};\n\t\t}\n\t}\n\n\tprivate async handleWrite(payload: Record<string, unknown>): Promise<TaskResult> {\n\t\tconst path = payload.path as string;\n\t\tconst content = payload.content as string;\n\n\t\tif (!path || content === undefined) {\n\t\t\treturn { success: false, error: \"Path and content are required\", errorCode: \"MISSING_PARAMS\" };\n\t\t}\n\n\t\ttry {\n\t\t\tawait mkdir(dirname(path), { recursive: true });\n\t\t\tawait writeFile(path, content, \"utf-8\");\n\t\t\treturn { success: true, output: `Written to ${path}` };\n\t\t} catch (error) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"WRITE_ERROR\",\n\t\t\t};\n\t\t}\n\t}\n\n\tprivate async handleList(payload: Record<string, unknown>): Promise<TaskResult> {\n\t\tconst path = (payload.path as string) || \".\";\n\t\tconst recursive = payload.recursive as boolean;\n\n\t\ttry {\n\t\t\tconst entries = await this.listDirectory(path, recursive);\n\t\t\treturn { success: true, output: JSON.stringify(entries, null, 2) };\n\t\t} catch (error) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"LIST_ERROR\",\n\t\t\t};\n\t\t}\n\t}\n\n\tprivate async listDirectory(dir: string, recursive: boolean): Promise<unknown[]> {\n\t\tconst entries = await readdir(dir, { withFileTypes: true });\n\t\tconst result: unknown[] = [];\n\n\t\tfor (const entry of entries) {\n\t\t\tconst fullPath = join(dir, entry.name);\n\t\t\tconst info: Record<string, unknown> = {\n\t\t\t\tname: entry.name,\n\t\t\t\tpath: fullPath,\n\t\t\t\ttype: entry.isDirectory() ? \"directory\" : \"file\",\n\t\t\t};\n\n\t\t\tif (entry.isFile()) {\n\t\t\t\ttry {\n\t\t\t\t\tconst stats = await stat(fullPath);\n\t\t\t\t\tinfo.size = stats.size;\n\t\t\t\t\tinfo.modified = stats.mtime;\n\t\t\t\t} catch {}\n\t\t\t}\n\n\t\t\tresult.push(info);\n\n\t\t\tif (recursive && entry.isDirectory()) {\n\t\t\t\tconst children = await this.listDirectory(fullPath, true);\n\t\t\t\t(info as Record<string, unknown>).children = children;\n\t\t\t}\n\t\t}\n\n\t\treturn result;\n\t}\n\n\tprivate async handleMkdir(payload: Record<string, unknown>): Promise<TaskResult> {\n\t\tconst path = payload.path as string;\n\t\tif (!path) {\n\t\t\treturn { success: false, error: \"Path is required\", errorCode: \"MISSING_PATH\" };\n\t\t}\n\n\t\ttry {\n\t\t\tawait mkdir(path, { recursive: true });\n\t\t\treturn { success: true, output: `Created directory ${path}` };\n\t\t} catch (error) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"MKDIR_ERROR\",\n\t\t\t};\n\t\t}\n\t}\n\n\tprivate async handleDelete(payload: Record<string, unknown>): Promise<TaskResult> {\n\t\tconst path = payload.path as string;\n\t\tconst recursive = payload.recursive as boolean;\n\n\t\tif (!path) {\n\t\t\treturn { success: false, error: \"Path is required\", errorCode: \"MISSING_PATH\" };\n\t\t}\n\n\t\ttry {\n\t\t\tawait rm(path, { recursive, force: true });\n\t\t\treturn { success: true, output: `Deleted ${path}` };\n\t\t} catch (error) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"DELETE_ERROR\",\n\t\t\t};\n\t\t}\n\t}\n\n\tprivate async handleCopy(payload: Record<string, unknown>): Promise<TaskResult> {\n\t\tconst src = payload.source as string;\n\t\tconst dest = payload.destination as string;\n\n\t\tif (!src || !dest) {\n\t\t\treturn { success: false, error: \"Source and destination are required\", errorCode: \"MISSING_PARAMS\" };\n\t\t}\n\n\t\ttry {\n\t\t\tconst content = await readFile(src);\n\t\t\tawait mkdir(dirname(dest), { recursive: true });\n\t\t\tawait writeFile(dest, content);\n\t\t\treturn { success: true, output: `Copied ${src} to ${dest}` };\n\t\t} catch (error) {\n\t\t\treturn {\n\t\t\t\tsuccess: false,\n\t\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t\t\terrorCode: \"COPY_ERROR\",\n\t\t\t};\n\t\t}\n\t}\n}\n"]}

package/dist/core/companion/scheduler/workers/io-worker.js

@@ -0,0 +1,197 @@
+import { mkdir, readdir, readFile, rm, stat, writeFile } from "fs/promises";
+import { cpus, totalmem } from "os";
+import { dirname, join } from "path";
+import { BaseWorker } from "./base-worker.js";
+export class IOWorker extends BaseWorker {
+    id;
+    type = "io";
+    capabilities;
+    constructor(id) {
+        super();
+        this.id = id || `io_${Date.now()}_${Math.random().toString(36).substr(2, 6)}`;
+        this.capabilities = {
+            cpuCores: cpus().length,
+            memoryMb: Math.floor(totalmem() / (1024 * 1024)),
+            hasGpu: false,
+            supportsTypes: ["io_heavy", "script"],
+        };
+    }
+    async execute(task, signal) {
+        this.markBusy(task.id);
+        const startTime = Date.now();
+        try {
+            const result = await this.executeIOTask(task, signal);
+            const executionTime = Date.now() - startTime;
+            this.updateStats(executionTime, result.success);
+            return result;
+        }
+        catch (error) {
+            const executionTime = Date.now() - startTime;
+            this.updateStats(executionTime, false);
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "IO_ERROR",
+            };
+        }
+        finally {
+            this.markIdle();
+        }
+    }
+    async executeIOTask(task, signal) {
+        const payload = task.payload;
+        const operation = payload.data?.operation;
+        if (signal?.aborted) {
+            return { success: false, error: "Task cancelled", errorCode: "CANCELLED" };
+        }
+        switch (operation) {
+            case "read":
+                return this.handleRead(payload);
+            case "write":
+                return this.handleWrite(payload);
+            case "list":
+                return this.handleList(payload);
+            case "mkdir":
+                return this.handleMkdir(payload);
+            case "delete":
+                return this.handleDelete(payload);
+            case "copy":
+                return this.handleCopy(payload);
+            default:
+                return { success: false, error: `Unknown IO operation: ${operation}`, errorCode: "UNKNOWN_OPERATION" };
+        }
+    }
+    async handleRead(payload) {
+        const path = payload.path;
+        if (!path) {
+            return { success: false, error: "Path is required", errorCode: "MISSING_PATH" };
+        }
+        try {
+            const content = await readFile(path, "utf-8");
+            return { success: true, output: content };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "READ_ERROR",
+            };
+        }
+    }
+    async handleWrite(payload) {
+        const path = payload.path;
+        const content = payload.content;
+        if (!path || content === undefined) {
+            return { success: false, error: "Path and content are required", errorCode: "MISSING_PARAMS" };
+        }
+        try {
+            await mkdir(dirname(path), { recursive: true });
+            await writeFile(path, content, "utf-8");
+            return { success: true, output: `Written to ${path}` };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "WRITE_ERROR",
+            };
+        }
+    }
+    async handleList(payload) {
+        const path = payload.path || ".";
+        const recursive = payload.recursive;
+        try {
+            const entries = await this.listDirectory(path, recursive);
+            return { success: true, output: JSON.stringify(entries, null, 2) };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "LIST_ERROR",
+            };
+        }
+    }
+    async listDirectory(dir, recursive) {
+        const entries = await readdir(dir, { withFileTypes: true });
+        const result = [];
+        for (const entry of entries) {
+            const fullPath = join(dir, entry.name);
+            const info = {
+                name: entry.name,
+                path: fullPath,
+                type: entry.isDirectory() ? "directory" : "file",
+            };
+            if (entry.isFile()) {
+                try {
+                    const stats = await stat(fullPath);
+                    info.size = stats.size;
+                    info.modified = stats.mtime;
+                }
+                catch { }
+            }
+            result.push(info);
+            if (recursive && entry.isDirectory()) {
+                const children = await this.listDirectory(fullPath, true);
+                info.children = children;
+            }
+        }
+        return result;
+    }
+    async handleMkdir(payload) {
+        const path = payload.path;
+        if (!path) {
+            return { success: false, error: "Path is required", errorCode: "MISSING_PATH" };
+        }
+        try {
+            await mkdir(path, { recursive: true });
+            return { success: true, output: `Created directory ${path}` };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "MKDIR_ERROR",
+            };
+        }
+    }
+    async handleDelete(payload) {
+        const path = payload.path;
+        const recursive = payload.recursive;
+        if (!path) {
+            return { success: false, error: "Path is required", errorCode: "MISSING_PATH" };
+        }
+        try {
+            await rm(path, { recursive, force: true });
+            return { success: true, output: `Deleted ${path}` };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "DELETE_ERROR",
+            };
+        }
+    }
+    async handleCopy(payload) {
+        const src = payload.source;
+        const dest = payload.destination;
+        if (!src || !dest) {
+            return { success: false, error: "Source and destination are required", errorCode: "MISSING_PARAMS" };
+        }
+        try {
+            const content = await readFile(src);
+            await mkdir(dirname(dest), { recursive: true });
+            await writeFile(dest, content);
+            return { success: true, output: `Copied ${src} to ${dest}` };
+        }
+        catch (error) {
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+                errorCode: "COPY_ERROR",
+            };
+        }
+    }
+}
+//# sourceMappingURL=io-worker.js.map