the-forest/client/node_modules/@pmmmwh/react-refresh-webpack-plugin/lib/utils/makeRefreshRuntimeModule.js
2024-09-17 20:35:18 -04:00

78 lines
3.2 KiB
JavaScript

const { getRefreshGlobalScope } = require('../globals');
const getRefreshGlobal = require('./getRefreshGlobal');
/**
* Makes a runtime module to intercept module execution for React Refresh.
* @param {import('webpack')} webpack The Webpack exports.
* @returns {ReactRefreshRuntimeModule} The runtime module class.
*/
function makeRefreshRuntimeModule(webpack) {
return class ReactRefreshRuntimeModule extends webpack.RuntimeModule {
constructor() {
// Second argument is the `stage` for this runtime module -
// we'll use the same stage as Webpack's HMR runtime module for safety.
super('react refresh', webpack.RuntimeModule.STAGE_BASIC);
}
/**
* @returns {string} runtime code
*/
generate() {
const { runtimeTemplate } = this.compilation;
const refreshGlobal = getRefreshGlobalScope(webpack.RuntimeGlobals);
return webpack.Template.asString([
`${webpack.RuntimeGlobals.interceptModuleExecution}.push(${runtimeTemplate.basicFunction(
'options',
[
`${
runtimeTemplate.supportsConst() ? 'const' : 'var'
} originalFactory = options.factory;`,
`options.factory = function (moduleObject, moduleExports, webpackRequire) {`,
webpack.Template.indent([
`${refreshGlobal}.setup(options.id);`,
'try {',
webpack.Template.indent(
'originalFactory.call(this, moduleObject, moduleExports, webpackRequire);'
),
'} finally {',
webpack.Template.indent([
`if (typeof Promise !== 'undefined' && moduleObject.exports instanceof Promise) {`,
webpack.Template.indent([
// The exports of the module are re-assigned -
// this ensures anything coming after us would wait for `cleanup` to fire.
// This is particularly important because `cleanup` restores the refresh global,
// maintaining consistency for mutable variables like `moduleId`.
// This `.then` clause is a ponyfill of the `Promise.finally` API -
// it is only part of the spec after ES2018,
// but Webpack's top level await implementation support engines down to ES2015.
'options.module.exports = options.module.exports.then(',
webpack.Template.indent([
`${runtimeTemplate.basicFunction('result', [
`${refreshGlobal}.cleanup(options.id);`,
'return result;',
])},`,
runtimeTemplate.basicFunction('reason', [
`${refreshGlobal}.cleanup(options.id);`,
'return Promise.reject(reason);',
]),
]),
`);`,
]),
'} else {',
webpack.Template.indent(`${refreshGlobal}.cleanup(options.id)`),
'}',
]),
'}',
]),
`};`,
]
)})`,
'',
getRefreshGlobal(webpack.Template, webpack.RuntimeGlobals, runtimeTemplate),
]);
}
};
}
module.exports = makeRefreshRuntimeModule;