react-refresh 0.3.0 → 0.4.2

package/cjs/react-refresh-babel.production.min.js

@@ -11,12 +11,13 @@
 c,b),!0;case "CallExpression":var f=b.get("arguments");if(void 0===f||0===f.length)break;var g=b.get("callee");switch(g.node.type){case "MemberExpression":case "Identifier":g=g.getSource();if(!m(a+"$"+g,f[0],d))return!1;d(a,c,b);return!0;default:return!1}case "VariableDeclarator":if(f=c.init,null!==f&&(g=c.id.name,t(g))){switch(f.type){case "ArrowFunctionExpression":case "FunctionExpression":break;case "CallExpression":c=f.callee;var e=c.type;if("Import"===e)return!1;if("Identifier"===e){if(0===c.name.indexOf("require")||
 0===c.name.indexOf("import"))return!1}else if("MemberExpression"!==e)return!1;break;case "TaggedTemplateExpression":break;default:return!1}c=b.get("init");if(m(a,c,d))return!0;g=b.scope.getBinding(g);if(void 0===g)return;b=!1;g=g.referencePaths;for(e=0;e<g.length;e++){var h=g[e];if("JSXIdentifier"===h.node.type||"Identifier"===h.node.type){h=h.parent;if("JSXOpeningElement"===h.type)b=!0;else if("CallExpression"===h.type){h=h.callee;var k=void 0;switch(h.type){case "Identifier":k=h.name;break;case "MemberExpression":k=
 h.property.name}switch(k){case "createElement":case "jsx":case "jsxDEV":case "jsxs":b=!0}}if(b)return d(a,f,c),!0}}}}return!1}function w(a){a=n.get(a);return void 0===a?null:{key:a.map(function(a){return a.name+"{"+a.key+"}"}).join("\n"),customHooks:a.filter(function(a){a:switch(a.name){case "useState":case "React.useState":case "useReducer":case "React.useReducer":case "useEffect":case "React.useEffect":case "useLayoutEffect":case "React.useLayoutEffect":case "useMemo":case "React.useMemo":case "useCallback":case "React.useCallback":case "useRef":case "React.useRef":case "useContext":case "React.useContext":case "useImperativeMethods":case "React.useImperativeMethods":case "useDebugValue":case "React.useDebugValue":a=
-!0;break a;default:a=!1}return!a}).map(function(a){return e.cloneDeep(a.callee)})}}function C(a){a=a.hub.file;var b=x.get(a);if(void 0!==b)return b;b=!1;for(var d=a.ast.comments,c=0;c<d.length;c++)if(-1!==d[c].value.indexOf("@refresh reset")){b=!0;break}x.set(a,b);return b}function u(a,b,d){var c=b.key;b=b.customHooks;var f=C(d.path),g=[];b.forEach(function(a){var b=void 0;switch(a.type){case "MemberExpression":"Identifier"===a.object.type&&(b=a.object.name);break;case "Identifier":b=a.name}d.hasBinding(b)?
-g.push(a):f=!0});a=[a,e.stringLiteral(c)];(f||0<g.length)&&a.push(e.booleanLiteral(f));0<g.length&&a.push(e.functionExpression(null,[],e.blockStatement([e.returnStatement(e.arrayExpression(g))])));return a}if("function"===typeof q.getEnv){var y=q.getEnv();if("development"!==y)throw Error('React Refresh Babel transform should only be enabled in development environment. Instead, the environment is: "'+y+'".');}var e=q.types,l=new Map,x=new WeakMap,k=new WeakSet,p=new WeakSet,z=new WeakSet,n=new WeakMap,
-D={CallExpression:function(a){var b=a.node.callee,d=null;switch(b.type){case "Identifier":d=b.name;break;case "MemberExpression":d=b.property.name}if(null!==d&&/^use[A-Z]/.test(d)&&(b=a.scope.getFunctionParent(),null!==b)){b=b.block;n.has(b)||n.set(b,[]);b=n.get(b);var c="";"VariableDeclarator"===a.parent.type&&(c=a.parentPath.get("id").getSource());var f=a.get("arguments");"useState"===d&&0<f.length?c+="("+f[0].getSource()+")":"useReducer"===d&&1<f.length&&(c+="("+f[1].getSource()+")");b.push({callee:a.node.callee,
-name:d,key:c})}}};return{visitor:{ExportDefaultDeclaration:function(a){var b=a.node,d=b.declaration,c=a.get("declaration");if("CallExpression"===d.type&&!k.has(b)){k.add(b);var f=a.parentPath;m("%default%",c,function(a,b,c){null!==c&&(a=r(f,a),c.replaceWith(e.assignmentExpression("=",a,b)))})}},FunctionDeclaration:{enter:function(a){var b=a.node,d=void 0,c=void 0;switch(a.parent.type){case "Program":c=a;d=a.parentPath;break;case "ExportNamedDeclaration":c=a.parentPath;d=c.parentPath;break;case "ExportDefaultDeclaration":c=
-a.parentPath;d=c.parentPath;break;default:return}var f=b.id;null!==f&&(f=f.name,t(f)&&!k.has(b)&&(k.add(b),m(f,a,function(a,b){a=r(d,a);c.insertAfter(e.expressionStatement(e.assignmentExpression("=",a,b)))})))},exit:function(a){var b=a.node,d=b.id;if(null!==d){var c=w(b);if(null!==c&&!p.has(b)){p.add(b);b=a.scope.generateUidIdentifier("_s");a.scope.parent.push({id:b,init:e.callExpression(e.identifier("$RefreshSig$"),[])});a.get("body").unshiftContainer("body",e.expressionStatement(e.callExpression(b,
-[])));var f=null;a.find(function(a){if(a.parentPath.isBlock())return f=a,!0});null!==f&&f.insertAfter(e.expressionStatement(e.callExpression(b,u(d,c,f.scope))))}}}},"ArrowFunctionExpression|FunctionExpression":{exit:function(a){var b=a.node,d=w(b);if(null!==d&&!p.has(b)){p.add(b);var c=a.scope.generateUidIdentifier("_s");a.scope.parent.push({id:c,init:e.callExpression(e.identifier("$RefreshSig$"),[])});a.get("body").unshiftContainer("body",e.expressionStatement(e.callExpression(c,[])));if("VariableDeclarator"===
-a.parent.type){var f=null;a.find(function(a){if(a.parentPath.isBlock())return f=a,!0});null!==f&&f.insertAfter(e.expressionStatement(e.callExpression(c,u(a.parent.id,d,f.scope))))}else a.replaceWith(e.callExpression(c,u(b,d,a.scope)))}}},VariableDeclaration:function(a){var b=a.node,d=void 0,c=void 0;switch(a.parent.type){case "Program":c=a;d=a.parentPath;break;case "ExportNamedDeclaration":c=a.parentPath;d=c.parentPath;break;case "ExportDefaultDeclaration":c=a.parentPath;d=c.parentPath;break;default:return}if(!k.has(b)&&
-(k.add(b),a=a.get("declarations"),1===a.length)){var f=a[0];m(f.node.id.name,f,function(a,b,h){null!==h&&(a=r(d,a),"ArrowFunctionExpression"!==b.type&&"FunctionExpression"!==b.type||"VariableDeclarator"!==h.parent.type?h.replaceWith(e.assignmentExpression("=",a,b)):c.insertAfter(e.expressionStatement(e.assignmentExpression("=",a,f.node.id))))})}},Program:{enter:function(a){a.traverse(D)},exit:function(a){var b=l.get(a);if(void 0!==b){var d=a.node;if(!z.has(d)){z.add(d);l.delete(a);var c=[];a.pushContainer("body",
-e.variableDeclaration("var",c));b.forEach(function(b){var d=b.handle;b=b.persistentID;a.pushContainer("body",e.expressionStatement(e.callExpression(e.identifier("$RefreshReg$"),[d,e.stringLiteral(b)])));c.push(e.variableDeclarator(d))})}}}}}}}var A={default:v},B=A&&v||A;module.exports=B.default||B;
+!0;break a;default:a=!1}return!a}).map(function(a){return e.cloneDeep(a.callee)})}}function D(a){a=a.hub.file;var b=x.get(a);if(void 0!==b)return b;b=!1;for(var d=a.ast.comments,c=0;c<d.length;c++)if(-1!==d[c].value.indexOf("@refresh reset")){b=!0;break}x.set(a,b);return b}function u(a,b,d){var c=b.key;b=b.customHooks;var f=D(d.path),g=[];b.forEach(function(a){switch(a.type){case "MemberExpression":if("Identifier"===a.object.type)var b=a.object.name;break;case "Identifier":b=a.name}d.hasBinding(b)?
+g.push(a):f=!0});b=c;"function"!==typeof require||y.emitFullSignatures||(b=require("crypto").createHash("sha1").update(c).digest("base64"));a=[a,e.stringLiteral(b)];(f||0<g.length)&&a.push(e.booleanLiteral(f));0<g.length&&a.push(e.functionExpression(null,[],e.blockStatement([e.returnStatement(e.arrayExpression(g))])));return a}var y=1<arguments.length&&void 0!==arguments[1]?arguments[1]:{};if("function"===typeof q.getEnv){var z=q.getEnv();if("development"!==z&&!y.skipEnvCheck)throw Error('React Refresh Babel transform should only be enabled in development environment. Instead, the environment is: "'+
+z+'". If you want to override this check, pass {skipEnvCheck: true} as plugin options.');}var e=q.types,l=new Map,x=new WeakMap,k=new WeakSet,p=new WeakSet,A=new WeakSet,n=new WeakMap,E={CallExpression:function(a){var b=a.node.callee,d=null;switch(b.type){case "Identifier":d=b.name;break;case "MemberExpression":d=b.property.name}if(null!==d&&/^use[A-Z]/.test(d)&&(b=a.scope.getFunctionParent(),null!==b)){b=b.block;n.has(b)||n.set(b,[]);b=n.get(b);var c="";"VariableDeclarator"===a.parent.type&&(c=a.parentPath.get("id").getSource());
+var f=a.get("arguments");"useState"===d&&0<f.length?c+="("+f[0].getSource()+")":"useReducer"===d&&1<f.length&&(c+="("+f[1].getSource()+")");b.push({callee:a.node.callee,name:d,key:c})}}};return{visitor:{ExportDefaultDeclaration:function(a){var b=a.node,d=b.declaration,c=a.get("declaration");if("CallExpression"===d.type&&!k.has(b)){k.add(b);var f=a.parentPath;m("%default%",c,function(a,b,c){null!==c&&(a=r(f,a),c.replaceWith(e.assignmentExpression("=",a,b)))})}},FunctionDeclaration:{enter:function(a){var b=
+a.node;switch(a.parent.type){case "Program":var d=a;var c=a.parentPath;break;case "ExportNamedDeclaration":d=a.parentPath;c=d.parentPath;break;case "ExportDefaultDeclaration":d=a.parentPath;c=d.parentPath;break;default:return}var f=b.id;null!==f&&(f=f.name,t(f)&&!k.has(b)&&(k.add(b),m(f,a,function(a,b){a=r(c,a);d.insertAfter(e.expressionStatement(e.assignmentExpression("=",a,b)))})))},exit:function(a){var b=a.node,d=b.id;if(null!==d){var c=w(b);if(null!==c&&!p.has(b)){p.add(b);b=a.scope.generateUidIdentifier("_s");
+a.scope.parent.push({id:b,init:e.callExpression(e.identifier("$RefreshSig$"),[])});a.get("body").unshiftContainer("body",e.expressionStatement(e.callExpression(b,[])));var f=null;a.find(function(a){if(a.parentPath.isBlock())return f=a,!0});null!==f&&f.insertAfter(e.expressionStatement(e.callExpression(b,u(d,c,f.scope))))}}}},"ArrowFunctionExpression|FunctionExpression":{exit:function(a){var b=a.node,d=w(b);if(null!==d&&!p.has(b)){p.add(b);var c=a.scope.generateUidIdentifier("_s");a.scope.parent.push({id:c,
+init:e.callExpression(e.identifier("$RefreshSig$"),[])});"BlockStatement"!==a.node.body.type&&(a.node.body=e.blockStatement([e.returnStatement(a.node.body)]));a.get("body").unshiftContainer("body",e.expressionStatement(e.callExpression(c,[])));if("VariableDeclarator"===a.parent.type){var f=null;a.find(function(a){if(a.parentPath.isBlock())return f=a,!0});null!==f&&f.insertAfter(e.expressionStatement(e.callExpression(c,u(a.parent.id,d,f.scope))))}else a.replaceWith(e.callExpression(c,u(b,d,a.scope)))}}},
+VariableDeclaration:function(a){var b=a.node;switch(a.parent.type){case "Program":var d=a;var c=a.parentPath;break;case "ExportNamedDeclaration":d=a.parentPath;c=d.parentPath;break;case "ExportDefaultDeclaration":d=a.parentPath;c=d.parentPath;break;default:return}if(!k.has(b)&&(k.add(b),a=a.get("declarations"),1===a.length)){var f=a[0];m(f.node.id.name,f,function(a,b,h){null!==h&&(a=r(c,a),"ArrowFunctionExpression"!==b.type&&"FunctionExpression"!==b.type||"VariableDeclarator"!==h.parent.type?h.replaceWith(e.assignmentExpression("=",
+a,b)):d.insertAfter(e.expressionStatement(e.assignmentExpression("=",a,f.node.id))))})}},Program:{enter:function(a){a.traverse(E)},exit:function(a){var b=l.get(a);if(void 0!==b){var d=a.node;if(!A.has(d)){A.add(d);l.delete(a);var c=[];a.pushContainer("body",e.variableDeclaration("var",c));b.forEach(function(b){var d=b.handle;b=b.persistentID;a.pushContainer("body",e.expressionStatement(e.callExpression(e.identifier("$RefreshReg$"),[d,e.stringLiteral(b)])));c.push(e.variableDeclarator(d))})}}}}}}}
+var B={default:v},C=B&&v||B;module.exports=C.default||C;

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

