File size: 4,106 Bytes
bf237c2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
92
93
94
95
96
97
/**
 * Makes a runtime module to intercept module execution for React Refresh.
 * This module creates an isolated `__webpack_require__` function for each module,
 * and injects a `$Refresh$` object into it for use by React Refresh.
 * @param {import('webpack')} webpack The Webpack exports.
 * @returns {new () => import('webpack').RuntimeModule} 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() {
      if (!this.compilation) throw new Error('Webpack compilation missing!');

      const { runtimeTemplate } = this.compilation;
      const declareVar = runtimeTemplate.supportsConst() ? 'const' : 'var';
      return webpack.Template.asString([
        `${declareVar} setup = ${runtimeTemplate.basicFunction('moduleId', [
          `${declareVar} refresh = {`,
          webpack.Template.indent([
            `moduleId: moduleId,`,
            `register: ${runtimeTemplate.basicFunction('type, id', [
              `${declareVar} typeId = moduleId + " " + id;`,
              `refresh.runtime.register(type, typeId);`,
            ])},`,
            `signature: ${runtimeTemplate.returningFunction(
              'refresh.runtime.createSignatureFunctionForTransform()'
            )},`,
            `runtime: {`,
            webpack.Template.indent([
              `createSignatureFunctionForTransform: ${runtimeTemplate.returningFunction(
                runtimeTemplate.returningFunction('type', 'type')
              )},`,
              `register: ${runtimeTemplate.emptyFunction()}`,
            ]),
            `},`,
          ]),
          `};`,
          `return refresh;`,
        ])}`,
        '',
        `${webpack.RuntimeGlobals.interceptModuleExecution}.push(${runtimeTemplate.basicFunction(
          'options',
          [
            `${declareVar} originalFactory = options.factory;`,
            // Using a function declaration -
            // ensures `this` would propagate for modules relying on it
            `options.factory = function(moduleObject, moduleExports, webpackRequire) {`,
            webpack.Template.indent([
              // Our require function delegates to the original require function
              `${declareVar} hotRequire = ${runtimeTemplate.returningFunction(
                'webpackRequire(request)',
                'request'
              )};`,
              // The propery descriptor factory below ensures all properties but `$Refresh$`
              // are proxied through to the original require function
              `${declareVar} createPropertyDescriptor = ${runtimeTemplate.basicFunction('name', [
                `return {`,
                webpack.Template.indent([
                  `configurable: true,`,
                  `enumerable: true,`,
                  `get: ${runtimeTemplate.returningFunction('webpackRequire[name]')},`,
                  `set: ${runtimeTemplate.basicFunction('value', [
                    'webpackRequire[name] = value;',
                  ])},`,
                ]),
                `};`,
              ])};`,
              `for (${declareVar} name in webpackRequire) {`,
              webpack.Template.indent([
                `if (Object.prototype.hasOwnProperty.call(webpackRequire, name) && name !== "$Refresh$") {`,
                webpack.Template.indent([
                  `Object.defineProperty(hotRequire, name, createPropertyDescriptor(name));`,
                ]),
                `}`,
              ]),
              `}`,
              `hotRequire.$Refresh$ = setup(options.id);`,
              `originalFactory.call(this, moduleObject, moduleExports, hotRequire);`,
            ]),
            '};',
          ]
        )});`,
      ]);
    }
  };
}

module.exports = makeRefreshRuntimeModule;