@trigger.dev/core 0.0.0-prerelease-20251007140052 → 0.0.0-prerelease-20251113143926

package/dist/commonjs/v3/realtime-streams-api.js

@@ -0,0 +1,23 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.realtimeStreams = void 0;
+// Split module-level variable definition into separate files to allow
+// tree-shaking on each api instance.
+const index_js_1 = require("./realtimeStreams/index.js");
+exports.realtimeStreams = index_js_1.RealtimeStreamsAPI.getInstance();
+__exportStar(require("./realtimeStreams/types.js"), exports);
+//# sourceMappingURL=realtime-streams-api.js.map

package/dist/commonjs/v3/realtime-streams-api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"realtime-streams-api.js","sourceRoot":"","sources":["../../../src/v3/realtime-streams-api.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,sEAAsE;AACtE,qCAAqC;AACrC,yDAAgE;AAEnD,QAAA,eAAe,GAAG,6BAAkB,CAAC,WAAW,EAAE,CAAC;AAEhE,6DAA2C"}

package/dist/commonjs/v3/realtimeStreams/index.d.ts

@@ -0,0 +1,10 @@
+import { RealtimeStreamOperationOptions, RealtimeStreamInstance, RealtimeStreamsManager } from "./types.js";
+export declare class RealtimeStreamsAPI implements RealtimeStreamsManager {
+    #private;
+    private static _instance?;
+    private constructor();
+    static getInstance(): RealtimeStreamsAPI;
+    setGlobalManager(manager: RealtimeStreamsManager): boolean;
+    pipe<T>(key: string, source: AsyncIterable<T> | ReadableStream<T>, options?: RealtimeStreamOperationOptions): RealtimeStreamInstance<T>;
+    append<TPart extends BodyInit>(key: string, part: TPart, options?: RealtimeStreamOperationOptions): Promise<void>;
+}

package/dist/commonjs/v3/realtimeStreams/index.js

@@ -0,0 +1,31 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RealtimeStreamsAPI = void 0;
+const globals_js_1 = require("../utils/globals.js");
+const noopManager_js_1 = require("./noopManager.js");
+const API_NAME = "realtime-streams";
+const NOOP_MANAGER = new noopManager_js_1.NoopRealtimeStreamsManager();
+class RealtimeStreamsAPI {
+    static _instance;
+    constructor() { }
+    static getInstance() {
+        if (!this._instance) {
+            this._instance = new RealtimeStreamsAPI();
+        }
+        return this._instance;
+    }
+    setGlobalManager(manager) {
+        return (0, globals_js_1.registerGlobal)(API_NAME, manager);
+    }
+    #getManager() {
+        return (0, globals_js_1.getGlobal)(API_NAME) ?? NOOP_MANAGER;
+    }
+    pipe(key, source, options) {
+        return this.#getManager().pipe(key, source, options);
+    }
+    append(key, part, options) {
+        return this.#getManager().append(key, part, options);
+    }
+}
+exports.RealtimeStreamsAPI = RealtimeStreamsAPI;
+//# sourceMappingURL=index.js.map

package/dist/commonjs/v3/realtimeStreams/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/realtimeStreams/index.ts"],"names":[],"mappings":";;;AAAA,oDAAgE;AAChE,qDAA8D;AAO9D,MAAM,QAAQ,GAAG,kBAAkB,CAAC;AAEpC,MAAM,YAAY,GAAG,IAAI,2CAA0B,EAAE,CAAC;AAEtD,MAAa,kBAAkB;IACrB,MAAM,CAAC,SAAS,CAAsB;IAE9C,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,kBAAkB,EAAE,CAAC;QAC5C,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,gBAAgB,CAAC,OAA+B;QAC9C,OAAO,IAAA,2BAAc,EAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,WAAW;QACT,OAAO,IAAA,sBAAS,EAAC,QAAQ,CAAC,IAAI,YAAY,CAAC;IAC7C,CAAC;IAEM,IAAI,CACT,GAAW,EACX,MAA4C,EAC5C,OAAwC;QAExC,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;IAEM,MAAM,CACX,GAAW,EACX,IAAW,EACX,OAAwC;QAExC,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;CACF;AApCD,gDAoCC"}

