react-refresh 0.4.2 → 0.4.3

package/cjs/react-refresh-runtime.development.js

@@ -462,22 +462,37 @@ function _getMountedRootCount() {
 
 function createSignatureFunctionForTransform() {
   {
-    var call = 0;
+    // We'll fill in the signature in two steps.
+    // First, we'll know the signature itself. This happens outside the component.
+    // Then, we'll know the references to custom Hooks. This happens inside the component.
+    // After that, the returned function will be a fast path no-op.
+    var status = 'needsSignature';
     var savedType;
     var hasCustomHooks;
     return function (type, key, forceReset, getCustomHooks) {
-      switch (call++) {
-        case 0:
-          savedType = type;
-          hasCustomHooks = typeof getCustomHooks === 'function';
-          setSignature(type, key, forceReset, getCustomHooks);
+      switch (status) {
+        case 'needsSignature':
+          if (type !== undefined) {
+            // If we received an argument, this is the initial registration call.
+            savedType = type;
+            hasCustomHooks = typeof getCustomHooks === 'function';
+            setSignature(type, key, forceReset, getCustomHooks); // The next call we expect is from inside a function, to fill in the custom Hooks.
+
+            status = 'needsCustomHooks';
+          }
+
           break;
 
-        case 1:
+        case 'needsCustomHooks':
           if (hasCustomHooks) {
             collectCustomHooksForSignature(savedType);
           }
 
+          status = 'resolved';
+          break;
+
+        case 'resolved':
+          // Do nothing. Fast path for all future renders.
           break;
       }
 

package/package.json

@@ -4,7 +4,7 @@
   "keywords": [
     "react"
   ],
-  "version": "0.4.2",
+  "version": "0.4.3",
   "homepage": "https://reactjs.org/",
   "bugs": "https://github.com/facebook/react/issues",
   "license": "MIT",