@trigger.dev/core 0.0.0-v3-prerelease-20250108141813 → 0.0.0-v4-prerelease-20250916125920

package/dist/commonjs/v3/serverOnly/shutdownManager.test.js

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const vitest_1 = require("vitest");
+const shutdownManager_js_1 = require("./shutdownManager.js");
+(0, vitest_1.describe)("ShutdownManager", { concurrent: false }, () => {
+    // Mock process.exit to prevent actual exit
+    const mockExit = vitest_1.vi.spyOn(process, "exit").mockImplementation(() => undefined);
+    (0, vitest_1.beforeEach)(() => {
+        vitest_1.vi.clearAllMocks();
+    });
+    (0, vitest_1.test)("should successfully register a new handler", () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const handler = vitest_1.vi.fn();
+        manager.register("test-handler", handler);
+        (0, vitest_1.expect)(manager._getHandlersForTesting().has("test-handler")).toBe(true);
+        const registeredHandler = manager._getHandlersForTesting().get("test-handler");
+        (0, vitest_1.expect)(registeredHandler?.handler).toBe(handler);
+        (0, vitest_1.expect)(registeredHandler?.signals).toEqual(["SIGTERM", "SIGINT"]);
+    });
+    (0, vitest_1.test)("should throw error when registering duplicate handler name", () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const handler = vitest_1.vi.fn();
+        manager.register("duplicate-handler", handler);
+        (0, vitest_1.expect)(() => {
+            manager.register("duplicate-handler", handler);
+        }).toThrow('Shutdown handler "duplicate-handler" already registered');
+    });
+    (0, vitest_1.test)("should register handler with custom signals", () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const handler = vitest_1.vi.fn();
+        manager.register("custom-signals", handler, ["SIGTERM"]);
+        const registeredHandler = manager._getHandlersForTesting().get("custom-signals");
+        (0, vitest_1.expect)(registeredHandler?.signals).toEqual(["SIGTERM"]);
+    });
+    (0, vitest_1.test)("should call registered handlers when shutdown is triggered", async () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const handler1 = vitest_1.vi.fn();
+        const handler2 = vitest_1.vi.fn();
+        manager.register("handler1", handler1);
+        manager.register("handler2", handler2);
+        await manager.shutdown("SIGTERM");
+        (0, vitest_1.expect)(handler1).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(handler2).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledWith(128 + 15); // SIGTERM number
+    });
+    (0, vitest_1.test)("should only call handlers registered for specific signal", async () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const handler1 = vitest_1.vi.fn();
+        const handler2 = vitest_1.vi.fn();
+        manager.register("handler1", handler1, ["SIGTERM"]);
+        manager.register("handler2", handler2, ["SIGINT"]);
+        await manager.shutdown("SIGTERM");
+        (0, vitest_1.expect)(handler1).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(handler2).not.toHaveBeenCalled();
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledWith(128 + 15);
+    });
+    (0, vitest_1.test)("should handle errors in shutdown handlers gracefully", async () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const successHandler = vitest_1.vi.fn();
+        const errorHandler = vitest_1.vi.fn().mockRejectedValue(new Error("Handler failed"));
+        manager.register("success-handler", successHandler);
+        manager.register("error-handler", errorHandler);
+        await manager.shutdown("SIGTERM");
+        (0, vitest_1.expect)(successHandler).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(errorHandler).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledWith(128 + 15);
+    });
+    (0, vitest_1.test)("should only run shutdown sequence once even if called multiple times", async () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const handler = vitest_1.vi.fn();
+        manager.register("test-handler", handler);
+        await Promise.all([manager.shutdown("SIGTERM"), manager.shutdown("SIGTERM")]);
+        (0, vitest_1.expect)(handler).toHaveBeenCalledTimes(1);
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledTimes(1);
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledWith(128 + 15);
+    });
+    (0, vitest_1.test)("should exit with correct signal number on SIGINT", async () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        manager.register("test-handler", vitest_1.vi.fn());
+        await manager.shutdown("SIGINT");
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledWith(128 + 2); // SIGINT number
+    });
+    (0, vitest_1.test)("should exit with correct signal number on SIGTERM", async () => {
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        manager.register("test-handler", vitest_1.vi.fn());
+        await manager.shutdown("SIGTERM");
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledWith(128 + 15); // SIGTERM number
+    });
+    (0, vitest_1.test)("should only exit after all handlers have finished", async () => {
+        const sequence = [];
+        const manager = new shutdownManager_js_1.ShutdownManager(false);
+        const handler1 = vitest_1.vi.fn().mockImplementation(async () => {
+            sequence.push("handler1 start");
+            await new Promise((resolve) => setTimeout(resolve, 10));
+            sequence.push("handler1 end");
+        });
+        const handler2 = vitest_1.vi.fn().mockImplementation(async () => {
+            sequence.push("handler2 start");
+            await new Promise((resolve) => setTimeout(resolve, 20));
+            sequence.push("handler2 end");
+        });
+        const handler3 = vitest_1.vi.fn().mockImplementation(async () => {
+            sequence.push("handler3 start");
+            await new Promise((resolve) => setTimeout(resolve, 5));
+            sequence.push("handler3 end");
+        });
+        // Store the current mock implementation
+        const currentExit = mockExit.getMockImplementation();
+        // Override with our sequence-tracking implementation
+        mockExit.mockImplementation((code) => {
+            sequence.push("exit");
+            return undefined;
+        });
+        manager.register("handler1", handler1);
+        manager.register("handler2", handler2);
+        manager.register("handler3", handler3);
+        await manager.shutdown("SIGTERM");
+        // Verify the execution order
+        (0, vitest_1.expect)(sequence).toEqual([
+            "handler1 start",
+            "handler2 start",
+            "handler3 start",
+            "handler3 end",
+            "handler1 end",
+            "handler2 end",
+            "exit",
+        ]);
+        // Verify the handlers were called with correct arguments
+        (0, vitest_1.expect)(handler1).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(handler2).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(handler3).toHaveBeenCalledWith("SIGTERM");
+        (0, vitest_1.expect)(mockExit).toHaveBeenCalledWith(128 + 15);
+        // Restore original mock implementation
+        if (currentExit) {
+            mockExit.mockImplementation(currentExit);
+        }
+    });
+});
+//# sourceMappingURL=shutdownManager.test.js.map

