twitter-status-bot/.gems/gems/faraday-0.9.0/lib/faraday/response.rb

94 lines
2.1 KiB
Ruby
Raw Normal View History

2014-09-03 08:49:59 +00:00
require 'forwardable'
module Faraday
class Response
# Used for simple response middleware.
class Middleware < Faraday::Middleware
def call(env)
@app.call(env).on_complete do |environment|
on_complete(environment)
end
end
# Override this to modify the environment after the response has finished.
# Calls the `parse` method if defined
def on_complete(env)
env.body = parse(env.body) if respond_to?(:parse) && env.parse_body?
end
end
extend Forwardable
extend MiddlewareRegistry
register_middleware File.expand_path('../response', __FILE__),
:raise_error => [:RaiseError, 'raise_error'],
:logger => [:Logger, 'logger']
def initialize(env = nil)
@env = Env.from(env) if env
@on_complete_callbacks = []
end
attr_reader :env
def_delegators :env, :to_hash
def status
finished? ? env.status : nil
end
def headers
finished? ? env.response_headers : {}
end
def_delegator :headers, :[]
def body
finished? ? env.body : nil
end
def finished?
!!env
end
def on_complete
if not finished?
@on_complete_callbacks << Proc.new
else
yield(env)
end
return self
end
def finish(env)
raise "response already finished" if finished?
@env = Env.from(env)
@on_complete_callbacks.each { |callback| callback.call(env) }
return self
end
def success?
finished? && env.success?
end
# because @on_complete_callbacks cannot be marshalled
def marshal_dump
!finished? ? nil : {
:status => @env.status, :body => @env.body,
:response_headers => @env.response_headers
}
end
def marshal_load(env)
@env = Env.from(env)
end
# Expand the env with more properties, without overriding existing ones.
# Useful for applying request params after restoring a marshalled Response.
def apply_request(request_env)
raise "response didn't finish yet" unless finished?
@env = Env.from(request_env).update(@env)
return self
end
end
end