Return all children

Author: eps1lon

packages/react-reconciler/src/ReactFiber.js

@@ -75,7 +75,6 @@ import {
   ViewTransitionComponent,
   ActivityComponent,
 } from './ReactWorkTags';
-import {createInitialOffscreenInstance} from './ReactFiberOffscreenComponent';
 import {getComponentNameFromOwner} from 'react-reconciler/src/getComponentNameFromFiber';
 import {isDevToolsPresent} from './ReactFiberDevToolsHook';
 import {
@@ -830,8 +829,6 @@ export function createFiberFromOffscreen(
 ): Fiber {
   const fiber = createFiber(OffscreenComponent, pendingProps, key, mode);
   fiber.lanes = lanes;
-  const primaryChildInstance = createInitialOffscreenInstance();
-  fiber.stateNode = primaryChildInstance;
   return fiber;
 }
 export function createFiberFromActivity(
@@ -879,10 +876,6 @@ export function createFiberFromLegacyHidden(
   const fiber = createFiber(LegacyHiddenComponent, pendingProps, key, mode);
   fiber.elementType = REACT_LEGACY_HIDDEN_TYPE;
   fiber.lanes = lanes;
-  // Adding a stateNode for legacy hidden because it's currently using
-  // the offscreen implementation, which depends on a state node
-  const instance = createInitialOffscreenInstance();
-  fiber.stateNode = instance;
   return fiber;
 }
 

packages/react-reconciler/src/ReactFiberBeginWork.js

@@ -278,7 +278,10 @@ import {
   createCapturedValueFromError,
   createCapturedValueAtFiber,
 } from './ReactCapturedValue';
-import {createInitialOffscreenInstance} from './ReactFiberOffscreenComponent';
+import {
+  OffscreenHidden,
+  OffscreenVisible,
+} from './ReactFiberOffscreenComponent';
 import {
   createClassErrorUpdate,
   initializeClassErrorUpdate,
@@ -619,17 +622,21 @@ function updateOffscreenComponent(
   const prevState: OffscreenState | null =
     current !== null ? current.memoizedState : null;
 
+  const nextHidden =
+    nextProps.mode === 'hidden' ||
+    (enableLegacyHidden && nextProps.mode === 'unstable-defer-without-hiding');
+
   if (current === null && workInProgress.stateNode === null) {
-    // We previously reset the work-in-progress.
-    // We need to create a new Offscreen instance.
-    const primaryChildInstance = createInitialOffscreenInstance();
+    const primaryChildInstance = {
+      _visibility: nextHidden ? OffscreenHidden : OffscreenVisible,
+      _pendingMarkers: null,
+      _retryCache: null,
+      _transitions: null,
+    };
     workInProgress.stateNode = primaryChildInstance;
   }
 
-  if (
-    nextProps.mode === 'hidden' ||
-    (enableLegacyHidden && nextProps.mode === 'unstable-defer-without-hiding')
-  ) {
+  if (nextHidden) {
     // Rendering a hidden tree.
 
     const didSuspend = (workInProgress.flags & DidCapture) !== NoFlags;
@@ -2806,7 +2813,7 @@ function mountSuspenseFallbackAfterRetryWithoutHydrating(
 ) {
   const fiberMode = workInProgress.mode;
   const primaryChildProps: OffscreenProps = {
-    mode: 'visible',
+    mode: 'hidden',
     children: primaryChildren,
   };
   const primaryChildFragment = mountWorkInProgressOffscreenFiber(
@@ -2835,7 +2842,7 @@ function mountSuspenseFallbackAfterRetryWithoutHydrating(
     reconcileChildFibers(workInProgress, current.child, null, renderLanes);
   }
 
-  return fallbackChildFragment;
+  return primaryChildFragment;
 }
 
 function mountDehydratedSuspenseComponent(

packages/react-reconciler/src/ReactFiberOffscreenComponent.js

@@ -50,6 +50,7 @@ export type OffscreenQueue = {
 
 type OffscreenVisibility = number;
 
+export const OffscreenHidden = /*                      */ 0b000;
 export const OffscreenVisible = /*                     */ 0b001;
 export const OffscreenPassiveEffectsConnected = /*     */ 0b010;
 
@@ -59,12 +60,3 @@ export type OffscreenInstance = {
   _transitions: Set<Transition> | null,
   _retryCache: WeakSet<Wakeable> | Set<Wakeable> | null,
 };
-
-export function createInitialOffscreenInstance(): OffscreenInstance {
-  return {
-    _visibility: OffscreenVisible,
-    _pendingMarkers: null,
-    _retryCache: null,
-    _transitions: null,
-  };
-}