package/dist/commonjs/v3/realtimeStreams/manager.d.ts

@@ -0,0 +1,14 @@
+import { ApiClient } from "../apiClient/index.js";
+import { RealtimeStreamInstance, RealtimeStreamOperationOptions, RealtimeStreamsManager } from "./types.js";
+export declare class StandardRealtimeStreamsManager implements RealtimeStreamsManager {
+    private apiClient;
+    private baseUrl;
+    private debug;
+    constructor(apiClient: ApiClient, baseUrl: string, debug?: boolean);
+    private activeStreams;
+    reset(): void;
+    pipe<T>(key: string, source: AsyncIterable<T> | ReadableStream<T>, options?: RealtimeStreamOperationOptions): RealtimeStreamInstance<T>;
+    append<TPart extends BodyInit>(key: string, part: TPart, options?: RealtimeStreamOperationOptions): Promise<void>;
+    hasActiveStreams(): boolean;
+    waitForAllStreams(timeout?: number): Promise<void>;
+}

package/dist/commonjs/v3/realtimeStreams/manager.js

@@ -0,0 +1,128 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StandardRealtimeStreamsManager = void 0;
+const asyncIterableStream_js_1 = require("../streams/asyncIterableStream.js");
+const task_context_api_js_1 = require("../task-context-api.js");
+const streamInstance_js_1 = require("./streamInstance.js");
+class StandardRealtimeStreamsManager {
+    apiClient;
+    baseUrl;
+    debug;
+    constructor(apiClient, baseUrl, debug = false) {
+        this.apiClient = apiClient;
+        this.baseUrl = baseUrl;
+        this.debug = debug;
+    }
+    // Track active streams - using a Set allows multiple streams for the same key to coexist
+    activeStreams = new Set();
+    reset() {
+        this.activeStreams.clear();
+    }
+    pipe(key, source, options) {
+        // Normalize ReadableStream to AsyncIterable
+        const readableStreamSource = (0, asyncIterableStream_js_1.ensureReadableStream)(source);
+        const runId = getRunIdForOptions(options);
+        if (!runId) {
+            throw new Error("Could not determine the target run ID for the realtime stream. Please specify a target run ID using the `target` option.");
+        }
+        // Create an AbortController for this stream
+        const abortController = new AbortController();
+        // Chain with user-provided signal if present
+        const combinedSignal = options?.signal
+            ? AbortSignal.any?.([options.signal, abortController.signal]) ?? abortController.signal
+            : abortController.signal;
+        const streamInstance = new streamInstance_js_1.StreamInstance({
+            apiClient: this.apiClient,
+            baseUrl: this.baseUrl,
+            runId,
+            key,
+            source: readableStreamSource,
+            signal: combinedSignal,
+            requestOptions: options?.requestOptions,
+            target: options?.target,
+            debug: this.debug,
+        });
+        // Register this stream
+        const streamInfo = { wait: () => streamInstance.wait(), abortController };
+        this.activeStreams.add(streamInfo);
+        // Clean up when stream completes
+        streamInstance.wait().finally(() => this.activeStreams.delete(streamInfo));
+        return {
+            wait: () => streamInstance.wait(),
+            stream: streamInstance.stream,
+        };
+    }
+    async append(key, part, options) {
+        const runId = getRunIdForOptions(options);
+        if (!runId) {
+            throw new Error("Could not determine the target run ID for the realtime stream. Please specify a target run ID using the `target` option.");
+        }
+        const result = await this.apiClient.appendToStream(runId, "self", key, part, options?.requestOptions);
+        if (!result.ok) {
+            throw new Error(`Failed to append to stream: ${result.message ?? "Unknown error"}`);
+        }
+    }
+    hasActiveStreams() {
+        return this.activeStreams.size > 0;
+    }
+    // Waits for all the streams to finish
+    async waitForAllStreams(timeout = 60_000) {
+        if (this.activeStreams.size === 0) {
+            return;
+        }
+        const promises = Array.from(this.activeStreams).map((stream) => stream.wait());
+        // Create a timeout promise that resolves to a special sentinel value
+        const TIMEOUT_SENTINEL = Symbol("timeout");
+        const timeoutPromise = new Promise((resolve) => setTimeout(() => resolve(TIMEOUT_SENTINEL), timeout));
+        // Race between all streams completing/rejecting and the timeout
+        const result = await Promise.race([Promise.all(promises), timeoutPromise]);
+        // Check if we timed out
+        if (result === TIMEOUT_SENTINEL) {
+            // Timeout occurred - abort all active streams
+            const abortedCount = this.activeStreams.size;
+            for (const streamInfo of this.activeStreams) {
+                streamInfo.abortController.abort();
+                this.activeStreams.delete(streamInfo);
+            }
+            throw new Error(`Timeout waiting for streams to finish after ${timeout}ms. Aborted ${abortedCount} active stream(s).`);
+        }
+        // If we reach here, Promise.all completed (either all resolved or one rejected)
+        // Any rejection from Promise.all will have already propagated
+    }
+}
+exports.StandardRealtimeStreamsManager = StandardRealtimeStreamsManager;
+function getRunIdForOptions(options) {
+    if (options?.target) {
+        if (options.target === "parent") {
+            return task_context_api_js_1.taskContext.ctx?.run?.parentTaskRunId ?? task_context_api_js_1.taskContext.ctx?.run?.id;
+        }
+        if (options.target === "root") {
+            return task_context_api_js_1.taskContext.ctx?.run?.rootTaskRunId ?? task_context_api_js_1.taskContext.ctx?.run?.id;
+        }
+        if (options.target === "self") {
+            return task_context_api_js_1.taskContext.ctx?.run?.id;
+        }
+        return options.target;
+    }
+    return task_context_api_js_1.taskContext.ctx?.run?.id;
+}
+function parseCreateStreamResponse(version, headers) {
+    if (version === "v1") {
+        return { version: "v1" };
+    }
+    const accessToken = headers?.["x-s2-access-token"];
+    const basin = headers?.["x-s2-basin"];
+    if (!accessToken || !basin) {
+        return { version: "v1" };
+    }
+    const flushIntervalMs = headers?.["x-s2-flush-interval-ms"];
+    const maxRetries = headers?.["x-s2-max-retries"];
+    return {
+        version: "v2",
+        accessToken,
+        basin,
+        flushIntervalMs: flushIntervalMs ? parseInt(flushIntervalMs) : undefined,
+        maxRetries: maxRetries ? parseInt(maxRetries) : undefined,
+    };
+}
+//# sourceMappingURL=manager.js.map

