@trigger.dev/core 0.0.0-buffer-fix-20231206151546 → 0.0.0-bun-prerelease-20260909132648

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

@@ -0,0 +1,109 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkerApiRunSnapshotsSinceResponseBody = exports.WorkerApiSuspendCompletionResponseBody = exports.WorkerApiDebugLogBody = exports.WorkerApiDebugLogBodyInput = exports.DebugLogPropertiesInput = exports.DebugLogProperties = exports.DebugLogPropertiesValue = exports.WorkerApiRunLatestSnapshotResponseBody = exports.WorkerApiRunAttemptCompleteResponseBody = exports.WorkerApiRunAttemptCompleteRequestBody = exports.WorkerApiRunAttemptStartResponseBody = exports.WorkerApiRunAttemptStartRequestBody = exports.WorkerApiRunHeartbeatResponseBody = exports.WorkerApiRunHeartbeatRequestBody = exports.WorkerApiDequeueResponseBody = exports.WorkerApiDequeueRequestBody = exports.WorkerQueueClass = exports.WorkerApiConnectResponseBody = exports.WorkerApiConnectRequestBody = exports.WorkerApiContinueRunExecutionRequestBody = exports.WorkerApiSuspendRunResponseBody = exports.WorkerApiSuspendRunRequestBody = exports.WorkerApiHeartbeatResponseBody = exports.WorkerApiHeartbeatRequestBody = void 0;
+const v4_1 = require("zod/v4");
+const zod_js_1 = require("../../utils/zod.js");
+const common_js_1 = require("../../schemas/common.js");
+const runEngine_js_1 = require("../../schemas/runEngine.js");
+exports.WorkerApiHeartbeatRequestBody = v4_1.z.object({
+    cpu: v4_1.z.object({
+        used: v4_1.z.number(),
+        available: v4_1.z.number(),
+    }),
+    memory: v4_1.z.object({
+        used: v4_1.z.number(),
+        available: v4_1.z.number(),
+    }),
+    tasks: v4_1.z.array(v4_1.z.string()),
+});
+exports.WorkerApiHeartbeatResponseBody = v4_1.z.object({
+    ok: v4_1.z.literal(true),
+});
+exports.WorkerApiSuspendRunRequestBody = (0, zod_js_1.discriminatedUnion)("success", [
+    v4_1.z.object({
+        success: v4_1.z.literal(true),
+        checkpoint: runEngine_js_1.CheckpointInput,
+    }),
+    v4_1.z.object({
+        success: v4_1.z.literal(false),
+        error: v4_1.z.string(),
+    }),
+]);
+exports.WorkerApiSuspendRunResponseBody = v4_1.z.object({
+    ok: v4_1.z.literal(true),
+});
+exports.WorkerApiContinueRunExecutionRequestBody = runEngine_js_1.ExecutionResult;
+exports.WorkerApiConnectRequestBody = v4_1.z.object({
+    metadata: v4_1.z.record(v4_1.z.string(), v4_1.z.any()),
+});
+exports.WorkerApiConnectResponseBody = v4_1.z.object({
+    ok: v4_1.z.literal(true),
+    workerGroup: v4_1.z.object({
+        type: v4_1.z.string(),
+        name: v4_1.z.string(),
+    }),
+});
+exports.WorkerQueueClass = v4_1.z.enum(["default", "scheduled"]);
+exports.WorkerApiDequeueRequestBody = v4_1.z.object({
+    maxResources: runEngine_js_1.MachineResources.optional(),
+    maxRunCount: v4_1.z.number().optional(),
+    /**
+     * Which class of worker queue this consumer pulls from. Absent or "default" =
+     * the worker group's region queue. "scheduled" targets the dedicated
+     * scheduled-lineage queue so a separate fleet can drain it independently. The
+     * server derives the actual queue name from the token, so this only ever
+     * selects between the authenticated worker's own queues.
+     */
+    queueClass: exports.WorkerQueueClass.optional(),
+});
+exports.WorkerApiDequeueResponseBody = runEngine_js_1.DequeuedMessage.array();
+exports.WorkerApiRunHeartbeatRequestBody = v4_1.z.object({
+    cpu: v4_1.z.number().optional(),
+    memory: v4_1.z.number().optional(),
+});
+exports.WorkerApiRunHeartbeatResponseBody = v4_1.z.object({
+    ok: v4_1.z.literal(true),
+});
+exports.WorkerApiRunAttemptStartRequestBody = v4_1.z.object({
+    isWarmStart: v4_1.z.boolean().optional(),
+});
+exports.WorkerApiRunAttemptStartResponseBody = runEngine_js_1.StartRunAttemptResult.and(v4_1.z.object({
+    envVars: v4_1.z.record(v4_1.z.string(), v4_1.z.string()),
+}));
+exports.WorkerApiRunAttemptCompleteRequestBody = v4_1.z.object({
+    completion: common_js_1.TaskRunExecutionResult,
+});
+exports.WorkerApiRunAttemptCompleteResponseBody = v4_1.z.object({
+    result: runEngine_js_1.CompleteRunAttemptResult,
+});
+exports.WorkerApiRunLatestSnapshotResponseBody = v4_1.z.object({
+    execution: runEngine_js_1.RunExecutionData,
+});
+exports.DebugLogPropertiesValue = v4_1.z.union([
+    v4_1.z.string(),
+    v4_1.z.number(),
+    v4_1.z.boolean(),
+    v4_1.z.array(v4_1.z.string().nullish()),
+    v4_1.z.array(v4_1.z.number().nullish()),
+    v4_1.z.array(v4_1.z.boolean().nullish()),
+]);
+exports.DebugLogProperties = v4_1.z.record(v4_1.z.string(), exports.DebugLogPropertiesValue.optional());
+exports.DebugLogPropertiesInput = v4_1.z.record(v4_1.z.string(), v4_1.z.unknown());
+exports.WorkerApiDebugLogBodyInput = v4_1.z.object({
+    time: v4_1.z.coerce.date(),
+    message: v4_1.z.string(),
+    properties: exports.DebugLogPropertiesInput.optional(),
+});
+exports.WorkerApiDebugLogBody = v4_1.z.object({
+    time: v4_1.z.coerce.date(),
+    message: v4_1.z.string(),
+    properties: exports.DebugLogProperties.optional(),
+});
+exports.WorkerApiSuspendCompletionResponseBody = v4_1.z.object({
+    success: v4_1.z.boolean(),
+    error: v4_1.z.string().optional(),
+});
+exports.WorkerApiRunSnapshotsSinceResponseBody = v4_1.z.object({
+    snapshots: v4_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,+BAA2B;AAC3B,+CAAwD;AACxD,uDAAiE;AACjE,6DAQoC;AAEvB,QAAA,6BAA6B,GAAG,MAAC,CAAC,MAAM,CAAC;IACpD,GAAG,EAAE,MAAC,CAAC,MAAM,CAAC;QACZ,IAAI,EAAE,MAAC,CAAC,MAAM,EAAE;QAChB,SAAS,EAAE,MAAC,CAAC,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,EAAE,MAAC,CAAC,MAAM,CAAC;QACf,IAAI,EAAE,MAAC,CAAC,MAAM,EAAE;QAChB,SAAS,EAAE,MAAC,CAAC,MAAM,EAAE;KACtB,CAAC;IACF,KAAK,EAAE,MAAC,CAAC,KAAK,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC;CAC3B,CAAC,CAAC;AAGU,QAAA,8BAA8B,GAAG,MAAC,CAAC,MAAM,CAAC;IACrD,EAAE,EAAE,MAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGU,QAAA,8BAA8B,GAAG,IAAA,2BAAkB,EAAC,SAAS,EAAE;IAC1E,MAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,MAAC,CAAC,OAAO,CAAC,IAAI,CAAC;QACxB,UAAU,EAAE,8BAAe;KAC5B,CAAC;IACF,MAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,MAAC,CAAC,OAAO,CAAC,KAAK,CAAC;QACzB,KAAK,EAAE,MAAC,CAAC,MAAM,EAAE;KAClB,CAAC;CACH,CAAC,CAAC;AAGU,QAAA,+BAA+B,GAAG,MAAC,CAAC,MAAM,CAAC;IACtD,EAAE,EAAE,MAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGU,QAAA,wCAAwC,GAAG,8BAAe,CAAC;AAK3D,QAAA,2BAA2B,GAAG,MAAC,CAAC,MAAM,CAAC;IAClD,QAAQ,EAAE,MAAC,CAAC,MAAM,CAAC,MAAC,CAAC,MAAM,EAAE,EAAE,MAAC,CAAC,GAAG,EAAE,CAAC;CACxC,CAAC,CAAC;AAGU,QAAA,4BAA4B,GAAG,MAAC,CAAC,MAAM,CAAC;IACnD,EAAE,EAAE,MAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IACnB,WAAW,EAAE,MAAC,CAAC,MAAM,CAAC;QACpB,IAAI,EAAE,MAAC,CAAC,MAAM,EAAE;QAChB,IAAI,EAAE,MAAC,CAAC,MAAM,EAAE;KACjB,CAAC;CACH,CAAC,CAAC;AAGU,QAAA,gBAAgB,GAAG,MAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC,CAAC;AAGpD,QAAA,2BAA2B,GAAG,MAAC,CAAC,MAAM,CAAC;IAClD,YAAY,EAAE,+BAAgB,CAAC,QAAQ,EAAE;IACzC,WAAW,EAAE,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAClC;;;;;;OAMG;IACH,UAAU,EAAE,QAAA,gBAAgB,CAAC,QAAQ,EAAE;CACxC,CAAC,CAAC;AAGU,QAAA,4BAA4B,GAAG,8BAAe,CAAC,KAAK,EAAE,CAAC;AAGvD,QAAA,gCAAgC,GAAG,MAAC,CAAC,MAAM,CAAC;IACvD,GAAG,EAAE,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC1B,MAAM,EAAE,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC9B,CAAC,CAAC;AAGU,QAAA,iCAAiC,GAAG,MAAC,CAAC,MAAM,CAAC;IACxD,EAAE,EAAE,MAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGU,QAAA,mCAAmC,GAAG,MAAC,CAAC,MAAM,CAAC;IAC1D,WAAW,EAAE,MAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;CACpC,CAAC,CAAC;AAKU,QAAA,oCAAoC,GAAG,oCAAqB,CAAC,GAAG,CAC3E,MAAC,CAAC,MAAM,CAAC;IACP,OAAO,EAAE,MAAC,CAAC,MAAM,CAAC,MAAC,CAAC,MAAM,EAAE,EAAE,MAAC,CAAC,MAAM,EAAE,CAAC;CAC1C,CAAC,CACH,CAAC;AAKW,QAAA,sCAAsC,GAAG,MAAC,CAAC,MAAM,CAAC;IAC7D,UAAU,EAAE,kCAAsB;CACnC,CAAC,CAAC;AAKU,QAAA,uCAAuC,GAAG,MAAC,CAAC,MAAM,CAAC;IAC9D,MAAM,EAAE,uCAAwB;CACjC,CAAC,CAAC;AAKU,QAAA,sCAAsC,GAAG,MAAC,CAAC,MAAM,CAAC;IAC7D,SAAS,EAAE,+BAAgB;CAC5B,CAAC,CAAC;AAKU,QAAA,uBAAuB,GAAG,MAAC,CAAC,KAAK,CAAC;IAC7C,MAAC,CAAC,MAAM,EAAE;IACV,MAAC,CAAC,MAAM,EAAE;IACV,MAAC,CAAC,OAAO,EAAE;IACX,MAAC,CAAC,KAAK,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,MAAC,CAAC,KAAK,CAAC,MAAC,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,MAAC,CAAC,KAAK,CAAC,MAAC,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC;CAC/B,CAAC,CAAC;AAEU,QAAA,kBAAkB,GAAG,MAAC,CAAC,MAAM,CAAC,MAAC,CAAC,MAAM,EAAE,EAAE,QAAA,uBAAuB,CAAC,QAAQ,EAAE,CAAC,CAAC;AAG9E,QAAA,uBAAuB,GAAG,MAAC,CAAC,MAAM,CAAC,MAAC,CAAC,MAAM,EAAE,EAAE,MAAC,CAAC,OAAO,EAAE,CAAC,CAAC;AAG5D,QAAA,0BAA0B,GAAG,MAAC,CAAC,MAAM,CAAC;IACjD,IAAI,EAAE,MAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IACrB,OAAO,EAAE,MAAC,CAAC,MAAM,EAAE;IACnB,UAAU,EAAE,QAAA,uBAAuB,CAAC,QAAQ,EAAE;CAC/C,CAAC,CAAC;AAGU,QAAA,qBAAqB,GAAG,MAAC,CAAC,MAAM,CAAC;IAC5C,IAAI,EAAE,MAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IACrB,OAAO,EAAE,MAAC,CAAC,MAAM,EAAE;IACnB,UAAU,EAAE,QAAA,kBAAkB,CAAC,QAAQ,EAAE;CAC1C,CAAC,CAAC;AAGU,QAAA,sCAAsC,GAAG,MAAC,CAAC,MAAM,CAAC;IAC7D,OAAO,EAAE,MAAC,CAAC,OAAO,EAAE;IACpB,KAAK,EAAE,MAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC7B,CAAC,CAAC;AAKU,QAAA,sCAAsC,GAAG,MAAC,CAAC,MAAM,CAAC;IAC7D,SAAS,EAAE,MAAC,CAAC,KAAK,CAAC,+BAAgB,CAAC;CACrC,CAAC,CAAC"}

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

@@ -0,0 +1,41 @@
+import { SupervisorHttpClient } from "./http.js";
+import type { PreDequeueFn, PreSkipFn, SupervisorClientCommonOptions } from "./types.js";
+import type { WorkerQueueClass } from "./schemas.js";
+import type { ScalingOptions } from "./consumerPool.js";
+import type { 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;
+    /** Which worker-queue class this supervisor's consumers pull from. Defaults to the region queue. */
+    queueClass?: WorkerQueueClass;
+    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,179 @@
+"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,
+                queueClass: opts.queueClass,
+            },
+            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, timing) {
+        this.logger.verbose("Dequeued messages with contents", { count: messages.length, messages });
+        for (const message of messages) {
+            this.emit("runQueueMessage", {
+                time: new Date(),
+                message,
+                dequeueResponseMs: timing?.dequeueResponseMs,
+                pollingIntervalMs: timing?.pollingIntervalMs,
+            });
+        }
+    }
+    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;AAQjD,uDAAyD;AAEzD,oDAAkC;AAClC,oDAA8C;AAE9C,uDAAsC;AAEtC,uCAAoD;AACpD,yDAA0D;AAC1D,yEAAyE;AAmBzE,uBAA+B,SAAQ,gBAA0B;IAa3C,IAAI;IAZR,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;oBADU,IAAI;QAGtB,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;gBAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;aAC5B;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,CACrB,QAAsC,EACtC,MAAiE;QAEjE,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;gBACP,iBAAiB,EAAE,MAAM,EAAE,iBAAiB;gBAC5C,iBAAiB,EAAE,MAAM,EAAE,iBAAiB;aAC7C,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"}

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

@@ -0,0 +1,22 @@
+import type { MachineResources } from "../../schemas/runEngine.js";
+export type SupervisorHttpRequestMetric = {
+    name: string;
+    method: string;
+    status: string;
+    outcome: "ok" | "http_error" | "invalid_response" | "network_error";
+    durationMs: number;
+};
+export type SupervisorClientCommonOptions = {
+    apiUrl: string;
+    workerToken: string;
+    instanceName: string;
+    deploymentId?: string;
+    managedWorkerSecret?: string;
+    sendRunDebugLogs?: boolean;
+    onHttpRequestComplete?: (metric: SupervisorHttpRequestMetric) => void;
+};
+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,2 @@
+import type { SupervisorClientCommonOptions } from "./types.js";
+export declare function getDefaultWorkerHeaders(options: SupervisorClientCommonOptions): Record<string, string>;

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

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getDefaultWorkerHeaders = getDefaultWorkerHeaders;
+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,
+    });
+}
+//# 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":";;;AAAA,4CAA8C;AAC9C,wCAA2C;AAG3C,iCACE,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"}

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

@@ -0,0 +1,53 @@
+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;
+    deploymentFriendlyId?: 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":";;;AAAA,6CAA6C;AAC7C,uBAA8B,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"}