module ActiveResourceOAuthClient

# File lib/aeolus_image/active_resource_oauth_client.rb, line 26
def request_with_oauth(method, path, *arguments)
  @oauth_config = Aeolus::Image::Factory::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 => @oauth_config[:site] )
    token = OAuth::AccessToken.new(oauth_consumer)
    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 =~ %r^OAuth /}.select{|h| h =~ %roauth_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

# File lib/aeolus_image/active_resource_oauth_client.rb, line 60
def use_oauth_for_url?(url)
  Aeolus::Image::Factory::Base.use_oauth? and
    url.include?(Aeolus::Image::Factory::Base.config[:site])
end