package/dist/commonjs/v3/serverOnly/shutdownManager.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shutdownManager.test.js","sourceRoot":"","sources":["../../../../src/v3/serverOnly/shutdownManager.test.ts"],"names":[],"mappings":";;AAAA,mCAAgE;AAChE,6DAAuD;AAEvD,IAAA,iBAAQ,EAAC,iBAAiB,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE;IACtD,2CAA2C;IAC3C,MAAM,QAAQ,GAAG,WAAE,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,SAAkB,CAAC,CAAC;IAExF,IAAA,mBAAU,EAAC,GAAG,EAAE;QACd,WAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,4CAA4C,EAAE,GAAG,EAAE;QACtD,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAE1C,IAAA,eAAM,EAAC,OAAO,CAAC,sBAAsB,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxE,MAAM,iBAAiB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;QAC/E,IAAA,eAAM,EAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,4DAA4D,EAAE,GAAG,EAAE;QACtE,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QAE/C,IAAA,eAAM,EAAC,GAAG,EAAE;YACV,OAAO,CAAC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC,OAAO,CAAC,yDAAyD,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,6CAA6C,EAAE,GAAG,EAAE;QACvD,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,gBAAgB,EAAE,OAAO,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QAEzD,MAAM,iBAAiB,GAAG,OAAO,CAAC,sBAAsB,EAAE,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QACjF,IAAA,eAAM,EAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QAEzB,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEvC,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,iBAAiB;IACpE,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QAC1E,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QAEzB,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;QACpD,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;QAEnD,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACxC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,cAAc,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QAC/B,MAAM,YAAY,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC,CAAC;QAE5E,OAAO,CAAC,QAAQ,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC;QACpD,OAAO,CAAC,QAAQ,CAAC,eAAe,EAAE,YAAY,CAAC,CAAC;QAEhD,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,IAAA,eAAM,EAAC,cAAc,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACvD,IAAA,eAAM,EAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACrD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACtF,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,OAAO,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC;QACxB,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QAE1C,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAE9E,IAAA,eAAM,EAAC,OAAO,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QACzC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC1C,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAE1C,MAAM,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACjC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,gBAAgB;IAClE,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,OAAO,CAAC,QAAQ,CAAC,cAAc,EAAE,WAAE,CAAC,EAAE,EAAE,CAAC,CAAC;QAE1C,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAClC,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC,CAAC,iBAAiB;IACpE,CAAC,CAAC,CAAC;IAEH,IAAA,aAAI,EAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,IAAI,oCAAe,CAAC,KAAK,CAAC,CAAC;QAE3C,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;YACxD,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,WAAE,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,KAAK,IAAI,EAAE;YACrD,QAAQ,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YAChC,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YACvD,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QAChC,CAAC,CAAC,CAAC;QAEH,wCAAwC;QACxC,MAAM,WAAW,GAAG,QAAQ,CAAC,qBAAqB,EAAE,CAAC;QAErD,qDAAqD;QACrD,QAAQ,CAAC,kBAAkB,CAAC,CAAC,IAA6B,EAAE,EAAE;YAC5D,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,OAAO,SAAkB,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QAEvC,MAAM,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAElC,6BAA6B;QAC7B,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,OAAO,CAAC;YACvB,gBAAgB;YAChB,gBAAgB;YAChB,gBAAgB;YAChB,cAAc;YACd,cAAc;YACd,cAAc;YACd,MAAM;SACP,CAAC,CAAC;QAEH,yDAAyD;QACzD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACjD,IAAA,eAAM,EAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,GAAG,GAAG,EAAE,CAAC,CAAC;QAEhD,uCAAuC;QACvC,IAAI,WAAW,EAAE,CAAC;YAChB,QAAQ,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/commonjs/v3/serverOnly/singleton.d.ts

@@ -0,0 +1 @@
+export declare function singleton<T>(name: string, getValue: () => T): T;

package/dist/commonjs/v3/serverOnly/singleton.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.singleton = singleton;
+function singleton(name, getValue) {
+    const thusly = globalThis;
+    thusly.__trigger_singletons ??= {};
+    thusly.__trigger_singletons[name] ??= getValue();
+    return thusly.__trigger_singletons[name];
+}
+//# sourceMappingURL=singleton.js.map

package/dist/commonjs/v3/serverOnly/singleton.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"singleton.js","sourceRoot":"","sources":["../../../../src/v3/serverOnly/singleton.ts"],"names":[],"mappings":";;AAAA,8BAOC;AAPD,SAAgB,SAAS,CAAI,IAAY,EAAE,QAAiB;IAC1D,MAAM,MAAM,GAAG,UAEd,CAAC;IACF,MAAM,CAAC,oBAAoB,KAAK,EAAE,CAAC;IACnC,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,KAAK,QAAQ,EAAE,CAAC;IACjD,OAAO,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,CAAC;AAC3C,CAAC"}

package/dist/commonjs/v3/streams/asyncIterableStream.d.ts

