78 lines
3.2 KiB
JavaScript
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;
|