@middy/http-content-encoding 3.0.0-alpha.0 → 3.0.0-alpha.4

package/index.js

@@ -0,0 +1,103 @@
+import { Readable, Writable, pipeline as pipelineCallback } from 'stream';
+import { promisify } from 'util';
+import { createBrotliCompress, createGzip, createDeflate } from 'zlib';
+import { normalizeHttpResponse } from '@middy/util';
+const pipeline = promisify(pipelineCallback);
+const contentEncodingStreams = {
+  br: (opts = {}) => createBrotliCompress(opts),
+  gzip: (opts = {}) => createGzip(opts),
+  deflate: (opts = {}) => createDeflate(opts)
+};
+const defaults = {
+  br: undefined,
+  gzip: undefined,
+  deflate: undefined,
+  overridePreferredEncoding: []
+};
+
+const httpContentEncodingMiddleware = opts => {
+  const options = { ...defaults,
+    ...opts
+  };
+  const supportedContentEncodings = Object.keys(contentEncodingStreams);
+
+  const httpContentEncodingMiddlewareAfter = async request => {
+    normalizeHttpResponse(request);
+    const {
+      event: {
+        preferredEncoding,
+        preferredEncodings
+      },
+      response
+    } = request;
+
+    if (response.isBase64Encoded || !preferredEncoding || !supportedContentEncodings.includes(preferredEncoding)) {
+      return;
+    }
+
+    const bodyIsString = typeof response.body === 'string';
+    let contentEncodingStream = contentEncodingStreams[preferredEncoding](options[preferredEncoding]);
+    let contentEncoding = preferredEncoding;
+
+    for (const encoding of options.overridePreferredEncoding) {
+      if (!preferredEncodings.includes(encoding)) continue;
+      contentEncodingStream = contentEncodingStreams[encoding](options[encoding]);
+      contentEncoding = encoding;
+      break;
+    }
+
+    if (bodyIsString) {
+      const readStream = Readable.from(response.body, {
+        objectMode: false
+      });
+      const chunks = [];
+      const writeStream = new Writable({
+        write(chunk, encoding, callback) {
+          chunks.push(chunk);
+          callback();
+        }
+
+      });
+      await pipeline(readStream, contentEncodingStream, writeStream);
+      const body = Buffer.concat(chunks).toString('base64');
+
+      if (body.length < response.body.length) {
+        response.headers['Content-Encoding'] = contentEncoding;
+        response.body = body;
+        response.isBase64Encoded = true;
+      }
+    } else if (isReadableStream(response.body)) {
+      response.headers['Content-Encoding'] = contentEncoding;
+      response.body = response.body.pipe(contentEncodingStream);
+      response.isBase64Encoded = true;
+    }
+
+    request.response = response;
+  };
+
+  const httpContentEncodingMiddlewareOnError = async request => {
+    if (request.response === undefined) return;
+    return httpContentEncodingMiddlewareAfter(request);
+  };
+
+  return {
+    after: httpContentEncodingMiddlewareAfter,
+    onError: httpContentEncodingMiddlewareOnError
+  };
+};
+
+const isReadableStream = stream => typeof (stream === null || stream === void 0 ? void 0 : stream.pipe) === 'function' && (stream === null || stream === void 0 ? void 0 : stream.readable) !== false && typeof (stream === null || stream === void 0 ? void 0 : stream._read) === 'function' && typeof (stream === null || stream === void 0 ? void 0 : stream._readableState) === 'object';
+
+const polyfillPipelinePromise = async () => {
+  if (process.version < 'v15.0.0') {
+    const stream = await import('stream');
+    const util = await import('util');
+    return util.promisify(stream.pipeline);
+  } else {
+    const stream = await import('stream/promises');
+    return stream.pipeline;
+  }
+};
+
+global.pipeline = await polyfillPipelinePromise();
+export default httpContentEncodingMiddleware;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@middy/http-content-encoding",
-  "version": "3.0.0-alpha.0",
+  "version": "3.0.0-alpha.4",
   "description": "Http content encoding middleware for the middy framework",
   "type": "module",
   "engines": {
@@ -13,11 +13,13 @@
   "exports": "./index.js",
   "types": "index.d.ts",
   "files": [
+    "index.js",
     "index.d.ts"
   ],
   "scripts": {
     "test": "npm run test:unit",
-    "test:unit": "ava"
+    "test:unit": "ava",
+    "test:benchmark": "node __benchmarks__/index.js"
   },
   "license": "MIT",
   "keywords": [
@@ -49,10 +51,10 @@
   },
   "homepage": "https://github.com/middyjs/middy#readme",
   "dependencies": {
-    "@middy/util": "^3.0.0-alpha.0"
+    "@middy/util": "^3.0.0-alpha.4"
   },
   "devDependencies": {
-    "@middy/core": "^3.0.0-alpha.0"
+    "@middy/core": "^3.0.0-alpha.4"
   },
-  "gitHead": "c533f62841c8a39d061d7b94f30ba178f002c8db"
+  "gitHead": "d4bea7f4e21f6a9bbb1f6f6908361169598b9e53"
 }