@trigger.dev/core 0.0.0-v3-prerelease-20250108141813 → 0.0.0-v4-prerelease-20250916125920

package/dist/commonjs/v3/runEngineWorker/supervisor/schemas.js

@@ -0,0 +1,100 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkerApiRunSnapshotsSinceResponseBody = exports.WorkerApiSuspendCompletionResponseBody = exports.WorkerApiDebugLogBody = exports.WorkerApiDebugLogBodyInput = exports.DebugLogPropertiesInput = exports.DebugLogProperties = exports.DebugLogPropertiesValue = exports.WorkerApiDequeueFromVersionResponseBody = exports.WorkerApiRunLatestSnapshotResponseBody = exports.WorkerApiRunAttemptCompleteResponseBody = exports.WorkerApiRunAttemptCompleteRequestBody = exports.WorkerApiRunAttemptStartResponseBody = exports.WorkerApiRunAttemptStartRequestBody = exports.WorkerApiRunHeartbeatResponseBody = exports.WorkerApiRunHeartbeatRequestBody = exports.WorkerApiDequeueResponseBody = exports.WorkerApiDequeueRequestBody = exports.WorkerApiConnectResponseBody = exports.WorkerApiConnectRequestBody = exports.WorkerApiContinueRunExecutionRequestBody = exports.WorkerApiSuspendRunResponseBody = exports.WorkerApiSuspendRunRequestBody = exports.WorkerApiHeartbeatResponseBody = exports.WorkerApiHeartbeatRequestBody = void 0;
+const zod_1 = require("zod");
+const common_js_1 = require("../../schemas/common.js");
+const runEngine_js_1 = require("../../schemas/runEngine.js");
+exports.WorkerApiHeartbeatRequestBody = zod_1.z.object({
+    cpu: zod_1.z.object({
+        used: zod_1.z.number(),
+        available: zod_1.z.number(),
+    }),
+    memory: zod_1.z.object({
+        used: zod_1.z.number(),
+        available: zod_1.z.number(),
+    }),
+    tasks: zod_1.z.array(zod_1.z.string()),
+});
+exports.WorkerApiHeartbeatResponseBody = zod_1.z.object({
+    ok: zod_1.z.literal(true),
+});
+exports.WorkerApiSuspendRunRequestBody = zod_1.z.discriminatedUnion("success", [
+    zod_1.z.object({
+        success: zod_1.z.literal(true),
+        checkpoint: runEngine_js_1.CheckpointInput,
+    }),
+    zod_1.z.object({
+        success: zod_1.z.literal(false),
+        error: zod_1.z.string(),
+    }),
+]);
+exports.WorkerApiSuspendRunResponseBody = zod_1.z.object({
+    ok: zod_1.z.literal(true),
+});
+exports.WorkerApiContinueRunExecutionRequestBody = runEngine_js_1.ExecutionResult;
+exports.WorkerApiConnectRequestBody = zod_1.z.object({
+    metadata: zod_1.z.record(zod_1.z.any()),
+});
+exports.WorkerApiConnectResponseBody = zod_1.z.object({
+    ok: zod_1.z.literal(true),
+    workerGroup: zod_1.z.object({
+        type: zod_1.z.string(),
+        name: zod_1.z.string(),
+    }),
+});
+exports.WorkerApiDequeueRequestBody = zod_1.z.object({
+    maxResources: runEngine_js_1.MachineResources.optional(),
+    maxRunCount: zod_1.z.number().optional(),
+});
+exports.WorkerApiDequeueResponseBody = runEngine_js_1.DequeuedMessage.array();
+exports.WorkerApiRunHeartbeatRequestBody = zod_1.z.object({
+    cpu: zod_1.z.number().optional(),
+    memory: zod_1.z.number().optional(),
+});
+exports.WorkerApiRunHeartbeatResponseBody = zod_1.z.object({
+    ok: zod_1.z.literal(true),
+});
+exports.WorkerApiRunAttemptStartRequestBody = zod_1.z.object({
+    isWarmStart: zod_1.z.boolean().optional(),
+});
+exports.WorkerApiRunAttemptStartResponseBody = runEngine_js_1.StartRunAttemptResult.and(zod_1.z.object({
+    envVars: zod_1.z.record(zod_1.z.string()),
+}));
+exports.WorkerApiRunAttemptCompleteRequestBody = zod_1.z.object({
+    completion: common_js_1.TaskRunExecutionResult,
+});
+exports.WorkerApiRunAttemptCompleteResponseBody = zod_1.z.object({
+    result: runEngine_js_1.CompleteRunAttemptResult,
+});
+exports.WorkerApiRunLatestSnapshotResponseBody = zod_1.z.object({
+    execution: runEngine_js_1.RunExecutionData,
+});
+exports.WorkerApiDequeueFromVersionResponseBody = runEngine_js_1.DequeuedMessage.array();
+exports.DebugLogPropertiesValue = zod_1.z.union([
+    zod_1.z.string(),
+    zod_1.z.number(),
+    zod_1.z.boolean(),
+    zod_1.z.array(zod_1.z.string().nullish()),
+    zod_1.z.array(zod_1.z.number().nullish()),
+    zod_1.z.array(zod_1.z.boolean().nullish()),
+]);
+exports.DebugLogProperties = zod_1.z.record(zod_1.z.string(), exports.DebugLogPropertiesValue.optional());
+exports.DebugLogPropertiesInput = zod_1.z.record(zod_1.z.string(), zod_1.z.unknown());
+exports.WorkerApiDebugLogBodyInput = zod_1.z.object({
+    time: zod_1.z.coerce.date(),
+    message: zod_1.z.string(),
+    properties: exports.DebugLogPropertiesInput.optional(),
+});
+exports.WorkerApiDebugLogBody = zod_1.z.object({
+    time: zod_1.z.coerce.date(),
+    message: zod_1.z.string(),
+    properties: exports.DebugLogProperties.optional(),
+});
+exports.WorkerApiSuspendCompletionResponseBody = zod_1.z.object({
+    success: zod_1.z.boolean(),
+    error: zod_1.z.string().optional(),
+});
+exports.WorkerApiRunSnapshotsSinceResponseBody = zod_1.z.object({
+    snapshots: zod_1.z.array(runEngine_js_1.RunExecutionData),
+});
+//# sourceMappingURL=schemas.js.map

