From 5affb856d8a1b28f169dabca20aac5630f5cebe5 Mon Sep 17 00:00:00 2001 From: Michelle Bu Date: Fri, 18 Oct 2013 20:04:05 -0700 Subject: [PATCH] Remove ugly switch --- lib/server.js | 31 +++++++++---------------------- 1 file changed, 9 insertions(+), 22 deletions(-) diff --git a/lib/server.js b/lib/server.js index 4f8657c..0e5d087 100644 --- a/lib/server.js +++ b/lib/server.js @@ -124,28 +124,15 @@ PeerServer.prototype._configureWS = function(socket, key, id, token) { try { var message = JSON.parse(data); - switch (message.type) { - case 'LEAVE': - // Clean up if a Peer sends a LEAVE. - if (!message.dst) { - self._removePeer(key, id); - } - break; - // ICE candidates - case 'CANDIDATE': - // Offer or answer between peers. - case 'OFFER': - case 'ANSWER': - // Use the ID we know to be correct to prevent spoofing. - self._handleTransmission(key, { - type: message.type, - src: id, - dst: message.dst, - payload: message.payload - }); - break; - default: - util.prettyError('Message unrecognized'); + if (['LEAVE', 'CANDIDATE', 'OFFER', 'ANSWER'].indexOf(message.type) !== -1) { + self._handleTransmission(key, { + type: message.type, + src: id, + dst: message.dst, + payload: message.payload + }); + } else { + util.prettyError('Message unrecognized'); } } catch(e) { util.log('Invalid message', data);