@taro-hmr-compile/plugin-webpack 0.0.1-alpha.6 → 0.0.1-alpha.8

package/dist/patch/hmr-patch-processor.js

@@ -0,0 +1,105 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HmrPatchProcessor = void 0;
+const generator_1 = __importDefault(require("@babel/generator"));
+const utils_1 = require("./utils");
+const crypto_1 = require("crypto");
+const constants_1 = require("../constants");
+class HmrPatchProcessor {
+    constructor(opts) {
+        this.bootstrapFactories = new Map();
+        this.bootstrapFactoriesByCompilation = new WeakMap();
+        this.moduleHashesByFullHash = new Map();
+        this.recentUpdatePatches = [];
+        this.opts = opts;
+    }
+    getChangedModuleIdsByFullHash(fromHash, toHash) {
+        const fromModuleHashes = this.moduleHashesByFullHash.get(fromHash);
+        const toModuleHashes = this.moduleHashesByFullHash.get(toHash);
+        // 任一编译缺少 Hash 基线时不做过滤，避免误丢有效的 hot-update 模块。
+        if (!fromModuleHashes || !toModuleHashes)
+            return;
+        // 同时检查前后两次编译的模块，确保新增、修改和删除都能被识别。
+        const moduleIds = new Set(fromModuleHashes.keys());
+        toModuleHashes.forEach((_, moduleId) => moduleIds.add(moduleId));
+        const changedModuleIds = new Set();
+        moduleIds.forEach(moduleId => {
+            var _a, _b;
+            const fromFactoryHash = (_a = fromModuleHashes.get(moduleId)) === null || _a === void 0 ? void 0 : _a.factoryHash;
+            const toFactoryHash = (_b = toModuleHashes.get(moduleId)) === null || _b === void 0 ? void 0 : _b.factoryHash;
+            if (fromFactoryHash !== toFactoryHash)
+                changedModuleIds.add(moduleId);
+        });
+        return changedModuleIds;
+    }
+    processUpdate(opt) {
+        var _a;
+        const { fromHash, toHash } = opt;
+        const tempPatchInfo = (0, utils_1.parseHotUpdateAssets)(opt);
+        const changedModuleIds = this.getChangedModuleIdsByFullHash(fromHash, toHash);
+        const moduleFactories = (() => {
+            if (!changedModuleIds)
+                return Array.from(tempPatchInfo.moduleFactories.values());
+            return Array.from(tempPatchInfo.moduleFactories.values()).filter(item => changedModuleIds.has(item.moduleId));
+        })();
+        const removedModules = (() => {
+            if (!changedModuleIds)
+                return Array.from(tempPatchInfo.removedModules.values());
+            return Array.from(tempPatchInfo.removedModules.values()).filter(item => changedModuleIds.has(item));
+        })();
+        this.recentUpdatePatches.push({ type: 'update', moduleFactories, fromHash, toHash, removedModules });
+        const patchHistoryLimit = Math.max(1, (_a = this.opts.patchHistoryLimit) !== null && _a !== void 0 ? _a : constants_1.DEFAULT_HMR_PATCH_HISTORY_LIMIT);
+        if (this.recentUpdatePatches.length <= patchHistoryLimit)
+            return;
+        const [oldestUpdatePatch, secondOldestUpdatePatch, ...rest] = this.recentUpdatePatches;
+        const mergedUpdatePatch = (0, utils_1.mergeUpdatePatches)(oldestUpdatePatch, secondOldestUpdatePatch);
+        this.recentUpdatePatches = [mergedUpdatePatch, ...rest];
+        this.moduleHashesByFullHash.delete(oldestUpdatePatch.toHash);
+    }
+    processChunkModules(opt) {
+        const { hash: fullHash } = opt.compilation;
+        const modules = (0, utils_1.parseChunkAsset)(opt.source);
+        modules.forEach(moduleFactory => {
+            var _a;
+            const { moduleId, factory } = moduleFactory;
+            if (opt.isBootstrapChunk) {
+                const compilationFactories = (_a = this.bootstrapFactoriesByCompilation.get(opt.compilation)) !== null && _a !== void 0 ? _a : new Map();
+                compilationFactories.set(moduleId, { moduleId, factory });
+                this.bootstrapFactoriesByCompilation.set(opt.compilation, compilationFactories);
+            }
+            if (!fullHash)
+                return;
+            const moduleHashes = (() => {
+                const hasCurFullHashModuleHashMap = this.moduleHashesByFullHash.has(fullHash);
+                if (!hasCurFullHashModuleHashMap)
+                    this.moduleHashesByFullHash.set(fullHash, new Map());
+                return this.moduleHashesByFullHash.get(fullHash);
+            })();
+            const { code: source } = (0, generator_1.default)(factory, { compact: true, comments: false, minified: true, });
+            const factoryHash = (0, crypto_1.createHash)('sha256').update(source).digest('hex');
+            moduleHashes.set(moduleId, { moduleId, factoryHash });
+        });
+    }
+    commitBootstrapFactories(compilation) {
+        var _a;
+        this.bootstrapFactories.clear();
+        (_a = this.bootstrapFactoriesByCompilation.get(compilation)) === null || _a === void 0 ? void 0 : _a.forEach(moduleFactory => {
+            this.bootstrapFactories.set(moduleFactory.moduleId, moduleFactory);
+        });
+        this.bootstrapFactoriesByCompilation.delete(compilation);
+    }
+    generate() {
+        const { bootstrapFactories, recentUpdatePatches } = this;
+        return (0, utils_1.generatePatchAssetSource)({ bootstrapFactories, recentUpdatePatches });
+    }
+    reset() {
+        this.bootstrapFactories.clear();
+        this.moduleHashesByFullHash.clear();
+        this.recentUpdatePatches = [];
+    }
+}
+exports.HmrPatchProcessor = HmrPatchProcessor;
+//# sourceMappingURL=hmr-patch-processor.js.map

