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.
		
		
		
		
		
			
		
			
				
					159 lines
				
				5.7 KiB
			
		
		
			
		
	
	
					159 lines
				
				5.7 KiB
			| 
											3 years ago
										 | "use strict"; | ||
|  | 
 | ||
|  | Object.defineProperty(exports, "__esModule", { | ||
|  |   value: true | ||
|  | }); | ||
|  | exports.addHook = addHook; | ||
|  | 
 | ||
|  | var _module = _interopRequireDefault(require("module")); | ||
|  | 
 | ||
|  | var _path = _interopRequireDefault(require("path")); | ||
|  | 
 | ||
|  | function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } | ||
|  | 
 | ||
|  | /* (c) 2015 Ari Porad (@ariporad) <http://ariporad.com>. License: ariporad.mit-license.org */ | ||
|  | const nodeModulesRegex = /^(?:.*[\\/])?node_modules(?:[\\/].*)?$/; // Guard against poorly mocked module constructors.
 | ||
|  | 
 | ||
|  | const Module = module.constructor.length > 1 ? module.constructor : _module.default; | ||
|  | const HOOK_RETURNED_NOTHING_ERROR_MESSAGE = '[Pirates] A hook returned a non-string, or nothing at all! This is a' + ' violation of intergalactic law!\n' + '--------------------\n' + 'If you have no idea what this means or what Pirates is, let me explain: ' + 'Pirates is a module that makes is easy to implement require hooks. One of' + " the require hooks you're using uses it. One of these require hooks" + " didn't return anything from it's handler, so we don't know what to" + ' do. You might want to debug this.'; | ||
|  | /** | ||
|  |  * @param {string} filename The filename to check. | ||
|  |  * @param {string[]} exts The extensions to hook. Should start with '.' (ex. ['.js']). | ||
|  |  * @param {Matcher|null} matcher A matcher function, will be called with path to a file. Should return truthy if the file should be hooked, falsy otherwise. | ||
|  |  * @param {boolean} ignoreNodeModules Auto-ignore node_modules. Independent of any matcher. | ||
|  |  */ | ||
|  | 
 | ||
|  | function shouldCompile(filename, exts, matcher, ignoreNodeModules) { | ||
|  |   if (typeof filename !== 'string') { | ||
|  |     return false; | ||
|  |   } | ||
|  | 
 | ||
|  |   if (exts.indexOf(_path.default.extname(filename)) === -1) { | ||
|  |     return false; | ||
|  |   } | ||
|  | 
 | ||
|  |   const resolvedFilename = _path.default.resolve(filename); | ||
|  | 
 | ||
|  |   if (ignoreNodeModules && nodeModulesRegex.test(resolvedFilename)) { | ||
|  |     return false; | ||
|  |   } | ||
|  | 
 | ||
|  |   if (matcher && typeof matcher === 'function') { | ||
|  |     return !!matcher(resolvedFilename); | ||
|  |   } | ||
|  | 
 | ||
|  |   return true; | ||
|  | } | ||
|  | /** | ||
|  |  * @callback Hook The hook. Accepts the code of the module and the filename. | ||
|  |  * @param {string} code | ||
|  |  * @param {string} filename | ||
|  |  * @returns {string} | ||
|  |  */ | ||
|  | 
 | ||
|  | /** | ||
|  |  * @callback Matcher A matcher function, will be called with path to a file. | ||
|  |  * | ||
|  |  * Should return truthy if the file should be hooked, falsy otherwise. | ||
|  |  * @param {string} path | ||
|  |  * @returns {boolean} | ||
|  |  */ | ||
|  | 
 | ||
|  | /** | ||
|  |  * @callback RevertFunction Reverts the hook when called. | ||
|  |  * @returns {void} | ||
|  |  */ | ||
|  | 
 | ||
|  | /** | ||
|  |  * @typedef {object} Options | ||
|  |  * @property {Matcher|null} [matcher=null] A matcher function, will be called with path to a file. | ||
|  |  * | ||
|  |  * Should return truthy if the file should be hooked, falsy otherwise. | ||
|  |  * | ||
|  |  * @property {string[]} [extensions=['.js']] The extensions to hook. Should start with '.' (ex. ['.js']). | ||
|  |  * @property {string[]} [exts=['.js']] The extensions to hook. Should start with '.' (ex. ['.js']). | ||
|  |  * | ||
|  |  * @property {string[]} [extension=['.js']] The extensions to hook. Should start with '.' (ex. ['.js']). | ||
|  |  * @property {string[]} [ext=['.js']] The extensions to hook. Should start with '.' (ex. ['.js']). | ||
|  |  * | ||
|  |  * @property {boolean} [ignoreNodeModules=true] Auto-ignore node_modules. Independent of any matcher. | ||
|  |  */ | ||
|  | 
 | ||
