refactor response handling for each concurrency model

This will give each concurrency model more control over
particular code paths and serving static files.

1 files changed, 5 insertions, 6 deletions

diff --git a/lib/rainbows/http_response.rb b/lib/rainbows/http_response.rb
index ddab2f8..564d2d0 100644
--- a/lib/rainbows/http_response.rb
+++ b/lib/rainbows/http_response.rb
@@ -4,20 +4,19 @@
 # Cramp 0.11 relies on this, and is only activated by Cramp
 if defined?(Cramp) && defined?(Rainbows::EventMachine::Client)
   class Rainbows::HttpResponse
-    class << self
-      include Rainbows::Response
-      alias write write_response
+    # dummy method for Cramp to alias_method_chain
+    def self.write(client, response, out)
     end
   end
 
   module Rainbows::EventMachine::CrampSocket
-    def write_header(_, response, out)
+    def em_write_response(response, alive = false)
       if websocket?
         write web_socket_upgrade_data
         web_socket_handshake!
-        out = nil # disable response headers
+        response[1] = nil # disable response headers
       end
-      super(self, response, out)
+      super
     end
   end