package/dist/commonjs/v3/runEngineWorker/supervisor/schemas.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schemas.js","sourceRoot":"","sources":["../../../../../src/v3/runEngineWorker/supervisor/schemas.ts"],"names":[],"mappings":";;;AAAA,6BAAwB;AACxB,uDAAiE;AACjE,6DAQoC;AAEvB,QAAA,6BAA6B,GAAG,OAAC,CAAC,MAAM,CAAC;IACpD,GAAG,EAAE,OAAC,CAAC,MAAM,CAAC;QACZ,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;QAChB,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC;QACf,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;QAChB,SAAS,EAAE,OAAC,CAAC,MAAM,EAAE;KACtB,CAAC;IACF,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;CAC3B,CAAC,CAAC;AAGU,QAAA,8BAA8B,GAAG,OAAC,CAAC,MAAM,CAAC;IACrD,EAAE,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGU,QAAA,8BAA8B,GAAG,OAAC,CAAC,kBAAkB,CAAC,SAAS,EAAE;IAC5E,OAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC;QACxB,UAAU,EAAE,8BAAe;KAC5B,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,OAAC,CAAC,OAAO,CAAC,KAAK,CAAC;QACzB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE;KAClB,CAAC;CACH,CAAC,CAAC;AAGU,QAAA,+BAA+B,GAAG,OAAC,CAAC,MAAM,CAAC;IACtD,EAAE,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGU,QAAA,wCAAwC,GAAG,8BAAe,CAAC;AAK3D,QAAA,2BAA2B,GAAG,OAAC,CAAC,MAAM,CAAC;IAClD,QAAQ,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;CAC5B,CAAC,CAAC;AAGU,QAAA,4BAA4B,GAAG,OAAC,CAAC,MAAM,CAAC;IACnD,EAAE,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IACnB,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC;QACpB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;QAChB,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;KACjB,CAAC;CACH,CAAC,CAAC;AAGU,QAAA,2BAA2B,GAAG,OAAC,CAAC,MAAM,CAAC;IAClD,YAAY,EAAE,+BAAgB,CAAC,QAAQ,EAAE;IACzC,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACnC,CAAC,CAAC;AAGU,QAAA,4BAA4B,GAAG,8BAAe,CAAC,KAAK,EAAE,CAAC;AAGvD,QAAA,gCAAgC,GAAG,OAAC,CAAC,MAAM,CAAC;IACvD,GAAG,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC1B,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC9B,CAAC,CAAC;AAGU,QAAA,iCAAiC,GAAG,OAAC,CAAC,MAAM,CAAC;IACxD,EAAE,EAAE,OAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGU,QAAA,mCAAmC,GAAG,OAAC,CAAC,MAAM,CAAC;IAC1D,WAAW,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;CACpC,CAAC,CAAC;AAKU,QAAA,oCAAoC,GAAG,oCAAqB,CAAC,GAAG,CAC3E,OAAC,CAAC,MAAM,CAAC;IACP,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;CAC9B,CAAC,CACH,CAAC;AAKW,QAAA,sCAAsC,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7D,UAAU,EAAE,kCAAsB;CACnC,CAAC,CAAC;AAKU,QAAA,uCAAuC,GAAG,OAAC,CAAC,MAAM,CAAC;IAC9D,MAAM,EAAE,uCAAwB;CACjC,CAAC,CAAC;AAKU,QAAA,sCAAsC,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7D,SAAS,EAAE,+BAAgB;CAC5B,CAAC,CAAC;AAKU,QAAA,uCAAuC,GAAG,8BAAe,CAAC,KAAK,EAAE,CAAC;AAKlE,QAAA,uBAAuB,GAAG,OAAC,CAAC,KAAK,CAAC;IAC7C,OAAC,CAAC,MAAM,EAAE;IACV,OAAC,CAAC,MAAM,EAAE;IACV,OAAC,CAAC,OAAO,EAAE;IACX,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC;CAC/B,CAAC,CAAC;AAEU,QAAA,kBAAkB,GAAG,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,+BAAuB,CAAC,QAAQ,EAAE,CAAC,CAAC;AAG9E,QAAA,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC;AAG5D,QAAA,0BAA0B,GAAG,OAAC,CAAC,MAAM,CAAC;IACjD,IAAI,EAAE,OAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IACrB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;IACnB,UAAU,EAAE,+BAAuB,CAAC,QAAQ,EAAE;CAC/C,CAAC,CAAC;AAGU,QAAA,qBAAqB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC5C,IAAI,EAAE,OAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IACrB,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;IACnB,UAAU,EAAE,0BAAkB,CAAC,QAAQ,EAAE;CAC1C,CAAC,CAAC;AAGU,QAAA,sCAAsC,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7D,OAAO,EAAE,OAAC,CAAC,OAAO,EAAE;IACpB,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC7B,CAAC,CAAC;AAKU,QAAA,sCAAsC,GAAG,OAAC,CAAC,MAAM,CAAC;IAC7D,SAAS,EAAE,OAAC,CAAC,KAAK,CAAC,+BAAgB,CAAC;CACrC,CAAC,CAAC"}

package/dist/commonjs/v3/runEngineWorker/supervisor/session.d.ts

@@ -0,0 +1,38 @@
+import { SupervisorHttpClient } from "./http.js";
+import { PreDequeueFn, PreSkipFn, SupervisorClientCommonOptions } from "./types.js";
+import { ScalingOptions } from "./consumerPool.js";
+import { WorkerEvents } from "./events.js";
+import EventEmitter from "events";
+import type { Registry } from "prom-client";
+type SupervisorSessionOptions = SupervisorClientCommonOptions & {
+    queueConsumerEnabled?: boolean;
+    runNotificationsEnabled?: boolean;
+    heartbeatIntervalSeconds: number;
+    dequeueIntervalMs: number;
+    dequeueIdleIntervalMs: number;
+    preDequeue?: PreDequeueFn;
+    preSkip?: PreSkipFn;
+    maxRunCount?: number;
+    sendRunDebugLogs?: boolean;
+    scaling: ScalingOptions;
+    metricsRegistry?: Registry;
+};
+export declare class SupervisorSession extends EventEmitter<WorkerEvents> {
+    private opts;
+    readonly httpClient: SupervisorHttpClient;
+    private readonly logger;
+    private readonly runNotificationsEnabled;
+    private runNotificationsSocket?;
+    private readonly queueConsumerEnabled;
+    private readonly consumerPool;
+    private readonly heartbeat;
+    constructor(opts: SupervisorSessionOptions);
+    private onDequeue;
+    subscribeToRunNotifications(runFriendlyIds: string[]): void;
+    unsubscribeFromRunNotifications(runFriendlyIds: string[]): void;
+    private createRunNotificationsSocket;
+    start(): Promise<void>;
+    stop(): Promise<void>;
+    private getHeartbeatBody;
+}
+export {};

