Support both enable/disable attribute for multi value headers on ALB.

Setting found in target groups of lambda type.
CodeGenieApp · Mar 25, 2021 · 64e9b7f · 64e9b7f
1 parent 671f914
commit 64e9b7f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/event-sources/aws/alb.js b/src/event-sources/aws/alb.js
@@ -2,16 +2,22 @@ const { getRequestValuesFromEvent, getMultiValueHeaders } = require('../utils')
 
 const getRequestValuesFromAlbEvent = ({ event }) => getRequestValuesFromEvent({ event })
 const getResponseToAlb = ({
+  event,
   statusCode,
   body,
-  headers,
+  headers: responseHeaders,
   isBase64Encoded
 }) => {
-  const multiValueHeaders = getMultiValueHeaders({ headers })
+  const multiValueHeaders = event.multiValueHeaders ? getMultiValueHeaders({ headers: responseHeaders }) : undefined
+  const headers = Object.entries(responseHeaders).reduce((acc, [k, v]) => {
+    acc[k] = Array.isArray(v) ? v[0] : v
+    return acc
+  }, {})
 
   return {
     statusCode,
     body,
+    headers,
     multiValueHeaders,
     isBase64Encoded
   }

diff --git a/src/transport.js b/src/transport.js
@@ -10,6 +10,7 @@ function forwardResponse ({
   response,
   resolver,
   eventSource,
+  event,
   log
 }) {
   const statusCode = response.statusCode
@@ -30,6 +31,7 @@ function forwardResponse ({
   })
 
   const successResponse = eventSource.getResponse({
+    event,
     statusCode,
     body,
     headers,
@@ -141,6 +143,7 @@ async function forwardRequestToNodeServer ({
     response,
     resolver,
     eventSource,
+    event,
     log
   })
   return response