fix: #1478 support react class-component hot-update (#1489)

* fix: #1478 support react class-component hot-update * refactor: 🔥 revert example code --------- Co-authored-by: pshu <[email protected]>
umijs · Aug 29, 2024 · 8775b18 · 8775b18
1 parent ec7a39b
commit 8775b18
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/crates/mako/src/visitors/react.rs b/crates/mako/src/visitors/react.rs
@@ -160,8 +160,21 @@ fn react_refresh_module_postfix(context: &Arc<Context>) -> Box<dyn VisitMut> {
         code: r#"
 if (prevRefreshReg) self.$RefreshReg$ = prevRefreshReg;
 if (prevRefreshSig) self.$RefreshSig$ = prevRefreshSig;
+function registerClassComponent(filename, moduleExports) {
+  for (const key in moduleExports) {
+    try {
+        if (key === "__esModule") continue;
+        const exportValue = moduleExports[key];
+        if (RefreshRuntime.isLikelyComponentType(exportValue) && exportValue.prototype && exportValue.prototype.isReactComponent) {
+            RefreshRuntime.register(exportValue, filename + " " + key);
+        }
+    } catch (e) {
+        // in case the moduleExports[key] is not accessible due depedence loop
+    }
+  }
+}
 function $RefreshIsReactComponentLike$(moduleExports) {
-  if (RefreshRuntime.isLikelyComponentType(moduleExports.default || moduleExports)) {
+  if (RefreshRuntime.isLikelyComponentType(moduleExports || moduleExports.default)) {
     return true;
   }
   for (var key in moduleExports) {
@@ -175,6 +188,7 @@ function $RefreshIsReactComponentLike$(moduleExports) {
   }
   return false;
 }
+registerClassComponent(module.id, module.exports);
 if ($RefreshIsReactComponentLike$(module.exports)) {
     module.meta.hot.accept();
     RefreshRuntime.performReactRefresh();