package/dist/commonjs/v3/runEngineWorker/supervisor/session.js

@@ -0,0 +1,176 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SupervisorSession = void 0;
+const http_js_1 = require("./http.js");
+const consumerPool_js_1 = require("./consumerPool.js");
+const events_1 = __importDefault(require("events"));
+const version_js_1 = require("../../../version.js");
+const socket_io_client_1 = require("socket.io-client");
+const util_js_1 = require("./util.js");
+const interval_js_1 = require("../../utils/interval.js");
+const structuredLogger_js_1 = require("../../utils/structuredLogger.js");
+class SupervisorSession extends events_1.default {
+    opts;
+    httpClient;
+    logger = new structuredLogger_js_1.SimpleStructuredLogger("supervisor-session");
+    runNotificationsEnabled;
+    runNotificationsSocket;
+    queueConsumerEnabled;
+    consumerPool;
+    heartbeat;
+    constructor(opts) {
+        super();
+        this.opts = opts;
+        this.runNotificationsEnabled = opts.runNotificationsEnabled ?? true;
+        this.queueConsumerEnabled = opts.queueConsumerEnabled ?? true;
+        this.httpClient = new http_js_1.SupervisorHttpClient(opts);
+        this.consumerPool = new consumerPool_js_1.RunQueueConsumerPool({
+            consumer: {
+                client: this.httpClient,
+                preDequeue: opts.preDequeue,
+                preSkip: opts.preSkip,
+                onDequeue: this.onDequeue.bind(this),
+                intervalMs: opts.dequeueIntervalMs,
+                idleIntervalMs: opts.dequeueIdleIntervalMs,
+                maxRunCount: opts.maxRunCount,
+            },
+            scaling: opts.scaling,
+            metricsRegistry: opts.metricsRegistry,
+        });
+        this.heartbeat = new interval_js_1.IntervalService({
+            onInterval: async () => {
+                this.logger.debug("Sending heartbeat");
+                const body = this.getHeartbeatBody();
+                const response = await this.httpClient.heartbeatWorker(body);
+                if (!response.success) {
+                    this.logger.error("Heartbeat failed", { error: response.error });
+                }
+            },
+            intervalMs: opts.heartbeatIntervalSeconds * 1000,
+            leadingEdge: false,
+            onError: async (error) => {
+                this.logger.error("Failed to send heartbeat", { error });
+            },
+        });
+    }
+    async onDequeue(messages) {
+        this.logger.verbose("Dequeued messages with contents", { count: messages.length, messages });
+        for (const message of messages) {
+            this.emit("runQueueMessage", {
+                time: new Date(),
+                message,
+            });
+        }
+    }
+    subscribeToRunNotifications(runFriendlyIds) {
+        this.logger.debug("Subscribing to run notifications", { runFriendlyIds });
+        if (!this.runNotificationsSocket) {
+            this.logger.error("Socket not connected");
+            return;
+        }
+        this.runNotificationsSocket.emit("run:subscribe", { version: "1", runFriendlyIds });
+        Promise.allSettled(runFriendlyIds.map((runFriendlyId) => this.httpClient.sendDebugLog(runFriendlyId, {
+            time: new Date(),
+            message: "run:subscribe supervisor -> platform",
+        })));
+    }
+    unsubscribeFromRunNotifications(runFriendlyIds) {
+        this.logger.debug("Unsubscribing from run notifications", { runFriendlyIds });
+        if (!this.runNotificationsSocket) {
+            this.logger.error("Socket not connected");
+            return;
+        }
+        this.runNotificationsSocket.emit("run:unsubscribe", { version: "1", runFriendlyIds });
+        Promise.allSettled(runFriendlyIds.map((runFriendlyId) => this.httpClient.sendDebugLog(runFriendlyId, {
+            time: new Date(),
+            message: "run:unsubscribe supervisor -> platform",
+            properties: {
+                runFriendlyIds,
+            },
+        })));
+    }
+    createRunNotificationsSocket() {
+        const wsUrl = new URL(this.opts.apiUrl);
+        wsUrl.pathname = "/worker";
+        const socket = (0, socket_io_client_1.io)(wsUrl.href, {
+            transports: ["websocket"],
+            extraHeaders: (0, util_js_1.getDefaultWorkerHeaders)(this.opts),
+        });
+        socket.on("run:notify", ({ version, run }) => {
+            this.logger.debug("[WS] Received run notification", { version, run });
+            this.emit("runNotification", { time: new Date(), run });
+            this.httpClient.sendDebugLog(run.friendlyId, {
+                time: new Date(),
+                message: "run:notify received by supervisor",
+            });
+        });
+        socket.on("connect", () => {
+            this.logger.log("[WS] Connected to platform");
+        });
+        socket.on("connect_error", (error) => {
+            this.logger.error("[WS] Connection error", { error });
+        });
+        socket.on("disconnect", (reason, description) => {
+            this.logger.log("[WS] Disconnected from platform", { reason, description });
+        });
+        return socket;
+    }
+    async start() {
+        const connect = await this.httpClient.connect({
+            metadata: {
+                workerVersion: version_js_1.VERSION,
+            },
+        });
+        if (!connect.success) {
+            this.logger.error("Failed to connect", { error: connect.error });
+            throw new Error("[SupervisorSession] Failed to connect");
+        }
+        const { workerGroup } = connect.data;
+        this.logger.log("Connected to platform", {
+            type: workerGroup.type,
+            name: workerGroup.name,
+        });
+        if (this.queueConsumerEnabled) {
+            this.logger.log("Queue consumer enabled", {
+                scalingStrategy: this.consumerPool["scalingStrategy"],
+                minConsumers: this.consumerPool["minConsumerCount"],
+                maxConsumers: this.consumerPool["maxConsumerCount"],
+            });
+            await this.consumerPool.start();
+            this.heartbeat.start();
+        }
+        else {
+            this.logger.warn("Queue consumer disabled");
+        }
+        if (this.runNotificationsEnabled) {
+            this.logger.log("Run notifications enabled");
+            this.runNotificationsSocket = this.createRunNotificationsSocket();
+        }
+        else {
+            this.logger.warn("Run notifications disabled");
+        }
+    }
+    async stop() {
+        await this.consumerPool.stop();
+        this.heartbeat.stop();
+        this.runNotificationsSocket?.disconnect();
+    }
+    getHeartbeatBody() {
+        return {
+            cpu: {
+                used: 0.5,
+                available: 0.5,
+            },
+            memory: {
+                used: 0.5,
+                available: 0.5,
+            },
+            tasks: [],
+        };
+    }
+}
+exports.SupervisorSession = SupervisorSession;
+//# sourceMappingURL=session.js.map

