'use strict'; const js = require('default-require-extensions/js'); module.exports = appendTransform; let count = 0; function appendTransform(transform, ext, extensions) { // Generate a unique key for this transform var key = __dirname + count; // eslint-disable-line count++; ext = ext || '.js'; extensions = extensions || require.extensions; let forwardGet; let forwardSet; const descriptor = Object.getOwnPropertyDescriptor(extensions, ext) || {value: js, configurable: true}; if ( ((descriptor.get || descriptor.set) && !(descriptor.get && descriptor.set)) || !descriptor.configurable ) { throw new Error('Somebody did bad things to require.extensions["' + ext + '"]'); } if (descriptor.get) { // Wrap a previous append-transform install and pass through to the getter/setter pair it created forwardGet = function () { return descriptor.get(); }; forwardSet = function (val) { descriptor.set(val); return forwardGet(); }; } else { forwardGet = function () { return descriptor.value; }; forwardSet = function (val) { descriptor.value = val; return val; }; } function wrapCustomHook(hook) { return function (module, filename) { // We wrap every added extension, but we only apply the transform to the one on top of the stack if (!module[key]) { module[key] = true; const originalCompile = module._compile; // eslint-disable-next-line func-name-matching func-names module._compile = function replacementCompile(code, filename) { module._compile = originalCompile; code = transform(code, filename); module._compile(code, filename); }; } hook(module, filename); }; } // Wrap the original forwardSet(wrapCustomHook(forwardGet())); const hooks = [forwardGet()]; function setCurrentHook(hook) { const restoreIndex = hooks.indexOf(hook); if (restoreIndex === -1) { hooks.push(forwardSet(wrapCustomHook(hook))); } else { // We have already scene this hook, and it is being reverted (proxyquire, etc) - don't wrap again. hooks.splice(restoreIndex + 1, hooks.length); forwardSet(hook); } } Object.defineProperty(extensions, ext, { configurable: true, enumerable: true, get: forwardGet, set: setCurrentHook }); }