OccurrenceChunkIdsPlugin.js 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  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"),
  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. /** @type {Chunk[]} */
  53. const chunksInOccurrenceOrder = [...chunks].sort((a, b) => {
  54. if (prioritiseInitial) {
  55. const aEntryOccurs =
  56. /** @type {number} */
  57. (occursInInitialChunksMap.get(a));
  58. const bEntryOccurs =
  59. /** @type {number} */
  60. (occursInInitialChunksMap.get(b));
  61. if (aEntryOccurs > bEntryOccurs) return -1;
  62. if (aEntryOccurs < bEntryOccurs) return 1;
  63. }
  64. const aOccurs = a.getNumberOfGroups();
  65. const bOccurs = b.getNumberOfGroups();
  66. if (aOccurs > bOccurs) return -1;
  67. if (aOccurs < bOccurs) return 1;
  68. return compareNatural(a, b);
  69. });
  70. assignAscendingChunkIds(chunksInOccurrenceOrder, compilation);
  71. });
  72. });
  73. }
  74. }
  75. module.exports = OccurrenceChunkIdsPlugin;