package/dist/commonjs/v3/runEngineWorker/supervisor/session.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"session.js","sourceRoot":"","sources":["../../../../../src/v3/runEngineWorker/supervisor/session.ts"],"names":[],"mappings":";;;;;;AAAA,uCAAiD;AAGjD,uDAAyE;AAEzE,oDAAkC;AAClC,oDAA8C;AAC9C,uDAA8C;AAE9C,uCAAoD;AACpD,yDAA0D;AAC1D,yEAAyE;AAiBzE,MAAa,iBAAkB,SAAQ,gBAA0B;IAa3C;IAZJ,UAAU,CAAuB;IAEhC,MAAM,GAAG,IAAI,4CAAsB,CAAC,oBAAoB,CAAC,CAAC;IAE1D,uBAAuB,CAAU;IAC1C,sBAAsB,CAAkE;IAE/E,oBAAoB,CAAU;IAC9B,YAAY,CAAuB;IAEnC,SAAS,CAAkB;IAE5C,YAAoB,IAA8B;QAChD,KAAK,EAAE,CAAC;QADU,SAAI,GAAJ,IAAI,CAA0B;QAGhD,IAAI,CAAC,uBAAuB,GAAG,IAAI,CAAC,uBAAuB,IAAI,IAAI,CAAC;QACpE,IAAI,CAAC,oBAAoB,GAAG,IAAI,CAAC,oBAAoB,IAAI,IAAI,CAAC;QAE9D,IAAI,CAAC,UAAU,GAAG,IAAI,8BAAoB,CAAC,IAAI,CAAC,CAAC;QAEjD,IAAI,CAAC,YAAY,GAAG,IAAI,sCAAoB,CAAC;YAC3C,QAAQ,EAAE;gBACR,MAAM,EAAE,IAAI,CAAC,UAAU;gBACvB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;gBACpC,UAAU,EAAE,IAAI,CAAC,iBAAiB;gBAClC,cAAc,EAAE,IAAI,CAAC,qBAAqB;gBAC1C,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B;YACD,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,eAAe,EAAE,IAAI,CAAC,eAAe;SACtC,CAAC,CAAC;QAEH,IAAI,CAAC,SAAS,GAAG,IAAI,6BAAe,CAAC;YACnC,UAAU,EAAE,KAAK,IAAI,EAAE;gBACrB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;gBAEvC,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACrC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;gBAE7D,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;oBACtB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;gBACnE,CAAC;YACH,CAAC;YACD,UAAU,EAAE,IAAI,CAAC,wBAAwB,GAAG,IAAI;YAChD,WAAW,EAAE,KAAK;YAClB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACvB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC3D,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,QAAsC;QAC5D,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,iCAAiC,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;QAE7F,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE;gBAC3B,IAAI,EAAE,IAAI,IAAI,EAAE;gBAChB,OAAO;aACR,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,2BAA2B,CAAC,cAAwB;QAClD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAE1E,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,eAAe,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC,CAAC;QAEpF,OAAO,CAAC,UAAU,CAChB,cAAc,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CACnC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,EAAE;YAC1C,IAAI,EAAE,IAAI,IAAI,EAAE;YAChB,OAAO,EAAE,sCAAsC;SAChD,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAED,+BAA+B,CAAC,cAAwB;QACtD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sCAAsC,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAE9E,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;YAC1C,OAAO;QACT,CAAC;QAED,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,cAAc,EAAE,CAAC,CAAC;QAEtF,OAAO,CAAC,UAAU,CAChB,cAAc,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CACnC,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,aAAa,EAAE;YAC1C,IAAI,EAAE,IAAI,IAAI,EAAE;YAChB,OAAO,EAAE,wCAAwC;YACjD,UAAU,EAAE;gBACV,cAAc;aACf;SACF,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,4BAA4B;QAClC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACxC,KAAK,CAAC,QAAQ,GAAG,SAAS,CAAC;QAE3B,MAAM,MAAM,GAAG,IAAA,qBAAE,EAAC,KAAK,CAAC,IAAI,EAAE;YAC5B,UAAU,EAAE,CAAC,WAAW,CAAC;YACzB,YAAY,EAAE,IAAA,iCAAuB,EAAC,IAAI,CAAC,IAAI,CAAC;SACjD,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE;YAC3C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;YACtE,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;YAExD,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE;gBAC3C,IAAI,EAAE,IAAI,IAAI,EAAE;gBAChB,OAAO,EAAE,mCAAmC;aAC7C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;YACxB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,KAAK,EAAE,EAAE;YACnC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,WAAW,EAAE,EAAE;YAC9C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,iCAAiC,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;QAC9E,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,KAAK;QACT,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC;YAC5C,QAAQ,EAAE;gBACR,aAAa,EAAE,oBAAO;aACvB;SACF,CAAC,CAAC;QAEH,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACrB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,mBAAmB,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;YACjE,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;QAErC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,uBAAuB,EAAE;YACvC,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,IAAI,EAAE,WAAW,CAAC,IAAI;SACvB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,wBAAwB,EAAE;gBACxC,eAAe,EAAE,IAAI,CAAC,YAAY,CAAC,iBAAiB,CAAC;gBACrD,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC;gBACnD,YAAY,EAAE,IAAI,CAAC,YAAY,CAAC,kBAAkB,CAAC;aACpD,CAAC,CAAC;YAEH,MAAM,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAChC,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACzB,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAC7C,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,4BAA4B,EAAE,CAAC;QACpE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;QACjD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC;QAC/B,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QACtB,IAAI,CAAC,sBAAsB,EAAE,UAAU,EAAE,CAAC;IAC5C,CAAC;IAEO,gBAAgB;QACtB,OAAO;YACL,GAAG,EAAE;gBACH,IAAI,EAAE,GAAG;gBACT,SAAS,EAAE,GAAG;aACf;YACD,MAAM,EAAE;gBACN,IAAI,EAAE,GAAG;gBACT,SAAS,EAAE,GAAG;aACf;YACD,KAAK,EAAE,EAAE;SACV,CAAC;IACJ,CAAC;CACF;AArMD,8CAqMC"}

