You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
967 B
36 lines
967 B
'use strict' |
|
|
|
const { promisify } = require('util') |
|
|
|
const handler = { |
|
get: function (target, prop, receiver) { |
|
if (typeof target[prop] !== 'function') { |
|
return target[prop] |
|
} |
|
if (target[prop][promisify.custom]) { |
|
return function () { |
|
return Reflect.get(target, prop, receiver)[promisify.custom].apply(target, arguments) |
|
} |
|
} |
|
return function () { |
|
return new Promise((resolve, reject) => { |
|
Reflect.get(target, prop, receiver).apply(target, [...arguments, function (err, result) { |
|
if (err) { |
|
return reject(err) |
|
} |
|
resolve(result) |
|
}]) |
|
}) |
|
} |
|
} |
|
} |
|
|
|
module.exports = function (thingToPromisify) { |
|
if (typeof thingToPromisify === 'function') { |
|
return promisify(thingToPromisify) |
|
} |
|
if (typeof thingToPromisify === 'object') { |
|
return new Proxy(thingToPromisify, handler) |
|
} |
|
throw new TypeError('Can only promisify functions or objects') |
|
}
|
|
|