@@ -24,93 +24,35 @@ var hasSymbol = typeof Symbol === 'function' && Symbol.for;
 
 
 
-
-
-// TODO: We don't use AsyncMode or ConcurrentMode anymore. They were temporary
+ // TODO: We don't use AsyncMode or ConcurrentMode anymore. They were temporary
 // (unstable) APIs that have been removed. Can we remove the symbols?
 
 
+
 var REACT_FORWARD_REF_TYPE = hasSymbol ? Symbol.for('react.forward_ref') : 0xead0;
 
 
 var REACT_MEMO_TYPE = hasSymbol ? Symbol.for('react.memo') : 0xead3;
 
-/**
- * Similar to invariant but only logs a warning if the condition is not met.
- * This can be used to log issues in development environments in critical
- * paths. Removing the logging code for production environments will keep the
- * same logic and follow the same code paths.
- */
-
-var warningWithoutStack = function () {};
-
-{
-  warningWithoutStack = function (condition, format) {
-    for (var _len = arguments.length, args = Array(_len > 2 ? _len - 2 : 0), _key = 2; _key < _len; _key++) {
-      args[_key - 2] = arguments[_key];
-    }
-
-    if (format === undefined) {
-      throw new Error('`warningWithoutStack(condition, format, ...args)` requires a warning ' + 'message argument');
-    }
-    if (args.length > 8) {
-      // Check before the condition to catch violations early.
-      throw new Error('warningWithoutStack() currently supports at most 8 arguments.');
-    }
-    if (condition) {
-      return;
-    }
-    if (typeof console !== 'undefined') {
-      var argsWithFormat = args.map(function (item) {
-        return '' + item;
-      });
-      argsWithFormat.unshift('Warning: ' + format);
-
-      // We intentionally don't use spread (or .apply) directly because it
-      // breaks IE9: https://github.com/facebook/react/issues/13610
-      Function.prototype.apply.call(console.error, console, argsWithFormat);
-    }
-    try {
-      // --- Welcome to debugging React ---
-      // This error was thrown as a convenience so that you can use this stack
-      // to find the callsite that caused this warning to fire.
-      var argIndex = 0;
-      var message = 'Warning: ' + format.replace(/%s/g, function () {
-        return args[argIndex++];
-      });
-      throw new Error(message);
-    } catch (x) {}
-  };
-}
-
-var warningWithoutStack$1 = warningWithoutStack;
-
-// In old environments, we'll leak previous types after every edit.
-var PossiblyWeakMap = typeof WeakMap === 'function' ? WeakMap : Map;
-
-// We never remove these associations.
+var PossiblyWeakMap = typeof WeakMap === 'function' ? WeakMap : Map; // We never remove these associations.
 // It's OK to reference families, but use WeakMap/Set for types.
