lambda-deadline-middleware 1.0.1 → 2.0.0

package/dist/middleware.d.ts.map

@@ -1 +1 @@
-{"mappings":"AAKA,cAAc,qBAAqB,0BAA0B,yBAAyB;AAGtF,cAIE,iCAEK;AAEP,OAAO,cAAM,kBAAmB,QAAQ,6BAA2B;AAqBnE,iBAAiB,cAAc;UACpB,YAAY;EACpB,OAAO;AACV;AAEA,OAAO,cAAM,sBACX,YAAY,mBACZ,QAAQ,6BACP;AAmBH,OAAO,cAAM,iBACX,UAAU,yBACV,UAAU,gBACT;AAKH,OAAO,cAAM;CACV;CAAsB;EACrB,QAAQ,6BACP,0BAA0B,OAAO","names":[],"sources":["src/middleware.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\nimport { getRemainingTimeInMillis } from \"./context-store.js\";\nimport { DeadlineExceededError } from \"./error.js\";\nimport type { DeadlineComputation, DeadlineMiddlewareConfig, RequestDeadlineMs } from \"./types.js\";\nimport { milliseconds } from \"./types.js\";\n\nimport type {\n  FinalizeHandler,\n  FinalizeHandlerArguments,\n  FinalizeHandlerOutput,\n  FinalizeRequestMiddleware,\n  HandlerExecutionContext,\n} from \"@smithy/types\";\n\nexport const computeDeadline = (config: DeadlineMiddlewareConfig): DeadlineComputation => {\n  const remaining = getRemainingTimeInMillis();\n\n  if (remaining === undefined) {\n    return { kind: \"no-context\" };\n  }\n\n  const deadline = remaining - config.flushBufferMs;\n\n  if (deadline <= 0) {\n    return {\n      kind: \"insufficient-time\",\n      remaining: milliseconds(remaining),\n      buffer: config.flushBufferMs,\n    };\n  }\n\n  // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded narrowing: deadline is validated > 0 above\n  return { kind: \"deadline\", value: deadline as RequestDeadlineMs };\n};\n\nexport interface DeadlineTimer {\n  readonly controller: AbortController;\n  [Symbol.dispose]: () => void;\n}\n\nexport const createDeadlineTimer = (\n  deadlineMs: RequestDeadlineMs,\n  config: DeadlineMiddlewareConfig,\n): DeadlineTimer => {\n  const controller = new AbortController();\n  const remaining = milliseconds(deadlineMs + config.flushBufferMs);\n  const error = new DeadlineExceededError({\n    deadlineMs: milliseconds(deadlineMs),\n    flushBufferMs: config.flushBufferMs,\n    remainingMs: remaining,\n  });\n  const timeoutId = setTimeout(() => {\n    controller.abort(error);\n  }, deadlineMs);\n  return {\n    controller,\n    [Symbol.dispose]() {\n      clearTimeout(timeoutId);\n    },\n  };\n};\n\nexport const composeSignals = (\n  existing: AbortSignal | undefined,\n  deadline: AbortSignal,\n): AbortSignal => {\n  if (existing === undefined) return deadline;\n  return AbortSignal.any([existing, deadline]);\n};\n\nexport const deadlineMiddlewareHandler =\n  <Input extends object, Output extends object>(\n    config: DeadlineMiddlewareConfig,\n  ): FinalizeRequestMiddleware<Input, Output> =>\n  (\n    next: FinalizeHandler<Input, Output>,\n    _context: HandlerExecutionContext,\n  ): FinalizeHandler<Input, Output> =>\n  // oxlint-disable-next-line typescript/consistent-return -- switch is exhaustive over DeadlineComputation discriminated union\n  async (args: FinalizeHandlerArguments<Input>): Promise<FinalizeHandlerOutput<Output>> => {\n    const computation = computeDeadline(config);\n\n    switch (computation.kind) {\n      case \"no-context\":\n        return next(args);\n\n      case \"insufficient-time\":\n        throw new DeadlineExceededError({\n          deadlineMs: milliseconds(0),\n          flushBufferMs: computation.buffer,\n          remainingMs: computation.remaining,\n        });\n\n      case \"deadline\": {\n        // `using` guarantees cleanup (clearTimeout) even if next() throws, the promise rejects,\n        // or an external abort signal fires — strictly more reliable than try/finally.\n        using timer = createDeadlineTimer(computation.value, config);\n        // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object; we access optional signal property\n        const request = args.request as { signal?: AbortSignal } | undefined;\n        const signal = composeSignals(request?.signal, timer.controller.signal);\n        const result = await next({\n          ...args,\n          // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal\n          request: { ...(args.request as object), signal },\n        });\n        return result;\n      }\n    }\n  };\n"]}
+{"mappings":"AAGA,cAKE,iBACK;AAIP,OAAO,cAAM;CAAsB;CAAsB;OAA0B,UACjF,OACA","names":[],"sources":["src/middleware.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\nimport type {\n  FinalizeHandler,\n  FinalizeHandlerArguments,\n  FinalizeHandlerOutput,\n  HandlerExecutionContext,\n  Pluggable,\n} from \"@smithy/types\";\n\nimport { getDeadlineSignal } from \"./context-store.js\";\n\nexport const deadlineMiddleware = <Input extends object, Output extends object>(): Pluggable<\n  Input,\n  Output\n> => ({\n  applyToStack(stack) {\n    stack.add(\n      (\n        next: FinalizeHandler<Input, Output>,\n        _context: HandlerExecutionContext,\n      ): FinalizeHandler<Input, Output> =>\n        async (args: FinalizeHandlerArguments<Input>): Promise<FinalizeHandlerOutput<Output>> => {\n          const deadlineSignal = getDeadlineSignal();\n          if (deadlineSignal === undefined) return next(args);\n\n          // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object\n          const request = args.request as { signal?: AbortSignal } | undefined;\n          const existing = request?.signal;\n          const signal = existing ? AbortSignal.any([existing, deadlineSignal]) : deadlineSignal;\n\n          return next({\n            ...args,\n            // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal\n            request: { ...(args.request as object), signal },\n          });\n        },\n      {\n        step: \"finalizeRequest\",\n        name: \"deadlineMiddleware\",\n        override: true,\n      },\n    );\n  },\n});\n"]}

