about summary refs log tree commit homepage
path: root/lib/rainbows
diff options
context:
space:
mode:
Diffstat (limited to 'lib/rainbows')
-rw-r--r--lib/rainbows/base.rb3
-rw-r--r--lib/rainbows/event_machine.rb9
-rw-r--r--lib/rainbows/rev/deferred_response.rb7
3 files changed, 5 insertions, 14 deletions
diff --git a/lib/rainbows/base.rb b/lib/rainbows/base.rb
index a773722..8e5031f 100644
--- a/lib/rainbows/base.rb
+++ b/lib/rainbows/base.rb
@@ -33,8 +33,7 @@ module Rainbows
     if IO.respond_to?(:copy_stream)
       def write_body(client, body)
         if body.respond_to?(:to_path)
-          io = body.respond_to?(:to_io) ? body.to_io : body.to_path
-          IO.copy_stream(io, client)
+          IO.copy_stream(Rainbows.body_to_io(body), client)
         else
           body.each { |chunk| client.write(chunk) }
         end
diff --git a/lib/rainbows/event_machine.rb b/lib/rainbows/event_machine.rb
index b19bdec..12d9e5e 100644
--- a/lib/rainbows/event_machine.rb
+++ b/lib/rainbows/event_machine.rb
@@ -112,18 +112,15 @@ module Rainbows
         end
 
         headers = Rack::Utils::HeaderHash.new(response[1])
-        path = body.to_path
-        io = body.to_io if body.respond_to?(:to_io)
-        io ||= IO.new($1.to_i) if path =~ %r{\A/dev/fd/(\d+)\z}
-        io ||= File.open(path, 'rb') # could be a named pipe
-
+        io = Rainbows.body_to_io(body)
         st = io.stat
+
         if st.file?
           headers.delete('Transfer-Encoding')
           headers['Content-Length'] ||= st.size.to_s
           response = [ response.first, headers.to_hash, [] ]
           HttpResponse.write(self, response, out)
-          stream = stream_file_data(path)
+          stream = stream_file_data(body.to_path)
           stream.callback { quit } unless alive
         elsif st.socket? || st.pipe?
           do_chunk = !!(headers['Transfer-Encoding'] =~ %r{\Achunked\z}i)
diff --git a/lib/rainbows/rev/deferred_response.rb b/lib/rainbows/rev/deferred_response.rb
index dd7a229..62a308c 100644
--- a/lib/rainbows/rev/deferred_response.rb
+++ b/lib/rainbows/rev/deferred_response.rb
@@ -17,12 +17,7 @@ module Rainbows
             return HttpResponse.write(client, response, out)
 
         headers = HH.new(headers)
-
-        # to_io is not part of the Rack spec, but make an exception
-        # here since we can't get here without checking to_path first
-        io = body.to_io if body.respond_to?(:to_io)
-        io ||= ::IO.new($1.to_i) if body.to_path =~ %r{\A/dev/fd/(\d+)\z}
-        io ||= File.open(body.to_path, 'rb')
+        io = Rainbows.body_to_io(body)
         st = io.stat
 
         if st.socket? || st.pipe?