@trigger.dev/core 3.3.16 → 4.0.0-v4-beta.0

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

@@ -0,0 +1,88 @@
+import { z } from "zod";
+import { TaskRunExecutionResult } from "../../schemas/common.js";
+import { MachineResources, DequeuedMessage, StartRunAttemptResult, CompleteRunAttemptResult, RunExecutionData, CheckpointInput, ExecutionResult, } from "../../schemas/runEngine.js";
+export const WorkerApiHeartbeatRequestBody = z.object({
+    cpu: z.object({
+        used: z.number(),
+        available: z.number(),
+    }),
+    memory: z.object({
+        used: z.number(),
+        available: z.number(),
+    }),
+    tasks: z.array(z.string()),
+});
+export const WorkerApiHeartbeatResponseBody = z.object({
+    ok: z.literal(true),
+});
+export const WorkerApiSuspendRunRequestBody = z.discriminatedUnion("success", [
+    z.object({
+        success: z.literal(true),
+        checkpoint: CheckpointInput,
+    }),
+    z.object({
+        success: z.literal(false),
+        error: z.string(),
+    }),
+]);
+export const WorkerApiSuspendRunResponseBody = z.object({
+    ok: z.literal(true),
+});
+export const WorkerApiContinueRunExecutionRequestBody = ExecutionResult;
+export const WorkerApiConnectRequestBody = z.object({
+    metadata: z.record(z.any()),
+});
+export const WorkerApiConnectResponseBody = z.object({
+    ok: z.literal(true),
+    workerGroup: z.object({
+        type: z.string(),
+        name: z.string(),
+    }),
+});
+export const WorkerApiDequeueRequestBody = z.object({
+    maxResources: MachineResources.optional(),
+    maxRunCount: z.number().optional(),
+});
+export const WorkerApiDequeueResponseBody = DequeuedMessage.array();
+export const WorkerApiRunHeartbeatRequestBody = z.object({
+    cpu: z.number().optional(),
+    memory: z.number().optional(),
+});
+export const WorkerApiRunHeartbeatResponseBody = z.object({
+    ok: z.literal(true),
+});
+export const WorkerApiRunAttemptStartRequestBody = z.object({
+    isWarmStart: z.boolean().optional(),
+});
+export const WorkerApiRunAttemptStartResponseBody = StartRunAttemptResult.and(z.object({
+    envVars: z.record(z.string()),
+}));
+export const WorkerApiRunAttemptCompleteRequestBody = z.object({
+    completion: TaskRunExecutionResult,
+});
+export const WorkerApiRunAttemptCompleteResponseBody = z.object({
+    result: CompleteRunAttemptResult,
+});
+export const WorkerApiRunLatestSnapshotResponseBody = z.object({
+    execution: RunExecutionData,
+});
+export const WorkerApiDequeueFromVersionResponseBody = DequeuedMessage.array();
+const AttributeValue = z.union([
+    z.string(),
+    z.number(),
+    z.boolean(),
+    z.array(z.string().nullable()),
+    z.array(z.number().nullable()),
+    z.array(z.boolean().nullable()),
+]);
+const Attributes = z.record(z.string(), AttributeValue.optional());
+export const WorkerApiDebugLogBody = z.object({
+    time: z.coerce.date(),
+    message: z.string(),
+    properties: Attributes.optional(),
+});
+export const WorkerApiSuspendCompletionResponseBody = z.object({
+    success: z.boolean(),
+    error: z.string().optional(),
+});
+//# sourceMappingURL=schemas.js.map