package/dist/patch/hmr-patch-processor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hmr-patch-processor.js","sourceRoot":"","sources":["../../src/patch/hmr-patch-processor.ts"],"names":[],"mappings":";;;;;;AACA,iEAAwC;AACxC,mCAIiB;AACjB,mCAAkC;AAElC,4CAA6D;AA0B7D,MAAa,iBAAiB;IAW1B,YAAY,IAAuC;QAVlC,uBAAkB,GAAuB,IAAI,GAAG,EAAE,CAAA;QAElD,oCAA+B,GAAG,IAAI,OAAO,EAAmC,CAAA;QAEhF,2BAAsB,GAA2B,IAAI,GAAG,EAAE,CAAA;QAEnE,wBAAmB,GAAwB,EAAE,CAAA;QAKjD,IAAI,CAAC,IAAI,GAAG,IAAI,CAAA;IACpB,CAAC;IAEO,6BAA6B,CAAC,QAAgB,EAAE,MAAc;QAClE,MAAM,gBAAgB,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QAElE,MAAM,cAAc,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAE9D,6CAA6C;QAC7C,IAAI,CAAC,gBAAgB,IAAI,CAAC,cAAc;YAAE,OAAM;QAEhD,iCAAiC;QACjC,MAAM,SAAS,GAAG,IAAI,GAAG,CAAC,gBAAgB,CAAC,IAAI,EAAE,CAAC,CAAA;QAClD,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAA;QAEhE,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAU,CAAA;QAC1C,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;;YACzB,MAAM,eAAe,GAAG,MAAA,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,0CAAE,WAAW,CAAA;YACnE,MAAM,aAAa,GAAG,MAAA,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,0CAAE,WAAW,CAAA;YAC/D,IAAI,eAAe,KAAK,aAAa;gBAAE,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QACzE,CAAC,CAAC,CAAA;QAEF,OAAO,gBAAgB,CAAA;IAC3B,CAAC;IAED,aAAa,CAAC,GAAyB;;QACnC,MAAM,EAAC,QAAQ,EAAE,MAAM,EAAC,GAAG,GAAG,CAAA;QAE9B,MAAM,aAAa,GAAG,IAAA,4BAAoB,EAAC,GAAG,CAAC,CAAA;QAE/C,MAAM,gBAAgB,GAAG,IAAI,CAAC,6BAA6B,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;QAE7E,MAAM,eAAe,GAAG,CAAC,GAAG,EAAE;YAC1B,IAAI,CAAC,gBAAgB;gBAAE,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAA;YAChF,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAA;QACjH,CAAC,CAAC,EAAE,CAAA;QAEJ,MAAM,cAAc,GAAG,CAAC,GAAG,EAAE;YACzB,IAAI,CAAC,gBAAgB;gBAAE,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAA;YAC/E,OAAO,KAAK,CAAC,IAAI,CAAC,aAAa,CAAC,cAAc,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAA;QACvG,CAAC,CAAC,EAAE,CAAA;QAEJ,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAC,IAAI,EAAE,QAAQ,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,EAAE,cAAc,EAAC,CAAC,CAAA;QAElG,MAAM,iBAAiB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAA,IAAI,CAAC,IAAI,CAAC,iBAAiB,mCAAI,2CAA+B,CAAC,CAAA;QAErG,IAAI,IAAI,CAAC,mBAAmB,CAAC,MAAM,IAAI,iBAAiB;YAAE,OAAM;QAEhE,MAAM,CAAC,iBAAiB,EAAE,uBAAuB,EAAE,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC,mBAAmB,CAAA;QAEtF,MAAM,iBAAiB,GAAG,IAAA,0BAAkB,EAAC,iBAAiB,EAAE,uBAAuB,CAAC,CAAA;QAExF,IAAI,CAAC,mBAAmB,GAAG,CAAC,iBAAiB,EAAE,GAAG,IAAI,CAAC,CAAA;QAEvD,IAAI,CAAC,sBAAsB,CAAC,MAAM,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAA;IAChE,CAAC;IAED,mBAAmB,CAAC,GAA+B;QAC/C,MAAM,EAAC,IAAI,EAAE,QAAQ,EAAC,GAAG,GAAG,CAAC,WAAW,CAAA;QACxC,MAAM,OAAO,GAAG,IAAA,uBAAe,EAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAC3C,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE;;YAC5B,MAAM,EAAC,QAAQ,EAAE,OAAO,EAAC,GAAG,aAAa,CAAA;YAEzC,IAAI,GAAG,CAAC,gBAAgB,EAAE,CAAC;gBACvB,MAAM,oBAAoB,GAAG,MAAA,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,mCAAI,IAAI,GAAG,EAAE,CAAA;gBACnG,oBAAoB,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAC,QAAQ,EAAE,OAAO,EAAC,CAAC,CAAA;gBACvD,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,oBAAoB,CAAC,CAAA;YACnF,CAAC;YAED,IAAI,CAAC,QAAQ;gBAAE,OAAM;YAErB,MAAM,YAAY,GAAG,CAAC,GAAG,EAAE;gBACvB,MAAM,2BAA2B,GAAG,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;gBAC7E,IAAI,CAAC,2BAA2B;oBAAE,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,GAAG,EAAE,CAAC,CAAA;gBACtF,OAAO,IAAI,CAAC,sBAAsB,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAA;YACrD,CAAC,CAAC,EAAE,CAAA;YAEJ,MAAM,EAAC,IAAI,EAAE,MAAM,EAAC,GAAG,IAAA,mBAAS,EAAC,OAAO,EAAE,EAAC,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,GAAE,CAAC,CAAA;YAC5F,MAAM,WAAW,GAAG,IAAA,mBAAU,EAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;YACrE,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAC,QAAQ,EAAE,WAAW,EAAC,CAAC,CAAA;QACvD,CAAC,CAAC,CAAA;IACN,CAAC;IAED,wBAAwB,CAAC,WAAwB;;QAC7C,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAA;QAC/B,MAAA,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,WAAW,CAAC,0CAAE,OAAO,CAAC,aAAa,CAAC,EAAE;YAC3E,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAA;QACtE,CAAC,CAAC,CAAA;QACF,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,WAAW,CAAC,CAAA;IAC5D,CAAC;IAED,QAAQ;QACJ,MAAM,EAAC,kBAAkB,EAAE,mBAAmB,EAAC,GAAG,IAAI,CAAA;QACtD,OAAO,IAAA,gCAAwB,EAAC,EAAC,kBAAkB,EAAE,mBAAmB,EAAC,CAAC,CAAA;IAC9E,CAAC;IAED,KAAK;QACD,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAA;QAC/B,IAAI,CAAC,sBAAsB,CAAC,KAAK,EAAE,CAAA;QACnC,IAAI,CAAC,mBAAmB,GAAG,EAAE,CAAA;IACjC,CAAC;CACJ;AAjHD,8CAiHC"}

