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
85
86
87
88
89
90
91
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const { find } = require("../util/SetHelpers");
const {
    compareModulesByPreOrderIndexOrIdentifier,
    compareModulesByPostOrderIndexOrIdentifier
} = require("../util/comparators");
 
/** @typedef {import("../Compiler")} Compiler */
 
/**
 * @typedef {Object} ChunkModuleIdRangePluginOptions
 * @property {string} name the chunk name
 * @property {("index" | "index2" | "preOrderIndex" | "postOrderIndex")=} order order
 * @property {number=} start start id
 * @property {number=} end end id
 */
 
class ChunkModuleIdRangePlugin {
    /**
     * @param {ChunkModuleIdRangePluginOptions} options options object
     */
    constructor(options) {
        this.options = options;
    }
 
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        const options = this.options;
        compiler.hooks.compilation.tap("ChunkModuleIdRangePlugin", compilation => {
            const moduleGraph = compilation.moduleGraph;
            compilation.hooks.moduleIds.tap("ChunkModuleIdRangePlugin", modules => {
                const chunkGraph = compilation.chunkGraph;
                const chunk = find(
                    compilation.chunks,
                    chunk => chunk.name === options.name
                );
                if (!chunk) {
                    throw new Error(
                        `ChunkModuleIdRangePlugin: Chunk with name '${options.name}"' was not found`
                    );
                }
 
                let chunkModules;
                if (options.order) {
                    let cmpFn;
                    switch (options.order) {
                        case "index":
                        case "preOrderIndex":
                            cmpFn = compareModulesByPreOrderIndexOrIdentifier(moduleGraph);
                            break;
                        case "index2":
                        case "postOrderIndex":
                            cmpFn = compareModulesByPostOrderIndexOrIdentifier(moduleGraph);
                            break;
                        default:
                            throw new Error(
                                "ChunkModuleIdRangePlugin: unexpected value of order"
                            );
                    }
                    chunkModules = chunkGraph.getOrderedChunkModules(chunk, cmpFn);
                } else {
                    chunkModules = Array.from(modules)
                        .filter(m => {
                            return chunkGraph.isModuleInChunk(m, chunk);
                        })
                        .sort(compareModulesByPreOrderIndexOrIdentifier(moduleGraph));
                }
 
                let currentId = options.start || 0;
                for (let i = 0; i < chunkModules.length; i++) {
                    const m = chunkModules[i];
                    if (m.needId && chunkGraph.getModuleId(m) === null) {
                        chunkGraph.setModuleId(m, currentId++);
                    }
                    if (options.end && currentId > options.end) break;
                }
            });
        });
    }
}
module.exports = ChunkModuleIdRangePlugin;