Hotify module.exports own properties

This is useful for ES6-style exports (.default), multiple component exports and some other use cases (e.g. OmniscientJS exporting `module.exports.jsx`).
gaearon · Dec 22, 2014 · 5e57a4b · 5e57a4b
1 parent 48e3ec8
commit 5e57a4b
Showing 1 changed file with 19 additions and 12 deletions.
diff --git a/index.js b/index.js
@@ -42,19 +42,26 @@ module.exports = function (source, map) {
   appendText = [
     '/* REACT HOT LOADER */',
     'if (module.hot) {',
-      acceptUpdates ? [
-        'module.hot.accept(function (err) {',
-          'if (err) {',
-            'console.error("Cannot not apply hot update to " + ' + JSON.stringify(filename) + ' + ": " + err.message);',
+      '(function () {',
+        acceptUpdates ? [
+          'module.hot.accept(function (err) {',
+            'if (err) {',
+              'console.error("Cannot not apply hot update to " + ' + JSON.stringify(filename) + ' + ": " + err.message);',
+            '}',
+          '});'
+        ].join(' ') : '',
+        'module.hot.dispose(function (data) {',
+        '  data.makeHot = module.makeHot;',
+        '});',
+        'if (module.exports && module.makeHot) {',
+          'module.exports = module.makeHot(module.exports, "__MODULE_EXPORTS");',
+          'for (var key in module.exports) {',
+            'if (module.exports[key] && module.exports.hasOwnProperty(key)) {',
+              'module.exports[key] = module.makeHot(module.exports[key], "__MODULE_EXPORTS_" + key);',
+            '}',
           '}',
-        '});'
-      ].join(' ') : '',
-      'module.hot.dispose(function (data) {',
-      '  data.makeHot = module.makeHot;',
-      '});',
-      'if (module.exports && module.makeHot) {',
-      '  module.exports = module.makeHot(module.exports, "__MODULE_EXPORTS")',
-      '}',
+        '}',
+      '})();',
     '}'
   ].join(' ');