package/dist/patch/types.d.ts

@@ -0,0 +1,15 @@
+import type { Expression } from "@babel/types";
+export interface ModuleFactory {
+    factory: Expression;
+    moduleId: string;
+}
+export interface HotUpdateManifest {
+    m?: Array<string | number>;
+}
+export interface UpdatePatchSource {
+    type: 'update';
+    fromHash: string;
+    toHash: string;
+    moduleFactories: ModuleFactory[];
+    removedModules: string[];
+}

package/dist/patch/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/patch/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/patch/types.ts"],"names":[],"mappings":""}

package/dist/patch/utils.d.ts

@@ -0,0 +1,26 @@
+import { Asset } from "webpack";
+import { ModuleFactory } from "./types";
+import { UpdatePatchSource } from "./types";
+export interface ParseHotUpdateAssetsOptions {
+    manifests: Readonly<Asset>[];
+    scripts: Readonly<Asset>[];
+}
+/**
+ * 解析热更新产物
+ * @param opt
+ */
+export declare const parseHotUpdateAssets: (opt: ParseHotUpdateAssetsOptions) => {
+    moduleFactories: Map<string, ModuleFactory>;
+    removedModules: Set<string>;
+};
+/**
+ * 解析 chunk 产物
+ * @param source
+ */
+export declare const parseChunkAsset: (source: string) => Map<string, ModuleFactory>;
+export declare const mergeUpdatePatches: (basePatch: UpdatePatchSource, nextPatch: UpdatePatchSource) => UpdatePatchSource;
+export interface generatePatchAssetSourceOpts {
+    bootstrapFactories: Map<string, ModuleFactory>;
+    recentUpdatePatches: UpdatePatchSource[];
+}
+export declare const generatePatchAssetSource: (opt: generatePatchAssetSourceOpts) => string;

package/dist/patch/utils.js

