twitter-status-bot/.gems/gems/faraday-0.9.0/lib/faraday/adapter/patron.rb

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