@middy/core 4.3.1 → 4.4.1

package/index.cjs

@@ -4,9 +4,11 @@ Object.defineProperty(exports, "__esModule", {
 });
 Object.defineProperty(module, "exports", {
     enumerable: true,
-    get: ()=>_default
+    get: function() {
+        return _default;
+    }
 });
-const _nodeStream = require("node:stream");
+const _nodestream = require("node:stream");
 const _promises = require("node:stream/promises");
 const _promises1 = require("node:timers/promises");
 const defaultLambdaHandler = ()=>{};
@@ -49,12 +51,18 @@ const middy = (lambdaHandler = defaultLambdaHandler, plugin = {})=>{
         ], plugin);
     };
     const middy = plugin.streamifyResponse ? awslambda.streamifyResponse(async (event, responseStream, context)=>{
-        const response = await middyHandler(event, context);
-        response.body ??= '';
-        let { body  } = response;
-        if (typeof body === 'string') {
+        const handlerResponse = await middyHandler(event, context);
+        let handlerBody = handlerResponse;
+        if (handlerResponse.statusCode) {
+            handlerBody = handlerResponse.body ?? '';
+            responseStream = awslambda.HttpResponseStream.from(responseStream, handlerResponse);
+        }
+        let handlerStream;
+        if (handlerBody._readableState) {
+            handlerStream = handlerBody;
+        } else if (typeof handlerBody === 'string') {
             function* iterator(input) {
-                const size = 16 * 1024;
+                const size = 16384;
                 let position = 0;
                 const length = input.length;
                 while(position < length){
@@ -62,10 +70,12 @@ const middy = (lambdaHandler = defaultLambdaHandler, plugin = {})=>{
                     position += size;
                 }
             }
-            body = _nodeStream.Readable.from(iterator(response.body));
+            handlerStream = _nodestream.Readable.from(iterator(handlerBody));
+        }
+        if (!handlerStream) {
+            throw new Error('handler response not a ReadableStream');
         }
-        responseStream = awslambda.HttpResponseStream.from(responseStream, response);
-        await (0, _promises.pipeline)(body, responseStream);
+        await (0, _promises.pipeline)(handlerStream, responseStream);
     }) : middyHandler;
     middy.use = (middlewares)=>{
         if (!Array.isArray(middlewares)) {

package/index.js

@@ -41,12 +41,18 @@ const middy = (lambdaHandler = defaultLambdaHandler, plugin = {})=>{
         ], plugin);
     };
     const middy = plugin.streamifyResponse ? awslambda.streamifyResponse(async (event, responseStream, context)=>{
-        const response = await middyHandler(event, context);
-        response.body ??= '';
-        let { body  } = response;
-        if (typeof body === 'string') {
+        const handlerResponse = await middyHandler(event, context);
+        let handlerBody = handlerResponse;
+        if (handlerResponse.statusCode) {
+            handlerBody = handlerResponse.body ?? '';
+            responseStream = awslambda.HttpResponseStream.from(responseStream, handlerResponse);
+        }
+        let handlerStream;
+        if (handlerBody._readableState) {
+            handlerStream = handlerBody;
+        } else if (typeof handlerBody === 'string') {
             function* iterator(input) {
-                const size = 16 * 1024;
+                const size = 16384;
                 let position = 0;
                 const length = input.length;
                 while(position < length){
@@ -54,10 +60,12 @@ const middy = (lambdaHandler = defaultLambdaHandler, plugin = {})=>{
                     position += size;
                 }
             }
-            body = Readable.from(iterator(response.body));
+            handlerStream = Readable.from(iterator(handlerBody));
+        }
+        if (!handlerStream) {
+            throw new Error('handler response not a ReadableStream');
         }
-        responseStream = awslambda.HttpResponseStream.from(responseStream, response);
-        await pipeline(body, responseStream);
+        await pipeline(handlerStream, responseStream);
     }) : middyHandler;
     middy.use = (middlewares)=>{
         if (!Array.isArray(middlewares)) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@middy/core",
-  "version": "4.3.1",
+  "version": "4.4.1",
   "description": "🛵 The stylish Node.js middleware engine for AWS Lambda (core package)",
   "type": "module",
   "engines": {
@@ -62,9 +62,9 @@
     "url": "https://github.com/sponsors/willfarrell"
   },
   "devDependencies": {
-    "@datastream/core": "0.0.29",
+    "@datastream/core": "0.0.30",
     "@types/aws-lambda": "^8.10.76",
     "@types/node": "^18.0.0"
   },
-  "gitHead": "e7576b56f165591acc427fd567b6cba8d4874340"
+  "gitHead": "51e5c1f83483a60d9e40f5db59b03af56d2ee710"
 }