Unify ReactFiberCurrentOwner and ReactCurrentFiber

Author: sebmarkbage

packages/react-dom/src/client/ReactDOMRootFB.js

@@ -61,7 +61,7 @@ import {LegacyRoot} from 'react-reconciler/src/ReactRootTags';
 import getComponentNameFromType from 'shared/getComponentNameFromType';
 import {has as hasInstance} from 'shared/ReactInstanceMap';
 
-import {currentOwner} from 'react-reconciler/src/ReactFiberCurrentOwner';
+import {current as currentOwner} from 'react-reconciler/src/ReactCurrentFiber';
 
 import assign from 'shared/assign';
 

packages/react-native-renderer/src/ReactNativePublicCompat.js

@@ -25,7 +25,7 @@ import {
 } from 'react-reconciler/src/ReactFiberReconciler';
 import {doesFiberContain} from 'react-reconciler/src/ReactFiberTreeReflection';
 import getComponentNameFromType from 'shared/getComponentNameFromType';
-import {currentOwner} from 'react-reconciler/src/ReactFiberCurrentOwner';
+import {current as currentOwner} from 'react-reconciler/src/ReactCurrentFiber';
 
 export function findHostInstance_DEPRECATED<TElementType: ElementType>(
   componentOrHandle: ?(ElementRef<TElementType> | number),

packages/react-reconciler/src/ReactCurrentFiber.js

@@ -41,21 +41,33 @@ function getCurrentFiberStackInDev(): string {
   return '';
 }
 
+export function resetCurrentDebugFiberInDEV() {
+  if (__DEV__) {
+    resetCurrentFiber();
+  }
+}
+
+export function setCurrentDebugFiberInDEV(fiber: Fiber | null) {
+  if (__DEV__) {
+    setCurrentFiber(fiber);
+  }
+}
+
 export function resetCurrentFiber() {
   if (__DEV__) {
     ReactSharedInternals.getCurrentStack = null;
-    current = null;
     isRendering = false;
   }
+  current = null;
 }
 
 export function setCurrentFiber(fiber: Fiber | null) {
   if (__DEV__) {
     ReactSharedInternals.getCurrentStack =
       fiber === null ? null : getCurrentFiberStackInDev;
-    current = fiber;
     isRendering = false;
   }
+  current = fiber;
 }
 
 export function getCurrentFiber(): Fiber | null {

packages/react-reconciler/src/ReactFiberAsyncDispatcher.js

@@ -16,7 +16,7 @@ import {CacheContext} from './ReactFiberCacheComponent';
 
 import {disableStringRefs} from 'shared/ReactFeatureFlags';
 
-import {currentOwner} from './ReactFiberCurrentOwner';
+import {current as currentOwner} from 'react-reconciler/src/ReactCurrentFiber';
 
 function getCacheForType<T>(resourceType: () => T): T {
   if (!enableCache) {

packages/react-reconciler/src/ReactFiberBeginWork.js

@@ -125,6 +125,7 @@ import {
 import {
   getCurrentFiberOwnerNameInDevOrNull,
   setIsRendering,
+  setCurrentFiber,
 } from './ReactCurrentFiber';
 import {
   resolveFunctionForHotReloading,
@@ -296,7 +297,6 @@ import {
   pushRootMarkerInstance,
   TransitionTracingMarker,
 } from './ReactFiberTracingMarkerComponent';
-import {setCurrentOwner} from './ReactFiberCurrentOwner';
 
 // A special exception that's used to unwind the stack when an update flows
 // into a dehydrated boundary.
@@ -432,7 +432,6 @@ function updateForwardRef(
     markComponentRenderStarted(workInProgress);
   }
   if (__DEV__) {
-    setCurrentOwner(workInProgress);
     setIsRendering(true);
     nextChildren = renderWithHooks(
       current,
@@ -1150,7 +1149,6 @@ function updateFunctionComponent(
     markComponentRenderStarted(workInProgress);
   }
   if (__DEV__) {
-    setCurrentOwner(workInProgress);
     setIsRendering(true);
     nextChildren = renderWithHooks(
       current,
@@ -1373,7 +1371,7 @@ function finishClassComponent(
 
   // Rerender
   if (__DEV__ || !disableStringRefs) {
-    setCurrentOwner(workInProgress);
+    setCurrentFiber(workInProgress);
   }
   let nextChildren;
   if (
@@ -3419,7 +3417,6 @@ function updateContextConsumer(
   }
   let newChildren;
   if (__DEV__) {
-    setCurrentOwner(workInProgress);
     setIsRendering(true);
     newChildren = render(newValue);
     setIsRendering(false);

packages/react-reconciler/src/ReactFiberCommitWork.js

@@ -101,8 +101,8 @@ import {
 } from './ReactFiberFlags';
 import getComponentNameFromFiber from 'react-reconciler/src/getComponentNameFromFiber';
 import {
-  resetCurrentFiber as resetCurrentDebugFiberInDEV,
-  setCurrentFiber as setCurrentDebugFiberInDEV,
+  resetCurrentDebugFiberInDEV,
+  setCurrentDebugFiberInDEV,
   getCurrentFiber as getCurrentDebugFiberInDEV,
 } from './ReactCurrentFiber';
 import {resolveClassComponentProps} from './ReactFiberClassComponent';

packages/react-reconciler/src/ReactFiberCurrentOwner.js

@@ -78,8 +78,8 @@ import {
 import {
   isRendering as ReactCurrentFiberIsRendering,
   current as ReactCurrentFiberCurrent,
-  resetCurrentFiber as resetCurrentDebugFiberInDEV,
-  setCurrentFiber as setCurrentDebugFiberInDEV,
+  resetCurrentDebugFiberInDEV,
+  setCurrentDebugFiberInDEV,
 } from './ReactCurrentFiber';
 import {StrictLegacyMode} from './ReactTypeOfMode';
 import {

packages/react-reconciler/src/ReactFiberReconciler.js

@@ -24,7 +24,7 @@ import {
   SuspenseComponent,
 } from './ReactWorkTags';
 import {NoFlags, Placement, Hydrating} from './ReactFiberFlags';
-import {currentOwner} from './ReactFiberCurrentOwner';
+import {current as currentOwner} from './ReactCurrentFiber';
 
 export function getNearestMountedFiber(fiber: Fiber): null | Fiber {
   let node = fiber;

packages/react-reconciler/src/ReactFiberTreeReflection.js

@@ -204,7 +204,6 @@ import {
   ContextOnlyDispatcher,
 } from './ReactFiberHooks';
 import {DefaultAsyncDispatcher} from './ReactFiberAsyncDispatcher';
-import {setCurrentOwner} from './ReactFiberCurrentOwner';
 import {
   createCapturedValueAtFiber,
   type CapturedValue,
@@ -230,8 +229,10 @@ import ReactStrictModeWarnings from './ReactStrictModeWarnings';
 import {
   isRendering as ReactCurrentDebugFiberIsRenderingInDEV,
   current as ReactCurrentFiberCurrent,
-  resetCurrentFiber as resetCurrentDebugFiberInDEV,
-  setCurrentFiber as setCurrentDebugFiberInDEV,
+  setCurrentDebugFiberInDEV,
+  resetCurrentDebugFiberInDEV,
+  resetCurrentFiber,
+  setCurrentFiber,
 } from './ReactCurrentFiber';
 import {
   isDevToolsPresent,
@@ -1683,9 +1684,8 @@ function handleThrow(root: FiberRoot, thrownValue: any): void {
   // These should be reset immediately because they're only supposed to be set
   // when React is executing user code.
   resetHooksAfterThrow();
-  resetCurrentDebugFiberInDEV();
   if (__DEV__ || !disableStringRefs) {
-    setCurrentOwner(null);
+    resetCurrentFiber();
   }
 
   if (thrownValue === SuspenseException) {
@@ -2377,18 +2377,16 @@ function performUnitOfWork(unitOfWork: Fiber): void {
     next = beginWork(current, unitOfWork, entangledRenderLanes);
   }
 
-  resetCurrentDebugFiberInDEV();
+  if (__DEV__ || !disableStringRefs) {
+    resetCurrentFiber();
+  }
   unitOfWork.memoizedProps = unitOfWork.pendingProps;
   if (next === null) {
     // If this doesn't spawn new work, complete the current work.
     completeUnitOfWork(unitOfWork);
   } else {
     workInProgress = next;
   }
-
-  if (__DEV__ || !disableStringRefs) {
-    setCurrentOwner(null);
-  }
 }
 
 function replaySuspendedUnitOfWork(unitOfWork: Fiber): void {
@@ -2399,7 +2397,6 @@ function replaySuspendedUnitOfWork(unitOfWork: Fiber): void {
   setCurrentDebugFiberInDEV(unitOfWork);
 
   let next;
-  setCurrentDebugFiberInDEV(unitOfWork);
   const isProfilingMode =
     enableProfilerTimer && (unitOfWork.mode & ProfileMode) !== NoMode;
   if (isProfilingMode) {
@@ -2492,18 +2489,16 @@ function replaySuspendedUnitOfWork(unitOfWork: Fiber): void {
   // The begin phase finished successfully without suspending. Return to the
   // normal work loop.
 
-  resetCurrentDebugFiberInDEV();
+  if (__DEV__ || !disableStringRefs) {
+    resetCurrentFiber();
+  }
   unitOfWork.memoizedProps = unitOfWork.pendingProps;
   if (next === null) {
     // If this doesn't spawn new work, complete the current work.
     completeUnitOfWork(unitOfWork);
   } else {
     workInProgress = next;
   }
-
-  if (__DEV__ || !disableStringRefs) {
-    setCurrentOwner(null);
-  }
 }
 
 function throwAndUnwindWorkLoop(
@@ -2893,11 +2888,6 @@ function commitRootImpl(
     const prevExecutionContext = executionContext;
     executionContext |= CommitContext;
 
-    // Reset this to null before calling lifecycles
-    if (__DEV__ || !disableStringRefs) {
-      setCurrentOwner(null);
-    }
-
     // The commit phase is broken into several sub-phases. We do a separate pass
     // of the effect list for each phase: all mutation effects come before all
     // layout effects, and so on.