@@ -0,0 +1,5 @@
+export type AsyncIterableStream<T> = AsyncIterable<T> & ReadableStream<T>;
+export declare function createAsyncIterableStream<S, T>(source: ReadableStream<S>, transformer: Transformer<S, T>): AsyncIterableStream<T>;
+export declare function createAsyncIterableReadable<S, T>(source: ReadableStream<S>, transformer: Transformer<S, T>, signal: AbortSignal): AsyncIterableStream<T>;
+export declare function createAsyncIterableStreamFromAsyncIterable<T>(asyncIterable: AsyncIterable<T>, transformer?: Transformer<T, T>, signal?: AbortSignal): AsyncIterableStream<T>;
+export declare function createAsyncIterableStreamFromAsyncGenerator<T>(asyncGenerator: AsyncGenerator<T, void, unknown>, transformer: Transformer<T, T>, signal?: AbortSignal): AsyncIterableStream<T>;

package/dist/commonjs/v3/streams/asyncIterableStream.js

@@ -0,0 +1,81 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createAsyncIterableStream = createAsyncIterableStream;
+exports.createAsyncIterableReadable = createAsyncIterableReadable;
+exports.createAsyncIterableStreamFromAsyncIterable = createAsyncIterableStreamFromAsyncIterable;
+exports.createAsyncIterableStreamFromAsyncGenerator = createAsyncIterableStreamFromAsyncGenerator;
+function createAsyncIterableStream(source, transformer) {
+    const transformedStream = source.pipeThrough(new TransformStream(transformer));
+    transformedStream[Symbol.asyncIterator] = () => {
+        const reader = transformedStream.getReader();
+        return {
+            async next() {
+                const { done, value } = await reader.read();
+                return done ? { done: true, value: undefined } : { done: false, value };
+            },
+        };
+    };
+    return transformedStream;
+}
+function createAsyncIterableReadable(source, transformer, signal) {
+    return new ReadableStream({
+        async start(controller) {
+            const transformedStream = source.pipeThrough(new TransformStream(transformer));
+            const reader = transformedStream.getReader();
+            signal.addEventListener("abort", () => {
+                queueMicrotask(() => {
+                    reader.cancel();
+                    controller.close();
+                });
+            });
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done) {
+                    controller.close();
+                    break;
+                }
+                controller.enqueue(value);
+            }
+        },
+    });
+}
+function createAsyncIterableStreamFromAsyncIterable(asyncIterable, transformer, signal) {
+    const stream = new ReadableStream({
+        async start(controller) {
+            try {
+                if (signal) {
+                    signal.addEventListener("abort", () => {
+                        controller.close();
+                    });
+                }
+                const iterator = asyncIterable[Symbol.asyncIterator]();
+                while (true) {
+                    if (signal?.aborted) {
+                        break;
+                    }
+                    const { done, value } = await iterator.next();
+                    if (done) {
+                        controller.close();
+                        break;
+                    }
+                    controller.enqueue(value);
+                }
+            }
+            catch (error) {
+                controller.error(error);
+            }
+        },
+        cancel() {
+            // If the stream is a tinyexec process with a kill method, kill it
+            if ("kill" in asyncIterable) {
+                asyncIterable.kill();
+            }
+        },
+    });
+    const transformedStream = stream.pipeThrough(new TransformStream(transformer));
+    return transformedStream;
+}
+function createAsyncIterableStreamFromAsyncGenerator(asyncGenerator, transformer, signal) {
+    return createAsyncIterableStreamFromAsyncIterable(asyncGenerator, transformer, signal);
+}
+//# sourceMappingURL=asyncIterableStream.js.map

package/dist/commonjs/v3/streams/asyncIterableStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"asyncIterableStream.js","sourceRoot":"","sources":["../../../../src/v3/streams/asyncIterableStream.ts"],"names":[],"mappings":";;AAEA,8DAiBC;AAED,kEA4BC;AAED,gGA6CC;AAED,kGAMC;AAtGD,SAAgB,yBAAyB,CACvC,MAAyB,EACzB,WAA8B;IAE9B,MAAM,iBAAiB,GAAQ,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;IAEpF,iBAAiB,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,GAAG,EAAE;QAC7C,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,CAAC;QAC7C,OAAO;YACL,KAAK,CAAC,IAAI;gBACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,OAAO,IAAI,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;YAC1E,CAAC;SACF,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,iBAAiB,CAAC;AAC3B,CAAC;AAED,SAAgB,2BAA2B,CACzC,MAAyB,EACzB,WAA8B,EAC9B,MAAmB;IAEnB,OAAO,IAAI,cAAc,CAAI;QAC3B,KAAK,CAAC,KAAK,CAAC,UAAU;YACpB,MAAM,iBAAiB,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;YAC/E,MAAM,MAAM,GAAG,iBAAiB,CAAC,SAAS,EAAE,CAAC;YAE7C,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;gBACpC,cAAc,CAAC,GAAG,EAAE;oBAClB,MAAM,CAAC,MAAM,EAAE,CAAC;oBAChB,UAAU,CAAC,KAAK,EAAE,CAAC;gBACrB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI,EAAE,CAAC;oBACT,UAAU,CAAC,KAAK,EAAE,CAAC;oBACnB,MAAM;gBACR,CAAC;gBAED,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC;KACF,CAA2B,CAAC;AAC/B,CAAC;AAED,SAAgB,0CAA0C,CACxD,aAA+B,EAC/B,WAA+B,EAC/B,MAAoB;IAEpB,MAAM,MAAM,GAAG,IAAI,cAAc,CAAI;QACnC,KAAK,CAAC,KAAK,CAAC,UAAU;YACpB,IAAI,CAAC;gBACH,IAAI,MAAM,EAAE,CAAC;oBACX,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE;wBACpC,UAAU,CAAC,KAAK,EAAE,CAAC;oBACrB,CAAC,CAAC,CAAC;gBACL,CAAC;gBAED,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;gBAEvD,OAAO,IAAI,EAAE,CAAC;oBACZ,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;wBACpB,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;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,UAAU,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;QACD,MAAM;YACJ,kEAAkE;YAClE,IAAI,MAAM,IAAI,aAAa,EAAE,CAAC;gBAC3B,aAAqB,CAAC,IAAI,EAAE,CAAC;YAChC,CAAC;QACH,CAAC;KACF,CAAC,CAAC;IAEH,MAAM,iBAAiB,GAAG,MAAM,CAAC,WAAW,CAAC,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;IAE/E,OAAO,iBAA2C,CAAC;AACrD,CAAC;AAED,SAAgB,2CAA2C,CACzD,cAAgD,EAChD,WAA8B,EAC9B,MAAoB;IAEpB,OAAO,0CAA0C,CAAC,cAAc,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC;AACzF,CAAC"}

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