package/dist/commonjs/v3/runEngineWorker/supervisor/types.d.ts

@@ -0,0 +1,14 @@
+import { MachineResources } from "../../schemas/runEngine.js";
+export type SupervisorClientCommonOptions = {
+    apiUrl: string;
+    workerToken: string;
+    instanceName: string;
+    deploymentId?: string;
+    managedWorkerSecret?: string;
+    sendRunDebugLogs?: boolean;
+};
+export type PreDequeueFn = () => Promise<{
+    maxResources?: MachineResources;
+    skipDequeue?: boolean;
+}>;
+export type PreSkipFn = () => Promise<void>;

package/dist/commonjs/v3/runEngineWorker/supervisor/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/commonjs/v3/runEngineWorker/supervisor/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/v3/runEngineWorker/supervisor/types.ts"],"names":[],"mappings":""}

package/dist/commonjs/v3/runEngineWorker/supervisor/util.d.ts

@@ -0,0 +1,3 @@
+import { SupervisorClientCommonOptions } from "./types.js";
+export declare function getDefaultWorkerHeaders(options: SupervisorClientCommonOptions): Record<string, string>;
+export declare function redactKeys<T extends Record<string, any>>(obj: T, keys: Array<keyof T>): T;

package/dist/commonjs/v3/runEngineWorker/supervisor/util.js

