init

马宇豪

2024-07-16 f591c27b57e2418c9495bc02ae8cfff84d35bc18

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