package/dist/esm/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,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,qBAAqB,EACrB,wBAAwB,EACxB,gBAAgB,EAChB,eAAe,EACf,eAAe,GAChB,MAAM,4BAA4B,CAAC;AAEpC,MAAM,CAAC,MAAM,6BAA6B,GAAG,CAAC,CAAC,MAAM,CAAC;IACpD,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC;QACZ,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;KACtB,CAAC;IACF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QACf,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE;KACtB,CAAC;IACF,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAC3B,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,CAAC,MAAM,CAAC;IACrD,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,8BAA8B,GAAG,CAAC,CAAC,kBAAkB,CAAC,SAAS,EAAE;IAC5E,CAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;QACxB,UAAU,EAAE,eAAe;KAC5B,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC;QACzB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;KAClB,CAAC;CACH,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,+BAA+B,GAAG,CAAC,CAAC,MAAM,CAAC;IACtD,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,wCAAwC,GAAG,eAAe,CAAC;AAKxE,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IAClD,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;CAC5B,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,CAAC,MAAM,CAAC;IACnD,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;IACnB,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;QACpB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;KACjB,CAAC;CACH,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,2BAA2B,GAAG,CAAC,CAAC,MAAM,CAAC;IAClD,YAAY,EAAE,gBAAgB,CAAC,QAAQ,EAAE;IACzC,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CACnC,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,4BAA4B,GAAG,eAAe,CAAC,KAAK,EAAE,CAAC;AAGpE,MAAM,CAAC,MAAM,gCAAgC,GAAG,CAAC,CAAC,MAAM,CAAC;IACvD,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC1B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC9B,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,iCAAiC,GAAG,CAAC,CAAC,MAAM,CAAC;IACxD,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC;CACpB,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,mCAAmC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC1D,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;CACpC,CAAC,CAAC;AAKH,MAAM,CAAC,MAAM,oCAAoC,GAAG,qBAAqB,CAAC,GAAG,CAC3E,CAAC,CAAC,MAAM,CAAC;IACP,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;CAC9B,CAAC,CACH,CAAC;AAKF,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7D,UAAU,EAAE,sBAAsB;CACnC,CAAC,CAAC;AAKH,MAAM,CAAC,MAAM,uCAAuC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9D,MAAM,EAAE,wBAAwB;CACjC,CAAC,CAAC;AAKH,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7D,SAAS,EAAE,gBAAgB;CAC5B,CAAC,CAAC;AAKH,MAAM,CAAC,MAAM,uCAAuC,GAAG,eAAe,CAAC,KAAK,EAAE,CAAC;AAK/E,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC;IAC7B,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,OAAO,EAAE;IACX,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC9B,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC;IAC9B,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC;CAChC,CAAC,CAAC;AAEH,MAAM,UAAU,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,QAAQ,EAAE,CAAC,CAAC;AAEnE,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IACrB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;IACnB,UAAU,EAAE,UAAU,CAAC,QAAQ,EAAE;CAClC,CAAC,CAAC;AAGH,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7D,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE;IACpB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC7B,CAAC,CAAC"}

package/dist/esm/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/esm/v3/runEngineWorker/supervisor/session.js

@@ -0,0 +1,167 @@
+import { SupervisorHttpClient } from "./http.js";
+import { RunQueueConsumer } from "./queueConsumer.js";
+import EventEmitter from "events";
+import { VERSION } from "../../../version.js";
+import { io } from "socket.io-client";
+import { getDefaultWorkerHeaders } from "./util.js";
+import { HeartbeatService } from "../../utils/heartbeat.js";
+export class SupervisorSession extends EventEmitter {
+    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 SupervisorHttpClient(opts);
+        this.queueConsumer = new 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 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 = io(wsUrl.href, {
+            transports: ["websocket"],
+            extraHeaders: 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,
+            },
+        });
+        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: [],
+        };
+    }
+}
+//# sourceMappingURL=session.js.map

package/dist/esm/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,OAAO,EAAE,oBAAoB,EAAE,MAAM,WAAW,CAAC;AAGjD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAEtD,OAAO,YAAY,MAAM,QAAQ,CAAC;AAClC,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,EAAE,EAAU,MAAM,kBAAkB,CAAC;AAE9C,OAAO,EAAE,uBAAuB,EAAE,MAAM,WAAW,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAY5D,MAAM,OAAO,iBAAkB,SAAQ,YAA0B;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,oBAAoB,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,CAAC,aAAa,GAAG,IAAI,gBAAgB,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,gBAAgB,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,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE;YAC5B,UAAU,EAAE,CAAC,WAAW,CAAC;YACzB,YAAY,EAAE,uBAAuB,CAAC,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,OAAO;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"}

package/dist/esm/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/esm/v3/runEngineWorker/supervisor/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/esm/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/esm/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/esm/v3/runEngineWorker/supervisor/util.js

@@ -0,0 +1,35 @@
+import { WORKER_HEADERS } from "../consts.js";
+import { createHeaders } from "../util.js";
+export function getDefaultWorkerHeaders(options) {
+    return createHeaders({
+        Authorization: `Bearer ${options.workerToken}`,
+        [WORKER_HEADERS.INSTANCE_NAME]: options.instanceName,
+        [WORKER_HEADERS.DEPLOYMENT_ID]: options.deploymentId,
+        [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);
+}
+export 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/esm/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,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAG3C,MAAM,UAAU,uBAAuB,CACrC,OAAsC;IAEtC,OAAO,aAAa,CAAC;QACnB,aAAa,EAAE,UAAU,OAAO,CAAC,WAAW,EAAE;QAC9C,CAAC,cAAc,CAAC,aAAa,CAAC,EAAE,OAAO,CAAC,YAAY;QACpD,CAAC,cAAc,CAAC,aAAa,CAAC,EAAE,OAAO,CAAC,YAAY;QACpD,CAAC,cAAc,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,MAAM,UAAU,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/esm/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/esm/v3/runEngineWorker/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

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

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

package/dist/esm/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/esm/v3/runEngineWorker/util.js

@@ -0,0 +1,12 @@
+/** Will ignore headers with falsey values */
+export 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/esm/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,MAAM,UAAU,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"}