From 85e24f53e532aff154993ada176d605ae90d0055 Mon Sep 17 00:00:00 2001
From: Tai NGUYEN <tai.nguyen@skello.io>
Date: Sat, 2 Jul 2022 19:33:32 +0200
Subject: [PATCH] Add options to Set headers when initializing client

---
 README.md                  | 5 +++++
 lib/exponent-server-sdk.rb | 5 +++--
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 51259ce..927ffca 100644
--- a/README.md
+++ b/README.md
@@ -36,6 +36,11 @@ Optional arguments: `gzip: true`
 ```ruby
 client = Exponent::Push::Client.new
 # client = Exponent::Push::Client.new(gzip: true)  # for compressed, faster requests
+# client = Exponent::Push::Client.new(headers: {
+#  'Content-Type' => 'application/json',
+#  'Accept' => 'application/json',
+#  'Authorization' => 'Bearer ACCESS_TOKEN',
+# })  # to set or update headers
 
 messages = [{
   to: "ExponentPushToken[xxxxxxxxxxxxxxxxxxxxxx]",
diff --git a/lib/exponent-server-sdk.rb b/lib/exponent-server-sdk.rb
index 2361259..351121c 100644
--- a/lib/exponent-server-sdk.rb
+++ b/lib/exponent-server-sdk.rb
@@ -42,6 +42,7 @@ def initialize(**args)
         # future versions will deprecate this
         @response_handler = args[:response_handler] || ResponseHandler.new
         @gzip             = args[:gzip] == true
+        @headers          = args[:headers] ? headers.merge!(args[:headers]) : headers
       end
 
       # returns a string response with parsed success json or error
@@ -77,7 +78,7 @@ def push_notifications(messages)
         @http_client.post(
           push_url,
           body: messages.to_json,
-          headers: headers,
+          headers: @headers,
           accept_encoding: @gzip
         )
       end
@@ -90,7 +91,7 @@ def get_receipts(receipt_ids)
         @http_client.post(
           receipts_url,
           body: { ids: receipt_ids }.to_json,
-          headers: headers,
+          headers: @headers,
           accept_encoding: @gzip
         )
       end