@middy/input-output-logger 6.2.4 → 6.3.1

package/index.js

@@ -1,4 +1,5 @@
 import { Transform } from "node:stream";
+import { TransformStream } from "node:stream/web";
 
 const defaults = {
 	logger: (message) => {
@@ -65,11 +66,19 @@ const inputOutputLoggerMiddleware = (opts = {}) => {
 		omitAndLog("event", request);
 	};
 	const inputOutputLoggerMiddlewareAfter = async (request) => {
+		// Check for Node.js stream
 		if (
 			request.response?._readableState ??
 			request.response?.body?._readableState
 		) {
 			passThrough(request, omitAndLog);
+		}
+		// Check for Web stream
+		else if (
+			request.response instanceof ReadableStream ||
+			request.response?.body instanceof ReadableStream
+		) {
+			passThroughWebStream(request, omitAndLog);
 		} else {
 			omitAndLog("response", request);
 		}
@@ -162,4 +171,39 @@ const passThrough = (request, omitAndLog) => {
 	}
 };
 
+// Handler for Web Streams API
+const passThroughWebStream = (request, omitAndLog) => {
+	const hasBody = request.response?.body;
+	let body = "";
+
+	const transformer = new TransformStream({
+		transform(chunk, controller) {
+			// For web streams, chunks could be various types
+			const textChunk =
+				typeof chunk === "string"
+					? chunk
+					: chunk instanceof Uint8Array
+						? new TextDecoder().decode(chunk)
+						: String(chunk);
+			body += textChunk;
+			controller.enqueue(chunk);
+		},
+		flush(controller) {
+			if (hasBody) {
+				omitAndLog("response", { response: { ...request.response, body } });
+			} else {
+				omitAndLog("response", { response: body });
+			}
+		},
+	});
+
+	if (hasBody) {
+		// Handle response with body property that's a ReadableStream
+		request.response.body = request.response.body.pipeThrough(transformer);
+	} else {
+		// Handle response that's directly a ReadableStream
+		request.response = request.response.pipeThrough(transformer);
+	}
+};
+
 export default inputOutputLoggerMiddleware;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@middy/input-output-logger",
-	"version": "6.2.4",
+	"version": "6.3.1",
 	"description": "Input and output logger middleware for the middy framework",
 	"type": "module",
 	"engines": {
@@ -62,7 +62,7 @@
 	},
 	"devDependencies": {
 		"@datastream/core": "0.0.40",
-		"@middy/core": "6.2.4",
+		"@middy/core": "6.3.1",
 		"@types/node": "^20.0.0"
 	},
 	"gitHead": "7a6c0fbb8ab71d6a2171e678697de9f237568431"