Fix incompatibility with faraday 0.6.2
[arvados.git] / lib / google / api_client.rb
index 913ce4d86a00ea56d88ce95251725024df333768..b2fa0b3c447158d1565608b00682745f9a4deb5a 100644 (file)
@@ -31,8 +31,9 @@ require 'google/api_client/media'
 require 'google/api_client/service_account'
 require 'google/api_client/batch'
 require 'google/api_client/gzip'
+require 'google/api_client/charset'
 require 'google/api_client/client_secrets'
-require 'google/api_client/railtie' if defined?(Rails::Railtie)
+require 'google/api_client/railtie' if defined?(Rails)
 
 module Google
 
@@ -40,7 +41,7 @@ module Google
   # This class manages APIs communication.
   class APIClient
     include Google::APIClient::Logging
-    
+
     ##
     # Creates a new Google API client.
     #
@@ -53,13 +54,16 @@ module Google
     #     <li><code>:two_legged_oauth_1</code></li>
     #     <li><code>:oauth_1</code></li>
     #     <li><code>:oauth_2</code></li>
+    #     <li><code>:google_app_default</code></li>
     #   </ul>
     # @option options [Boolean] :auto_refresh_token (true)
     #   The setting that controls whether or not the api client attempts to
-    #   refresh authorization when a 401 is hit in #execute. If the token does 
+    #   refresh authorization when a 401 is hit in #execute. If the token does
     #   not support it, this option is ignored.
     # @option options [String] :application_name
     #   The name of the application using the client.
+    # @option options [String | Array | nil] :scope
+    #   The scope(s) used when using google application default credentials
     # @option options [String] :application_version
     #   The version number of the application using the client.
     # @option options [String] :user_agent
@@ -75,9 +79,14 @@ module Google
     # @option options [String] :ca_file
     #   Optional set of root certificates to use when validating SSL connections.
     #   By default, a bundled set of trusted roots will be used.
+    # @options options[Hash] :force_encoding
+    #   Experimental option. True if response body should be force encoded into the charset
+    #   specified in the Content-Type header. Mostly intended for compressed content.
+    # @options options[Hash] :faraday_options
+    #   Pass through of options to set on the Faraday connection
     def initialize(options={})
       logger.debug { "#{self.class} - Initializing client with options #{options}" }
-      
+
       # Normalize key to String to allow indifferent access.
       options = options.inject({}) do |accu, (key, value)|
         accu[key.to_sym] = value
@@ -97,14 +106,20 @@ module Google
       else
         logger.warn { "#{self.class} - Please provide :application_name and :application_version when initializing the client" }
       end
+
+      proxy = options[:proxy] || Object::ENV["http_proxy"]
+
       self.user_agent = options[:user_agent] || (
         "#{application_string} " +
-        "google-api-ruby-client/#{Google::APIClient::VERSION::STRING} #{ENV::OS_VERSION} (gzip)"
+        "google-api-ruby-client/#{Google::APIClient::VERSION::STRING} #{ENV::OS_VERSION}".strip + " (gzip)"
       ).strip
       # The writer method understands a few Symbols and will generate useful
       # default authentication mechanisms.
       self.authorization =
         options.key?(:authorization) ? options[:authorization] : :oauth_2
+      if !options['scope'].nil? and self.authorization.respond_to?(:scope=)
+        self.authorization.scope = options['scope']
+      end
       self.auto_refresh_token = options.fetch(:auto_refresh_token) { true }
       self.key = options[:key]
       self.user_ip = options[:user_ip]
@@ -115,11 +130,24 @@ module Google
       @discovered_apis = {}
       ca_file = options[:ca_file] || File.expand_path('../../cacerts.pem', __FILE__)
       self.connection = Faraday.new do |faraday|
+        faraday.response :charset if options[:force_encoding]
         faraday.response :gzip
         faraday.options.params_encoder = Faraday::FlatParamsEncoder
         faraday.ssl.ca_file = ca_file
         faraday.ssl.verify = true
+        if faraday.respond_to?(:proxy=)
+          # faraday >= 0.6.2
+          faraday.proxy = proxy
+        else
+          # older versions of faraday
+          faraday.proxy proxy
+        end
         faraday.adapter Faraday.default_adapter
+        if options[:faraday_option].is_a?(Hash)
+          options[:faraday_option].each_pair do |option, value|
+            faraday.options.send("#{option}=", value)
+          end
+        end
       end
       return self
     end
@@ -158,6 +186,10 @@ module Google
           :client_credential_secret => nil,
           :two_legged => true
         )
+      when :google_app_default
+        require 'googleauth'
+        new_authorization = Google::Auth.get_application_default
+
       when :oauth_2
         require 'signet/oauth_2/client'
         # NOTE: Do not rely on this default value, as it may change
@@ -188,7 +220,7 @@ module Google
 
     ##
     # The setting that controls whether or not the api client attempts to