package/dist/commonjs/v3/realtimeStreams/manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.js","sourceRoot":"","sources":["../../../../src/v3/realtimeStreams/manager.ts"],"names":[],"mappings":";;;AACA,8EAA8F;AAC9F,gEAAqD;AACrD,2DAAqD;AAOrD,MAAa,8BAA8B;IAE/B;IACA;IACA;IAHV,YACU,SAAoB,EACpB,OAAe,EACf,QAAiB,KAAK;QAFtB,cAAS,GAAT,SAAS,CAAW;QACpB,YAAO,GAAP,OAAO,CAAQ;QACf,UAAK,GAAL,KAAK,CAAiB;IAC7B,CAAC;IACJ,yFAAyF;IACjF,aAAa,GAAG,IAAI,GAAG,EAG3B,CAAC;IAEL,KAAK;QACH,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;IAC7B,CAAC;IAEM,IAAI,CACT,GAAW,EACX,MAA4C,EAC5C,OAAwC;QAExC,4CAA4C;QAC5C,MAAM,oBAAoB,GAAG,IAAA,6CAAoB,EAAC,MAAM,CAAC,CAAC;QAE1D,MAAM,KAAK,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAE1C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CACb,0HAA0H,CAC3H,CAAC;QACJ,CAAC;QAED,4CAA4C;QAC5C,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC9C,6CAA6C;QAC7C,MAAM,cAAc,GAAG,OAAO,EAAE,MAAM;YACpC,CAAC,CAAC,WAAW,CAAC,GAAG,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,eAAe,CAAC,MAAM,CAAC,CAAC,IAAI,eAAe,CAAC,MAAM;YACvF,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC;QAE3B,MAAM,cAAc,GAAG,IAAI,kCAAc,CAAC;YACxC,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK;YACL,GAAG;YACH,MAAM,EAAE,oBAAoB;YAC5B,MAAM,EAAE,cAAc;YACtB,cAAc,EAAE,OAAO,EAAE,cAAc;YACvC,MAAM,EAAE,OAAO,EAAE,MAAM;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAC;QAEH,uBAAuB;QACvB,MAAM,UAAU,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,EAAE,eAAe,EAAE,CAAC;QAC1E,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAEnC,iCAAiC;QACjC,cAAc,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC;QAE3E,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE;YACjC,MAAM,EAAE,cAAc,CAAC,MAAM;SAC9B,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,MAAM,CACjB,GAAW,EACX,IAAW,EACX,OAAwC;QAExC,MAAM,KAAK,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAE1C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CACb,0HAA0H,CAC3H,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,cAAc,CAChD,KAAK,EACL,MAAM,EACN,GAAG,EACH,IAAI,EACJ,OAAO,EAAE,cAAc,CACxB,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC;YACf,MAAM,IAAI,KAAK,CAAC,+BAA+B,MAAM,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC,CAAC;QACtF,CAAC;IACH,CAAC;IAEM,gBAAgB;QACrB,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,GAAG,CAAC,CAAC;IACrC,CAAC;IAED,sCAAsC;IAC/B,KAAK,CAAC,iBAAiB,CAAC,UAAkB,MAAM;QACrD,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAClC,OAAO;QACT,CAAC;QAED,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;QAE/E,qEAAqE;QACrE,MAAM,gBAAgB,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3C,MAAM,cAAc,GAAG,IAAI,OAAO,CAA0B,CAAC,OAAO,EAAE,EAAE,CACtE,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,OAAO,CAAC,CACrD,CAAC;QAEF,gEAAgE;QAChE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC;QAE3E,wBAAwB;QACxB,IAAI,MAAM,KAAK,gBAAgB,EAAE,CAAC;YAChC,8CAA8C;YAC9C,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC;YAC7C,KAAK,MAAM,UAAU,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBAC5C,UAAU,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;gBACnC,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACxC,CAAC;YAED,MAAM,IAAI,KAAK,CACb,+CAA+C,OAAO,eAAe,YAAY,oBAAoB,CACtG,CAAC;QACJ,CAAC;QAED,gFAAgF;QAChF,8DAA8D;IAChE,CAAC;CACF;AAhID,wEAgIC;AAED,SAAS,kBAAkB,CAAC,OAAwC;IAClE,IAAI,OAAO,EAAE,MAAM,EAAE,CAAC;QACpB,IAAI,OAAO,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,iCAAW,CAAC,GAAG,EAAE,GAAG,EAAE,eAAe,IAAI,iCAAW,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC;QAC3E,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC9B,OAAO,iCAAW,CAAC,GAAG,EAAE,GAAG,EAAE,aAAa,IAAI,iCAAW,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC;QACzE,CAAC;QAED,IAAI,OAAO,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC9B,OAAO,iCAAW,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC;QAClC,CAAC;QAED,OAAO,OAAO,CAAC,MAAM,CAAC;IACxB,CAAC;IAED,OAAO,iCAAW,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC;AAClC,CAAC;AAcD,SAAS,yBAAyB,CAChC,OAAe,EACf,OAA2C;IAE3C,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACrB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,WAAW,GAAG,OAAO,EAAE,CAAC,mBAAmB,CAAC,CAAC;IACnD,MAAM,KAAK,GAAG,OAAO,EAAE,CAAC,YAAY,CAAC,CAAC;IAEtC,IAAI,CAAC,WAAW,IAAI,CAAC,KAAK,EAAE,CAAC;QAC3B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,eAAe,GAAG,OAAO,EAAE,CAAC,wBAAwB,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,OAAO,EAAE,CAAC,kBAAkB,CAAC,CAAC;IAEjD,OAAO;QACL,OAAO,EAAE,IAAI;QACb,WAAW;QACX,KAAK;QACL,eAAe,EAAE,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,SAAS;QACxE,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS;KAC1D,CAAC;AACJ,CAAC"}

