@trigger.dev/core 3.3.17 → 4.0.0-v4-beta.1

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

@@ -0,0 +1,91 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WorkerApiSuspendCompletionResponseBody = exports.WorkerApiDebugLogBody = 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();
+const AttributeValue = zod_1.z.union([
+    zod_1.z.string(),
+    zod_1.z.number(),
+    zod_1.z.boolean(),
+    zod_1.z.array(zod_1.z.string().nullable()),
+    zod_1.z.array(zod_1.z.number().nullable()),
+    zod_1.z.array(zod_1.z.boolean().nullable()),
+]);
+const Attributes = zod_1.z.record(zod_1.z.string(), AttributeValue.optional());
+exports.WorkerApiDebugLogBody = zod_1.z.object({
+    time: zod_1.z.coerce.date(),
+    message: zod_1.z.string(),
+    properties: Attributes.optional(),
+});
+exports.WorkerApiSuspendCompletionResponseBody = zod_1.z.object({
+    success: zod_1.z.boolean(),
+    error: zod_1.z.string().optional(),
+});
+//# 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;AAK/E,MAAM,cAAc,GAAG,OAAC,CAAC,KAAK,CAAC;IAC7B,OAAC,CAAC,MAAM,EAAE;IACV,OAAC,CAAC,MAAM,EAAE;IACV,OAAC,CAAC,OAAO,EAAE;IACX,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC9B,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC9B,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC;CAChC,CAAC,CAAC;AAEH,MAAM,UAAU,GAAG,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,QAAQ,EAAE,CAAC,CAAC;AAEtD,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,UAAU,CAAC,QAAQ,EAAE;CAClC,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"}

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

