From bfa8a82ac4c044b2c92608d0e505df7e5be0ec73 Mon Sep 17 00:00:00 2001
From: coderaiser <mnemonic.enemy@gmail.com>
Date: Wed, 14 May 2014 03:53:36 -0400
Subject: [PATCH] refactor(cloudfunc) addListener

---
 lib/cloudfunc.js | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/lib/cloudfunc.js b/lib/cloudfunc.js
index b9e6ee09..c6bff040 100644
--- a/lib/cloudfunc.js
+++ b/lib/cloudfunc.js
@@ -51,9 +51,11 @@ var Util;
         this.MAX_FILE_SIZE          = 500 * 1024;
         
         this.addListener            = function(name, func, allListeners, socket) {
-             var listeners, obj;
+            var listeners, obj,
+                type    = Util.getType(name);
             
-            if (Util.isString(name)) {
+            switch(type) {
+            case 'string':
                 listeners       = allListeners[name];
                 
                 if (!listeners)
@@ -63,15 +65,21 @@ var Util;
                 
                 if (func && socket)
                     socket.on(name, func);
+                
+                break;
             
-            } else if (Util.isObject(name)) {
+            case 'object':
                 obj = name;
                 
-                for (name in obj) {
+                Object.keys(obj).forEach(function(name) {
                     func = obj[name];
                     CloudFunc.addListener(name, func, allListeners, socket);
-                }
+                });
+                
+                break;
             }
+            
+            return this;
         };
         
         this.removeListener         = function(name, func, allListeners, socket) {