package/dist/commonjs/v3/realtimeStreams/noopManager.d.ts

@@ -0,0 +1,5 @@
+import { RealtimeStreamOperationOptions, RealtimeStreamInstance, RealtimeStreamsManager } from "./types.js";
+export declare class NoopRealtimeStreamsManager implements RealtimeStreamsManager {
+    pipe<T>(key: string, source: AsyncIterable<T> | ReadableStream<T>, options?: RealtimeStreamOperationOptions): RealtimeStreamInstance<T>;
+    append<TPart extends BodyInit>(key: string, part: TPart, options?: RealtimeStreamOperationOptions): Promise<void>;
+}

package/dist/commonjs/v3/realtimeStreams/noopManager.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NoopRealtimeStreamsManager = void 0;
+const asyncIterableStream_js_1 = require("../streams/asyncIterableStream.js");
+class NoopRealtimeStreamsManager {
+    pipe(key, source, options) {
+        return {
+            wait: () => Promise.resolve(),
+            get stream() {
+                return (0, asyncIterableStream_js_1.createAsyncIterableStreamFromAsyncIterable)(source);
+            },
+        };
+    }
+    async append(key, part, options) { }
+}
+exports.NoopRealtimeStreamsManager = NoopRealtimeStreamsManager;
+//# sourceMappingURL=noopManager.js.map

