From 7bdc6f17a4ffba5ada317fc701305ca6de62c6bb Mon Sep 17 00:00:00 2001
From: Mikko Leino <mikko.leino@reaktor.com>
Date: Fri, 30 Nov 2018 16:47:48 +0100
Subject: [PATCH] Add support for origin custom headers

---
 README.md | 3 +++
 index.js  | 9 ++++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index cf42659..39f2949 100644
--- a/README.md
+++ b/README.md
@@ -57,6 +57,9 @@ custom:
       - page
       - per_page
     priceClass: PriceClass_100
+    originCustomHeaders:
+      - x-api-key: foo
+      - bar: quux
 ```
 
 ### Notes
diff --git a/index.js b/index.js
index dfe3337..95a77d3 100644
--- a/index.js
+++ b/index.js
@@ -98,7 +98,14 @@ class ServerlessApiCloudFrontPlugin {
   }
 
   prepareOrigins(distributionConfig) {
-    distributionConfig.Origins[0].OriginPath = `/${this.options.stage}`;
+    const origin = _.head(distributionConfig.Origins)
+    const originCustomHeaders = this.getConfig('originCustomHeaders', [])
+
+    origin.OriginCustomHeaders = originCustomHeaders
+      .map(_.toPairs)
+      .map(_.head)
+      .map(_.partial(_.zipObject, ['HeaderName', 'HeaderValue']))
+    origin.OriginPath = `/${this.options.stage}`;
   }
 
   prepareCookies(distributionConfig) {