94 lines
3.1 KiB
JavaScript
94 lines
3.1 KiB
JavaScript
|
/*
|
||
|
|
Copyright (c) 2014 Google Inc. All rights reserved.
|
||
|
|
Copyright (c) 2012-2013 Johannes Ewald.
|
||
|
|
|
||
|
|
Use of this source code is governed by the MIT License, available in this package's LICENSE file
|
||
|
|
or at http://opensource.org/licenses/MIT.
|
||
|
|
*/
|
||
|
|
/** @module lib/requizzle */
|
||
|
|
|
||
|
|
const loader = require('./loader');
|
||
|
|
const Module = require('module');
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Function that returns text to swizzle into the module.
|
||
|
|
*
|
||
|
|
* @typedef module:lib/requizzle~wrapperFunction
|
||
|
|
* @type {function}
|
||
|
|
* @param {string} targetPath - The path to the target module.
|
||
|
|
* @param {string} parentModulePath - The path to the module that is requiring the target module.
|
||
|
|
* @return {string} The text to insert before or after the module's source code.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Options for the wrappers that will be swizzled into the target module.
|
||
|
|
*
|
||
|
|
* @typedef module:lib/requizzle~options
|
||
|
|
* @type {Object}
|
||
|
|
* @property {Object=} options.extras - Functions that generate text to swizzle into the target
|
||
|
|
* module.
|
||
|
|
* @property {module:lib/requizzle~wrapperFunction} options.extras.after - Function that returns
|
||
|
|
* text to insert after the module's source code.
|
||
|
|
* @property {module:lib/requizzle~wrapperFunction} options.extras.before - Function that returns
|
||
|
|
* text to insert before the module's source code.
|
||
|
|
* @property {(Array.<string>|string)} options.requirePaths - Additional paths to search when
|
||
|
|
* resolving module paths in the target module.
|
||
|
|
*/
|
||
|
|
|
||
|
|
function isNativeModule(targetPath, parentModule) {
|
||
|
|
const lookupPaths = Module._resolveLookupPaths(targetPath, parentModule, true);
|
||
|
|
|
||
|
|
/* istanbul ignore next */
|
||
|
|
return lookupPaths === null ||
|
||
|
|
(lookupPaths.length === 2 &&
|
||
|
|
lookupPaths[1].length === 0 &&
|
||
|
|
lookupPaths[0] === targetPath);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Create a `Requizzle` instance. If you provide options, Requizzle will default to those options
|
||
|
|
* when you call {@link Requizzle#requizzle}.
|
||
|
|
*
|
||
|
|
* @class
|
||
|
|
* @param {!module:lib/requizzle~options} options - Options for the wrappers that will be swizzled
|
||
|
|
* into the target module.
|
||
|
|
* @param {Object=} cache - For internal use.
|
||
|
|
*/
|
||
|
|
class Requizzle {
|
||
|
|
constructor(options, cache) {
|
||
|
|
this._options = options;
|
||
|
|
this._cache = cache || {
|
||
|
|
module: {},
|
||
|
|
source: {}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Load the module, swizzling in the requested changes.
|
||
|
|
*
|
||
|
|
* @param {!string} targetPath - The path to the module that will be loaded.
|
||
|
|
* @return {Module} The swizzled module.
|
||
|
|
*/
|
||
|
|
requizzle(targetPath) {
|
||
|
|
const options = this._options;
|
||
|
|
const parentModule = options.parent;
|
||
|
|
let targetModule;
|
||
|
|
let wrapper;
|
||
|
|
|
||
|
|
// Don't interfere with native modules
|
||
|
|
if (isNativeModule(targetPath, parentModule)) {
|
||
|
|
return require(targetPath);
|
||
|
|
}
|
||
|
|
|
||
|
|
// Resolve the filename relative to the parent module
|
||
|
|
targetPath = Module._resolveFilename(targetPath, parentModule);
|
||
|
|
|
||
|
|
wrapper = loader.createWrapper(targetPath, parentModule, this._cache, this._options);
|
||
|
|
targetModule = loader.load(targetPath, parentModule, wrapper, this._cache, this._options);
|
||
|
|
|
||
|
|
return targetModule.exports;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = Requizzle;
|