@@ -0,0 +1,39 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getDefaultWorkerHeaders = getDefaultWorkerHeaders;
+exports.redactKeys = redactKeys;
+const consts_js_1 = require("../consts.js");
+const util_js_1 = require("../util.js");
+function getDefaultWorkerHeaders(options) {
+    return (0, util_js_1.createHeaders)({
+        Authorization: `Bearer ${options.workerToken}`,
+        [consts_js_1.WORKER_HEADERS.INSTANCE_NAME]: options.instanceName,
+        [consts_js_1.WORKER_HEADERS.DEPLOYMENT_ID]: options.deploymentId,
+        [consts_js_1.WORKER_HEADERS.MANAGED_SECRET]: options.managedWorkerSecret,
+    });
+}
+function redactString(value, end = 10) {
+    return value.slice(0, end) + "*".repeat(value.length - end);
+}
+function redactNumber(value, end = 10) {
+    const str = String(value);
+    const redacted = redactString(str, end);
+    return Number(redacted);
+}
+function redactKeys(obj, keys) {
+    const redacted = { ...obj };
+    for (const key of keys) {
+        const value = obj[key];
+        if (typeof value === "number") {
+            redacted[key] = redactNumber(value);
+        }
+        else if (typeof value === "string") {
+            redacted[key] = redactString(value);
+        }
+        else {
+            continue;
+        }
+    }
+    return redacted;
+}
+//# sourceMappingURL=util.js.map

