69 lines
2.1 KiB
Ruby
69 lines
2.1 KiB
Ruby
|
module Faraday
|
||
|
|
class Adapter
|
||
|
|
class Patron < Faraday::Adapter
|
||
|
|
dependency 'patron'
|
||
|
|
|
||
|
|
def initialize(app, &block)
|
||
|
|
super(app)
|
||
|
|
@block = block
|
||
|
|
end
|
||
|
|
|
||
|
|
def call(env)
|
||
|
|
super
|
||
|
|
|
||
|
|
# TODO: support streaming requests
|
||
|
|
env[:body] = env[:body].read if env[:body].respond_to? :read
|
||
|
|
|
||
|
|
session = @session ||= create_session
|
||
|
|
|
||
|
|
if req = env[:request]
|
||
|
|
session.timeout = session.connect_timeout = req[:timeout] if req[:timeout]
|
||
|
|
session.connect_timeout = req[:open_timeout] if req[:open_timeout]
|
||
|
|
|
||
|
|
if proxy = req[:proxy]
|
||
|
|
proxy_uri = proxy[:uri].dup
|
||
|
|
proxy_uri.user = proxy[:user] && Utils.escape(proxy[:user]).gsub('+', '%20')
|
||
|
|
proxy_uri.password = proxy[:password] && Utils.escape(proxy[:password]).gsub('+', '%20')
|
||
|
|
session.proxy = proxy_uri.to_s
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
response = begin
|
||
|
|
data = env[:body] ? env[:body].to_s : nil
|
||
|
|
session.request(env[:method], env[:url].to_s, env[:request_headers], :data => data)
|
||
|
|
rescue Errno::ECONNREFUSED, ::Patron::ConnectionFailed
|
||
|
|
raise Error::ConnectionFailed, $!
|
||
|
|
end
|
||
|
|
|
||
|
|
save_response(env, response.status, response.body, response.headers)
|
||
|
|
|
||
|
|
@app.call env
|
||
|
|
rescue ::Patron::TimeoutError => err
|
||
|
|
raise Faraday::Error::TimeoutError, err
|
||
|
|
rescue ::Patron::Error => err
|
||
|
|
if err.message.include?("code 407")
|
||
|
|
raise Error::ConnectionFailed, %{407 "Proxy Authentication Required "}
|
||
|
|
else
|
||
|
|
raise Error::ConnectionFailed, err
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
if loaded? && defined?(::Patron::Request::VALID_ACTIONS)
|
||
|
|
# HAX: helps but doesn't work completely
|
||
|
|
# https://github.com/toland/patron/issues/34
|
||
|
|
::Patron::Request::VALID_ACTIONS.tap do |actions|
|
||
|
|
actions << :patch unless actions.include? :patch
|
||
|
|
actions << :options unless actions.include? :options
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
def create_session
|
||
|
|
session = ::Patron::Session.new
|
||
|
|
session.insecure = true
|
||
|
|
@block.call(session) if @block
|
||
|
|
session
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|