+
 var allFamiliesByID = new Map();
 var allFamiliesByType = new PossiblyWeakMap();
-var allSignaturesByType = new PossiblyWeakMap();
-// This WeakMap is read by React, so we only put families
+var allSignaturesByType = new PossiblyWeakMap(); // This WeakMap is read by React, so we only put families
 // that have actually been edited here. This keeps checks fast.
 // $FlowIssue
-var updatedFamiliesByType = new PossiblyWeakMap();
 
-// This is cleared on every performReactRefresh() call.
+var updatedFamiliesByType = new PossiblyWeakMap(); // This is cleared on every performReactRefresh() call.
 // It is an array of [Family, NextType] tuples.
-var pendingUpdates = [];
 
-// This is injected by the renderer via DevTools global hook.
-var setRefreshHandler = null;
-var scheduleRefresh = null;
-var scheduleRoot = null;
-var findHostInstancesForRefresh = null;
+var pendingUpdates = []; // This is injected by the renderer via DevTools global hook.
+
+var helpersByRendererID = new Map();
+var helpersByRoot = new Map(); // We keep track of mounted roots so we can schedule updates.
+
+var mountedRoots = new Set(); // If a root captures an error, we add its element to this Map so we can retry on edit.
 
-// We keep track of mounted roots so we can schedule updates.
-var mountedRoots = new Set();
-// If a root captures an error, we add its element to this Map so we can retry on edit.
 var failedRoots = new Map();
 var didSomeRootFailOnMount = false;
 
