Merge pull request dougmoscrop#1 from eelayoubi/feature/create-lambda-edge-type-for-aws

feat: add lambda edge origin request to AWS

Author: eelayoubi

lib/provider/aws/clean-up-event.js renamed to lib/provider/aws/api-gw/clean-up-event.js

@@ -2,7 +2,7 @@
 
 const URL = require('url');
 
-const Request = require('../../request');
+const Request = require('../../../request');
 
 function requestMethod(event) {
   if (event.version === '2.0') {

lib/provider/aws/create-request.js renamed to lib/provider/aws/api-gw/create-request.js

@@ -1,7 +1,7 @@
 'use strict';
 
 const isBinary = require('./is-binary');
-const Response = require('../../response');
+const Response = require('../../../response');
 const sanitizeHeaders = require('./sanitize-headers');
 
 module.exports = (event, response, options) => {

lib/provider/aws/format-response.js renamed to lib/provider/aws/api-gw/format-response.js

@@ -0,0 +1,15 @@
+const cleanUpEvent = require('./clean-up-event');
+
+const createRequest = require('./create-request');
+const formatResponse = require('./format-response');
+
+module.exports = options => {
+  return getResponse => async (event_, context = {}) => {
+    const event = cleanUpEvent(event_, options);
+
+    const request = createRequest(event, options);
+    const response = await getResponse(request, event, context);
+
+    return formatResponse(event, response, options);
+  };
+};

lib/provider/aws/api-gw/index.js

@@ -1,15 +1,10 @@
-const cleanUpEvent = require('./clean-up-event');
-
-const createRequest = require('./create-request');
-const formatResponse = require('./format-response');
+const apiGw = require('./api-gw');
+const lambdaEdgeOriginRequest = require('./lambda-edge-origin-request');
 
 module.exports = options => {
-  return getResponse => async (event_, context = {}) => {
-    const event = cleanUpEvent(event_, options);
-
-    const request = createRequest(event, options);
-    const response = await getResponse(request, event, context);
-
-    return formatResponse(event, response, options);
-  };
+  if (options.type === 'lambda-edge-origin-request') {
+    return lambdaEdgeOriginRequest(options)
+  } else {
+    return apiGw(options)
+  }
 };

lib/provider/aws/is-binary.js renamed to lib/provider/aws/api-gw/is-binary.js

@@ -0,0 +1,31 @@
+'use strict';
+
+function getPath({ uri }) {
+  return typeof uri === 'string' ? uri : '/';
+}
+
+module.exports = function cleanupEvent(evt, options) {
+  const event = (evt &&
+    evt.Records &&
+    evt.Records.length > 0 &&
+    evt.Records[0].cf) ||
+    {};
+
+  event.config = event.config || {};
+  
+  event.request = event.request || {};
+  event.request.body = event.request.body || '';
+  event.request.headers = event.request.headers || {};
+  event.request.method = event.request.method || 'GET';
+  event.request.uri = getPath(event.request);
+
+  if (options.basePath) {
+    const basePathIndex = event.path.indexOf(options.basePath);
+
+    if (basePathIndex > -1) {
+      event.path = event.path.substr(basePathIndex + options.basePath.length);
+    }
+  }
+
+  return event;
+};

lib/provider/aws/sanitize-headers.js renamed to lib/provider/aws/api-gw/sanitize-headers.js

@@ -0,0 +1,62 @@
+'use strict';
+
+const crypto = require('crypto');
+const Request = require('../../../request');
+
+function requestHeaders(event) {
+  let headers = Object.keys(event.request.headers).reduce((headers, key) => {
+    headers[event.request.headers[key][0].key.toLowerCase()] = event.request.headers[key][0].value;
+    return headers;
+  }, {});
+
+  headers['x-request-id'] = crypto.randomBytes(30).toString('base64')
+
+  return headers;
+}
+
+function requestBody(event) {
+  const type = typeof event.request.body;
+  const body = event && event.request && event.request.body;
+
+  if (!body) return '';
+
+  if (Buffer.isBuffer(event.request.body.data)) {
+    return event.request.body.data;
+  } else if (type === 'string') {
+    return Buffer.from(event.request.body.data, event.request.body.encoding === 'base64' ? 'base64' : 'utf8');
+  } else if (type === 'object') {
+    return Buffer.from(JSON.stringify(event.request.body.data));
+  }
+
+  throw new Error(`Unexpected event.body type: ${typeof event.request.body.data}`);
+}
+
+function getUrl(path, queryString) {
+  if (queryString) {
+    return `${path}?${queryString}`
+  }
+
+  return path;
+}
+
+module.exports = (event, options) => {
+  const method = event.request.method;
+  const remoteAddress = event.request.clientIp;
+  const headers = requestHeaders(event);
+  const body = requestBody(event);
+
+  if (typeof options.requestId === 'string' && options.requestId.length > 0) {
+    const header = options.requestId.toLowerCase();
+    headers[header] = headers[header] || event.config.requestId;
+  }
+
+  const req = new Request({
+    method,
+    headers,
+    body,
+    remoteAddress,
+    url: getUrl(event.request.uri, event.request.querystring)
+  });
+
+  return req;
+};

lib/provider/aws/index.js

@@ -0,0 +1,27 @@
+'use strict';
+const http = require('http');
+
+const isBinary = require('./is-binary');
+const Response = require('../../../response');
+const sanitizeHeaders = require('./sanitize-headers');
+
+module.exports = (response, options) => {
+  const { statusCode } = response;
+  const headers = Response.headers(response);
+
+  if (headers['transfer-encoding'] === 'chunked' || response.chunkedEncoding) {
+    throw new Error('chunked encoding not supported');
+  }
+
+  const isBase64Encoded = isBinary(headers, options);
+  const encoding = isBase64Encoded ? 'base64' : 'utf8';
+  let body = Response.body(response).toString(encoding);
+
+  return {
+    status: statusCode.toString(),
+    statusDescription: http.STATUS_CODES[statusCode],
+    headers: sanitizeHeaders(headers),
+    body,
+    bodyEncoding: isBase64Encoded ? 'base64' : 'text'
+  };
+};