class Faye::WebSocket::RainbowsClient

def app_call(*args)
  @env['em.connection'] = self
  if args.first == NULL_IO and @hp.content_length == 0 and websocket?
    prepare_request_body
  else
    super
  end
end

def on_read(data)
  if @state == :body and websocket? and @hp.body_eof?
    @state = :websocket
    @input.rewind
    app_call StringIO.new(@buf)
  else
    super
  end
end

def receive_data(data)
  return super unless @state == :websocket
  socket_stream.receive(data) if socket_stream
end

def unbind
  super
ensure
  socket_stream.fail if socket_stream
end

def write_headers(status, headers, *args)
  super unless socket_connection? and status == 101
end