lib/rapgenius/client.rb in rapgenius-1.0.5 vs lib/rapgenius/client.rb in rapgenius-1.1.0

@@ -1,15 +1,18 @@
 require 'httparty'
 
 module RapGenius
   module Client
+    class << self
+      attr_accessor :access_token
+    end
+
     class HTTPClient
       include HTTParty
 
       format   :json
       base_uri 'https://api.rapgenius.com'
-      headers  'User-Agent' => "rapgenius.rb v#{RapGenius::VERSION}"
     end
 
     BASE_URL = HTTPClient.base_uri + "/".freeze
     PLAIN_TEXT_FORMAT = "plain".freeze
     DOM_TEXT_FORMAT = "dom".freeze
@@ -29,9 +32,12 @@
     end
 
     def fetch(url)
       response = HTTPClient.get(url, query: {
         text_format: "#{DOM_TEXT_FORMAT},#{PLAIN_TEXT_FORMAT}"
+      }, headers: {
+        'Authorization' => "Bearer #{RapGenius::Client.access_token}",
+        'User-Agent' => "rapgenius.rb v#{RapGenius::VERSION}"
       })
 
       if response.code != 200
         if response.code == 404
           raise RapGenius::NotFoundError