module WebSocket::Driver::EventEmitter

def on(event, callable = nil, &block)
  if callable
    add_listener(event, callable)
  else
    add_listener(event, &block)
  end
end