exit-hook-plus
Version:
Do something before the program exits or when the program crashes!
297 lines (296 loc) • 8.42 kB
JavaScript
;
var __awaiter =
(this && this.__awaiter) ||
function (thisArg, _arguments, P, generator) {
function adopt(value) {
return value instanceof P
? value
: new P(function (resolve) {
resolve(value);
});
}
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) {
try {
step(generator.next(value));
} catch (e) {
reject(e);
}
}
function rejected(value) {
try {
step(generator['throw'](value));
} catch (e) {
reject(e);
}
}
function step(result) {
result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected);
}
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __generator =
(this && this.__generator) ||
function (thisArg, body) {
var _ = {
label: 0,
sent: function () {
if (t[0] & 1) throw t[1];
return t[1];
},
trys: [],
ops: []
},
f,
y,
t,
g;
return (
(g = { next: verb(0), throw: verb(1), return: verb(2) }),
typeof Symbol === 'function' &&
(g[Symbol.iterator] = function () {
return this;
}),
g
);
function verb(n) {
return function (v) {
return step([n, v]);
};
}
function step(op) {
if (f) throw new TypeError('Generator is already executing.');
while (_)
try {
if (
((f = 1),
y &&
(t =
op[0] & 2
? y['return']
: op[0]
? y['throw'] || ((t = y['return']) && t.call(y), 0)
: y.next) &&
!(t = t.call(y, op[1])).done)
)
return t;
if (((y = 0), t)) op = [op[0] & 2, t.value];
switch (op[0]) {
case 0:
case 1:
t = op;
break;
case 4:
_.label++;
return { value: op[1], done: false };
case 5:
_.label++;
y = op[1];
op = [0];
continue;
case 7:
op = _.ops.pop();
_.trys.pop();
continue;
default:
if (
!((t = _.trys), (t = t.length > 0 && t[t.length - 1])) &&
(op[0] === 6 || op[0] === 2)
) {
_ = 0;
continue;
}
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) {
_.label = op[1];
break;
}
if (op[0] === 6 && _.label < t[1]) {
_.label = t[1];
t = op;
break;
}
if (t && _.label < t[2]) {
_.label = t[2];
_.ops.push(op);
break;
}
if (t[2]) _.ops.pop();
_.trys.pop();
continue;
}
op = body.call(thisArg, _);
} catch (e) {
op = [6, e];
y = 0;
} finally {
f = t = 0;
}
if (op[0] & 5) throw op[1];
return { value: op[0] ? op[1] : void 0, done: true };
}
};
var __spreadArrays =
(this && this.__spreadArrays) ||
function () {
for (var s = 0, i = 0, il = arguments.length; i < il; i++) s += arguments[i].length;
for (var r = Array(s), k = 0, i = 0; i < il; i++)
for (var a = arguments[i], j = 0, jl = a.length; j < jl; j++, k++) r[k] = a[j];
return r;
};
Object.defineProperty(exports, '__esModule', { value: true });
exports.executeAllHooksAndTerminate = exports.removeExitHook = exports.addExitHook = exports.disableDefaultExitLogger = void 0;
var TRIGGER_ERRORS = ['unhandledRejection', 'uncaughtException'];
var TRIGGER_SIGNALS = [
['SIGHUP', 128 + 1],
['SIGINT', 128 + 2],
['SIGTERM', 128 + 15],
['SIGBREAK', 128 + 21]
];
var defaultExitLogger = function (reason) {
switch (reason.category) {
case 'exception':
var stack = reason.errorOrReason.stack;
console.error(
'the program is now exiting due to unhandled exception: ' +
(stack !== null && stack !== void 0 ? stack : reason.errorOrReason)
);
break;
case 'signal':
console.error('the program is now exiting due to receiving signal: ' + reason.signal);
break;
}
};
var exitHooks = [defaultExitLogger];
function disableDefaultExitLogger() {
removeExitHook(defaultExitLogger);
}
exports.disableDefaultExitLogger = disableDefaultExitLogger;
function addExitHook(hook) {
if (typeof hook !== 'function') {
throw new Error('invalid argument, expected function but got ' + typeof hook);
}
exitHooks.push(hook);
}
exports.addExitHook = addExitHook;
function removeExitHook(hook) {
exitHooks = exitHooks.filter(function (h) {
return h !== hook;
});
}
exports.removeExitHook = removeExitHook;
function executeAllHooksAndTerminate(exitCode, extra) {
if (exitCode === void 0) {
exitCode = 0;
}
runHooks({ category: 'manual', extra: extra }).finally(function () {
return process.exit(exitCode);
});
}
exports.executeAllHooksAndTerminate = executeAllHooksAndTerminate;
process.on('beforeExit', function (code) {
return __awaiter(void 0, void 0, void 0, function () {
return __generator(this, function (_a) {
switch (_a.label) {
case 0:
return [4 /*yield*/, runHooks({ category: 'trivial', exitCode: code })];
case 1:
_a.sent();
return [2 /*return*/];
}
});
});
});
process.on('exit', function (code) {
runSyncHooks({ category: 'trivial', exitCode: code });
});
TRIGGER_ERRORS.forEach(function (errorType) {
process.on(errorType, function (errorOrReason) {
return __awaiter(void 0, void 0, void 0, function () {
return __generator(this, function (_a) {
switch (_a.label) {
case 0:
return [4 /*yield*/, runHooks({ category: 'exception', errorOrReason: errorOrReason })];
case 1:
_a.sent();
process.exit(1);
return [2 /*return*/];
}
});
});
});
});
TRIGGER_SIGNALS.forEach(function (_a) {
var signal = _a[0],
exitCode = _a[1];
process.on(signal, function (signal) {
return __awaiter(void 0, void 0, void 0, function () {
return __generator(this, function (_a) {
switch (_a.label) {
case 0:
return [4 /*yield*/, runHooks({ category: 'signal', signal: signal })];
case 1:
_a.sent();
process.exit(exitCode);
return [2 /*return*/];
}
});
});
});
});
function runSyncHooks(reason) {
var hooks = __spreadArrays(exitHooks)
.filter(function (h) {
return !h._executed;
})
.filter(function (h) {
return h.constructor.name !== 'AsyncFunction';
});
for (var _i = 0, hooks_1 = hooks; _i < hooks_1.length; _i++) {
var hook = hooks_1[_i];
hook._executed = true;
try {
hook(reason);
} catch (_a) {}
}
}
function runHooks(reason) {
return __awaiter(this, void 0, void 0, function () {
var hooks, _i, hooks_2, hook, _a;
return __generator(this, function (_b) {
switch (_b.label) {
case 0:
hooks = __spreadArrays(exitHooks).filter(function (h) {
return !h._executed;
});
(_i = 0), (hooks_2 = hooks);
_b.label = 1;
case 1:
if (!(_i < hooks_2.length)) return [3 /*break*/, 8];
hook = hooks_2[_i];
hook._executed = true;
_b.label = 2;
case 2:
_b.trys.push([2, 6, , 7]);
if (!(hook.constructor.name === 'AsyncFunction')) return [3 /*break*/, 4];
return [4 /*yield*/, hook(reason)];
case 3:
_b.sent();
return [3 /*break*/, 5];
case 4:
hook(reason);
_b.label = 5;
case 5:
return [3 /*break*/, 7];
case 6:
_a = _b.sent();
return [3 /*break*/, 7];
case 7:
_i++;
return [3 /*break*/, 1];
case 8:
return [2 /*return*/];
}
});
});
}