|  | /** | ||
|  |  * Add a require hook. | ||
|  |  * | ||
|  |  * @param {Hook} hook The hook. Accepts the code of the module and the filename. Required. | ||
|  |  * @param {Options} [opts] Options | ||
|  |  * @returns {RevertFunction} The `revert` function. Reverts the hook when called. | ||
|  |  */ | ||
|  | 
 | ||
|  | 
 | ||
|  | function addHook(hook, opts = {}) { | ||
|  |   let reverted = false; | ||
|  |   const loaders = []; | ||
|  |   const oldLoaders = []; | ||
|  |   let exts; // We need to do this to fix #15. Basically, if you use a non-standard extension (ie. .jsx), then
 | ||
|  |   // We modify the .js loader, then use the modified .js loader for as the base for .jsx.
 | ||
|  |   // This prevents that.
 | ||
|  | 
 | ||
|  |   const originalJSLoader = Module._extensions['.js']; | ||
|  |   const matcher = opts.matcher || null; | ||
|  |   const ignoreNodeModules = opts.ignoreNodeModules !== false; | ||
|  |   exts = opts.extensions || opts.exts || opts.extension || opts.ext || ['.js']; | ||
|  | 
 | ||
|  |   if (!Array.isArray(exts)) { | ||
|  |     exts = [exts]; | ||
|  |   } | ||
|  | 
 | ||
|  |   exts.forEach(ext => { | ||
|  |     if (typeof ext !== 'string') { | ||
|  |       throw new TypeError(`Invalid Extension: ${ext}`); | ||
|  |     } | ||
|  | 
 | ||
|  |     const oldLoader = Module._extensions[ext] || originalJSLoader; | ||
|  |     oldLoaders[ext] = Module._extensions[ext]; | ||
|  | 
 | ||
|  |     loaders[ext] = Module._extensions[ext] = function newLoader(mod, filename) { | ||
|  |       let compile; | ||
|  | 
 | ||
|  |       if (!reverted) { | ||
|  |         if (shouldCompile(filename, exts, matcher, ignoreNodeModules)) { | ||
|  |           compile = mod._compile; | ||
|  | 
 | ||
|  |           mod._compile = function _compile(code) { | ||
|  |             // reset the compile immediately as otherwise we end up having the
 | ||
|  |             // compile function being changed even though this loader might be reverted
 | ||
|  |             // Not reverting it here leads to long useless compile chains when doing
 | ||
|  |             // addHook -> revert -> addHook -> revert -> ...
 | ||
|  |             // The compile function is also anyway created new when the loader is called a second time.
 | ||
|  |             mod._compile = compile; | ||
|  |             const newCode = hook(code, filename); | ||
|  | 
 | ||
|  |             if (typeof newCode !== 'string') { | ||
|  |               throw new Error(HOOK_RETURNED_NOTHING_ERROR_MESSAGE); | ||
|  |             } | ||
|  | 
 | ||
|  |             return mod._compile(newCode, filename); | ||
|  |           }; | ||
|  |         } | ||
|  |       } | ||
|  | 
 | ||
|  |       oldLoader(mod, filename); | ||
|  |     }; | ||
|  |   }); | ||
|  |   return function revert() { | ||
|  |     if (reverted) return; | ||
|  |     reverted = true; | ||
|  |     exts.forEach(ext => { | ||
|  |       // if the current loader for the extension is our loader then unregister it and set the oldLoader again
 | ||
|  |       // if not we can not do anything as we cannot remove a loader from within the loader-chain
 | ||
|  |       if (Module._extensions[ext] === loaders[ext]) { | ||
|  |         if (!oldLoaders[ext]) { | ||
|  |           delete Module._extensions[ext]; | ||
|  |         } else { | ||
|  |           Module._extensions[ext] = oldLoaders[ext]; | ||
|  |         } | ||
|  |       } | ||
|  |     }); | ||
|  |   }; | ||
|  | } |