1
0
Fork 0
management/front/dkha-web-sz-main/node_modules/thenify/index.js

78 lines
1.9 KiB
JavaScript
Raw Normal View History

2023-12-18 13:12:25 +08:00
var Promise = require('any-promise')
var assert = require('assert')
module.exports = thenify
/**
* Turn async functions into promises
*
2024-01-16 21:26:16 +08:00
* @param {Function} fn
2023-12-18 13:12:25 +08:00
* @return {Function}
* @api public
*/
2024-01-16 21:26:16 +08:00
function thenify(fn, options) {
assert(typeof fn === 'function')
return createWrapper(fn, options)
2023-12-18 13:12:25 +08:00
}
/**
* Turn async functions into promises and backward compatible with callback
*
2024-01-16 21:26:16 +08:00
* @param {Function} fn
2023-12-18 13:12:25 +08:00
* @return {Function}
* @api public
*/
2024-01-16 21:26:16 +08:00
thenify.withCallback = function (fn, options) {
assert(typeof fn === 'function')
2023-12-18 13:12:25 +08:00
options = options || {}
options.withCallback = true
2024-01-16 21:26:16 +08:00
return createWrapper(fn, options)
2023-12-18 13:12:25 +08:00
}
function createCallback(resolve, reject, multiArgs) {
2024-01-16 21:26:16 +08:00
// default to true
if (multiArgs === undefined) multiArgs = true
2023-12-18 13:12:25 +08:00
return function(err, value) {
if (err) return reject(err)
var length = arguments.length
if (length <= 2 || !multiArgs) return resolve(value)
if (Array.isArray(multiArgs)) {
var values = {}
for (var i = 1; i < length; i++) values[multiArgs[i - 1]] = arguments[i]
return resolve(values)
}
var values = new Array(length - 1)
for (var i = 1; i < length; ++i) values[i - 1] = arguments[i]
resolve(values)
}
}
2024-01-16 21:26:16 +08:00
function createWrapper(fn, options) {
2023-12-18 13:12:25 +08:00
options = options || {}
2024-01-16 21:26:16 +08:00
var name = fn.name;
name = (name || '').replace(/\s|bound(?!$)/g, '')
var newFn = function () {
var self = this
var len = arguments.length
if (options.withCallback) {
var lastType = typeof arguments[len - 1]
if (lastType === 'function') return fn.apply(self, arguments)
}
var args = new Array(len + 1)
for (var i = 0; i < len; ++i) args[i] = arguments[i]
var lastIndex = i
return new Promise(function (resolve, reject) {
args[lastIndex] = createCallback(resolve, reject, options.multiArgs)
fn.apply(self, args)
})
}
Object.defineProperty(newFn, 'name', { value: name })
return newFn
2023-12-18 13:12:25 +08:00
}