@@ -0,0 +1,173 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generatePatchAssetSource = exports.mergeUpdatePatches = exports.parseChunkAsset = exports.parseHotUpdateAssets = void 0;
+const t = __importStar(require("@babel/types"));
+const parser_1 = require("@babel/parser");
+const traverse_1 = __importDefault(require("@babel/traverse"));
+const template_1 = __importDefault(require("@babel/template"));
+const generator_1 = __importDefault(require("@babel/generator"));
+/**
+ * 解析模块 Factory
+ * @param expression
+ */
+const parseModuleFactories = (expression) => {
+    const moduleFactories = new Map();
+    if (!t.isObjectExpression(expression))
+        return moduleFactories.values();
+    expression.properties.forEach(property => {
+        if (!t.isObjectProperty(property) || !t.isExpression(property.value))
+            throw new Error('[taro-hmr-compile] HMR 模块对象包含无法识别的属性');
+        const { key, value: factory } = property;
+        const moduleId = (() => {
+            if (t.isStringLiteral(key) || t.isNumericLiteral(key))
+                return String(key.value);
+            if (t.isIdentifier(key) && !property.computed)
+                return key.name;
+            throw new Error('[taro-hmr-compile] Chunk 模块对象包含无法识别的模块 ID');
+        })();
+        moduleFactories.set(moduleId, { factory, moduleId });
+    });
+    return moduleFactories.values();
+};
+/**
+ * 解析热更新产物
+ * @param opt
+ */
+const parseHotUpdateAssets = (opt) => {
+    const { manifests, scripts } = opt;
+    const moduleFactories = new Map();
+    scripts.forEach(asset => {
+        const ast = (0, parser_1.parse)(asset.source.source().toString(), { sourceType: 'script' });
+        (0, traverse_1.default)(ast, {
+            CallExpression(nodePath) {
+                const { callee } = nodePath.node;
+                if (!t.isMemberExpression(callee) || !t.isIdentifier(callee.object, { name: 'wx' }))
+                    return;
+                if (!t.isStringLiteral(callee.property) || !callee.property.value.startsWith('webpackHotUpdate'))
+                    return;
+                const [_, modules] = nodePath.node.arguments;
+                if (!t.isObjectExpression(modules))
+                    return;
+                parseModuleFactories(modules).forEach(module => moduleFactories.set(module.moduleId, module));
+            },
+        });
+    });
+    const removedModules = new Set();
+    manifests.forEach((module) => {
+        const manifest = JSON.parse(module.source.source().toString());
+        const { m: tempRemoveModules = [] } = manifest;
+        tempRemoveModules.forEach(removedModule => removedModules.add(String(removedModule)));
+    }, []);
+    return { moduleFactories, removedModules };
+};
+exports.parseHotUpdateAssets = parseHotUpdateAssets;
+/**
+ * 解析 chunk 产物
+ * @param source
+ */
+const parseChunkAsset = (source) => {
+    const ast = (0, parser_1.parse)(`(${source})`, { sourceType: 'script' });
+    const [statement] = ast.program.body;
+    if (!t.isExpressionStatement(statement) || !t.isObjectExpression(statement.expression))
+        throw new Error('[taro-hmr-compile] Chunk 模块段不是对象表达式');
+    const moduleFactories = new Map();
+    parseModuleFactories(statement.expression).forEach(module => moduleFactories.set(module.moduleId, module));
+    return moduleFactories;
+};
+exports.parseChunkAsset = parseChunkAsset;
+const mergeUpdatePatches = (basePatch, nextPatch) => {
+    if (basePatch.toHash !== nextPatch.fromHash)
+        throw new Error(`[taro-hmr-compile] 无法合并不连续的 HMR patch：${basePatch.toHash} !== ${nextPatch.fromHash}`);
+    const moduleFactories = new Map(basePatch.moduleFactories.map(moduleFactory => [moduleFactory.moduleId, moduleFactory]));
+    const removedModules = new Set(basePatch.removedModules);
+    nextPatch.moduleFactories.forEach(moduleFactory => {
+        const { moduleId } = moduleFactory;
+        moduleFactories.set(moduleId, moduleFactory);
+        removedModules.delete(moduleId);
+    });
+    nextPatch.removedModules.forEach(moduleId => {
+        moduleFactories.delete(moduleId);
+        removedModules.add(moduleId);
+    });
+    return {
+        type: 'update',
+        fromHash: basePatch.fromHash,
+        toHash: nextPatch.toHash,
+        moduleFactories: [...moduleFactories.values()],
+        removedModules: [...removedModules],
+    };
+};
+exports.mergeUpdatePatches = mergeUpdatePatches;
+const generatePatchAssetSource = (opt) => {
+    const { bootstrapFactories, recentUpdatePatches } = opt;
+    const buildBootstrapPatch = template_1.default.expression(`({
+        type: 'bootstrap',
+        modules: MODULES
+    })`);
+    const buildUpdatePatch = template_1.default.expression(`({
+        type: 'update',
+        fromHash: "FROM_HASH",
+        toHash: "TO_HASH",
+        modules: MODULES,
+        removedModules: REMOVED_MODULES
+    })`);
+    const buildPatchAsset = template_1.default.program('module.exports = PATCHES;');
+    const bootstrapModules = Array.from(bootstrapFactories.values()).map((moduleFactory) => {
+        const { moduleId, factory } = moduleFactory;
+        return t.objectProperty(t.stringLiteral(moduleId), factory);
+    });
+    const patchNodes = [
+        buildBootstrapPatch({ MODULES: t.objectExpression(bootstrapModules) }),
+        ...recentUpdatePatches.map(patch => {
+            const updateModules = patch.moduleFactories.map((moduleFactory) => {
+                const { moduleId, factory } = moduleFactory;
+                return t.objectProperty(t.stringLiteral(moduleId), factory);
+            });
+            return buildUpdatePatch({
+                FROM_HASH: patch.fromHash,
+                TO_HASH: patch.toHash,
+                MODULES: t.objectExpression(updateModules),
+                REMOVED_MODULES: t.arrayExpression(patch.removedModules.map(moduleId => t.stringLiteral(moduleId))),
+            });
+        }),
+    ];
+    return (0, generator_1.default)(buildPatchAsset({ PATCHES: t.arrayExpression(patchNodes) })).code;
+};
+exports.generatePatchAssetSource = generatePatchAssetSource;
+//# sourceMappingURL=utils.js.map

