78 lines
2.6 KiB
JavaScript
78 lines
2.6 KiB
JavaScript
|
/*
|
||
|
|
MIT License http://www.opensource.org/licenses/mit-license.php
|
||
|
|
Author Tobias Koppers @sokra
|
||
|
|
*/
|
||
|
|
"use strict";
|
||
|
|
|
||
|
|
const RequireContextDependency = require("./RequireContextDependency");
|
||
|
|
const ContextElementDependency = require("./ContextElementDependency");
|
||
|
|
|
||
|
|
const RequireContextDependencyParserPlugin = require("./RequireContextDependencyParserPlugin");
|
||
|
|
|
||
|
|
class RequireContextPlugin {
|
||
|
|
constructor(modulesDirectories, extensions) {
|
||
|
|
if(!Array.isArray(modulesDirectories))
|
||
|
|
throw new Error("modulesDirectories must be an array");
|
||
|
|
if(!Array.isArray(extensions))
|
||
|
|
throw new Error("extensions must be an array");
|
||
|
|
this.modulesDirectories = modulesDirectories;
|
||
|
|
this.extensions = extensions;
|
||
|
|
}
|
||
|
|
|
||
|
|
apply(compiler) {
|
||
|
|
const modulesDirectories = this.modulesDirectories;
|
||
|
|
const extensions = this.extensions;
|
||
|
|
compiler.plugin("compilation", (compilation, params) => {
|
||
|
|
const contextModuleFactory = params.contextModuleFactory;
|
||
|
|
const normalModuleFactory = params.normalModuleFactory;
|
||
|
|
|
||
|
|
compilation.dependencyFactories.set(RequireContextDependency, contextModuleFactory);
|
||
|
|
compilation.dependencyTemplates.set(RequireContextDependency, new RequireContextDependency.Template());
|
||
|
|
|
||
|
|
compilation.dependencyFactories.set(ContextElementDependency, normalModuleFactory);
|
||
|
|
|
||
|
|
params.normalModuleFactory.plugin("parser", (parser, parserOptions) => {
|
||
|
|
|
||
|
|
if(typeof parserOptions.requireContext !== "undefined" && !parserOptions.requireContext)
|
||
|
|
return;
|
||
|
|
|
||
|
|
parser.apply(new RequireContextDependencyParserPlugin());
|
||
|
|
});
|
||
|
|
|
||
|
|
params.contextModuleFactory.plugin("alternatives", (items, callback) => {
|
||
|
|
if(items.length === 0) return callback(null, items);
|
||
|
|
|
||
|
|
callback(null, items.map((obj) => {
|
||
|
|
return extensions.filter((ext) => {
|
||
|
|
const l = obj.request.length;
|
||
|
|
return l > ext.length && obj.request.substr(l - ext.length, l) === ext;
|
||
|
|
}).map((ext) => {
|
||
|
|
const l = obj.request.length;
|
||
|
|
return {
|
||
|
|
context: obj.context,
|
||
|
|
request: obj.request.substr(0, l - ext.length)
|
||
|
|
};
|
||
|
|
}).concat(obj);
|
||
|
|
}).reduce((a, b) => a.concat(b), []));
|
||
|
|
});
|
||
|
|
|
||
|
|
params.contextModuleFactory.plugin("alternatives", (items, callback) => {
|
||
|
|
if(items.length === 0) return callback(null, items);
|
||
|
|
|
||
|
|
callback(null, items.map((obj) => {
|
||
|
|
for(let i = 0; i < modulesDirectories.length; i++) {
|
||
|
|
const dir = modulesDirectories[i];
|
||
|
|
const idx = obj.request.indexOf("./" + dir + "/");
|
||
|
|
if(idx === 0) {
|
||
|
|
obj.request = obj.request.slice(dir.length + 3);
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return obj;
|
||
|
|
}));
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
module.exports = RequireContextPlugin;
|