@@ -120,7 +62,8 @@ function computeFullKey(signature) {
   }
 
   var fullKey = signature.ownKey;
-  var hooks = void 0;
+  var hooks;
+
   try {
     hooks = signature.getCustomHooks();
   } catch (err) {
@@ -134,22 +77,28 @@ function computeFullKey(signature) {
 
   for (var i = 0; i < hooks.length; i++) {
     var hook = hooks[i];
+
     if (typeof hook !== 'function') {
       // Something's wrong. Assume we need to remount.
       signature.forceReset = true;
       signature.fullKey = fullKey;
       return fullKey;
     }
+
     var nestedHookSignature = allSignaturesByType.get(hook);
+
     if (nestedHookSignature === undefined) {
       // No signature means Hook wasn't in the source code, e.g. in a library.
       // We'll skip it because we can assume it won't change during this session.
       continue;
     }
+
     var nestedHookKey = computeFullKey(nestedHookSignature);
+
     if (nestedHookSignature.forceReset) {
       signature.forceReset = true;
     }
+
     fullKey += '\n---\n' + nestedHookKey;
   }
 
@@ -164,12 +113,15 @@ function haveEqualSignatures(prevType, nextType) {
   if (prevSignature === undefined && nextSignature === undefined) {
     return true;
   }
+
   if (prevSignature === undefined || nextSignature === undefined) {
     return false;
   }
+
   if (computeFullKey(prevSignature) !== computeFullKey(nextSignature)) {
     return false;
   }
+
   if (nextSignature.forceReset) {
     return false;
   }
@@ -185,9 +137,11 @@ function canPreserveStateBetween(prevType, nextType) {
   if (isReactClass(prevType) || isReactClass(nextType)) {
     return false;
   }
+
   if (haveEqualSignatures(prevType, nextType)) {
     return true;
   }
+
   return false;
 }
 
@@ -204,118 +158,119 @@ function performReactRefresh() {
 
     var staleFamilies = new Set();
     var updatedFamilies = new Set();
-
     var updates = pendingUpdates;
     pendingUpdates = [];
     updates.forEach(function (_ref) {
       var family = _ref[0],
           nextType = _ref[1];
-
       // Now that we got a real edit, we can create associations
       // that will be read by the React reconciler.
       var prevType = family.current;
       updatedFamiliesByType.set(prevType, family);
       updatedFamiliesByType.set(nextType, family);
-      family.current = nextType;
+      family.current = nextType; // Determine whether this should be a re-render or a re-mount.
 
-      // Determine whether this should be a re-render or a re-mount.
       if (canPreserveStateBetween(prevType, nextType)) {
         updatedFamilies.add(family);
       } else {
         staleFamilies.add(family);
       }
-    });
+    }); // TODO: rename these fields to something more meaningful.
 
-    // TODO: rename these fields to something more meaningful.
     var update = {
-      updatedFamilies: updatedFamilies, // Families that will re-render preserving state
+      updatedFamilies: updatedFamilies,
+      // Families that will re-render preserving state
       staleFamilies: staleFamilies // Families that will be remounted
-    };
-
-    if (typeof setRefreshHandler !== 'function') {
-      warningWithoutStack$1(false, 'Could not find the setRefreshHandler() implementation. ' + 'This likely means that injectIntoGlobalHook() was either ' + 'called before the global DevTools hook was set up, or after the ' + 'renderer has already initialized. Please file an issue with a reproducing case.');
-      return null;
-    }
-
-    if (typeof scheduleRefresh !== 'function') {
-      warningWithoutStack$1(false, 'Could not find the scheduleRefresh() implementation. ' + 'This likely means that injectIntoGlobalHook() was either ' + 'called before the global DevTools hook was set up, or after the ' + 'renderer has already initialized. Please file an issue with a reproducing case.');
-      return null;
-    }
-    if (typeof scheduleRoot !== 'function') {
-      warningWithoutStack$1(false, 'Could not find the scheduleRoot() implementation. ' + 'This likely means that injectIntoGlobalHook() was either ' + 'called before the global DevTools hook was set up, or after the ' + 'renderer has already initialized. Please file an issue with a reproducing case.');
-      return null;
-    }
-    var scheduleRefreshForRoot = scheduleRefresh;
-    var scheduleRenderForRoot = scheduleRoot;
-
-    // Even if there are no roots, set the handler on first update.
-    // This ensures that if *new* roots are mounted, they'll use the resolve handler.
-    setRefreshHandler(resolveFamily);
 
+    };
+    helpersByRendererID.forEach(function (helpers) {
+      // Even if there are no roots, set the handler on first update.
+      // This ensures that if *new* roots are mounted, they'll use the resolve handler.
+      helpers.setRefreshHandler(resolveFamily);
+    });
     var didError = false;
     var firstError = null;
     failedRoots.forEach(function (element, root) {
+      var helpers = helpersByRoot.get(root);
+
+      if (helpers === undefined) {
+        throw new Error('Could not find helpers for a root. This is a bug in React Refresh.');
+      }
+
       try {
-        scheduleRenderForRoot(root, element);
+        helpers.scheduleRoot(root, element);
       } catch (err) {
         if (!didError) {
           didError = true;
           firstError = err;
-        }
-        // Keep trying other roots.
+        } // Keep trying other roots.
+
       }
     });
     mountedRoots.forEach(function (root) {
+      var helpers = helpersByRoot.get(root);
+
+      if (helpers === undefined) {
+        throw new Error('Could not find helpers for a root. This is a bug in React Refresh.');
+      }
+
       try {
-        scheduleRefreshForRoot(root, update);
+        helpers.scheduleRefresh(root, update);
       } catch (err) {
         if (!didError) {
           didError = true;
           firstError = err;
-        }
-        // Keep trying other roots.
+        } // Keep trying other roots.
+
       }
     });