package/dist/patch/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/patch/utils.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,gDAAkC;AAClC,0CAAoC;AACpC,+DAAuC;AACvC,+DAAuC;AACvC,iEAAyC;AAOzC;;;GAGG;AACH,MAAM,oBAAoB,GAAG,CAAC,UAA4B,EAAE,EAAE;IAE1D,MAAM,eAAe,GAAG,IAAI,GAAG,EAAyB,CAAA;IAExD,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC;QAAE,OAAO,eAAe,CAAC,MAAM,EAAE,CAAA;IAEtE,UAAU,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;QACrC,IAAI,CAAC,CAAC,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAA;QAE7H,MAAM,EAAC,GAAG,EAAE,KAAK,EAAE,OAAO,EAAC,GAAG,QAAQ,CAAA;QAEtC,MAAM,QAAQ,GAAG,CAAC,GAAG,EAAE;YACnB,IAAI,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC;gBAAE,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;YAC/E,IAAI,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ;gBAAE,OAAO,GAAG,CAAC,IAAI,CAAA;YAC9D,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAA;QAChE,CAAC,CAAC,EAAE,CAAA;QAEJ,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,EAAC,OAAO,EAAE,QAAQ,EAAC,CAAC,CAAA;IACtD,CAAC,CAAC,CAAA;IAEF,OAAO,eAAe,CAAC,MAAM,EAAE,CAAA;AACnC,CAAC,CAAA;AAQD;;;GAGG;AACI,MAAM,oBAAoB,GAAG,CAAC,GAAgC,EAAE,EAAE;IACrE,MAAM,EAAC,SAAS,EAAE,OAAO,EAAC,GAAG,GAAG,CAAA;IAEhC,MAAM,eAAe,GAAG,IAAI,GAAG,EAAyB,CAAA;IAExD,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;QAEpB,MAAM,GAAG,GAAG,IAAA,cAAK,EAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,EAAC,UAAU,EAAE,QAAQ,EAAC,CAAC,CAAA;QAE3E,IAAA,kBAAQ,EAAC,GAAG,EAAE;YACV,cAAc,CAAC,QAAQ;gBACnB,MAAM,EAAC,MAAM,EAAC,GAAG,QAAQ,CAAC,IAAI,CAAA;gBAE9B,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,EAAE,EAAC,IAAI,EAAE,IAAI,EAAC,CAAC;oBAAE,OAAM;gBAEzF,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,UAAU,CAAC,kBAAkB,CAAC;oBAAE,OAAM;gBAExG,MAAM,CAAC,CAAC,EAAE,OAAO,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAA;gBAE5C,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,CAAC;oBAAE,OAAM;gBAE1C,oBAAoB,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAA;YACjG,CAAC;SACJ,CAAC,CAAA;IACN,CAAC,CAAC,CAAA;IAEF,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAA;IAExC,SAAS,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;QAEzB,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAsB,CAAA;QAEnF,MAAM,EAAC,CAAC,EAAE,iBAAiB,GAAG,EAAE,EAAC,GAAG,QAAQ,CAAA;QAE5C,iBAAiB,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAA;IACzF,CAAC,EAAE,EAAE,CAAC,CAAA;IAEN,OAAO,EAAC,eAAe,EAAE,cAAc,EAAC,CAAA;AAC5C,CAAC,CAAA;AAtCY,QAAA,oBAAoB,wBAsChC;AAED;;;GAGG;AACI,MAAM,eAAe,GAAG,CAAC,MAAc,EAA8B,EAAE;IAC1E,MAAM,GAAG,GAAG,IAAA,cAAK,EAAC,IAAI,MAAM,GAAG,EAAE,EAAC,UAAU,EAAE,QAAQ,EAAC,CAAC,CAAA;IAExD,MAAM,CAAC,SAAS,CAAC,GAAG,GAAG,CAAC,OAAO,CAAC,IAAI,CAAA;IAEpC,IAAI,CAAC,CAAC,CAAC,qBAAqB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,SAAS,CAAC,UAAU,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAA;IAE9I,MAAM,eAAe,GAAG,IAAI,GAAG,EAAyB,CAAA;IAExD,oBAAoB,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAA;IAE1G,OAAO,eAAe,CAAA;AAC1B,CAAC,CAAA;AAZY,QAAA,eAAe,mBAY3B;AAEM,MAAM,kBAAkB,GAAG,CAAC,SAA4B,EAAE,SAA4B,EAAsB,EAAE;IACjH,IAAI,SAAS,CAAC,MAAM,KAAK,SAAS,CAAC,QAAQ;QAAE,MAAM,IAAI,KAAK,CAAC,yCAAyC,SAAS,CAAC,MAAM,QAAQ,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAA;IAEnJ,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,eAAe,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,aAAa,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,CAAA;IAExH,MAAM,cAAc,GAAG,IAAI,GAAG,CAAC,SAAS,CAAC,cAAc,CAAC,CAAA;IAExD,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE;QAC9C,MAAM,EAAC,QAAQ,EAAC,GAAG,aAAa,CAAA;QAChC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAA;QAC5C,cAAc,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IACnC,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE;QACxC,eAAe,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;QAChC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;IAChC,CAAC,CAAC,CAAA;IAEF,OAAO;QACH,IAAI,EAAE,QAAQ;QACd,QAAQ,EAAE,SAAS,CAAC,QAAQ;QAC5B,MAAM,EAAE,SAAS,CAAC,MAAM;QACxB,eAAe,EAAE,CAAC,GAAG,eAAe,CAAC,MAAM,EAAE,CAAC;QAC9C,cAAc,EAAE,CAAC,GAAG,cAAc,CAAC;KACtC,CAAA;AACL,CAAC,CAAA;AAzBY,QAAA,kBAAkB,sBAyB9B;AAOM,MAAM,wBAAwB,GAAG,CAAC,GAAiC,EAAU,EAAE;IAClF,MAAM,EAAC,kBAAkB,EAAE,mBAAmB,EAAC,GAAG,GAAG,CAAA;IACrD,MAAM,mBAAmB,GAAG,kBAAQ,CAAC,UAAU,CAAC;;;OAG7C,CAAC,CAAA;IACJ,MAAM,gBAAgB,GAAG,kBAAQ,CAAC,UAAU,CAAC;;;;;;OAM1C,CAAC,CAAA;IACJ,MAAM,eAAe,GAAG,kBAAQ,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAA;IACrE,MAAM,gBAAgB,GAAG,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE;QACnF,MAAM,EAAC,QAAQ,EAAE,OAAO,EAAC,GAAG,aAAa,CAAA;QACzC,OAAO,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAA;IAC/D,CAAC,CAAC,CAAA;IACF,MAAM,UAAU,GAAG;QACf,mBAAmB,CAAC,EAAC,OAAO,EAAE,CAAC,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,EAAC,CAAC;QACpE,GAAG,mBAAmB,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YAC/B,MAAM,aAAa,GAAG,KAAK,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE;gBAC9D,MAAM,EAAC,QAAQ,EAAE,OAAO,EAAC,GAAG,aAAa,CAAA;gBACzC,OAAO,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC,CAAA;YAC/D,CAAC,CAAC,CAAA;YACF,OAAO,gBAAgB,CAAC;gBACpB,SAAS,EAAE,KAAK,CAAC,QAAQ;gBACzB,OAAO,EAAE,KAAK,CAAC,MAAM;gBACrB,OAAO,EAAE,CAAC,CAAC,gBAAgB,CAAC,aAAa,CAAC;gBAC1C,eAAe,EAAE,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC,CAAC;aACtG,CAAC,CAAA;QACN,CAAC,CAAC;KACL,CAAA;IAED,OAAO,IAAA,mBAAS,EAAC,eAAe,CAAC,EAAC,OAAO,EAAE,CAAC,CAAC,eAAe,CAAC,UAAU,CAAC,EAAC,CAAC,CAAC,CAAC,IAAI,CAAA;AACpF,CAAC,CAAA;AAnCY,QAAA,wBAAwB,4BAmCpC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@taro-hmr-compile/plugin-webpack",
-  "version": "0.0.1-alpha.6",
+  "version": "0.0.1-alpha.8",
   "description": "",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -44,7 +44,7 @@
     "@babel/template": "^7.29.7",
     "@babel/traverse": "^7.29.8",
     "@babel/types": "^7.29.8",
