You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

32 lines
608 B
JavaScript

6 years ago
'use strict';
module.exports = function (fn, errMsg) {
if (typeof fn !== 'function') {
throw new TypeError('Expected a function');
}
var ret;
var called = false;
var fnName = fn.displayName || fn.name || (/function ([^\(]+)/.exec(fn.toString()) || [])[1];
var onetime = function () {
if (called) {
if (errMsg === true) {
fnName = fnName ? fnName + '()' : 'Function';
throw new Error(fnName + ' can only be called once.');
}
return ret;
}
called = true;
ret = fn.apply(this, arguments);
fn = null;
return ret;
};
onetime.displayName = fnName;
return onetime;
};