+
     if (didError) {
       throw firstError;
     }
+
     return update;
   }
 }
-
 function register(type, id) {
   {
     if (type === null) {
       return;
     }
+
     if (typeof type !== 'function' && typeof type !== 'object') {
       return;
-    }
-
-    // This can happen in an edge case, e.g. if we register
+    } // This can happen in an edge case, e.g. if we register
     // return value of a HOC but it returns a cached component.
     // Ignore anything but the first registration for each type.
+
+
     if (allFamiliesByType.has(type)) {
       return;
-    }
-    // Create family or remember to update it.
+    } // Create family or remember to update it.
     // None of this bookkeeping affects reconciliation
     // until the first performReactRefresh() call above.
+
+
     var family = allFamiliesByID.get(id);
+
     if (family === undefined) {
-      family = { current: type };
+      family = {
+        current: type
+      };
       allFamiliesByID.set(id, family);
     } else {
       pendingUpdates.push([family, type]);
     }
-    allFamiliesByType.set(type, family);
 
-    // Visit inner types because we might not have registered them.
+    allFamiliesByType.set(type, family); // Visit inner types because we might not have registered them.
+
     if (typeof type === 'object' && type !== null) {
       switch (type.$$typeof) {
         case REACT_FORWARD_REF_TYPE:
           register(type.render, id + '$render');
           break;
+
         case REACT_MEMO_TYPE:
           register(type.type, id + '$type');
           break;
@@ -323,10 +278,9 @@ function register(type, id) {
     }
   }
 }
