3rdparty/modules/aviator/lib/puppet/feature/faraday/adapter/patron.rb

   1 module Faraday
   2   class Adapter
   3     class Patron < Faraday::Adapter
   4       dependency 'patron'
   5
   6       def initialize(app, &block)
   7         super(app)
   8         @block = block
   9       end
  10
  11       def call(env)
  12         super
  13
  14         # TODO: support streaming requests
  15         env[:body] = env[:body].read if env[:body].respond_to? :read
  16
  17         session = @session ||= create_session
  18
  19         if req = env[:request]
  20           session.timeout = session.connect_timeout = req[:timeout] if req[:timeout]
  21           session.connect_timeout = req[:open_timeout]              if req[:open_timeout]
  22
  23           if proxy = req[:proxy]
  24             proxy_uri = proxy[:uri].dup
  25             proxy_uri.user = proxy[:user] && Utils.escape(proxy[:user]).gsub('+', '%20')
  26             proxy_uri.password = proxy[:password] && Utils.escape(proxy[:password]).gsub('+', '%20')
  27             session.proxy = proxy_uri.to_s
  28           end
  29         end
  30
  31         response = begin
  32           data = env[:body] ? env[:body].to_s : nil
  33           session.request(env[:method], env[:url].to_s, env[:request_headers], :data => data)
  34         rescue Errno::ECONNREFUSED, ::Patron::ConnectionFailed
  35           raise Error::ConnectionFailed, $!
  36         end
  37
  38         save_response(env, response.status, response.body, response.headers)
  39
  40         @app.call env
  41       rescue ::Patron::TimeoutError => err
  42         if err.message == "Connection time-out"
  43           raise Faraday::Error::ConnectionFailed, err
  44         else
  45           raise Faraday::Error::TimeoutError, err
  46         end
  47       rescue ::Patron::Error => err
  48         if err.message.include?("code 407")
  49           raise Error::ConnectionFailed, %{407 "Proxy Authentication Required "}
  50         else
  51           raise Error::ConnectionFailed, err
  52         end
  53       end
  54
  55       if loaded? && defined?(::Patron::Request::VALID_ACTIONS)
  56         # HAX: helps but doesn't work completely
  57         # https://github.com/toland/patron/issues/34
  58         ::Patron::Request::VALID_ACTIONS.tap do |actions|
  59           actions << :patch unless actions.include? :patch
  60           actions << :options unless actions.include? :options
  61         end
  62       end
  63
  64       def create_session
  65         session = ::Patron::Session.new
  66         session.insecure = true
  67         @block.call(session) if @block
  68         session
  69       end
  70     end
  71   end
  72 end