class Faye::WebSocket::Stream

def each(&callback)
  @stream_send ||= callback
end

def fail
  @web_socket.close(1006, '', false)
end

def initialize(web_socket)
  @web_socket  = web_socket
  @connection  = web_socket.env['em.connection']
  @stream_send = web_socket.env['stream.send']
  
  @connection.socket_stream = self if @connection.respond_to?(:socket_stream)
end

def receive(data)
  @web_socket.__send__(:parse, data)
end

def write(data)
  return unless @stream_send
  @stream_send.call(data) rescue nil
end