@@ -0,0 +1,32 @@
+import { SupervisorHttpClient } from "./http.js";
+import { PreDequeueFn, PreSkipFn, SupervisorClientCommonOptions } from "./types.js";
+import { WorkerEvents } from "./events.js";
+import EventEmitter from "events";
+type SupervisorSessionOptions = SupervisorClientCommonOptions & {
+    queueConsumerEnabled?: boolean;
+    runNotificationsEnabled?: boolean;
+    heartbeatIntervalSeconds?: number;
+    dequeueIntervalMs?: number;
+    preDequeue?: PreDequeueFn;
+    preSkip?: PreSkipFn;
+    maxRunCount?: number;
+};
+export declare class SupervisorSession extends EventEmitter<WorkerEvents> {
+    private opts;
+    readonly httpClient: SupervisorHttpClient;
+    private readonly runNotificationsEnabled;
+    private runNotificationsSocket?;
+    private readonly queueConsumerEnabled;
+    private readonly queueConsumer;
+    private readonly heartbeatService;
+    private readonly heartbeatIntervalSeconds;
+    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,174 @@
+"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 queueConsumer_js_1 = require("./queueConsumer.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 heartbeat_js_1 = require("../../utils/heartbeat.js");
+class SupervisorSession extends events_1.default {
+    opts;
+    httpClient;
+    runNotificationsEnabled;
+    runNotificationsSocket;
+    queueConsumerEnabled;
+    queueConsumer;
+    heartbeatService;
+    heartbeatIntervalSeconds;
+    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.queueConsumer = new queueConsumer_js_1.RunQueueConsumer({
+            client: this.httpClient,
+            preDequeue: opts.preDequeue,
+            preSkip: opts.preSkip,
+            onDequeue: this.onDequeue.bind(this),
+            intervalMs: opts.dequeueIntervalMs,
+            maxRunCount: opts.maxRunCount,
+        });
+        // TODO: This should be dynamic and set by (or at least overridden by) the platform
+        this.heartbeatIntervalSeconds = opts.heartbeatIntervalSeconds || 30;
+        this.heartbeatService = new heartbeat_js_1.HeartbeatService({
+            heartbeat: async () => {
+                console.debug("[SupervisorSession] Sending heartbeat");
+                const body = this.getHeartbeatBody();
+                const response = await this.httpClient.heartbeatWorker(body);
+                if (!response.success) {
+                    console.error("[SupervisorSession] Heartbeat failed", { error: response.error });
+                }
+            },
+            intervalMs: this.heartbeatIntervalSeconds * 1000,
+            leadingEdge: false,
+            onError: async (error) => {
+                console.error("[SupervisorSession] Failed to send heartbeat", { error });
+            },
+        });
+    }
+    async onDequeue(messages) {
+        // Incredibly verbose logging for debugging purposes
+        // console.log("[SupervisorSession] Dequeued messages", { count: messages.length });
+        // console.debug("[SupervisorSession] Dequeued messages with contents", messages);
+        for (const message of messages) {
+            console.log("[SupervisorSession] Emitting message", { message });
+            this.emit("runQueueMessage", {
+                time: new Date(),
+                message,
+            });
+        }
+    }
+    subscribeToRunNotifications(runFriendlyIds) {
+        console.log("[SupervisorSession] Subscribing to run notifications", { runFriendlyIds });
+        if (!this.runNotificationsSocket) {
+            console.error("[SupervisorSession] 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) {
+        console.log("[SupervisorSession] Unsubscribing from run notifications", { runFriendlyIds });
+        if (!this.runNotificationsSocket) {
+            console.error("[SupervisorSession] 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 }) => {
+            console.log("[SupervisorSession][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",
+            })
+                .catch((error) => {
+                console.error("[SupervisorSession] Failed to send debug log", { error });
+            });
+        });
+        socket.on("connect", () => {
+            console.log("[SupervisorSession][WS] Connected to platform");
+        });
+        socket.on("connect_error", (error) => {
+            console.error("[SupervisorSession][WS] Connection error", { error });
+        });
+        socket.on("disconnect", (reason, description) => {
+            console.log("[SupervisorSession][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) {
+            console.error("[SupervisorSession][HTTP] Failed to connect", { error: connect.error });
+            throw new Error("[SupervisorSession][HTTP] Failed to connect");
+        }
+        const { workerGroup } = connect.data;
+        console.log("[SupervisorSession][HTTP] Connected to platform", {
+            type: workerGroup.type,
+            name: workerGroup.name,
+        });
+        if (this.queueConsumerEnabled) {
+            console.log("[SupervisorSession] Queue consumer enabled");
+            this.queueConsumer.start();
+            this.heartbeatService.start();
+        }
+        else {
+            console.warn("[SupervisorSession] Queue consumer disabled");
+        }
+        if (this.runNotificationsEnabled) {
+            console.log("[SupervisorSession] Run notifications enabled");
+            this.runNotificationsSocket = this.createRunNotificationsSocket();
+        }
+        else {
+            console.warn("[SupervisorSession] Run notifications disabled");
+        }
+    }
+    async stop() {
+        this.heartbeatService.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,yDAAsD;AAEtD,oDAAkC;AAClC,oDAA8C;AAC9C,uDAA8C;AAE9C,uCAAoD;AACpD,2DAA4D;AAY5D,MAAa,iBAAkB,SAAQ,gBAA0B;IAY3C;IAXJ,UAAU,CAAuB;IAEhC,uBAAuB,CAAU;IAC1C,sBAAsB,CAAkE;IAE/E,oBAAoB,CAAU;IAC9B,aAAa,CAAmB;IAEhC,gBAAgB,CAAmB;IACnC,wBAAwB,CAAS;IAElD,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;QACjD,IAAI,CAAC,aAAa,GAAG,IAAI,mCAAgB,CAAC;YACxC,MAAM,EAAE,IAAI,CAAC,UAAU;YACvB,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;YACpC,UAAU,EAAE,IAAI,CAAC,iBAAiB;YAClC,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC,CAAC;QAEH,mFAAmF;QACnF,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC,wBAAwB,IAAI,EAAE,CAAC;QACpE,IAAI,CAAC,gBAAgB,GAAG,IAAI,+BAAgB,CAAC;YAC3C,SAAS,EAAE,KAAK,IAAI,EAAE;gBACpB,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBAEvD,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,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;gBACnF,CAAC;YACH,CAAC;YACD,UAAU,EAAE,IAAI,CAAC,wBAAwB,GAAG,IAAI;YAChD,WAAW,EAAE,KAAK;YAClB,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE;gBACvB,OAAO,CAAC,KAAK,CAAC,8CAA8C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC3E,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,QAAsC;QAC5D,oDAAoD;QACpD,oFAAoF;QACpF,kFAAkF;QAElF,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;YAC/B,OAAO,CAAC,GAAG,CAAC,sCAAsC,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YACjE,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,OAAO,CAAC,GAAG,CAAC,sDAAsD,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAExF,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;YAC1D,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,OAAO,CAAC,GAAG,CAAC,0DAA0D,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;QAE5F,IAAI,CAAC,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACjC,OAAO,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;YAC1D,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,OAAO,CAAC,GAAG,CAAC,mDAAmD,EAAE,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;YACnF,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,EAAE,IAAI,EAAE,IAAI,IAAI,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;YAExD,IAAI,CAAC,UAAU;iBACZ,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE;gBAC5B,IAAI,EAAE,IAAI,IAAI,EAAE;gBAChB,OAAO,EAAE,mCAAmC;aAC7C,CAAC;iBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACf,OAAO,CAAC,KAAK,CAAC,8CAA8C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC3E,CAAC,CAAC,CAAC;QACP,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE;YACxB,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,KAAK,EAAE,EAAE;YACnC,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,EAAE,CAAC,YAAY,EAAE,CAAC,MAAM,EAAE,WAAW,EAAE,EAAE;YAC9C,OAAO,CAAC,GAAG,CAAC,oDAAoD,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;QAC7F,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,OAAO,CAAC,KAAK,CAAC,6CAA6C,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;YACvF,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;QAErC,OAAO,CAAC,GAAG,CAAC,iDAAiD,EAAE;YAC7D,IAAI,EAAE,WAAW,CAAC,IAAI;YACtB,IAAI,EAAE,WAAW,CAAC,IAAI;SACvB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC9B,OAAO,CAAC,GAAG,CAAC,4CAA4C,CAAC,CAAC;YAC1D,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;YAC3B,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,CAAC;QAChC,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC9D,CAAC;QAED,IAAI,IAAI,CAAC,uBAAuB,EAAE,CAAC;YACjC,OAAO,CAAC,GAAG,CAAC,+CAA+C,CAAC,CAAC;YAC7D,IAAI,CAAC,sBAAsB,GAAG,IAAI,CAAC,4BAA4B,EAAE,CAAC;QACpE,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,gDAAgD,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC;QAC7B,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;AAjMD,8CAiMC"}

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

@@ -0,0 +1,13 @@
+import { MachineResources } from "../../schemas/runEngine.js";
+export type SupervisorClientCommonOptions = {
+    apiUrl: string;
+    workerToken: string;
+    instanceName: string;
+    deploymentId?: string;
+    managedWorkerSecret?: string;
+};
+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"}