Cleanup methods

lib/intervention/client.rb

@@ -1,6 +1,5 @@
 module Intervention
   class Client < EventMachine::Connection
-
     attr_reader :server, :parser
 
     def inspect
@@ -14,14 +13,13 @@ def post_init
     end
 
     def receive_data data
-      @parser.parse data
+      @parser.parse_data data
     end
 
-    def on_body_complete parser
+    def on_message_complete parser
       callback :request
-
       @parser.headers['host'] = Intervention.config.host_address
-      @request.headers['accept-encoding'] = "deflate"
+      @parser.headers['accept-encoding'] = "deflate"
       @server.send_data @parser.raw_data
     end
 

lib/intervention/server.rb

@@ -1,6 +1,5 @@
 module Intervention
   class Server < EventMachine::Connection
-
     attr_reader :client, :parser
 
     def inspect
@@ -18,15 +17,13 @@ def post_init
     end
 
     def receive_data data
-      @parser.parse data
+      @parser.parse_data data
     end
 
-    def on_message_complete
+    def on_message_complete parser
       callback :response
-
-      @response.headers.delete("Transfer-Encoding")
-      @response.headers['Content-Length'] = @response.body.length.to_s
       @client.send_data @parser.raw_data
+      @client.close_connection_after_writing
     end
 
     private