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 {
    compareModulesByPreOrderIndexOrIdentifier
} = require("../util/comparators");
const createSchemaValidation = require("../util/create-schema-validation");
const createHash = require("../util/createHash");
const {
    getUsedModuleIdsAndModules,
    getFullModuleName
} = require("./IdHelpers");
 
/** @typedef {import("../../declarations/plugins/HashedModuleIdsPlugin").HashedModuleIdsPluginOptions} HashedModuleIdsPluginOptions */
/** @typedef {import("../Compiler")} Compiler */
 
const validate = createSchemaValidation(
    require("../../schemas/plugins/HashedModuleIdsPlugin.check.js"),
    () => require("../../schemas/plugins/HashedModuleIdsPlugin.json"),
    {
        name: "Hashed Module Ids Plugin",
        baseDataPath: "options"
    }
);
 
class HashedModuleIdsPlugin {
    /**
     * @param {HashedModuleIdsPluginOptions=} options options object
     */
    constructor(options = {}) {
        validate(options);
 
        /** @type {HashedModuleIdsPluginOptions} */
        this.options = {
            context: null,
            hashFunction: "md4",
            hashDigest: "base64",
            hashDigestLength: 4,
            ...options
        };
    }
 
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        const options = this.options;
        compiler.hooks.compilation.tap("HashedModuleIdsPlugin", compilation => {
            compilation.hooks.moduleIds.tap("HashedModuleIdsPlugin", () => {
                const chunkGraph = compilation.chunkGraph;
                const context = this.options.context
                    ? this.options.context
                    : compiler.context;
 
                const [usedIds, modules] = getUsedModuleIdsAndModules(compilation);
                const modulesInNaturalOrder = modules.sort(
                    compareModulesByPreOrderIndexOrIdentifier(compilation.moduleGraph)
                );
                for (const module of modulesInNaturalOrder) {
                    const ident = getFullModuleName(module, context, compiler.root);
                    const hash = createHash(options.hashFunction);
                    hash.update(ident || "");
                    const hashId = /** @type {string} */ (
                        hash.digest(options.hashDigest)
                    );
                    let len = options.hashDigestLength;
                    while (usedIds.has(hashId.slice(0, len)))
                        /** @type {number} */ (len)++;
                    const moduleId = hashId.slice(0, len);
                    chunkGraph.setModuleId(module, moduleId);
                    usedIds.add(moduleId);
                }
            });
        });
    }
}
 
module.exports = HashedModuleIdsPlugin;