@middy/s3-object-response 3.0.0-alpha.0 → 3.0.0-alpha.1

package/index.js

@@ -0,0 +1,103 @@
+import __https from 'https';
+import { URL } from 'url';
+import { canPrefetch, createPrefetchClient, createClient } from '@middy/util';
+import S3 from 'aws-sdk/clients/s3.js';
+const defaults = {
+  AwsClient: S3,
+  awsClientOptions: {},
+  awsClientAssumeRole: undefined,
+  awsClientCapture: undefined,
+  httpsCapture: undefined,
+  disablePrefetch: false,
+  bodyType: undefined,
+  __https
+};
+let https = __https;
+
+const s3ObjectResponseMiddleware = (opts = {}) => {
+  const options = { ...defaults,
+    ...opts
+  };
+
+  if (!['stream', 'promise'].includes(options.bodyType)) {
+    throw new Error('bodyType is invalid.');
+  }
+
+  if (options.httpsCapture) {
+    https = options.httpsCapture(options.__https);
+  }
+
+  let client;
+
+  if (canPrefetch(options)) {
+    client = createPrefetchClient(options);
+  }
+
+  const s3ObjectResponseMiddlewareBefore = async request => {
+    const {
+      inputS3Url,
+      outputRoute,
+      outputToken
+    } = request.event.getObjectContext;
+    request.internal.s3ObjectResponse = {
+      RequestRoute: outputRoute,
+      RequestToken: outputToken
+    };
+    const parsedInputS3Url = new URL(inputS3Url);
+    const fetchOptions = {
+      method: 'GET',
+      host: parsedInputS3Url.hostname,
+      path: parsedInputS3Url.pathname
+    };
+    request.context.s3Object = fetchType(options.bodyType, fetchOptions);
+  };
+
+  const s3ObjectResponseMiddlewareAfter = async request => {
+    var _request$response;
+
+    if (!client) {
+      client = await createClient(options, request);
+    }
+
+    (_request$response = request.response).Body ?? (_request$response.Body = request.response.body);
+    delete request.response.body;
+    return client.writeGetObjectResponse({ ...request.response,
+      ...request.internal.s3ObjectResponse
+    }).promise().then(() => ({
+      statusCode: 200
+    }));
+  };
+
+  return {
+    before: s3ObjectResponseMiddlewareBefore,
+    after: s3ObjectResponseMiddlewareAfter
+  };
+};
+
+const fetchType = (type, fetchOptions) => {
+  if (type === 'stream') {
+    return fetchStream(fetchOptions);
+  } else if (type === 'promise') {
+    return fetchPromise(fetchOptions);
+  }
+
+  return null;
+};
+
+const fetchStream = fetchOptions => {
+  return https.request(fetchOptions);
+};
+
+const fetchPromise = fetchOptions => {
+  return new Promise((resolve, reject) => {
+    let data = '';
+    const stream = fetchStream(fetchOptions);
+    stream.on('data', chunk => {
+      data += chunk;
+    });
+    stream.on('end', () => resolve(data));
+    stream.on('error', error => reject(error));
+  });
+};
+
+export default s3ObjectResponseMiddleware;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@middy/s3-object-response",
-  "version": "3.0.0-alpha.0",
+  "version": "3.0.0-alpha.1",
   "description": "S3 object response handling middleware for the middy framework",
   "type": "module",
   "engines": {
@@ -13,6 +13,7 @@
   "exports": "./index.js",
   "types": "index.d.ts",
   "files": [
+    "index.js",
     "index.d.ts"
   ],
   "scripts": {
@@ -46,12 +47,12 @@
   },
   "homepage": "https://github.com/middyjs/middy#readme",
   "dependencies": {
-    "@middy/util": "^3.0.0-alpha.0"
+    "@middy/util": "^3.0.0-alpha.1"
   },
   "devDependencies": {
-    "@middy/core": "^3.0.0-alpha.0",
+    "@middy/core": "^3.0.0-alpha.1",
     "aws-sdk": "^2.939.0",
     "aws-xray-sdk": "^3.3.3"
   },
-  "gitHead": "c533f62841c8a39d061d7b94f30ba178f002c8db"
+  "gitHead": "a14125c6b2e21b181824f9985a919a47f1e4711f"
 }