package/dist/middleware.js

@@ -1,76 +1,25 @@
-// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors
-// SPDX-License-Identifier: MIT
-import { getRemainingTimeInMillis } from "./context-store.js";
-import { DeadlineExceededError } from "./error.js";
-import { milliseconds } from "./types.js";
-export const computeDeadline = (config) => {
-	const remaining = getRemainingTimeInMillis();
-	if (remaining === undefined) {
-		return { kind: "no-context" };
-	}
-	const deadline = remaining - config.flushBufferMs;
-	if (deadline <= 0) {
-		return {
-			kind: "insufficient-time",
-			remaining: milliseconds(remaining),
-			buffer: config.flushBufferMs
-		};
-	}
-	// oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded narrowing: deadline is validated > 0 above
-	return {
-		kind: "deadline",
-		value: deadline
-	};
-};
-export const createDeadlineTimer = (deadlineMs, config) => {
-	const controller = new AbortController();
-	const remaining = milliseconds(deadlineMs + config.flushBufferMs);
-	const error = new DeadlineExceededError({
-		deadlineMs: milliseconds(deadlineMs),
-		flushBufferMs: config.flushBufferMs,
-		remainingMs: remaining
-	});
-	const timeoutId = setTimeout(() => {
-		controller.abort(error);
-	}, deadlineMs);
-	return {
-		controller,
-		[Symbol.dispose]() {
-			clearTimeout(timeoutId);
-		}
-	};
-};
-export const composeSignals = (existing, deadline) => {
-	if (existing === undefined) return deadline;
-	return AbortSignal.any([existing, deadline]);
-};
-export const deadlineMiddlewareHandler = (config) => (next, _context) => async (args) => {
-	const computation = computeDeadline(config);
-	switch (computation.kind) {
-		case "no-context": return next(args);
-		case "insufficient-time": throw new DeadlineExceededError({
-			deadlineMs: milliseconds(0),
-			flushBufferMs: computation.buffer,
-			remainingMs: computation.remaining
+import { getDeadlineSignal } from "./context-store.js";
+export const deadlineMiddleware = () => ({ applyToStack(stack) {
+	stack.add((next, _context) => async (args) => {
+		const deadlineSignal = getDeadlineSignal();
+		if (deadlineSignal === undefined) return next(args);
+		// oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object
+		const request = args.request;
+		const existing = request?.signal;
+		const signal = existing ? AbortSignal.any([existing, deadlineSignal]) : deadlineSignal;
+		return next({
+			...args,
+			// oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal
+			request: {
+				...args.request,
+				signal
+			}
 		});
-		case "deadline": {
-			// `using` guarantees cleanup (clearTimeout) even if next() throws, the promise rejects,
-			// or an external abort signal fires — strictly more reliable than try/finally.
-			using timer = createDeadlineTimer(computation.value, config);
-			// oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object; we access optional signal property
-			const request = args.request;
-			const signal = composeSignals(request?.signal, timer.controller.signal);
-			const result = await next({
-				...args,
-				// oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal
-				request: {
-					...args.request,
-					signal
-				}
-			});
-			return result;
-		}
-	}
-};
+	}, {
+		step: "finalizeRequest",
+		name: "deadlineMiddleware",
+		override: true
+	});
+} });
 
 //# sourceMappingURL=middleware.js.map

package/dist/middleware.js.map

@@ -1 +1 @@
-{"mappings":";;AAGA,SAAS,gCAAgC;AACzC,SAAS,6BAA6B;AAEtC,SAAS,oBAAoB;AAU7B,OAAO,MAAM,mBAAmB,WAA0D;CACxF,MAAM,YAAY,yBAAyB;CAE3C,IAAI,cAAc,WAAW;EAC3B,OAAO,EAAE,MAAM,aAAa;CAC9B;CAEA,MAAM,WAAW,YAAY,OAAO;CAEpC,IAAI,YAAY,GAAG;EACjB,OAAO;GACL,MAAM;GACN,WAAW,aAAa,SAAS;GACjC,QAAQ,OAAO;EACjB;CACF;;CAGA,OAAO;EAAE,MAAM;EAAY,OAAO;CAA8B;AAClE;AAOA,OAAO,MAAM,uBACX,YACA,WACkB;CAClB,MAAM,aAAa,IAAI,gBAAgB;CACvC,MAAM,YAAY,aAAa,aAAa,OAAO,aAAa;CAChE,MAAM,QAAQ,IAAI,sBAAsB;EACtC,YAAY,aAAa,UAAU;EACnC,eAAe,OAAO;EACtB,aAAa;CACf,CAAC;CACD,MAAM,YAAY,iBAAiB;EACjC,WAAW,MAAM,KAAK;CACxB,GAAG,UAAU;CACb,OAAO;EACL;EACA,CAAC,OAAO,WAAW;GACjB,aAAa,SAAS;EACxB;CACF;AACF;AAEA,OAAO,MAAM,kBACX,UACA,aACgB;CAChB,IAAI,aAAa,WAAW,OAAO;CACnC,OAAO,YAAY,IAAI,CAAC,UAAU,QAAQ,CAAC;AAC7C;AAEA,OAAO,MAAM,6BAET,YAGA,MACA,aAGF,OAAO,SAAkF;CACvF,MAAM,cAAc,gBAAgB,MAAM;CAE1C,QAAQ,YAAY,MAApB;EACE,KAAK,cACH,OAAO,KAAK,IAAI;EAElB,KAAK,qBACH,MAAM,IAAI,sBAAsB;GAC9B,YAAY,aAAa,CAAC;GAC1B,eAAe,YAAY;GAC3B,aAAa,YAAY;EAC3B,CAAC;EAEH,KAAK,YAAY;;;GAGf,MAAM,QAAQ,oBAAoB,YAAY,OAAO,MAAM;;GAE3D,MAAM,UAAU,KAAK;GACrB,MAAM,SAAS,eAAe,SAAS,QAAQ,MAAM,WAAW,MAAM;GACtE,MAAM,SAAS,MAAM,KAAK;IACxB,GAAG;;IAEH,SAAS;KAAE,GAAI,KAAK;KAAoB;IAAO;GACjD,CAAC;GACD,OAAO;EACT;CACF;AACF","names":[],"sources":["src/middleware.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\nimport { getRemainingTimeInMillis } from \"./context-store.js\";\nimport { DeadlineExceededError } from \"./error.js\";\nimport type { DeadlineComputation, DeadlineMiddlewareConfig, RequestDeadlineMs } from \"./types.js\";\nimport { milliseconds } from \"./types.js\";\n\nimport type {\n  FinalizeHandler,\n  FinalizeHandlerArguments,\n  FinalizeHandlerOutput,\n  FinalizeRequestMiddleware,\n  HandlerExecutionContext,\n} from \"@smithy/types\";\n\nexport const computeDeadline = (config: DeadlineMiddlewareConfig): DeadlineComputation => {\n  const remaining = getRemainingTimeInMillis();\n\n  if (remaining === undefined) {\n    return { kind: \"no-context\" };\n  }\n\n  const deadline = remaining - config.flushBufferMs;\n\n  if (deadline <= 0) {\n    return {\n      kind: \"insufficient-time\",\n      remaining: milliseconds(remaining),\n      buffer: config.flushBufferMs,\n    };\n  }\n\n  // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded narrowing: deadline is validated > 0 above\n  return { kind: \"deadline\", value: deadline as RequestDeadlineMs };\n};\n\nexport interface DeadlineTimer {\n  readonly controller: AbortController;\n  [Symbol.dispose]: () => void;\n}\n\nexport const createDeadlineTimer = (\n  deadlineMs: RequestDeadlineMs,\n  config: DeadlineMiddlewareConfig,\n): DeadlineTimer => {\n  const controller = new AbortController();\n  const remaining = milliseconds(deadlineMs + config.flushBufferMs);\n  const error = new DeadlineExceededError({\n    deadlineMs: milliseconds(deadlineMs),\n    flushBufferMs: config.flushBufferMs,\n    remainingMs: remaining,\n  });\n  const timeoutId = setTimeout(() => {\n    controller.abort(error);\n  }, deadlineMs);\n  return {\n    controller,\n    [Symbol.dispose]() {\n      clearTimeout(timeoutId);\n    },\n  };\n};\n\nexport const composeSignals = (\n  existing: AbortSignal | undefined,\n  deadline: AbortSignal,\n): AbortSignal => {\n  if (existing === undefined) return deadline;\n  return AbortSignal.any([existing, deadline]);\n};\n\nexport const deadlineMiddlewareHandler =\n  <Input extends object, Output extends object>(\n    config: DeadlineMiddlewareConfig,\n  ): FinalizeRequestMiddleware<Input, Output> =>\n  (\n    next: FinalizeHandler<Input, Output>,\n    _context: HandlerExecutionContext,\n  ): FinalizeHandler<Input, Output> =>\n  // oxlint-disable-next-line typescript/consistent-return -- switch is exhaustive over DeadlineComputation discriminated union\n  async (args: FinalizeHandlerArguments<Input>): Promise<FinalizeHandlerOutput<Output>> => {\n    const computation = computeDeadline(config);\n\n    switch (computation.kind) {\n      case \"no-context\":\n        return next(args);\n\n      case \"insufficient-time\":\n        throw new DeadlineExceededError({\n          deadlineMs: milliseconds(0),\n          flushBufferMs: computation.buffer,\n          remainingMs: computation.remaining,\n        });\n\n      case \"deadline\": {\n        // `using` guarantees cleanup (clearTimeout) even if next() throws, the promise rejects,\n        // or an external abort signal fires — strictly more reliable than try/finally.\n        using timer = createDeadlineTimer(computation.value, config);\n        // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object; we access optional signal property\n        const request = args.request as { signal?: AbortSignal } | undefined;\n        const signal = composeSignals(request?.signal, timer.controller.signal);\n        const result = await next({\n          ...args,\n          // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal\n          request: { ...(args.request as object), signal },\n        });\n        return result;\n      }\n    }\n  };\n"]}
+{"mappings":"AAWA,SAAS,yBAAyB;AAElC,OAAO,MAAM,4BAGP,EACJ,aAAa,OAAO;CAClB,MAAM,KAEF,MACA,aAEA,OAAO,SAAkF;EACvF,MAAM,iBAAiB,kBAAkB;EACzC,IAAI,mBAAmB,WAAW,OAAO,KAAK,IAAI;;EAGlD,MAAM,UAAU,KAAK;EACrB,MAAM,WAAW,SAAS;EAC1B,MAAM,SAAS,WAAW,YAAY,IAAI,CAAC,UAAU,cAAc,CAAC,IAAI;EAExE,OAAO,KAAK;GACV,GAAG;;GAEH,SAAS;IAAE,GAAI,KAAK;IAAoB;GAAO;EACjD,CAAC;CACH,GACF;EACE,MAAM;EACN,MAAM;EACN,UAAU;CACZ,CACF;AACF,EACF","names":[],"sources":["src/middleware.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\nimport type {\n  FinalizeHandler,\n  FinalizeHandlerArguments,\n  FinalizeHandlerOutput,\n  HandlerExecutionContext,\n  Pluggable,\n} from \"@smithy/types\";\n\nimport { getDeadlineSignal } from \"./context-store.js\";\n\nexport const deadlineMiddleware = <Input extends object, Output extends object>(): Pluggable<\n  Input,\n  Output\n> => ({\n  applyToStack(stack) {\n    stack.add(\n      (\n        next: FinalizeHandler<Input, Output>,\n        _context: HandlerExecutionContext,\n      ): FinalizeHandler<Input, Output> =>\n        async (args: FinalizeHandlerArguments<Input>): Promise<FinalizeHandlerOutput<Output>> => {\n          const deadlineSignal = getDeadlineSignal();\n          if (deadlineSignal === undefined) return next(args);\n\n          // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object\n          const request = args.request as { signal?: AbortSignal } | undefined;\n          const existing = request?.signal;\n          const signal = existing ? AbortSignal.any([existing, deadlineSignal]) : deadlineSignal;\n\n          return next({\n            ...args,\n            // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal\n            request: { ...(args.request as object), signal },\n          });\n        },\n      {\n        step: \"finalizeRequest\",\n        name: \"deadlineMiddleware\",\n        override: true,\n      },\n    );\n  },\n});\n"]}

package/dist/types.d.ts

@@ -1,33 +1,5 @@
-declare const BrandSymbol: unique symbol;
-type Brand<
-	T,
-	B extends string
-> = T & {
-	readonly [BrandSymbol]: B;
-};
-export type Milliseconds = Brand<number, "Milliseconds">;
-export type FlushBufferMs = Brand<number, "FlushBufferMs">;
-export type RequestDeadlineMs = Brand<number, "RequestDeadlineMs">;
-export declare const milliseconds: (value: number) => Milliseconds;
-export declare const flushBufferMs: (value: number) => FlushBufferMs;
-export type DeadlineComputation = {
-	readonly kind: "deadline";
-	readonly value: RequestDeadlineMs;
-} | {
-	readonly kind: "insufficient-time";
-	readonly remaining: Milliseconds;
-	readonly buffer: FlushBufferMs;
-} | {
-	readonly kind: "no-context";
-};
-export interface DeadlineMiddlewareConfig {
-	readonly flushBufferMs: FlushBufferMs;
-	readonly telemetryEnabled: boolean;
-}
 export interface DeadlineOptions {
 	readonly flushBufferMs?: number;
-	readonly telemetryEnabled?: boolean;
 }
-export {};
 
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"mappings":"AAKA,cAAc;KAET;CAAM;CAAG;IAAoB,IAAI;WAAY,cAAc;AAAE;AAElE,YAAY,eAAe,cAAc;AAEzC,YAAY,gBAAgB,cAAc;AAE1C,YAAY,oBAAoB,cAAc;AAE9C,OAAO,cAAM,eAAgB,kBAAgB;AAQ7C,OAAO,cAAM,gBAAiB,kBAAgB;AAW9C,YAAY,sBACR;UAAW,MAAM;UAAqB,OAAO;AAAkB,IAC/D;UACW,MAAM;UACN,WAAW;UACX,QAAQ;AACnB,IACA;UAAW,MAAM;AAAa;AAElC,iBAAiB,yBAAyB;UAC/B,eAAe;UACf;AACX;AAEA,iBAAiB,gBAAgB;UACtB;UACA;AACX","names":[],"sources":["src/types.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\n// Branded types prevent interchange errors at compile time (e.g. passing seconds where milliseconds are expected).\n// Zero runtime cost. Smart constructors below validate at the boundary and brand the value.\ndeclare const BrandSymbol: unique symbol;\n\ntype Brand<T, B extends string> = T & { readonly [BrandSymbol]: B };\n\nexport type Milliseconds = Brand<number, \"Milliseconds\">;\n\nexport type FlushBufferMs = Brand<number, \"FlushBufferMs\">;\n\nexport type RequestDeadlineMs = Brand<number, \"RequestDeadlineMs\">;\n\nexport const milliseconds = (value: number): Milliseconds => {\n  if (!Number.isFinite(value)) {\n    throw new TypeError(`milliseconds value must be finite, received: ${value}`);\n  }\n  // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded type constructor: value is validated above\n  return value as Milliseconds;\n};\n\nexport const flushBufferMs = (value: number): FlushBufferMs => {\n  if (!Number.isFinite(value)) {\n    throw new TypeError(`flushBufferMs value must be finite, received: ${value}`);\n  }\n  if (value < 0) {\n    throw new TypeError(`flushBufferMs must be non-negative, received: ${value}`);\n  }\n  // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded type constructor: value is validated above\n  return value as FlushBufferMs;\n};\n\nexport type DeadlineComputation =\n  | { readonly kind: \"deadline\"; readonly value: RequestDeadlineMs }\n  | {\n      readonly kind: \"insufficient-time\";\n      readonly remaining: Milliseconds;\n      readonly buffer: FlushBufferMs;\n    }\n  | { readonly kind: \"no-context\" };\n\nexport interface DeadlineMiddlewareConfig {\n  readonly flushBufferMs: FlushBufferMs;\n  readonly telemetryEnabled: boolean;\n}\n\nexport interface DeadlineOptions {\n  readonly flushBufferMs?: number;\n  readonly telemetryEnabled?: boolean;\n}\n"]}
+{"mappings":"AAGA,iBAAiB,gBAAgB;UACtB;AACX","names":[],"sources":["src/types.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\nexport interface DeadlineOptions {\n  readonly flushBufferMs?: number;\n}\n"]}

package/dist/types.js

@@ -1,19 +1,3 @@
-export const milliseconds = (value) => {
-	if (!Number.isFinite(value)) {
-		throw new TypeError(`milliseconds value must be finite, received: ${value}`);
-	}
-	// oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded type constructor: value is validated above
-	return value;
-};
-export const flushBufferMs = (value) => {
-	if (!Number.isFinite(value)) {
-		throw new TypeError(`flushBufferMs value must be finite, received: ${value}`);
-	}
-	if (value < 0) {
-		throw new TypeError(`flushBufferMs must be non-negative, received: ${value}`);
-	}
-	// oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded type constructor: value is validated above
-	return value;
-};
+export {};
 
 //# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -1 +1 @@
-{"mappings":"AAeA,OAAO,MAAM,gBAAgB,UAAgC;CAC3D,IAAI,CAAC,OAAO,SAAS,KAAK,GAAG;EAC3B,MAAM,IAAI,UAAU,gDAAgD,OAAO;CAC7E;;CAEA,OAAO;AACT;AAEA,OAAO,MAAM,iBAAiB,UAAiC;CAC7D,IAAI,CAAC,OAAO,SAAS,KAAK,GAAG;EAC3B,MAAM,IAAI,UAAU,iDAAiD,OAAO;CAC9E;CACA,IAAI,QAAQ,GAAG;EACb,MAAM,IAAI,UAAU,iDAAiD,OAAO;CAC9E;;CAEA,OAAO;AACT","names":[],"sources":["src/types.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\n// Branded types prevent interchange errors at compile time (e.g. passing seconds where milliseconds are expected).\n// Zero runtime cost. Smart constructors below validate at the boundary and brand the value.\ndeclare const BrandSymbol: unique symbol;\n\ntype Brand<T, B extends string> = T & { readonly [BrandSymbol]: B };\n\nexport type Milliseconds = Brand<number, \"Milliseconds\">;\n\nexport type FlushBufferMs = Brand<number, \"FlushBufferMs\">;\n\nexport type RequestDeadlineMs = Brand<number, \"RequestDeadlineMs\">;\n\nexport const milliseconds = (value: number): Milliseconds => {\n  if (!Number.isFinite(value)) {\n    throw new TypeError(`milliseconds value must be finite, received: ${value}`);\n  }\n  // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded type constructor: value is validated above\n  return value as Milliseconds;\n};\n\nexport const flushBufferMs = (value: number): FlushBufferMs => {\n  if (!Number.isFinite(value)) {\n    throw new TypeError(`flushBufferMs value must be finite, received: ${value}`);\n  }\n  if (value < 0) {\n    throw new TypeError(`flushBufferMs must be non-negative, received: ${value}`);\n  }\n  // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded type constructor: value is validated above\n  return value as FlushBufferMs;\n};\n\nexport type DeadlineComputation =\n  | { readonly kind: \"deadline\"; readonly value: RequestDeadlineMs }\n  | {\n      readonly kind: \"insufficient-time\";\n      readonly remaining: Milliseconds;\n      readonly buffer: FlushBufferMs;\n    }\n  | { readonly kind: \"no-context\" };\n\nexport interface DeadlineMiddlewareConfig {\n  readonly flushBufferMs: FlushBufferMs;\n  readonly telemetryEnabled: boolean;\n}\n\nexport interface DeadlineOptions {\n  readonly flushBufferMs?: number;\n  readonly telemetryEnabled?: boolean;\n}\n"]}
+{"mappings":"","names":[],"sources":["src/types.ts"],"version":3,"sourcesContent":["// SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors\n// SPDX-License-Identifier: MIT\n\nexport interface DeadlineOptions {\n  readonly flushBufferMs?: number;\n}\n"]}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "lambda-deadline-middleware",
-  "version": "1.0.1",
-  "description": "AWS SDK v3 middleware for automatic Lambda deadline propagation via AbortController-based timeouts",
+  "version": "2.0.0",
+  "description": "Zero-dependency AWS SDK v3 middleware that propagates Lambda deadlines to outgoing SDK calls via AbortSignal",
   "license": "MIT",
   "repository": {
     "type": "git",
@@ -28,19 +28,19 @@
   },
   "scripts": {
     "build": "node --experimental-strip-types scripts/build.ts",
-    "test": "vitest run",
-    "test:watch": "vitest",
-    "bench": "vitest bench --run --outputJson=bench-results.json && node --experimental-strip-types scripts/validate-bench.ts",
+    "test": "vitest run --config .config/vitest.config.ts",
+    "test:watch": "vitest --config .config/vitest.config.ts",
+    "bench": "vitest bench --run --config .config/vitest.config.ts --outputJson=bench-results.json && node --experimental-strip-types scripts/validate-bench.ts",
     "typecheck": "tsc",
-    "lint": "oxlint --type-aware --type-check --fix .",
-    "lint:check": "oxlint --type-aware --type-check .",
+    "lint": "oxlint --fix .",
+    "lint:check": "oxlint .",
     "fmt": "oxfmt .",
     "fmt:check": "oxfmt --check .",
-    "lint:knip": "knip",
+    "lint:knip": "knip --config .config/knip.js",
     "sast": "scripts/ensure-opengrep.sh scan .",
     "sast:check": "scripts/ensure-opengrep.sh scan --error .",
-    "sca": "podman run --rm -v \"$(pwd):/src:Z\" -v trivy-cache:/root/.cache/trivy:Z ghcr.io/aquasecurity/trivy:0.71.0@sha256:016eae51fdcf989332a5404af7e8f625cd5d95d7c0907a221d080a996f556500 fs --scanners vuln --include-dev-deps --exit-code 1 --severity HIGH,CRITICAL /src",
-    "sca:full": "podman run --rm -v \"$(pwd):/src:Z\" -v trivy-cache:/root/.cache/trivy:Z ghcr.io/aquasecurity/trivy:0.71.0@sha256:016eae51fdcf989332a5404af7e8f625cd5d95d7c0907a221d080a996f556500 fs --scanners vuln,secret,misconfig --include-dev-deps /src",
+    "sca": "podman run --rm -v \"$(pwd):/src:Z\" -v trivy-cache:/root/.cache/trivy:Z ghcr.io/aquasecurity/trivy:0.71.0@sha256:016eae51fdcf989332a5404af7e8f625cd5d95d7c0907a221d080a996f556500 fs --config /src/.config/trivy.yaml --ignorefile /src/.config/.trivyignore --scanners vuln --include-dev-deps --exit-code 1 --severity HIGH,CRITICAL /src",
+    "sca:full": "podman run --rm -v \"$(pwd):/src:Z\" -v trivy-cache:/root/.cache/trivy:Z ghcr.io/aquasecurity/trivy:0.71.0@sha256:016eae51fdcf989332a5404af7e8f625cd5d95d7c0907a221d080a996f556500 fs --config /src/.config/trivy.yaml --ignorefile /src/.config/.trivyignore --scanners vuln,secret,misconfig --include-dev-deps /src",
     "actionlint": "podman run --rm -w /repo -v \"$(pwd):/repo:Z\" docker.io/rhysd/actionlint:1.7.12@sha256:9d36088643581e728c969f35141f88139fec77280b2be23c1f66f8e40e1025e7",
     "security": "pnpm run sast:check && pnpm run sca && pnpm run actionlint",
     "prepare": "lefthook install",
@@ -48,9 +48,9 @@
     "release": "semantic-release"
   },
   "devDependencies": {
-    "@aws-sdk/client-dynamodb": "^3.1064.0",
-    "@aws-sdk/client-s3": "^3.1064.0",
-    "@aws-sdk/client-sqs": "^3.1064.0",
+    "@aws-sdk/client-dynamodb": "^3.1073.0",
+    "@aws-sdk/client-s3": "^3.1073.0",
+    "@aws-sdk/client-sqs": "^3.1073.0",
     "@commitlint/cli": "^21.0.2",
     "@commitlint/config-conventional": "^21.0.2",
     "@fast-check/vitest": "^0.4.1",
@@ -58,19 +58,19 @@
     "@semantic-release/github": "12.0.8",
     "@semantic-release/npm": "13.1.5",
     "@semantic-release/release-notes-generator": "14.1.1",
-    "@smithy/types": "^4.14.3",
-    "@types/node": "^25.9.2",
-    "@vitest/coverage-v8": "^4.1.8",
+    "@smithy/types": "^4.15.0",
+    "@types/node": "^25.9.4",
+    "@vitest/coverage-v8": "^4.1.9",
     "conventional-changelog-conventionalcommits": "^9.3.1",
-    "knip": "^6.16.1",
+    "knip": "^6.17.1",
     "lefthook": "^2.1.9",
-    "oxc-transform": "^0.134.0",
-    "oxfmt": "^0.53.0",
-    "oxlint": "^1.69.0",
+    "oxc-transform": "^0.135.0",
+    "oxfmt": "^0.54.0",
+    "oxlint": "^1.70.0",
     "oxlint-tsgolint": "^0.23.0",
-    "semantic-release": "25.0.3",
+    "semantic-release": "25.0.5",
     "typescript": "^6.0.3",
-    "vitest": "^4.1.8"
+    "vitest": "^4.1.9"
   },
   "engines": {
     "node": ">=24"

package/src/context-store.ts

@@ -3,36 +3,88 @@
 
 import { AsyncLocalStorage } from "node:async_hooks";
 
-// AsyncLocalStorage propagates the Lambda context through the entire async call chain without parameter threading.
-// SDK middleware executes deep in the Smithy stack where we can't pass the context through function signatures.
+import { DeadlineExceededError } from "./error.js";
+
+import type { DeadlineOptions } from "./types.js";
+
+// AsyncLocalStorage propagates the deadline signal through the entire async call chain.
+// SDK middleware executes deep in the Smithy stack where we can't pass it through function signatures.
 export interface LambdaContextLike {
   getRemainingTimeInMillis?: () => number;
 }
 
-// Sentinel allows AsyncLocalStorage.run() to accept null/undefined context
-// without throwing, while the accessor can distinguish "no context stored"
-// from "context present but missing the method".
-const NO_CONTEXT: unique symbol = Symbol("no-context");
-
-type StoreValue = LambdaContextLike | typeof NO_CONTEXT;
+interface DeadlineStore {
+  signal?: AbortSignal;
+}
 
-const contextStorage = new AsyncLocalStorage<StoreValue>();
+const contextStorage = new AsyncLocalStorage<DeadlineStore>();
 
-export const run = <T>(context: LambdaContextLike | null | undefined, fn: () => T): T => {
-  const value: StoreValue = context ?? NO_CONTEXT;
-  return contextStorage.run(value, fn);
+/**
+ * Store an external AbortSignal for the current invocation.
+ * When set, the SDK middleware uses this signal directly instead of
+ * the auto-computed deadline signal.
+ *
+ * Call this at the start of your handler, before any SDK calls.
+ * The signal is scoped to the current async context (AsyncLocalStorage).
+ */
+export const setDeadlineSignal = (signal: AbortSignal): void => {
+  const store = contextStorage.getStore();
+  if (store === undefined) {
+    throw new Error("setDeadlineSignal() must be called within a withLambdaDeadline() scope");
+  }
+  store.signal = signal;
 };
 
-export const getRemainingTimeInMillis = (): number | undefined => {
+/**
+ * Retrieve the deadline signal for the current invocation, if one exists.
+ */
+export const getDeadlineSignal = (): AbortSignal | undefined => {
   const store = contextStorage.getStore();
+  if (store === undefined) return undefined;
+  return store.signal;
+};
 
-  if (store === undefined || store === NO_CONTEXT) {
-    return undefined;
-  }
+// Handler wrapper — computes the deadline signal once at invocation start and stores it via AsyncLocalStorage.
+type AsyncHandler<TEvent, TContext extends LambdaContextLike, TResult> = (
+  event: TEvent,
+  context: TContext,
+) => Promise<TResult>;
 
-  if (typeof store.getRemainingTimeInMillis !== "function") {
-    return undefined;
-  }
+export const withLambdaDeadline =
+  <TEvent, TContext extends LambdaContextLike, TResult>(
+    handler: AsyncHandler<TEvent, TContext, TResult>,
+    options?: DeadlineOptions,
+  ): AsyncHandler<TEvent, TContext, TResult> =>
+  async (event: TEvent, context: TContext): Promise<TResult> => {
+    const store: DeadlineStore = {};
 
-  return store.getRemainingTimeInMillis();
-};
+    // Compute the auto-deadline signal once, up front.
+    // If the user calls setDeadlineSignal() later, it overwrites this.
+    // oxlint-disable-next-line typescript/no-unnecessary-condition -- runtime safety: context may be null/undefined despite types (e.g. untyped callers)
+    if (context !== null && context !== undefined) {
+      const remaining =
+        typeof context.getRemainingTimeInMillis === "function"
+          ? context.getRemainingTimeInMillis()
+          : undefined;
+
+      if (remaining !== undefined) {
+        const rawBuffer = options?.flushBufferMs ?? 1000;
+        if (rawBuffer < 0) {
+          throw new TypeError(`flushBufferMs option must be non-negative, received: ${rawBuffer}`);
+        }
+        const deadline = remaining - rawBuffer;
+
+        if (deadline <= 0) {
+          throw new DeadlineExceededError({
+            deadlineMs: 0,
+            flushBufferMs: rawBuffer,
+            remainingMs: remaining,
+          });
+        }
+
+        store.signal = AbortSignal.timeout(deadline);
+      }
+    }
+
+    return contextStorage.run(store, async () => handler(event, context));
+  };

package/src/error.ts

@@ -1,19 +1,17 @@
 // SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors
 // SPDX-License-Identifier: MIT
 
-import type { FlushBufferMs, Milliseconds } from "./types.js";
-
 interface DeadlineExceededInit {
-  readonly deadlineMs: Milliseconds;
-  readonly flushBufferMs: FlushBufferMs;
-  readonly remainingMs: Milliseconds;
+  readonly deadlineMs: number;
+  readonly flushBufferMs: number;
+  readonly remainingMs: number;
 }
 
 export class DeadlineExceededError extends Error {
   override readonly name = "DeadlineExceededError" as const;
-  readonly deadlineMs: Milliseconds;
-  readonly flushBufferMs: FlushBufferMs;
-  readonly remainingMs: Milliseconds;
+  readonly deadlineMs: number;
+  readonly flushBufferMs: number;
+  readonly remainingMs: number;
 
   constructor(init: DeadlineExceededInit) {
     super(

package/src/index.ts

@@ -1,18 +1,9 @@
 // SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors
 // SPDX-License-Identifier: MIT
 
-export { withLambdaDeadline } from "./handler-wrapper.js";
-export { deadlineMiddleware } from "./registration.js";
+export { withLambdaDeadline, setDeadlineSignal } from "./context-store.js";
+export { deadlineMiddleware } from "./middleware.js";
 export { DeadlineExceededError, isDeadlineExceeded } from "./error.js";
-export { getRemainingTimeInMillis } from "./context-store.js";
-
-export type {
-  Milliseconds,
-  FlushBufferMs,
-  RequestDeadlineMs,
-  DeadlineComputation,
-  DeadlineMiddlewareConfig,
-  DeadlineOptions,
-} from "./types.js";
 
+export type { DeadlineOptions } from "./types.js";
 export type { LambdaContextLike } from "./context-store.js";

package/src/middleware.ts

@@ -1,111 +1,46 @@
 // SPDX-FileCopyrightText: 2026 lambda-deadline-middleware contributors
 // SPDX-License-Identifier: MIT
 
-import { getRemainingTimeInMillis } from "./context-store.js";
-import { DeadlineExceededError } from "./error.js";
-import type { DeadlineComputation, DeadlineMiddlewareConfig, RequestDeadlineMs } from "./types.js";
-import { milliseconds } from "./types.js";
-
 import type {
   FinalizeHandler,
   FinalizeHandlerArguments,
   FinalizeHandlerOutput,
-  FinalizeRequestMiddleware,
   HandlerExecutionContext,
+  Pluggable,
 } from "@smithy/types";
 
-export const computeDeadline = (config: DeadlineMiddlewareConfig): DeadlineComputation => {
-  const remaining = getRemainingTimeInMillis();
-
-  if (remaining === undefined) {
-    return { kind: "no-context" };
-  }
-
-  const deadline = remaining - config.flushBufferMs;
-
-  if (deadline <= 0) {
-    return {
-      kind: "insufficient-time",
-      remaining: milliseconds(remaining),
-      buffer: config.flushBufferMs,
-    };
-  }
-
-  // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- branded narrowing: deadline is validated > 0 above
-  return { kind: "deadline", value: deadline as RequestDeadlineMs };
-};
-
-export interface DeadlineTimer {
-  readonly controller: AbortController;
-  [Symbol.dispose]: () => void;
-}
-
-export const createDeadlineTimer = (
-  deadlineMs: RequestDeadlineMs,
-  config: DeadlineMiddlewareConfig,
-): DeadlineTimer => {
-  const controller = new AbortController();
-  const remaining = milliseconds(deadlineMs + config.flushBufferMs);
-  const error = new DeadlineExceededError({
-    deadlineMs: milliseconds(deadlineMs),
-    flushBufferMs: config.flushBufferMs,
-    remainingMs: remaining,
-  });
-  const timeoutId = setTimeout(() => {
-    controller.abort(error);
-  }, deadlineMs);
-  return {
-    controller,
-    [Symbol.dispose]() {
-      clearTimeout(timeoutId);
-    },
-  };
-};
-
-export const composeSignals = (
-  existing: AbortSignal | undefined,
-  deadline: AbortSignal,
-): AbortSignal => {
-  if (existing === undefined) return deadline;
-  return AbortSignal.any([existing, deadline]);
-};
-
-export const deadlineMiddlewareHandler =
-  <Input extends object, Output extends object>(
-    config: DeadlineMiddlewareConfig,
-  ): FinalizeRequestMiddleware<Input, Output> =>
-  (
-    next: FinalizeHandler<Input, Output>,
-    _context: HandlerExecutionContext,
-  ): FinalizeHandler<Input, Output> =>
-  // oxlint-disable-next-line typescript/consistent-return -- switch is exhaustive over DeadlineComputation discriminated union
-  async (args: FinalizeHandlerArguments<Input>): Promise<FinalizeHandlerOutput<Output>> => {
-    const computation = computeDeadline(config);
-
-    switch (computation.kind) {
-      case "no-context":
-        return next(args);
-
-      case "insufficient-time":
-        throw new DeadlineExceededError({
-          deadlineMs: milliseconds(0),
-          flushBufferMs: computation.buffer,
-          remainingMs: computation.remaining,
-        });
-
-      case "deadline": {
-        // `using` guarantees cleanup (clearTimeout) even if next() throws, the promise rejects,
-        // or an external abort signal fires — strictly more reliable than try/finally.
-        using timer = createDeadlineTimer(computation.value, config);
-        // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object; we access optional signal property
-        const request = args.request as { signal?: AbortSignal } | undefined;
-        const signal = composeSignals(request?.signal, timer.controller.signal);
-        const result = await next({
-          ...args,
-          // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal
-          request: { ...(args.request as object), signal },
-        });
-        return result;
-      }
-    }
-  };
+import { getDeadlineSignal } from "./context-store.js";
+
+export const deadlineMiddleware = <Input extends object, Output extends object>(): Pluggable<
+  Input,
+  Output
+> => ({
+  applyToStack(stack) {
+    stack.add(
+      (
+        next: FinalizeHandler<Input, Output>,
+        _context: HandlerExecutionContext,
+      ): FinalizeHandler<Input, Output> =>
+        async (args: FinalizeHandlerArguments<Input>): Promise<FinalizeHandlerOutput<Output>> => {
+          const deadlineSignal = getDeadlineSignal();
+          if (deadlineSignal === undefined) return next(args);
+
+          // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- Smithy request is an opaque object
+          const request = args.request as { signal?: AbortSignal } | undefined;
+          const existing = request?.signal;
+          const signal = existing ? AbortSignal.any([existing, deadlineSignal]) : deadlineSignal;
+
+          return next({
+            ...args,
+            // oxlint-disable-next-line typescript/no-unsafe-type-assertion -- spreading opaque Smithy request to add signal
+            request: { ...(args.request as object), signal },
+          });
+        },
+      {
+        step: "finalizeRequest",
+        name: "deadlineMiddleware",
+        override: true,
+      },
+    );
+  },
+});