OccurrenceChunkIdsPlugin.js 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Tobias Koppers @sokra
  4. */
  5. "use strict";
  6. const { compareChunksNatural } = require("../util/comparators");
  7. const createSchemaValidation = require("../util/create-schema-validation");
  8. const { assignAscendingChunkIds } = require("./IdHelpers");
  9. /** @typedef {import("../../declarations/plugins/ids/OccurrenceChunkIdsPlugin").OccurrenceChunkIdsPluginOptions} OccurrenceChunkIdsPluginOptions */
  10. /** @typedef {import("../Chunk")} Chunk */
  11. /** @typedef {import("../Compiler")} Compiler */
  12. /** @typedef {import("../Module")} Module */
  13. const validate = createSchemaValidation(
  14. require("../../schemas/plugins/ids/OccurrenceChunkIdsPlugin.check.js"),
  15. () => require("../../schemas/plugins/ids/OccurrenceChunkIdsPlugin.json"),
  16. {
  17. name: "Occurrence Order Chunk Ids Plugin",
  18. baseDataPath: "options"
  19. }
  20. );
  21. const PLUGIN_NAME = "OccurrenceChunkIdsPlugin";
  22. class OccurrenceChunkIdsPlugin {
  23. /**
  24. * @param {OccurrenceChunkIdsPluginOptions=} options options object
  25. */
  26. constructor(options = {}) {
  27. validate(options);
  28. this.options = options;
  29. }
  30. /**
  31. * Apply the plugin
  32. * @param {Compiler} compiler the compiler instance
  33. * @returns {void}
  34. */
  35. apply(compiler) {
  36. const prioritiseInitial = this.options.prioritiseInitial;
  37. compiler.hooks.compilation.tap(PLUGIN_NAME, compilation => {
  38. compilation.hooks.chunkIds.tap(PLUGIN_NAME, chunks => {
  39. const chunkGraph = compilation.chunkGraph;
  40. /** @type {Map<Chunk, number>} */
  41. const occursInInitialChunksMap = new Map();
  42. const compareNatural = compareChunksNatural(chunkGraph);
  43. for (const c of chunks) {
  44. let occurs = 0;
  45. for (const chunkGroup of c.groupsIterable) {
  46. for (const parent of chunkGroup.parentsIterable) {
  47. if (parent.isInitial()) occurs++;
  48. }
  49. }
  50. occursInInitialChunksMap.set(c, occurs);
  51. }
  52. const chunksInOccurrenceOrder = Array.from(chunks).sort((a, b) => {
  53. if (prioritiseInitial) {
  54. const aEntryOccurs =
  55. /** @type {number} */
  56. (occursInInitialChunksMap.get(a));
  57. const bEntryOccurs =
  58. /** @type {number} */
  59. (occursInInitialChunksMap.get(b));
  60. if (aEntryOccurs > bEntryOccurs) return -1;
  61. if (aEntryOccurs < bEntryOccurs) return 1;
  62. }
  63. const aOccurs = a.getNumberOfGroups();
  64. const bOccurs = b.getNumberOfGroups();
  65. if (aOccurs > bOccurs) return -1;
  66. if (aOccurs < bOccurs) return 1;
  67. return compareNatural(a, b);
  68. });
  69. assignAscendingChunkIds(chunksInOccurrenceOrder, compilation);
  70. });
  71. });
  72. }
  73. }
  74. module.exports = OccurrenceChunkIdsPlugin;