package/dist/commonjs/v3/realtimeStreams/noopManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"noopManager.js","sourceRoot":"","sources":["../../../../src/v3/realtimeStreams/noopManager.ts"],"names":[],"mappings":";;;AAAA,8EAG2C;AAO3C,MAAa,0BAA0B;IAC9B,IAAI,CACT,GAAW,EACX,MAA4C,EAC5C,OAAwC;QAExC,OAAO;YACL,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE;YAC7B,IAAI,MAAM;gBACR,OAAO,IAAA,mEAA0C,EAAC,MAAM,CAAC,CAAC;YAC5D,CAAC;SACF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,MAAM,CACjB,GAAW,EACX,IAAW,EACX,OAAwC,IACxB,CAAC;CACpB;AAnBD,gEAmBC"}

package/dist/commonjs/v3/realtimeStreams/streamInstance.d.ts

@@ -0,0 +1,23 @@
+import { ApiClient } from "../apiClient/index.js";
+import { AsyncIterableStream } from "../streams/asyncIterableStream.js";
+import { AnyZodFetchOptions } from "../zodfetch.js";
+import { StreamsWriter } from "./types.js";
+export type StreamInstanceOptions<T> = {
+    apiClient: ApiClient;
+    baseUrl: string;
+    runId: string;
+    key: string;
+    source: ReadableStream<T>;
+    signal?: AbortSignal;
+    requestOptions?: AnyZodFetchOptions;
+    target?: "self" | "parent" | "root" | string;
+    debug?: boolean;
+};
+export declare class StreamInstance<T> implements StreamsWriter {
+    private options;
+    private streamPromise;
+    constructor(options: StreamInstanceOptions<T>);
+    private initializeWriter;
+    wait(): Promise<void>;
+    get stream(): AsyncIterableStream<T>;
+}

package/dist/commonjs/v3/realtimeStreams/streamInstance.js

