RequireContextPlugin.js 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Tobias Koppers @sokra
  4. */
  5. "use strict";
  6. const {
  7. JAVASCRIPT_MODULE_TYPE_AUTO,
  8. JAVASCRIPT_MODULE_TYPE_DYNAMIC
  9. } = require("../ModuleTypeConstants");
  10. const { cachedSetProperty } = require("../util/cleverMerge");
  11. const ContextElementDependency = require("./ContextElementDependency");
  12. const RequireContextDependency = require("./RequireContextDependency");
  13. const RequireContextDependencyParserPlugin = require("./RequireContextDependencyParserPlugin");
  14. /** @typedef {import("../../declarations/WebpackOptions").ResolveOptions} ResolveOptions */
  15. /** @typedef {import("../Compiler")} Compiler */
  16. /** @type {ResolveOptions} */
  17. const EMPTY_RESOLVE_OPTIONS = {};
  18. const PLUGIN_NAME = "RequireContextPlugin";
  19. class RequireContextPlugin {
  20. /**
  21. * Apply the plugin
  22. * @param {Compiler} compiler the compiler instance
  23. * @returns {void}
  24. */
  25. apply(compiler) {
  26. compiler.hooks.compilation.tap(
  27. PLUGIN_NAME,
  28. (compilation, { contextModuleFactory, normalModuleFactory }) => {
  29. compilation.dependencyFactories.set(
  30. RequireContextDependency,
  31. contextModuleFactory
  32. );
  33. compilation.dependencyTemplates.set(
  34. RequireContextDependency,
  35. new RequireContextDependency.Template()
  36. );
  37. compilation.dependencyFactories.set(
  38. ContextElementDependency,
  39. normalModuleFactory
  40. );
  41. const handler = (parser, parserOptions) => {
  42. if (
  43. parserOptions.requireContext !== undefined &&
  44. !parserOptions.requireContext
  45. )
  46. return;
  47. new RequireContextDependencyParserPlugin().apply(parser);
  48. };
  49. normalModuleFactory.hooks.parser
  50. .for(JAVASCRIPT_MODULE_TYPE_AUTO)
  51. .tap(PLUGIN_NAME, handler);
  52. normalModuleFactory.hooks.parser
  53. .for(JAVASCRIPT_MODULE_TYPE_DYNAMIC)
  54. .tap(PLUGIN_NAME, handler);
  55. contextModuleFactory.hooks.alternativeRequests.tap(
  56. PLUGIN_NAME,
  57. (items, options) => {
  58. if (items.length === 0) return items;
  59. const finalResolveOptions = compiler.resolverFactory.get(
  60. "normal",
  61. cachedSetProperty(
  62. options.resolveOptions || EMPTY_RESOLVE_OPTIONS,
  63. "dependencyType",
  64. options.category
  65. )
  66. ).options;
  67. let newItems;
  68. if (!finalResolveOptions.fullySpecified) {
  69. newItems = [];
  70. for (const item of items) {
  71. const { request, context } = item;
  72. for (const ext of finalResolveOptions.extensions) {
  73. if (request.endsWith(ext)) {
  74. newItems.push({
  75. context,
  76. request: request.slice(0, -ext.length)
  77. });
  78. }
  79. }
  80. if (!finalResolveOptions.enforceExtension) {
  81. newItems.push(item);
  82. }
  83. }
  84. items = newItems;
  85. newItems = [];
  86. for (const obj of items) {
  87. const { request, context } = obj;
  88. for (const mainFile of finalResolveOptions.mainFiles) {
  89. if (request.endsWith(`/${mainFile}`)) {
  90. newItems.push({
  91. context,
  92. request: request.slice(0, -mainFile.length)
  93. });
  94. newItems.push({
  95. context,
  96. request: request.slice(0, -mainFile.length - 1)
  97. });
  98. }
  99. }
  100. newItems.push(obj);
  101. }
  102. items = newItems;
  103. }
  104. newItems = [];
  105. for (const item of items) {
  106. let hideOriginal = false;
  107. for (const modulesItems of finalResolveOptions.modules) {
  108. if (Array.isArray(modulesItems)) {
  109. for (const dir of modulesItems) {
  110. if (item.request.startsWith(`./${dir}/`)) {
  111. newItems.push({
  112. context: item.context,
  113. request: item.request.slice(dir.length + 3)
  114. });
  115. hideOriginal = true;
  116. }
  117. }
  118. } else {
  119. const dir = modulesItems.replace(/\\/g, "/");
  120. const fullPath =
  121. item.context.replace(/\\/g, "/") + item.request.slice(1);
  122. if (fullPath.startsWith(dir)) {
  123. newItems.push({
  124. context: item.context,
  125. request: fullPath.slice(dir.length + 1)
  126. });
  127. }
  128. }
  129. }
  130. if (!hideOriginal) {
  131. newItems.push(item);
  132. }
  133. }
  134. return newItems;
  135. }
  136. );
  137. }
  138. );
  139. }
  140. }
  141. module.exports = RequireContextPlugin;