-    "@taro-hmr-compile/core": "0.0.1-alpha.6",
+    "@taro-hmr-compile/core": "0.0.1-alpha.8",
     "debug": "^4.4.3",
     "tapable": "^2.3.3",
     "webpack-sources": "^3.5.1"

package/src/compile/hmr-compiler-loader.ts

@@ -1,17 +1,24 @@
 import generator from '@babel/generator'
 import * as parser from '@babel/parser'
 import template from '@babel/template'
-import traverse from '@babel/traverse'
+import traverse, {NodePath} from '@babel/traverse'
 import * as t from '@babel/types'
 import type {LoaderDefinitionFunction} from 'webpack'
 import webpack from 'webpack'
 
 const CREATE_PAGE_CONFIG = 'createPageConfig'
+
 const WITH_HMR_PAGE_CONFIG = 'withHmrPageConfig'
-const CORE_RUNTIME_REQUEST = require.resolve('@taro-hmr-compile/core/runtime')
+
+const TARO_RUNTIME_REQUEST = '@tarojs/runtime'
+
 const LOG_REQUEST = require.resolve('../log')
 
+const CORE_RUNTIME_REQUEST = require.resolve('@taro-hmr-compile/core/runtime')
+
 export const injectAppHmrRuntime = (source: string, debugNamespaces = '') => {
+
+
     const ast = parser.parse(source, {sourceType: 'module'})
 
     const buildHmrCode = template.statements(`
@@ -40,7 +47,7 @@ export const injectAppHmrRuntime = (source: string, debugNamespaces = '') => {
     traverse(ast, {
         Program(path) {
             const body = path.get('body')
-            const hasInjected = body.some(item => item.isImportDeclaration() && item.node.source.value === CORE_RUNTIME_REQUEST)
+            const hasInjected = body.some(item => item.isImportDeclaration() && t.isStringLiteral(item.node.source, {value: CORE_RUNTIME_REQUEST}))
             if (hasInjected) return
 
             const nodes = buildHmrCode()
@@ -55,54 +62,67 @@ export const injectAppHmrRuntime = (source: string, debugNamespaces = '') => {
 }
 
 export const transformPageConfig = (source: string, sourceFileName?: string) => {
+
     const ast = parser.parse(source, {sourceType: 'module', plugins: ['typescript', 'jsx']})
-    let transformed = false
+
+    const hasImportWithHmrPageConfig = (programPath: NodePath<t.Program>) => {
+        return programPath.node.body.some(node => {
+            if (!t.isImportDeclaration(node)) return false
+            if (!t.isStringLiteral(node.source, {value: CORE_RUNTIME_REQUEST})) return false
+            return node.specifiers.some(specifier => {
+                return t.isImportSpecifier(specifier) && t.isIdentifier(specifier.imported, {name: WITH_HMR_PAGE_CONFIG})
+            })
+        })
+    }
 
     traverse(ast, {
         Program(programPath) {
+
+            // 判断是否存在 createPageConfig 引用
             const createPageConfigBinding = programPath.scope.getBinding(CREATE_PAGE_CONFIG)
             if (!createPageConfigBinding || !createPageConfigBinding.path.isImportSpecifier()) return
-            const importDeclaration = createPageConfigBinding.path.parentPath
-            if (!importDeclaration.isImportDeclaration() || importDeclaration.node.source.value !== '@tarojs/runtime') return
 
-            const wrapperIdentifier = programPath.scope.hasBinding(WITH_HMR_PAGE_CONFIG)
-                ? programPath.scope.generateUidIdentifier(WITH_HMR_PAGE_CONFIG)
-                : t.identifier(WITH_HMR_PAGE_CONFIG)
+            // 判断 createPageConfig 引用是否来自 @tarojs/runtime
+            const {parentPath: createPageConfigImportDeclaration} = createPageConfigBinding.path
+            if (!t.isImportDeclaration(createPageConfigImportDeclaration.node)) return
+            if (!t.isStringLiteral(createPageConfigImportDeclaration.node.source, {value: TARO_RUNTIME_REQUEST})) return
 
-            programPath.traverse({
-                VariableDeclarator(variablePath) {
-                    if (!t.isIdentifier(variablePath.node.id, {name: 'taroOption'})) return
-                    const init = variablePath.node.init
-                    if (!t.isCallExpression(init) || !t.isIdentifier(init.callee, {name: CREATE_PAGE_CONFIG})) return
-                    if (init.arguments.length < 2 || t.isSpreadElement(init.arguments[1])) return
-                    const templateCode = '%%WRAPPER%%(%%PAGE_CONFIG%%, %%PAGE_NAME%%)'
-                    variablePath.node.init = template.expression(templateCode)({WRAPPER: wrapperIdentifier, PAGE_CONFIG: init, PAGE_NAME: init.arguments[1]})
-                    transformed = true
-                    variablePath.stop()
-                },
-            })
+            // 判断是否已经注入了 withHmrPageConfig
+            if (hasImportWithHmrPageConfig(programPath)) return
+
+            // 判断是否存在 withHmrPageConfig 引用
+            const hasWithHmrPageConfigBinding = programPath.scope.hasBinding(WITH_HMR_PAGE_CONFIG)
+
+            // 生成 withHmrPageConfig 的别名
+            const wrapperIdentifier = hasWithHmrPageConfigBinding ? programPath.scope.generateUidIdentifier(WITH_HMR_PAGE_CONFIG) : t.identifier(WITH_HMR_PAGE_CONFIG)
 
-            if (!transformed) return
+            // 生成 runtime import
+            const runtimeImport = (() => {
+                if (!hasWithHmrPageConfigBinding) return template.statement(`import { ${WITH_HMR_PAGE_CONFIG} } from ${JSON.stringify(CORE_RUNTIME_REQUEST)};`)()
+                return template.statement(`import { ${WITH_HMR_PAGE_CONFIG} as %%WRAPPER%% } from ${JSON.stringify(CORE_RUNTIME_REQUEST)};`)({WRAPPER: wrapperIdentifier})
+            })()
 
-            const existingImport = programPath.get('body').find(item => {
-                return item.isImportDeclaration() && item.node.source.value === CORE_RUNTIME_REQUEST
+            programPath.traverse({
+                CallExpression(CallExpressionPath) {
+                    const {node: CallExpressionNode} = CallExpressionPath
+                    if (!t.isIdentifier(CallExpressionNode.callee, {name: CREATE_PAGE_CONFIG})) return
+                    const [_, pageName] = CallExpressionNode.arguments
+                    const wrappedPageConfig = template.expression(`%%WRAPPER%%(%%PAGE_CONFIG%%, %%PAGE_NAME%%)`)({
+                        WRAPPER: wrapperIdentifier,
+                        PAGE_CONFIG: CallExpressionNode,
+                        PAGE_NAME: pageName,
+                    })
+
+                    if (!hasImportWithHmrPageConfig(programPath)) programPath.unshiftContainer('body', runtimeImport)
+
+                    CallExpressionPath.replaceWith(wrappedPageConfig)
+                    CallExpressionPath.skip()
+                }
             })
-            const runtimeImport = wrapperIdentifier.name === WITH_HMR_PAGE_CONFIG
-                ? template.statement(`import { withHmrPageConfig } from ${JSON.stringify(CORE_RUNTIME_REQUEST)}`)()
-                : template.statement(`import { withHmrPageConfig as %%WRAPPER%% } from ${JSON.stringify(CORE_RUNTIME_REQUEST)}`)({WRAPPER: wrapperIdentifier})
-            if (!t.isImportDeclaration(runtimeImport)) throw new TypeError('Expected runtime import template to create an import declaration')
-            const hasNamespaceSpecifier = existingImport?.isImportDeclaration() && existingImport.node.specifiers.some(specifier => t.isImportNamespaceSpecifier(specifier))
-
-            if (existingImport?.isImportDeclaration() && !hasNamespaceSpecifier) {
-                existingImport.pushContainer('specifiers', runtimeImport.specifiers[0])
-            } else {
-                programPath.unshiftContainer('body', runtimeImport)
-            }
+
         },
     })
 
-    if (!transformed) return
-
     return generator(ast, {sourceMaps: true, sourceFileName}, source).code
 }
 
@@ -113,11 +133,9 @@ interface LoaderOptions {
 
 const hmrCompilerLoader: LoaderDefinitionFunction<LoaderOptions> = function (source) {
     const {miniType, debugNamespaces} = this.getOptions()
-
     if (miniType === 'ENTRY') return injectAppHmrRuntime(source, debugNamespaces)
     if (miniType === 'PAGE') return transformPageConfig(source, this.resourcePath) ?? source
-
-    throw new Error(`unsupported HMR entry miniType: ${miniType}`)
+    return source
 }
 
 export default hmrCompilerLoader

package/src/compile/hmr-compiler-plugin.ts

@@ -29,12 +29,16 @@ export interface HmrCompilerPluginOptions {
     debugNamespaces?: string
 }
 
+type CompileRequested = SyncHook<[page: string]>
+
+type ChunkModulesRendered = SyncHook<[source: string, isPageChunk: boolean]>
+
 export interface HmrCompilerHooks {
-    compileRequested: SyncHook<[page: string]>
+    compileRequested: CompileRequested
 }
 
 export interface HmrCompilationHooks {
-    pageChunkModules: SyncHook<[source: string]>
+    chunkModulesRendered: ChunkModulesRendered
 }
 
 export const applyPatchBeforeStartup = (source: Source, request: string) => {
@@ -67,21 +71,19 @@ export class HmrCompilerPlugin {
     }
 
     static getCompilerHooks(compiler: Compiler): HmrCompilerHooks {
-        let hooks = HmrCompilerPlugin.compilerHooksMap.get(compiler)
-        if (!hooks) {
-            hooks = {compileRequested: new SyncHook(['page'])}
-            HmrCompilerPlugin.compilerHooksMap.set(compiler, hooks)
+        if (!HmrCompilerPlugin.compilerHooksMap.get(compiler)) {
+            const compileRequested: CompileRequested = new SyncHook(['page'])
+            HmrCompilerPlugin.compilerHooksMap.set(compiler, {compileRequested})
         }
-        return hooks
+        return HmrCompilerPlugin.compilerHooksMap.get(compiler)!
     }
 
     static getCompilationHooks(compilation: Compilation): HmrCompilationHooks {
-        let hooks = HmrCompilerPlugin.compilationHooksMap.get(compilation)
-        if (!hooks) {
-            hooks = {pageChunkModules: new SyncHook(['source'])}
-            HmrCompilerPlugin.compilationHooksMap.set(compilation, hooks)
+        if (!HmrCompilerPlugin.compilationHooksMap.get(compilation)) {
+            const chunkModulesRendered: ChunkModulesRendered = new SyncHook(['source', 'isBootstrapChunk'])
+            HmrCompilerPlugin.compilationHooksMap.set(compilation, {chunkModulesRendered})
         }
-        return hooks
+        return HmrCompilerPlugin.compilationHooksMap.get(compilation)!
     }
 
     apply(compiler: Compiler) {
@@ -124,16 +126,16 @@ export class HmrCompilerPlugin {
 
                     const loaderOptions = {miniType: normalModule.miniType, debugNamespaces: this.options.debugNamespaces}
                     // Loader 从右向左执行；插到 taro-loader 左侧才能处理它生成的入口注册代码。
-                    normalModule.loaders.splice(taroLoaderIndex, 0, { loader: HMR_COMPILER_LOADER, options: loaderOptions})
+                    normalModule.loaders.splice(taroLoaderIndex, 0, {loader: HMR_COMPILER_LOADER, options: loaderOptions})
                 },
             )
 
             // 只修改 Page Chunk 的 JS transport。Webpack 的 ModuleGraph / ChunkGraph 保持不变，
             // 因而 hash、HMR diff 和 hot-update 仍以完整模块图为准。
             javascriptHooks.renderChunk.tap({name: PLUGIN_NAME, stage: -10_000}, (source, {chunk}) => {
-                if (!isPageChunk(chunk)) return source
-                hmrHooks.pageChunkModules.call(source.source().toString())
-                return new RawSource('{}')
+                const isBootstrapChunk = isPageChunk(chunk)
+                hmrHooks.chunkModulesRendered.call(source.source().toString(), isBootstrapChunk)
+                return isBootstrapChunk ? new RawSource('{}') : source
             })
 
             javascriptHooks.renderStartup.tap(PLUGIN_NAME, (source, _module, {chunk}) => {