@@ -0,0 +1,106 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.StreamInstance = void 0;
+const streamsWriterV1_js_1 = require("./streamsWriterV1.js");
+const streamsWriterV2_js_1 = require("./streamsWriterV2.js");
+class StreamInstance {
+    options;
+    streamPromise;
+    constructor(options) {
+        this.options = options;
+        this.streamPromise = this.initializeWriter();
+    }
+    async initializeWriter() {
+        const { version, headers } = await this.options.apiClient.createStream(this.options.runId, "self", this.options.key, this.options?.requestOptions);
+        const parsedResponse = parseCreateStreamResponse(version, headers);
+        const streamWriter = parsedResponse.version === "v1"
+            ? new streamsWriterV1_js_1.StreamsWriterV1({
+                key: this.options.key,
+                runId: this.options.runId,
+                source: this.options.source,
+                baseUrl: this.options.baseUrl,
+                headers: this.options.apiClient.getHeaders(),
+                signal: this.options.signal,
+                version,
+                target: "self",
+            })
+            : new streamsWriterV2_js_1.StreamsWriterV2({
+                basin: parsedResponse.basin,
+                stream: parsedResponse.streamName ?? this.options.key,
+                accessToken: parsedResponse.accessToken,
+                source: this.options.source,
+                signal: this.options.signal,
+                debug: this.options.debug,
+                flushIntervalMs: parsedResponse.flushIntervalMs,
+                maxRetries: parsedResponse.maxRetries,
+            });
+        return streamWriter;
+    }
+    async wait() {
+        return this.streamPromise.then((writer) => writer.wait());
+    }
+    get stream() {
+        const self = this;
+        return new ReadableStream({
+            async start(controller) {
+                const streamWriter = await self.streamPromise;
+                const iterator = streamWriter[Symbol.asyncIterator]();
+                while (true) {
+                    if (self.options.signal?.aborted) {
+                        controller.close();
+                        break;
+                    }
+                    const { done, value } = await iterator.next();
+                    if (done) {
+                        controller.close();
+                        break;
+                    }
+                    controller.enqueue(value);
+                }
+            },
+        });
+    }
+}
+exports.StreamInstance = StreamInstance;
+function parseCreateStreamResponse(version, headers) {
+    if (version === "v1") {
+        return { version: "v1" };
+    }
+    const accessToken = headers?.["x-s2-access-token"];
+    const basin = headers?.["x-s2-basin"];
+    if (!accessToken || !basin) {
+        return { version: "v1" };
+    }
+    const flushIntervalMs = headers?.["x-s2-flush-interval-ms"];
+    const maxRetries = headers?.["x-s2-max-retries"];
+    const streamName = headers?.["x-s2-stream-name"];
+    return {
+        version: "v2",
+        accessToken,
+        basin,
+        flushIntervalMs: flushIntervalMs ? parseInt(flushIntervalMs) : undefined,
+        maxRetries: maxRetries ? parseInt(maxRetries) : undefined,
+        streamName,
+    };
+}
+async function* streamToAsyncIterator(stream) {
+    const reader = stream.getReader();
+    try {
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                return;
+            yield value;
+        }
+    }
+    finally {
+        safeReleaseLock(reader);
+    }
+}
+function safeReleaseLock(reader) {
+    try {
+        reader.releaseLock();
+    }
+    catch (error) { }
+}
+//# sourceMappingURL=streamInstance.js.map