@@ -9,7 +9,9 @@ export declare class TaskContextAPI {
     get isInsideTask(): boolean;
     get ctx(): TaskRunContext | undefined;
     get worker(): ServerBackgroundWorker | undefined;
+    get isWarmStart(): boolean | undefined;
     get attributes(): Attributes;
+    get resourceAttributes(): Attributes;
     get workerAttributes(): Attributes;
     get contextAttributes(): Attributes;
     disable(): void;

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

@@ -1,8 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TaskContextAPI = void 0;
-const globals_js_1 = require("../utils/globals.js");
 const semanticInternalAttributes_js_1 = require("../semanticInternalAttributes.js");
+const globals_js_1 = require("../utils/globals.js");
 const API_NAME = "task-context";
 class TaskContextAPI {
     static _instance;
@@ -22,11 +22,34 @@ class TaskContextAPI {
     get worker() {
         return this.#getTaskContext()?.worker;
     }
+    get isWarmStart() {
+        return this.#getTaskContext()?.isWarmStart;
+    }
     get attributes() {
         if (this.ctx) {
             return {
                 ...this.contextAttributes,
                 ...this.workerAttributes,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.WARM_START]: !!this.isWarmStart,
+            };
+        }
+        return {};
+    }
+    get resourceAttributes() {
+        if (this.ctx) {
+            return {
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ENVIRONMENT_ID]: this.ctx.environment.id,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ENVIRONMENT_TYPE]: this.ctx.environment.type,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ORGANIZATION_ID]: this.ctx.organization.id,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.PROJECT_ID]: this.ctx.project.id,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.PROJECT_REF]: this.ctx.project.ref,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.PROJECT_NAME]: this.ctx.project.name,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ORGANIZATION_SLUG]: this.ctx.organization.slug,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ORGANIZATION_NAME]: this.ctx.organization.name,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_NAME]: this.ctx.machine?.name,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_CPU]: this.ctx.machine?.cpu,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_MEMORY]: this.ctx.machine?.memory,
+                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_CENTS_PER_MS]: this.ctx.machine?.centsPerMs,
             };
         }
         return {};
@@ -43,29 +66,15 @@ class TaskContextAPI {
     get contextAttributes() {
         if (this.ctx) {
             return {
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ATTEMPT_ID]: this.ctx.attempt.id,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.ATTEMPT_NUMBER]: this.ctx.attempt.number,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.TASK_SLUG]: this.ctx.task.id,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.TASK_PATH]: this.ctx.task.filePath,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.TASK_EXPORT_NAME]: this.ctx.task.exportName,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.QUEUE_NAME]: this.ctx.queue.name,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.QUEUE_ID]: this.ctx.queue.id,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ENVIRONMENT_ID]: this.ctx.environment.id,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ENVIRONMENT_TYPE]: this.ctx.environment.type,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ORGANIZATION_ID]: this.ctx.organization.id,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.PROJECT_ID]: this.ctx.project.id,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.PROJECT_REF]: this.ctx.project.ref,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.PROJECT_NAME]: this.ctx.project.name,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.RUN_ID]: this.ctx.run.id,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.RUN_IS_TEST]: this.ctx.run.isTest,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ORGANIZATION_SLUG]: this.ctx.organization.slug,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ORGANIZATION_NAME]: this.ctx.organization.name,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.BATCH_ID]: this.ctx.batch?.id,
                 [semanticInternalAttributes_js_1.SemanticInternalAttributes.IDEMPOTENCY_KEY]: this.ctx.run.idempotencyKey,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_NAME]: this.ctx.machine?.name,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_CPU]: this.ctx.machine?.cpu,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_MEMORY]: this.ctx.machine?.memory,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.MACHINE_PRESET_CENTS_PER_MS]: this.ctx.machine?.centsPerMs,
             };
         }
         return {};

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

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/index.ts"],"names":[],"mappings":";;;AAEA,oDAAkF;AAElF,oFAA8E;AAE9E,MAAM,QAAQ,GAAG,cAAc,CAAC;AAEhC,MAAa,cAAc;IACjB,MAAM,CAAC,SAAS,CAAkB;IAE1C,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,eAAe,EAAE,KAAK,SAAS,CAAC;IAC9C,CAAC;IAED,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,CAAC;IACrC,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,MAAM,CAAC;IACxC,CAAC;IAED,IAAI,UAAU;QACZ,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,GAAG,IAAI,CAAC,iBAAiB;gBACzB,GAAG,IAAI,CAAC,gBAAgB;aACzB,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,gBAAgB;QAClB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;gBACL,CAAC,0DAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;gBACtD,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;aACjE,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,iBAAiB;QACnB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBAC5D,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;gBACpE,CAAC,0DAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBACxD,CAAC,0DAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ;gBAC9D,CAAC,0DAA0B,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU;gBACvE,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI;gBAC5D,CAAC,0DAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBACxD,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;gBACpE,CAAC,0DAA0B,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI;gBACxE,CAAC,0DAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE;gBACtE,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBAC5D,CAAC,0DAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG;gBAC9D,CAAC,0DAA0B,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI;gBAChE,CAAC,0DAA0B,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBACpD,CAAC,0DAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM;gBAC7D,CAAC,0DAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0DAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0DAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE;gBACzD,CAAC,0DAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc;gBACzE,CAAC,0DAA0B,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI;gBACxE,CAAC,0DAA0B,CAAC,kBAAkB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG;gBACtE,CAAC,0DAA0B,CAAC,qBAAqB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM;gBAC5E,CAAC,0DAA0B,CAAC,2BAA2B,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU;aACvF,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,OAAO;QACZ,IAAA,6BAAgB,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,oBAAoB,CAAC,WAAwB;QAClD,OAAO,IAAA,2BAAc,EAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC/C,CAAC;IAED,eAAe;QACb,OAAO,IAAA,sBAAS,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;CACF;AA1FD,wCA0FC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/index.ts"],"names":[],"mappings":";;;AAEA,oFAA8E;AAC9E,oDAAkF;AAGlF,MAAM,QAAQ,GAAG,cAAc,CAAC;AAEhC,MAAa,cAAc;IACjB,MAAM,CAAC,SAAS,CAAkB;IAE1C,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,eAAe,EAAE,KAAK,SAAS,CAAC;IAC9C,CAAC;IAED,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,GAAG,CAAC;IACrC,CAAC;IAED,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,MAAM,CAAC;IACxC,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,eAAe,EAAE,EAAE,WAAW,CAAC;IAC7C,CAAC;IAED,IAAI,UAAU;QACZ,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,GAAG,IAAI,CAAC,iBAAiB;gBACzB,GAAG,IAAI,CAAC,gBAAgB;gBACxB,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW;aAC5D,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,kBAAkB;QACpB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;gBACpE,CAAC,0DAA0B,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI;gBACxE,CAAC,0DAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,EAAE;gBACtE,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBAC5D,CAAC,0DAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG;gBAC9D,CAAC,0DAA0B,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI;gBAChE,CAAC,0DAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0DAA0B,CAAC,iBAAiB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI;gBAC1E,CAAC,0DAA0B,CAAC,mBAAmB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,IAAI;gBACxE,CAAC,0DAA0B,CAAC,kBAAkB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG;gBACtE,CAAC,0DAA0B,CAAC,qBAAqB,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM;gBAC5E,CAAC,0DAA0B,CAAC,2BAA2B,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,UAAU;aACvF,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,gBAAgB;QAClB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO;gBACL,CAAC,0DAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE;gBACtD,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO;aACjE,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,IAAI,iBAAiB;QACnB,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO;gBACL,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;gBACpE,CAAC,0DAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBACxD,CAAC,0DAA0B,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ;gBAC9D,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI;gBAC5D,CAAC,0DAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBACxD,CAAC,0DAA0B,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBACpD,CAAC,0DAA0B,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM;gBAC7D,CAAC,0DAA0B,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE;gBACzD,CAAC,0DAA0B,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,cAAc;aAC1E,CAAC;QACJ,CAAC;QAED,OAAO,EAAE,CAAC;IACZ,CAAC;IAEM,OAAO;QACZ,IAAA,6BAAgB,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,oBAAoB,CAAC,WAAwB;QAClD,OAAO,IAAA,2BAAc,EAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC/C,CAAC;IAED,eAAe;QACb,OAAO,IAAA,sBAAS,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;CACF;AAtGD,wCAsGC"}

