lib/rat_hole.rb in mikehale-rat-hole-0.1.3 vs lib/rat_hole.rb in mikehale-rat-hole-0.1.4

@@ -4,21 +4,21 @@
 require 'delegate'
 require 'util'
 
 class RatHole
 
-  VERSION = '0.1.3'
+  VERSION = '0.1.4'
 
   def initialize(host)
     @host = host
   end
 
   def process_user_request(rack_request)
     rack_request
   end
 
-  def process_server_response(rack_response)
+  def process_server_response(rack_response, rack_request)
     rack_response
   end
 
   def call(env)
     Net::HTTP.start(@host) do |http|
@@ -42,10 +42,10 @@
       code = response.code.to_i
       headers = response.to_hash
       body = response.body || ''
       headers.delete('transfer-encoding')
 
-      process_server_response(Rack::Response.new(body, code, headers)).finish
+      process_server_response(Rack::Response.new(body, code, headers), source_request).finish
     end
   end
 
   def request_headers(env)
     env.select{|k,v| k =~ /^HTTP/}.inject({}) do |h, e|