goatify

benchmark/benchmark.js

@@ -37,6 +37,8 @@
 
   var promisified = bluebird.promisify(callbackFun)
 
+  var goatified = pg.goatify(callbackFun)
+
   var callSuite = new benchmark.Suite('call time')
   callSuite
   .add('polygoat promise', function () {
@@ -54,6 +56,9 @@
   .add('bluebird promisified (eval on Node.js, closure on browser)', function () {
     promisified().then(function () {})
   })
+  .add('polygoat goatified', function () {
+    goatified().then(function () {})
+  })
   .on('cycle', function (event) {
     console.log(String(event.target))
   })
@@ -84,6 +89,11 @@
       notAsync(cb)
     })
   })
+  .add('polygoat goatify', function () {
+    return pg.goatify(function (cb) {
+      notAsync(cb)
+    })
+  })
   .on('cycle', function (event) {
     console.log(String(event.target))
   })

index.js

@@ -20,6 +20,32 @@
     }
   }
 
+  function goatify (fn) {
+    return function () {
+      var args = Array.prototype.slice.call(arguments)
+      var callback = typeof args[args.length - 1] === 'function' ? args.pop() : null
+      return polygoat(function (done) {
+        args.push(done)
+        fn.apply(null, args)
+      }, callback)
+    }
+  }
+
+  function goatifyAll (obj) {
+    for (var k in obj) {
+      if (obj.hasOwnProperty(k)) {
+        var v = obj[k]
+        if (typeof v === 'function') { // FIXME IIRC this fails on old browsers
+          obj[k] = goatify(v)
+        }
+      }
+    }
+    return obj
+  }
+
+  polygoat.goatify = goatify
+  polygoat.goatifyAll = goatifyAll
+
   if (typeof module !== 'undefined' && module.exports) {
     module.exports = polygoat
   } else {