RequireContextPlugin.js 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Tobias Koppers @sokra
  4. */
  5. "use strict";
  6. const RequireContextDependency = require("./RequireContextDependency");
  7. const ContextElementDependency = require("./ContextElementDependency");
  8. const RequireContextDependencyParserPlugin = require("./RequireContextDependencyParserPlugin");
  9. class RequireContextPlugin {
  10. constructor(modulesDirectories, extensions) {
  11. if(!Array.isArray(modulesDirectories))
  12. throw new Error("modulesDirectories must be an array");
  13. if(!Array.isArray(extensions))
  14. throw new Error("extensions must be an array");
  15. this.modulesDirectories = modulesDirectories;
  16. this.extensions = extensions;
  17. }
  18. apply(compiler) {
  19. const modulesDirectories = this.modulesDirectories;
  20. const extensions = this.extensions;
  21. compiler.plugin("compilation", (compilation, params) => {
  22. const contextModuleFactory = params.contextModuleFactory;
  23. const normalModuleFactory = params.normalModuleFactory;
  24. compilation.dependencyFactories.set(RequireContextDependency, contextModuleFactory);
  25. compilation.dependencyTemplates.set(RequireContextDependency, new RequireContextDependency.Template());
  26. compilation.dependencyFactories.set(ContextElementDependency, normalModuleFactory);
  27. params.normalModuleFactory.plugin("parser", (parser, parserOptions) => {
  28. if(typeof parserOptions.requireContext !== "undefined" && !parserOptions.requireContext)
  29. return;
  30. parser.apply(new RequireContextDependencyParserPlugin());
  31. });
  32. params.contextModuleFactory.plugin("alternatives", (items, callback) => {
  33. if(items.length === 0) return callback(null, items);
  34. callback(null, items.map((obj) => {
  35. return extensions.filter((ext) => {
  36. const l = obj.request.length;
  37. return l > ext.length && obj.request.substr(l - ext.length, l) === ext;
  38. }).map((ext) => {
  39. const l = obj.request.length;
  40. return {
  41. context: obj.context,
  42. request: obj.request.substr(0, l - ext.length)
  43. };
  44. }).concat(obj);
  45. }).reduce((a, b) => a.concat(b), []));
  46. });
  47. params.contextModuleFactory.plugin("alternatives", (items, callback) => {
  48. if(items.length === 0) return callback(null, items);
  49. callback(null, items.map((obj) => {
  50. for(let i = 0; i < modulesDirectories.length; i++) {
  51. const dir = modulesDirectories[i];
  52. const idx = obj.request.indexOf("./" + dir + "/");
  53. if(idx === 0) {
  54. obj.request = obj.request.slice(dir.length + 3);
  55. break;
  56. }
  57. }
  58. return obj;
  59. }));
  60. });
  61. });
  62. }
  63. }
  64. module.exports = RequireContextPlugin;