package/dist/commonjs/v3/runEngineWorker/supervisor/util.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"util.js","sourceRoot":"","sources":["../../../../../src/v3/runEngineWorker/supervisor/util.ts"],"names":[],"mappings":";;AAIA,0DASC;AAYD,gCAcC;AAvCD,4CAA8C;AAC9C,wCAA2C;AAG3C,SAAgB,uBAAuB,CACrC,OAAsC;IAEtC,OAAO,IAAA,uBAAa,EAAC;QACnB,aAAa,EAAE,UAAU,OAAO,CAAC,WAAW,EAAE;QAC9C,CAAC,0BAAc,CAAC,aAAa,CAAC,EAAE,OAAO,CAAC,YAAY;QACpD,CAAC,0BAAc,CAAC,aAAa,CAAC,EAAE,OAAO,CAAC,YAAY;QACpD,CAAC,0BAAc,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC,mBAAmB;KAC7D,CAAC,CAAC;AACL,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,GAAG,GAAG,EAAE;IAC3C,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,GAAG,CAAC,CAAC;AAC9D,CAAC;AAED,SAAS,YAAY,CAAC,KAAa,EAAE,GAAG,GAAG,EAAE;IAC3C,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC1B,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IACxC,OAAO,MAAM,CAAC,QAAQ,CAAC,CAAC;AAC1B,CAAC;AAED,SAAgB,UAAU,CAAgC,GAAM,EAAE,IAAoB;IACpF,MAAM,QAAQ,GAAG,EAAE,GAAG,GAAG,EAAE,CAAC;IAC5B,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC;QAEvB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,QAAQ,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC,KAAK,CAAQ,CAAC;QAC7C,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACrC,QAAQ,CAAC,GAAG,CAAC,GAAG,YAAY,CAAC,KAAK,CAAQ,CAAC;QAC7C,CAAC;aAAM,CAAC;YACN,SAAS;QACX,CAAC;IACH,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC"}

package/dist/commonjs/v3/runEngineWorker/types.d.ts

@@ -0,0 +1,52 @@
+export interface WorkerServerToClientEvents {
+    "run:notify": (message: {
+        version: "1";
+        run: {
+            friendlyId: string;
+        };
+    }) => void;
+}
+export interface WorkerClientToServerEvents {
+    "run:subscribe": (message: {
+        version: "1";
+        runFriendlyIds: string[];
+    }) => void;
+    "run:unsubscribe": (message: {
+        version: "1";
+        runFriendlyIds: string[];
+    }) => void;
+}
+export interface WorkloadServerToClientEvents {
+    "run:notify": (message: {
+        version: "1";
+        run: {
+            friendlyId: string;
+        };
+    }) => void;
+}
+export interface WorkloadClientToServerEvents {
+    "run:start": (message: {
+        version: "1";
+        run: {
+            friendlyId: string;
+        };
+        snapshot: {
+            friendlyId: string;
+        };
+    }) => void;
+    "run:stop": (message: {
+        version: "1";
+        run: {
+            friendlyId: string;
+        };
+        snapshot: {
+            friendlyId: string;
+        };
+    }) => void;
+}
+export type WorkloadClientSocketData = {
+    deploymentId: string;
+    runnerId: string;
+    runFriendlyId?: string;
+    snapshotId?: string;
+};

package/dist/commonjs/v3/runEngineWorker/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/commonjs/v3/runEngineWorker/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/v3/runEngineWorker/types.ts"],"names":[],"mappings":""}

package/dist/commonjs/v3/runEngineWorker/util.d.ts

@@ -0,0 +1,4 @@
+/** Will ignore headers with falsey values */
+export declare function createHeaders(headersInit: Record<string, string | undefined | null>): {
+    [k: string]: string;
+};

package/dist/commonjs/v3/runEngineWorker/util.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createHeaders = createHeaders;
+/** Will ignore headers with falsey values */
+function createHeaders(headersInit) {
+    const headers = new Headers();
+    for (const [key, value] of Object.entries(headersInit)) {
+        if (!value) {
+            continue;
+        }
+        headers.set(key, value);
+    }
+    return Object.fromEntries(headers.entries());
+}
+//# sourceMappingURL=util.js.map

package/dist/commonjs/v3/runEngineWorker/util.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"util.js","sourceRoot":"","sources":["../../../../src/v3/runEngineWorker/util.ts"],"names":[],"mappings":";;AACA,sCAWC;AAZD,6CAA6C;AAC7C,SAAgB,aAAa,CAAC,WAAsD;IAClF,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;IAE9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;QACvD,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,SAAS;QACX,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IAC1B,CAAC;IAED,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;AAC/C,CAAC"}