Merge pull request #2 from pexels/local-headers

Allow headers to be sent to the API to be defined locally

.gitignore

@@ -1,3 +1,4 @@
 .gems.up.to.date
 .env
 *.gem
+.headers

lib/pexels.rb

@@ -4,6 +4,21 @@ module Pexels
 
   class << self
     attr_reader :api_base_url, :api_version
+
+    # Local headers can be defined inside a `.headers` file at the project root,
+    # with the following format:
+    #
+    # header1=value
+    # header2=value
+    # etc.
+    #
+    def local_headers
+      @local_headers ||= if File.exist?('.headers')
+          File.read('.headers').split.to_h { |header| header.split('=') }
+        else
+          {}
+        end
+    end
   end
 end
 

lib/pexels/client.rb

@@ -4,7 +4,6 @@ class Pexels::Client
   attr_reader :api_key,
               :ratelimit_remaining
 
-
   def initialize(api_key = ENV['PEXELS_API_KEY'])
     @api_key = api_key
   end
@@ -19,18 +18,20 @@ def videos
 
   def request(path, method: 'GET', params: {})
     url = File.join(Pexels.api_base_url, path)
-    puts "Requesting #{url}" if ENV['DEBUG']
+    headers = {
+      'Authorization' => api_key
+    }.merge(Pexels.local_headers)
+
+    puts "Requesting #{url} with #{headers}" if ENV['DEBUG']
 
     results = Requests.request(
       method,
       url,
       params: params,
-      headers: {
-        'Authorization' => api_key
-      }
+      headers: headers
     )
 
-    @ratelimit_remaining = results.headers['x-ratelimit-remaining'].first.to_i
+    @ratelimit_remaining = results.headers['x-ratelimit-remaining']&.first&.to_i
 
     return JSON.parse(results.body)
   rescue StandardError => exception