package/dist/commonjs/v3/taskContext/otelProcessors.d.ts

@@ -1,9 +1,10 @@
-import { LogRecord, LogRecordProcessor } from "@opentelemetry/sdk-logs";
-import { Span, SpanProcessor } from "@opentelemetry/sdk-trace-base";
 import { Context } from "@opentelemetry/api";
+import { LogRecordProcessor, SdkLogRecord } from "@opentelemetry/sdk-logs";
+import { Span, SpanProcessor } from "@opentelemetry/sdk-trace-base";
 export declare class TaskContextSpanProcessor implements SpanProcessor {
     private _innerProcessor;
-    constructor(innerProcessor: SpanProcessor);
+    private _tracer;
+    constructor(version: string, innerProcessor: SpanProcessor);
     onStart(span: Span, parentContext: Context): void;
     onEnd(span: Span): void;
     shutdown(): Promise<void>;
@@ -13,6 +14,6 @@ export declare class TaskContextLogProcessor implements LogRecordProcessor {
     private _innerProcessor;
     constructor(innerProcessor: LogRecordProcessor);
     forceFlush(): Promise<void>;
-    onEmit(logRecord: LogRecord, context?: Context | undefined): void;
+    onEmit(logRecord: SdkLogRecord, context?: Context | undefined): void;
     shutdown(): Promise<void>;
 }

package/dist/commonjs/v3/taskContext/otelProcessors.js

@@ -1,21 +1,25 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.TaskContextLogProcessor = exports.TaskContextSpanProcessor = void 0;
+const api_1 = require("@opentelemetry/api");
 const semanticInternalAttributes_js_1 = require("../semanticInternalAttributes.js");
-const flattenAttributes_js_1 = require("../utils/flattenAttributes.js");
 const task_context_api_js_1 = require("../task-context-api.js");
+const flattenAttributes_js_1 = require("../utils/flattenAttributes.js");
 class TaskContextSpanProcessor {
     _innerProcessor;
-    constructor(innerProcessor) {
+    _tracer;
+    constructor(version, innerProcessor) {
+        this._tracer = api_1.trace.getTracer("trigger-dev-worker", version);
         this._innerProcessor = innerProcessor;
     }
     // Called when a span starts
     onStart(span, parentContext) {
         if (task_context_api_js_1.taskContext.ctx) {
-            span.setAttributes((0, flattenAttributes_js_1.flattenAttributes)({
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ATTEMPT_ID]: task_context_api_js_1.taskContext.ctx.attempt.id,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ATTEMPT_NUMBER]: task_context_api_js_1.taskContext.ctx.attempt.number,
-            }, semanticInternalAttributes_js_1.SemanticInternalAttributes.METADATA));
+            span.setAttributes((0, flattenAttributes_js_1.flattenAttributes)(task_context_api_js_1.taskContext.attributes, semanticInternalAttributes_js_1.SemanticInternalAttributes.METADATA));
+        }
+        if (!isPartialSpan(span) && !skipPartialSpan(span)) {
+            const partialSpan = createPartialSpan(this._tracer, span, parentContext);
+            partialSpan.end();
         }
         this._innerProcessor.onStart(span, parentContext);
     }
@@ -31,6 +35,30 @@ class TaskContextSpanProcessor {
     }
 }
 exports.TaskContextSpanProcessor = TaskContextSpanProcessor;
