@trigger.dev/core 0.0.0-v3-prerelease-20250108142426 → 0.0.0-v4-prerelease-20250917185130

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

@@ -0,0 +1,97 @@
+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();
+export const DebugLogPropertiesValue = z.union([
+    z.string(),
+    z.number(),
+    z.boolean(),
+    z.array(z.string().nullish()),
+    z.array(z.number().nullish()),
+    z.array(z.boolean().nullish()),
+]);
+export const DebugLogProperties = z.record(z.string(), DebugLogPropertiesValue.optional());
+export const DebugLogPropertiesInput = z.record(z.string(), z.unknown());
+export const WorkerApiDebugLogBodyInput = z.object({
+    time: z.coerce.date(),
+    message: z.string(),
+    properties: DebugLogPropertiesInput.optional(),
+});
+export const WorkerApiDebugLogBody = z.object({
+    time: z.coerce.date(),
+    message: z.string(),
+    properties: DebugLogProperties.optional(),
+});
+export const WorkerApiSuspendCompletionResponseBody = z.object({
+    success: z.boolean(),
+    error: z.string().optional(),
+});
+export const WorkerApiRunSnapshotsSinceResponseBody = z.object({
+    snapshots: z.array(RunExecutionData),
+});
+//# 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,MAAM,uBAAuB,GAAG,CAAC,CAAC,KAAK,CAAC;IAC7C,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,MAAM,EAAE;IACV,CAAC,CAAC,OAAO,EAAE;IACX,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,CAAC;CAC/B,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,QAAQ,EAAE,CAAC,CAAC;AAG3F,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;AAGzE,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,EAAE;IACrB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;IACnB,UAAU,EAAE,uBAAuB,CAAC,QAAQ,EAAE;CAC/C,CAAC,CAAC;AAGH,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,kBAAkB,CAAC,QAAQ,EAAE;CAC1C,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;AAKH,MAAM,CAAC,MAAM,sCAAsC,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7D,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,gBAAgB,CAAC;CACrC,CAAC,CAAC"}

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

@@ -0,0 +1,169 @@
+import { SupervisorHttpClient } from "./http.js";
+import { RunQueueConsumerPool } from "./consumerPool.js";
+import EventEmitter from "events";
+import { VERSION } from "../../../version.js";
+import { io } from "socket.io-client";
+import { getDefaultWorkerHeaders } from "./util.js";
+import { IntervalService } from "../../utils/interval.js";
+import { SimpleStructuredLogger } from "../../utils/structuredLogger.js";
+export class SupervisorSession extends EventEmitter {
+    opts;
+    httpClient;
+    logger = new 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 SupervisorHttpClient(opts);
+        this.consumerPool = new 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 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 = io(wsUrl.href, {
+            transports: ["websocket"],
+            extraHeaders: 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,
+            },
+        });
+        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: [],
+        };
+    }
+}
+//# 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,oBAAoB,EAAkB,MAAM,mBAAmB,CAAC;AAEzE,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,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAiBzE,MAAM,OAAO,iBAAkB,SAAQ,YAA0B;IAa3C;IAZJ,UAAU,CAAuB;IAEhC,MAAM,GAAG,IAAI,sBAAsB,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,oBAAoB,CAAC,IAAI,CAAC,CAAC;QAEjD,IAAI,CAAC,YAAY,GAAG,IAAI,oBAAoB,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,eAAe,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,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,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,OAAO;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/esm/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/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"}