module ActiveResourceOAuthClient
Monkey-patch ActiveResource to allow us to merge our OAuth headers in. Portions of the below are taken from Active Resource which is MIT licensed; hence this whole file is being licensed under the MIT License to err on the side of safety.
Public Instance Methods
default_header()
click to toggle source
# File lib/deadwood/active_resource_oauth_client.rb, line 74 def default_header @default_header ||= {} config = Deadwood::Katello::Base.config || {} @default_header = {'HTTP_KATELLO_USER' => config[:katello_user]} end
request_with_oauth(method, path, *arguments)
click to toggle source
# File lib/deadwood/active_resource_oauth_client.rb, line 29 def request_with_oauth(method, path, *arguments) @oauth_config = Deadwood::Katello::Base.config || {} # Take care to fall back to the standard request method if we don't have full OAuth credentials unless use_oauth_for_url?("#{site.scheme}://#{site.host}:#{site.port}#{path}") return request_without_oauth(method, path, *arguments) end result = ActiveSupport::Notifications.instrument("request.active_resource") do |payload| payload[:method] = method payload[:request_uri] = "#{site.scheme}://#{site.host}:#{site.port}#{path}" oauth_consumer = OAuth::Consumer.new( @oauth_config[:consumer_key], @oauth_config[:consumer_secret], :site => "#{site.scheme}://#{site.host}:#{site.port}" ) token = OAuth::AccessToken.new(oauth_consumer) # Katello objects must not include the id attribute on updates if method.to_s.include?('put') obj = JSON.parse(arguments.first) obj[obj.keys.first].delete('id') arguments[0] = obj.to_json end base_request = oauth_consumer.create_signed_request(method, path, token, {}, *arguments) payload[:result] = http.request(base_request) end # Error-handling code from OAuth # http://wiki.oauth.net/w/page/12238543/ProblemReporting auth_header = result.to_hash['www-authenticate'] problem_header = auth_header ? auth_header.select{|h| h =~ /^OAuth /}.select{|h| h =~ /oauth_problem/}.first : nil if auth_header && problem_header params = OAuth::Helper.parse_header(problem_header) raise OAuth::Problem.new(params.delete("oauth_problem"), result, params) end # Error-handling code from ActiveResource handle_response(result) rescue Timeout::Error => e raise TimeoutError.new(e.message) rescue OpenSSL::SSL::SSLError => e raise SSLError.new(e.message) end
use_oauth_for_url?(url)
click to toggle source
# File lib/deadwood/active_resource_oauth_client.rb, line 69 def use_oauth_for_url?(url) Deadwood::Katello::Base.use_oauth? and url.include?(Deadwood::Katello::Base.config[:site]) end