-    # refresh authorization when a 401 is hit in #execute. 
+    # refresh authorization when a 401 is hit in #execute.
     #
     # @return [Boolean]
     attr_accessor :auto_refresh_token
@@ -235,7 +267,7 @@ module Google
 
     ##
     # Number of times to retry on recoverable errors
-    # 
+    #
     # @return [FixNum]
     #  Number of retries
     attr_accessor :retries
@@ -262,10 +294,12 @@ module Google
     # @param [String, Symbol] api The API name.
     # @param [String] version The desired version of the API.
     # @param [Addressable::URI] uri The URI of the discovery document.
+    # @return [Google::APIClient::API] The service object.
     def register_discovery_uri(api, version, uri)
       api = api.to_s
       version = version || 'v1'
       @discovery_uris["#{api}:#{version}"] = uri
+      discovered_api(api, version)
     end
 
     ##
@@ -294,6 +328,7 @@ module Google
     # @param [String] version The desired version of the API.
     # @param [String, StringIO] discovery_document
     #   The contents of the discovery document.
+    # @return [Google::APIClient::API] The service object.
     def register_discovery_document(api, version, discovery_document)
       api = api.to_s
       version = version || 'v1'
@@ -308,6 +343,7 @@ module Google
       end
       @discovery_documents["#{api}:#{version}"] =
         MultiJson.load(discovery_document)
+      discovered_api(api, version)
     end
 
     ##
@@ -441,7 +477,7 @@ module Google
     # Verifies an ID token against a server certificate. Used to ensure that
     # an ID token supplied by an untrusted client-side mechanism is valid.
     # Raises an error if the token is invalid or missing.
-    # 
+    #
     # @deprecated Use the google-id-token gem for verifying JWTs
     def verify_id_token!
       require 'jwt'
@@ -460,7 +496,7 @@ module Google
       else
         check_cached_certs = lambda do
           valid = false
-          for key, cert in @certificates
+          for _key, cert in @certificates
             begin
               self.authorization.decoded_id_token(cert.public_key)
               valid = true
@@ -550,7 +586,7 @@ module Google
     #     - (TrueClass, FalseClass) :authenticated (default: true) -
     #       `true` if the request must be signed or somehow
     #       authenticated, `false` otherwise.
-    #     - (TrueClass, FalseClass) :gzip (default: true) - 
+    #     - (TrueClass, FalseClass) :gzip (default: true) -
     #       `true` if gzip enabled, `false` otherwise.
     #     - (FixNum) :retries -
     #       # of times to retry on recoverable errors
@@ -590,7 +626,7 @@ module Google
         options.update(params.shift) if params.size > 0
         request = self.generate_request(options)
       end
-      
+
       request.headers['User-Agent'] ||= '' + self.user_agent unless self.user_agent.nil?
       request.headers['Accept-Encoding'] ||= 'gzip' unless options[:gzip] == false
       request.headers['Content-Type'] ||= ''
@@ -599,11 +635,11 @@ module Google
 
       connection = options[:connection] || self.connection
       request.authorization = options[:authorization] || self.authorization unless options[:authenticated] == false
-      
+
       tries = 1 + (options[:retries] || self.retries)
       attempt = 0
 
-      Retriable.retriable :tries => tries, 
+      Retriable.retriable :tries => tries,
                           :on => [TransmissionError],
                           :on_retry => client_error_handler,
                           :interval => lambda {|attempts| (2 ** attempts) + rand} do
@@ -612,10 +648,9 @@ module Google
         # This 2nd level retriable only catches auth errors, and supports 1 retry, which allows
         # auth to be re-attempted without having to retry all sorts of other failures like
         # NotFound, etc
-        Retriable.retriable :tries => ((expired_auth_retry || tries > 1) && attempt == 1) ? 2 : 1, 
+        Retriable.retriable :tries => ((expired_auth_retry || tries > 1) && attempt == 1) ? 2 : 1,
                             :on => [AuthorizationError],
-                            :on_retry => authorization_error_handler(request.authorization),
-                            :interval => lambda {|attempts| (2 ** attempts) + rand} do
+                            :on_retry => authorization_error_handler(request.authorization) do
           result = request.send(connection, true)
 
           case result.status
@@ -680,7 +715,7 @@ module Google
       end
       return Addressable::Template.new(@base_uri + template).expand(mapping)
     end
-    
+
 
     ##
     # Returns on proc for special processing of retries for authorization errors
@@ -690,7 +725,7 @@ module Google
     #   OAuth 2 credentials
     # @return [Proc]
     def authorization_error_handler(authorization)
-      can_refresh = authorization.respond_to?(:refresh_token) && auto_refresh_token 
+      can_refresh = authorization.respond_to?(:refresh_token) && auto_refresh_token
       Proc.new do |exception, tries|
         next unless exception.kind_of?(AuthorizationError)
         if can_refresh
@@ -719,5 +754,3 @@ module Google
   end
 
 end
-
-require 'google/api_client/version'