1 files changed, 14 insertions, 7 deletions
diff --git a/lib/rainbows/http_response.rb b/lib/rainbows/http_response.rb
index 811a793..677b5a7 100644
--- a/lib/rainbows/http_response.rb
+++ b/lib/rainbows/http_response.rb
@@ -6,7 +6,8 @@ module Rainbows::HttpResponse
  
    CODES = Unicorn::HttpResponse::CODES
  
-  def self.header_string(status, headers, out)
+  def response_header(response, out)
+    status, headers = response
      status = CODES[status.to_i] || status
  
      headers.each do |key, value|
@@ -25,13 +26,19 @@ module Rainbows::HttpResponse
      "#{out.join('')}\r\n"
    end
  
-  def self.write(socket, rack_response, out = [])
-    status, headers, body = rack_response
-    out and socket.write(header_string(status, headers, out))
+  def write_header(socket, response, out)
+    out and socket.write(response_header(response, out))
+  end
+
+  def write_response(socket, response, out)
+    write_header(socket, response, out)
+    write_body(socket, response[2])
+  end
  
-    body.each { |chunk| socket.write(chunk) }
-    ensure
-      body.respond_to?(:close) and body.close
+  # called after forking
+  def self.setup(klass)
+    require('rainbows/http_response/body') and
+      klass.__send__(:include, Rainbows::HttpResponse::Body)
    end
  end
  # :startdoc: