promisify only what's needed

Author: scottinet

src/kuzzle.js

@@ -205,7 +205,14 @@ module.exports = Kuzzle = function (url, options, cb) {
   construct.call(this, url, cb);
 
   if (this.bluebird) {
-    return this.bluebird.promisifyAll(this, {suffix: 'Promise'});
+    return this.bluebird.promisifyAll(this, {
+      suffix: 'Promise',
+      filter: function (name, func, target, passes) {
+        var whitelist = ['getAllStatistics', 'getStatistics', 'listCollections', 'now', 'query'];
+
+        return passes && whitelist.indexOf(name) !== -1;
+      }
+    });
   }
 
   return this;

src/kuzzleDataCollection.js

@@ -38,7 +38,14 @@ function KuzzleDataCollection(kuzzle, collection) {
   });
 
   if (this.kuzzle.bluebird) {
-    return this.kuzzle.bluebird.promisifyAll(this, {suffix: 'Promise'});
+    return this.kuzzle.bluebird.promisifyAll(this, {
+      suffix: 'Promise',
+      filter: function (name, func, target, passes) {
+        var blacklist = ['publish', 'setHeaders', 'subscribe'];
+
+        return passes && blacklist.indexOf(name) === -1;
+      }
+    });
   }
 
   return this;

src/kuzzleDataMapping.js

@@ -43,7 +43,14 @@ function KuzzleDataMapping(kuzzleDataCollection) {
   });
 
   if (this.kuzzle.bluebird) {
-    return this.kuzzle.bluebird.promisifyAll(this, {suffix: 'Promise'});
+    return this.kuzzle.bluebird.promisifyAll(this, {
+      suffix: 'Promise',
+      filter: function (name, func, target, passes) {
+        var blacklist = ['set', 'setHeaders'];
+
+        return passes && blacklist.indexOf(name) === -1;
+      }
+    });
   }
 
   return this;

src/kuzzleDocument.js

@@ -94,7 +94,14 @@ function KuzzleDocument(kuzzleDataCollection, documentId, content) {
 
   // promisifying
   if (this.kuzzle.bluebird) {
-    return this.kuzzle.bluebird.promisifyAll(this, {suffix: 'Promise'});
+    return this.kuzzle.bluebird.promisifyAll(this, {
+      suffix: 'Promise',
+      filter: function (name, func, target, passes) {
+        var whitelist = ['delete', 'refresh', 'save'];
+
+        return passes && whitelist.indexOf(name) !== -1;
+      }
+    });
   }
 
   return this;

src/kuzzleRoom.js

@@ -96,7 +96,14 @@ function KuzzleRoom(kuzzleDataCollection, options) {
   });
 
   if (this.kuzzle.bluebird) {
-    return this.kuzzle.bluebird.promisifyAll(this, {suffix: 'Promise'});
+    return this.kuzzle.bluebird.promisifyAll(this, {
+      suffix: 'Promise',
+      filter: function (name, func, target, passes) {
+        var whitelist = ['count'];
+
+        return passes && whitelist.indexOf(name) !== -1;
+      }
+    });
   }
 
   return this;