package/dist/commonjs/v3/realtimeStreams/streamInstance.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streamInstance.js","sourceRoot":"","sources":["../../../../src/v3/realtimeStreams/streamInstance.ts"],"names":[],"mappings":";;;AAGA,6DAAuD;AACvD,6DAAuD;AAiBvD,MAAa,cAAc;IAGL;IAFZ,aAAa,CAAoC;IAEzD,YAAoB,OAAiC;QAAjC,YAAO,GAAP,OAAO,CAA0B;QACnD,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;IAC/C,CAAC;IAEO,KAAK,CAAC,gBAAgB;QAC5B,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,YAAY,CACpE,IAAI,CAAC,OAAO,CAAC,KAAK,EAClB,MAAM,EACN,IAAI,CAAC,OAAO,CAAC,GAAG,EAChB,IAAI,CAAC,OAAO,EAAE,cAAc,CAC7B,CAAC;QAEF,MAAM,cAAc,GAAG,yBAAyB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAEnE,MAAM,YAAY,GAChB,cAAc,CAAC,OAAO,KAAK,IAAI;YAC7B,CAAC,CAAC,IAAI,oCAAe,CAAC;gBAClB,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;gBACrB,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;gBACzB,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC3B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;gBAC7B,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,UAAU,EAAE;gBAC5C,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC3B,OAAO;gBACP,MAAM,EAAE,MAAM;aACf,CAAC;YACJ,CAAC,CAAC,IAAI,oCAAe,CAAC;gBAClB,KAAK,EAAE,cAAc,CAAC,KAAK;gBAC3B,MAAM,EAAE,cAAc,CAAC,UAAU,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG;gBACrD,WAAW,EAAE,cAAc,CAAC,WAAW;gBACvC,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC3B,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM;gBAC3B,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK;gBACzB,eAAe,EAAE,cAAc,CAAC,eAAe;gBAC/C,UAAU,EAAE,cAAc,CAAC,UAAU;aACtC,CAAC,CAAC;QAET,OAAO,YAAY,CAAC;IACtB,CAAC;IAEM,KAAK,CAAC,IAAI;QACf,OAAO,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC;IAED,IAAW,MAAM;QACf,MAAM,IAAI,GAAG,IAAI,CAAC;QAElB,OAAO,IAAI,cAAc,CAAI;YAC3B,KAAK,CAAC,KAAK,CAAC,UAAU;gBACpB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC;gBAE9C,MAAM,QAAQ,GAAG,YAAY,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;gBAEtD,OAAO,IAAI,EAAE,CAAC;oBACZ,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;wBACjC,UAAU,CAAC,KAAK,EAAE,CAAC;wBACnB,MAAM;oBACR,CAAC;oBAED,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;oBAE9C,IAAI,IAAI,EAAE,CAAC;wBACT,UAAU,CAAC,KAAK,EAAE,CAAC;wBACnB,MAAM;oBACR,CAAC;oBAED,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;gBAC5B,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC;CACF;AA1ED,wCA0EC;AAeD,SAAS,yBAAyB,CAChC,OAAe,EACf,OAA2C;IAE3C,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACrB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,WAAW,GAAG,OAAO,EAAE,CAAC,mBAAmB,CAAC,CAAC;IACnD,MAAM,KAAK,GAAG,OAAO,EAAE,CAAC,YAAY,CAAC,CAAC;IAEtC,IAAI,CAAC,WAAW,IAAI,CAAC,KAAK,EAAE,CAAC;QAC3B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC3B,CAAC;IAED,MAAM,eAAe,GAAG,OAAO,EAAE,CAAC,wBAAwB,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,OAAO,EAAE,CAAC,kBAAkB,CAAC,CAAC;IACjD,MAAM,UAAU,GAAG,OAAO,EAAE,CAAC,kBAAkB,CAAC,CAAC;IAEjD,OAAO;QACL,OAAO,EAAE,IAAI;QACb,WAAW;QACX,KAAK;QACL,eAAe,EAAE,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,SAAS;QACxE,UAAU,EAAE,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS;QACzD,UAAU;KACX,CAAC;AACJ,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,qBAAqB,CAAI,MAAyB;IAChE,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;IAClC,IAAI,CAAC;QACH,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,OAAO;YACjB,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;YAAS,CAAC;QACT,eAAe,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CAAC,MAAwC;IAC/D,IAAI,CAAC;QACH,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC,CAAA,CAAC;AACpB,CAAC"}

package/dist/commonjs/v3/realtimeStreams/streamsWriterV1.d.ts

@@ -0,0 +1,49 @@
+import { StreamsWriter } from "./types.js";
+export type StreamsWriterV1Options<T> = {
+    baseUrl: string;
+    runId: string;
+    key: string;
+    source: ReadableStream<T>;
+    headers?: Record<string, string>;
+    signal?: AbortSignal;
+    version?: string;
+    target?: "self" | "parent" | "root";
+    maxRetries?: number;
+    maxBufferSize?: number;
+    clientId?: string;
+};
+export declare class StreamsWriterV1<T> implements StreamsWriter {
+    private options;
+    private controller;
+    private serverStream;
+    private consumerStream;
+    private streamPromise;
+    private retryCount;
+    private readonly maxRetries;
+    private currentChunkIndex;
+    private readonly baseDelayMs;
+    private readonly maxDelayMs;
+    private readonly maxBufferSize;
+    private readonly clientId;
+    private ringBuffer;
+    private bufferStartIndex;
+    private highestBufferedIndex;
+    private streamReader;
+    private bufferReaderTask;
+    private streamComplete;
+    constructor(options: StreamsWriterV1Options<T>);
+    private generateClientId;
+    private startBuffering;
+    private makeRequest;
+    private initializeServerStream;
+    wait(): Promise<void>;
+    [Symbol.asyncIterator](): AsyncIterableIterator<T>;
+    private buildUrl;
+    private isRetryableError;
+    private isRetryableStatusCode;
+    private delay;
+    private calculateBackoffDelay;
+    private addToRingBuffer;
+    private getChunksFromBuffer;
+    private queryServerLastChunkIndex;
+}