+function isPartialSpan(span) {
+    return span.attributes[semanticInternalAttributes_js_1.SemanticInternalAttributes.SPAN_PARTIAL] === true;
+}
+function skipPartialSpan(span) {
+    return span.attributes[semanticInternalAttributes_js_1.SemanticInternalAttributes.SKIP_SPAN_PARTIAL] === true;
+}
+function createPartialSpan(tracer, span, parentContext) {
+    const partialSpan = tracer.startSpan(span.name, {
+        attributes: {
+            [semanticInternalAttributes_js_1.SemanticInternalAttributes.SPAN_PARTIAL]: true,
+            [semanticInternalAttributes_js_1.SemanticInternalAttributes.SPAN_ID]: span.spanContext().spanId,
+            ...span.attributes,
+        },
+    }, parentContext);
+    if (task_context_api_js_1.taskContext.ctx) {
+        partialSpan.setAttributes((0, flattenAttributes_js_1.flattenAttributes)(task_context_api_js_1.taskContext.attributes, semanticInternalAttributes_js_1.SemanticInternalAttributes.METADATA));
+    }
+    if (span.events) {
+        for (const event of span.events) {
+            partialSpan.addEvent(event.name, event.attributes, event.time);
+        }
+    }
+    return partialSpan;
+}
 class TaskContextLogProcessor {
     _innerProcessor;
     constructor(innerProcessor) {
@@ -42,10 +70,7 @@ class TaskContextLogProcessor {
     onEmit(logRecord, context) {
         // Adds in the context attributes to the log record
         if (task_context_api_js_1.taskContext.ctx) {
-            logRecord.setAttributes((0, flattenAttributes_js_1.flattenAttributes)({
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ATTEMPT_ID]: task_context_api_js_1.taskContext.ctx.attempt.id,
-                [semanticInternalAttributes_js_1.SemanticInternalAttributes.ATTEMPT_NUMBER]: task_context_api_js_1.taskContext.ctx.attempt.number,
-            }, semanticInternalAttributes_js_1.SemanticInternalAttributes.METADATA));
+            logRecord.setAttributes((0, flattenAttributes_js_1.flattenAttributes)(task_context_api_js_1.taskContext.attributes, semanticInternalAttributes_js_1.SemanticInternalAttributes.METADATA));
         }
         this._innerProcessor.onEmit(logRecord, context);
     }

package/dist/commonjs/v3/taskContext/otelProcessors.js.map

@@ -1 +1 @@
-{"version":3,"file":"otelProcessors.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/otelProcessors.ts"],"names":[],"mappings":";;;AAEA,oFAA8E;AAE9E,wEAAkE;AAClE,gEAAqD;AAErD,MAAa,wBAAwB;IAC3B,eAAe,CAAgB;IAEvC,YAAY,cAA6B;QACvC,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IAED,4BAA4B;IAC5B,OAAO,CAAC,IAAU,EAAE,aAAsB;QACxC,IAAI,iCAAW,CAAC,GAAG,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,CAChB,IAAA,wCAAiB,EACf;gBACE,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,iCAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnE,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,iCAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;aAC5E,EACD,0DAA0B,CAAC,QAAQ,CACpC,CACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACpD,CAAC;IAED,4DAA4D;IAE5D,KAAK,CAAC,IAAU;QACd,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;CACF;AArCD,4DAqCC;AAED,MAAa,uBAAuB;IAC1B,eAAe,CAAqB;IAE5C,YAAY,cAAkC;QAC5C,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IACD,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;IACD,MAAM,CAAC,SAAoB,EAAE,OAA6B;QACxD,mDAAmD;QACnD,IAAI,iCAAW,CAAC,GAAG,EAAE,CAAC;YACpB,SAAS,CAAC,aAAa,CACrB,IAAA,wCAAiB,EACf;gBACE,CAAC,0DAA0B,CAAC,UAAU,CAAC,EAAE,iCAAW,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE;gBACnE,CAAC,0DAA0B,CAAC,cAAc,CAAC,EAAE,iCAAW,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM;aAC5E,EACD,0DAA0B,CAAC,QAAQ,CACpC,CACF,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IACD,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;CACF;AA5BD,0DA4BC"}
+{"version":3,"file":"otelProcessors.js","sourceRoot":"","sources":["../../../../src/v3/taskContext/otelProcessors.ts"],"names":[],"mappings":";;;AAAA,4CAA4D;AAG5D,oFAA8E;AAC9E,gEAAqD;AACrD,wEAAkE;AAElE,MAAa,wBAAwB;IAC3B,eAAe,CAAgB;IAC/B,OAAO,CAAS;IAExB,YAAY,OAAe,EAAE,cAA6B;QACxD,IAAI,CAAC,OAAO,GAAG,WAAK,CAAC,SAAS,CAAC,oBAAoB,EAAE,OAAO,CAAC,CAAC;QAC9D,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IAED,4BAA4B;IAC5B,OAAO,CAAC,IAAU,EAAE,aAAsB;QACxC,IAAI,iCAAW,CAAC,GAAG,EAAE,CAAC;YACpB,IAAI,CAAC,aAAa,CAChB,IAAA,wCAAiB,EAAC,iCAAW,CAAC,UAAU,EAAE,0DAA0B,CAAC,QAAQ,CAAC,CAC/E,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,CAAC;YACnD,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,EAAE,aAAa,CAAC,CAAC;YACzE,WAAW,CAAC,GAAG,EAAE,CAAC;QACpB,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC;IACpD,CAAC;IAED,4DAA4D;IAE5D,KAAK,CAAC,IAAU;QACd,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACnC,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;IAED,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;CACF;AAtCD,4DAsCC;AAED,SAAS,aAAa,CAAC,IAAU;IAC/B,OAAO,IAAI,CAAC,UAAU,CAAC,0DAA0B,CAAC,YAAY,CAAC,KAAK,IAAI,CAAC;AAC3E,CAAC;AAED,SAAS,eAAe,CAAC,IAAU;IACjC,OAAO,IAAI,CAAC,UAAU,CAAC,0DAA0B,CAAC,iBAAiB,CAAC,KAAK,IAAI,CAAC;AAChF,CAAC;AAED,SAAS,iBAAiB,CAAC,MAAc,EAAE,IAAU,EAAE,aAAsB;IAC3E,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAClC,IAAI,CAAC,IAAI,EACT;QACE,UAAU,EAAE;YACV,CAAC,0DAA0B,CAAC,YAAY,CAAC,EAAE,IAAI;YAC/C,CAAC,0DAA0B,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM;YAC/D,GAAG,IAAI,CAAC,UAAU;SACnB;KACF,EACD,aAAa,CACd,CAAC;IAEF,IAAI,iCAAW,CAAC,GAAG,EAAE,CAAC;QACpB,WAAW,CAAC,aAAa,CACvB,IAAA,wCAAiB,EAAC,iCAAW,CAAC,UAAU,EAAE,0DAA0B,CAAC,QAAQ,CAAC,CAC/E,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACjE,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAa,uBAAuB;IAC1B,eAAe,CAAqB;IAE5C,YAAY,cAAkC;QAC5C,IAAI,CAAC,eAAe,GAAG,cAAc,CAAC;IACxC,CAAC;IACD,UAAU;QACR,OAAO,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;IAC3C,CAAC;IACD,MAAM,CAAC,SAAuB,EAAE,OAA6B;QAC3D,mDAAmD;QACnD,IAAI,iCAAW,CAAC,GAAG,EAAE,CAAC;YACpB,SAAS,CAAC,aAAa,CACrB,IAAA,wCAAiB,EAAC,iCAAW,CAAC,UAAU,EAAE,0DAA0B,CAAC,QAAQ,CAAC,CAC/E,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IACD,QAAQ;QACN,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;IACzC,CAAC;CACF;AAtBD,0DAsBC"}

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

@@ -2,4 +2,5 @@ import { ServerBackgroundWorker, TaskRunContext } from "../schemas/index.js";
 export type TaskContext = {
     ctx: TaskRunContext;
     worker: ServerBackgroundWorker;
+    isWarmStart?: boolean;
 };

package/dist/commonjs/v3/timeout/api.d.ts

@@ -5,7 +5,8 @@ export declare class TimeoutAPI implements TimeoutManager {
     private constructor();
     static getInstance(): TimeoutAPI;
     get signal(): AbortSignal | undefined;
-    abortAfterTimeout(timeoutInSeconds: number): AbortSignal;
+    abortAfterTimeout(timeoutInSeconds?: number): AbortController;
     setGlobalManager(manager: TimeoutManager): boolean;
     disable(): void;
+    reset(): void;
 }

package/dist/commonjs/v3/timeout/api.js

@@ -5,8 +5,9 @@ const globals_js_1 = require("../utils/globals.js");
 const API_NAME = "timeout";
 class NoopTimeoutManager {
     abortAfterTimeout(timeoutInSeconds) {
-        return new AbortController().signal;
+        return new AbortController();
     }
+    reset() { }
 }
 const NOOP_TIMEOUT_MANAGER = new NoopTimeoutManager();
 class TimeoutAPI {
@@ -19,10 +20,10 @@ class TimeoutAPI {
         return this._instance;
     }
     get signal() {
-        return this.#getManagerManager().signal;
+        return this.#getManager().signal;
     }
     abortAfterTimeout(timeoutInSeconds) {
-        return this.#getManagerManager().abortAfterTimeout(timeoutInSeconds);
+        return this.#getManager().abortAfterTimeout(timeoutInSeconds);
     }
     setGlobalManager(manager) {
         return (0, globals_js_1.registerGlobal)(API_NAME, manager);
@@ -30,7 +31,11 @@ class TimeoutAPI {
     disable() {
         (0, globals_js_1.unregisterGlobal)(API_NAME);
     }
-    #getManagerManager() {
+    reset() {
+        this.#getManager().reset();
+        this.disable();
+    }
+    #getManager() {
         return (0, globals_js_1.getGlobal)(API_NAME) ?? NOOP_TIMEOUT_MANAGER;
     }
 }

package/dist/commonjs/v3/timeout/api.js.map

@@ -1 +1 @@
-{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../src/v3/timeout/api.ts"],"names":[],"mappings":";;;AAAA,oDAAkF;AAGlF,MAAM,QAAQ,GAAG,SAAS,CAAC;AAE3B,MAAM,kBAAkB;IACtB,iBAAiB,CAAC,gBAAwB;QACxC,OAAO,IAAI,eAAe,EAAE,CAAC,MAAM,CAAC;IACtC,CAAC;CACF;AAED,MAAM,oBAAoB,GAAG,IAAI,kBAAkB,EAAE,CAAC;AAEtD,MAAa,UAAU;IACb,MAAM,CAAC,SAAS,CAAc;IAEtC,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,UAAU,EAAE,CAAC;QACpC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAW,MAAM;QACf,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC,MAAM,CAAC;IAC1C,CAAC;IAEM,iBAAiB,CAAC,gBAAwB;QAC/C,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;IACvE,CAAC;IAEM,gBAAgB,CAAC,OAAuB;QAC7C,OAAO,IAAA,2BAAc,EAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAEM,OAAO;QACZ,IAAA,6BAAgB,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAED,kBAAkB;QAChB,OAAO,IAAA,sBAAS,EAAC,QAAQ,CAAC,IAAI,oBAAoB,CAAC;IACrD,CAAC;CACF;AAhCD,gCAgCC"}
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../src/v3/timeout/api.ts"],"names":[],"mappings":";;;AAAA,oDAAkF;AAGlF,MAAM,QAAQ,GAAG,SAAS,CAAC;AAE3B,MAAM,kBAAkB;IACtB,iBAAiB,CAAC,gBAAyB;QACzC,OAAO,IAAI,eAAe,EAAE,CAAC;IAC/B,CAAC;IAED,KAAK,KAAI,CAAC;CACX;AAED,MAAM,oBAAoB,GAAG,IAAI,kBAAkB,EAAE,CAAC;AAEtD,MAAa,UAAU;IACb,MAAM,CAAC,SAAS,CAAc;IAEtC,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,UAAU,EAAE,CAAC;QACpC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,IAAW,MAAM;QACf,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC;IACnC,CAAC;IAEM,iBAAiB,CAAC,gBAAyB;QAChD,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,CAAC;IAChE,CAAC;IAEM,gBAAgB,CAAC,OAAuB;QAC7C,OAAO,IAAA,2BAAc,EAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAC3C,CAAC;IAEM,OAAO;QACZ,IAAA,6BAAgB,EAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAEM,KAAK;QACV,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,OAAO,EAAE,CAAC;IACjB,CAAC;IAED,WAAW;QACT,OAAO,IAAA,sBAAS,EAAC,QAAQ,CAAC,IAAI,oBAAoB,CAAC;IACrD,CAAC;CACF;AArCD,gCAqCC"}

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

@@ -1,6 +1,7 @@
 export interface TimeoutManager {
-    abortAfterTimeout: (timeoutInSeconds: number) => AbortSignal;
+    abortAfterTimeout: (timeoutInSeconds?: number) => AbortController;
     signal?: AbortSignal;
+    reset: () => void;
 }
 export declare class TaskRunExceededMaxDuration extends Error {
     readonly timeoutInSeconds: number;

package/dist/commonjs/v3/timeout/types.js.map

@@ -1 +1 @@
-{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/v3/timeout/types.ts"],"names":[],"mappings":";;;AAKA,MAAa,0BAA2B,SAAQ,KAAK;IAEjC;IACA;IAFlB,YACkB,gBAAwB,EACxB,cAAsB;QAEtC,KAAK,CAAC,+BAA+B,gBAAgB,UAAU,CAAC,CAAC;QAHjD,qBAAgB,GAAhB,gBAAgB,CAAQ;QACxB,mBAAc,GAAd,cAAc,CAAQ;QAGtC,IAAI,CAAC,IAAI,GAAG,4BAA4B,CAAC;IAC3C,CAAC;CACF;AARD,gEAQC"}
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/v3/timeout/types.ts"],"names":[],"mappings":";;;AAMA,MAAa,0BAA2B,SAAQ,KAAK;IAEjC;IACA;IAFlB,YACkB,gBAAwB,EACxB,cAAsB;QAEtC,KAAK,CAAC,+BAA+B,gBAAgB,UAAU,CAAC,CAAC;QAHjD,qBAAgB,GAAhB,gBAAgB,CAAQ;QACxB,mBAAc,GAAd,cAAc,CAAQ;QAGtC,IAAI,CAAC,IAAI,GAAG,4BAA4B,CAAC;IAC3C,CAAC;CACF;AARD,gEAQC"}

package/dist/commonjs/v3/timeout/usageTimeoutManager.d.ts

@@ -4,7 +4,9 @@ export declare class UsageTimeoutManager implements TimeoutManager {
     private readonly usageManager;
     private _abortController;
     private _abortSignal;
+    private _intervalId;
     constructor(usageManager: UsageManager);
     get signal(): AbortSignal | undefined;
-    abortAfterTimeout(timeoutInSeconds: number): AbortSignal;
+    reset(): void;
+    abortAfterTimeout(timeoutInSeconds?: number): AbortController;
 }

package/dist/commonjs/v3/timeout/usageTimeoutManager.js

@@ -6,6 +6,7 @@ class UsageTimeoutManager {
     usageManager;
     _abortController;
     _abortSignal;
+    _intervalId;
     constructor(usageManager) {
         this.usageManager = usageManager;
         this._abortController = new AbortController();
@@ -13,19 +14,33 @@ class UsageTimeoutManager {
     get signal() {
         return this._abortSignal;
     }
+    reset() {
+        this._abortController = new AbortController();
+        this._abortSignal = undefined;
+        if (this._intervalId) {
+            clearInterval(this._intervalId);
+            this._intervalId = undefined;
+        }
+    }
     abortAfterTimeout(timeoutInSeconds) {
         this._abortSignal = this._abortController.signal;
+        if (!timeoutInSeconds) {
+            return this._abortController;
+        }
+        if (this._intervalId) {
+            clearInterval(this._intervalId);
+        }
         // Now we need to start an interval that will measure usage and abort the signal if the usage is too high
-        const intervalId = setInterval(() => {
+        this._intervalId = setInterval(() => {
             const sample = this.usageManager.sample();
             if (sample) {
                 if (sample.cpuTime > timeoutInSeconds * 1000) {
-                    clearInterval(intervalId);
+                    clearInterval(this._intervalId);
                     this._abortController.abort(new types_js_1.TaskRunExceededMaxDuration(timeoutInSeconds, sample.cpuTime / 1000));
                 }
             }
         }, 1000);
-        return this._abortSignal;
+        return this._abortController;
     }
 }
 exports.UsageTimeoutManager = UsageTimeoutManager;