-
 function setSignature(type, key) {
   var forceReset = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
-  var getCustomHooks = arguments[3];
+  var getCustomHooks = arguments.length > 3 ? arguments[3] : undefined;
 
   {
     allSignaturesByType.set(type, {
@@ -338,41 +292,39 @@ function setSignature(type, key) {
       }
     });
   }
-}
-
-// This is lazily called during first render for a type.
+} // This is lazily called during first render for a type.
 // It captures Hook list at that time so inline requires don't break comparisons.
+
 function collectCustomHooksForSignature(type) {
   {
     var signature = allSignaturesByType.get(type);
+
     if (signature !== undefined) {
       computeFullKey(signature);
     }
   }
 }
-
 function getFamilyByID(id) {
   {
     return allFamiliesByID.get(id);
   }
 }
-
 function getFamilyByType(type) {
   {
     return allFamiliesByType.get(type);
   }
 }
-
 function findAffectedHostInstances(families) {
   {
-    if (typeof findHostInstancesForRefresh !== 'function') {
-      warningWithoutStack$1(false, 'Could not find the findHostInstancesForRefresh() implementation. ' + 'This likely means that injectIntoGlobalHook() was either ' + 'called before the global DevTools hook was set up, or after the ' + 'renderer has already initialized. Please file an issue with a reproducing case.');
-      return new Set();
-    }
-    var findInstances = findHostInstancesForRefresh;
     var affectedInstances = new Set();
     mountedRoots.forEach(function (root) {
-      var instancesForRoot = findInstances(root, families);
+      var helpers = helpersByRoot.get(root);
+
+      if (helpers === undefined) {
+        throw new Error('Could not find helpers for a root. This is a bug in React Refresh.');
+      }
+
+      var instancesForRoot = helpers.findHostInstancesForRefresh(root, families);
       instancesForRoot.forEach(function (inst) {
         affectedInstances.add(inst);
       });
@@ -380,44 +332,56 @@ function findAffectedHostInstances(families) {
     return affectedInstances;
   }
 }
-
 function injectIntoGlobalHook(globalObject) {
   {
     // For React Native, the global hook will be set up by require('react-devtools-core').
     // That code will run before us. So we need to monkeypatch functions on existing hook.
-
     // For React Web, the global hook will be set up by the extension.
     // This will also run before us.
     var hook = globalObject.__REACT_DEVTOOLS_GLOBAL_HOOK__;
+
     if (hook === undefined) {
       // However, if there is no DevTools extension, we'll need to set up the global hook ourselves.
       // Note that in this case it's important that renderer code runs *after* this method call.
       // Otherwise, the renderer will think that there is no global hook, and won't do the injection.
+      var nextID = 0;
       globalObject.__REACT_DEVTOOLS_GLOBAL_HOOK__ = hook = {
         supportsFiber: true,
-        inject: function () {},
+        inject: function (injected) {
+          return nextID++;
+        },
         onCommitFiberRoot: function (id, root, maybePriorityLevel, didError) {},
         onCommitFiberUnmount: function () {}
       };
-    }
+    } // Here, we just want to get a reference to scheduleRefresh.
+
 
-    // Here, we just want to get a reference to scheduleRefresh.
     var oldInject = hook.inject;
+
     hook.inject = function (injected) {
-      findHostInstancesForRefresh = injected.findHostInstancesForRefresh;
-      scheduleRefresh = injected.scheduleRefresh;
-      scheduleRoot = injected.scheduleRoot;
-      setRefreshHandler = injected.setRefreshHandler;
-      return oldInject.apply(this, arguments);
-    };
+      var id = oldInject.apply(this, arguments);
+
+      if (typeof injected.scheduleRefresh === 'function' && typeof injected.setRefreshHandler === 'function') {
+        // This version supports React Refresh.
+        helpersByRendererID.set(id, injected);
+      }
+
+      return id;
+    }; // We also want to track currently mounted roots.
+
 
-    // We also want to track currently mounted roots.
     var oldOnCommitFiberRoot = hook.onCommitFiberRoot;
+
     hook.onCommitFiberRoot = function (id, root, maybePriorityLevel, didError) {
-      var current = root.current;
-      var alternate = current.alternate;
+      var helpers = helpersByRendererID.get(id);
+
+      if (helpers === undefined) {
+        return;
+      }
 
-      // We need to determine whether this root has just (un)mounted.
+      helpersByRoot.set(root, helpers);
+      var current = root.current;
+      var alternate = current.alternate; // We need to determine whether this root has just (un)mounted.
       // This logic is copy-pasted from similar logic in the DevTools backend.
       // If this breaks with some refactoring, you'll want to update DevTools too.
 
@@ -429,16 +393,18 @@ function injectIntoGlobalHook(globalObject) {
           // Mount a new root.
           mountedRoots.add(root);
           failedRoots.delete(root);
-        } else if (wasMounted && isMounted) {
-          // Update an existing root.
+        } else if (wasMounted && isMounted) {// Update an existing root.
           // This doesn't affect our mounted root Set.
         } else if (wasMounted && !isMounted) {
           // Unmount an existing root.
           mountedRoots.delete(root);
+
           if (didError) {
             // We'll remount it on future edits.
             // Remember what was rendered so we can restore it.
             failedRoots.set(root, alternate.memoizedState.element);
+          } else {
+            helpersByRoot.delete(root);
           }
         } else if (!wasMounted && !isMounted) {
           if (didError && !failedRoots.has(root)) {
@@ -463,19 +429,15 @@ function injectIntoGlobalHook(globalObject) {
     };
   }
 }
-
 function hasUnrecoverableErrors() {
   return didSomeRootFailOnMount;
-}
+} // Exposed for testing.
 
-// Exposed for testing.
 function _getMountedRootCount() {
   {
     return mountedRoots.size;
   }
-}
-
-// This is a wrapper over more primitive functions for setting signature.
+} // This is a wrapper over more primitive functions for setting signature.
 // Signatures let us decide whether the Hook order has changed on refresh.
 //
 // This function is intended to be used as a transform target, e.g.:
@@ -497,11 +459,12 @@ function _getMountedRootCount() {
 //   'useState{[foo, setFoo]}(0)',
 //   () => [useCustomHook], /* Lazy to avoid triggering inline requires */
 // );
+
 function createSignatureFunctionForTransform() {
   {
     var call = 0;
-    var savedType = void 0;
-    var hasCustomHooks = void 0;
+    var savedType;
+    var hasCustomHooks;
     return function (type, key, forceReset, getCustomHooks) {
       switch (call++) {
         case 0:
@@ -509,17 +472,19 @@ function createSignatureFunctionForTransform() {
           hasCustomHooks = typeof getCustomHooks === 'function';
           setSignature(type, key, forceReset, getCustomHooks);
           break;
+
         case 1:
           if (hasCustomHooks) {
             collectCustomHooksForSignature(savedType);
           }
+
           break;
       }
+
       return type;
     };
   }
 }
-
 function isLikelyComponentType(type) {
   {
     switch (typeof type) {
@@ -531,23 +496,28 @@ function isLikelyComponentType(type) {
               // React class.
               return true;
             }
+
             var ownNames = Object.getOwnPropertyNames(type.prototype);
+
             if (ownNames.length > 1 || ownNames[0] !== 'constructor') {
               // This looks like a class.
               return false;
-            }
-            // eslint-disable-next-line no-proto
+            } // eslint-disable-next-line no-proto
+
+
             if (type.prototype.__proto__ !== Object.prototype) {
               // It has a superclass.
               return false;
-            }
-            // Pass through.
+            } // Pass through.
             // This looks like a regular function with empty prototype.
-          }
-          // For plain functions and arrows, use name as a heuristic.
+
+          } // For plain functions and arrows, use name as a heuristic.
+
+
           var name = type.name || type.displayName;
           return typeof name === 'string' && /^[A-Z]/.test(name);
         }
+
       case 'object':
         {
           if (type != null) {
@@ -556,12 +526,15 @@ function isLikelyComponentType(type) {
               case REACT_MEMO_TYPE:
                 // Definitely React components.
                 return true;
+
               default:
                 return false;
             }
           }
+
           return false;
         }
+
       default:
         {
           return false;
@@ -586,6 +559,8 @@ var ReactFreshRuntime = Object.freeze({
 });
 
 // This is hacky but makes it work with both Rollup and Jest.
+
+
 var runtime = ReactFreshRuntime.default || ReactFreshRuntime;
 
 module.exports = runtime;

package/package.json

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