only check if _mainTreeNodeRef is passed in

It can be that it is passed in but still `null`, that's fine because it
means it will be passed in later once the DOM is ready.
This commit is contained in:
Robin Malfait
2023-08-02 13:43:28 +02:00
parent 8a37854da3
commit a3fa86b6f4
@@ -1,6 +1,5 @@
import React, { useRef, useMemo, MutableRefObject } from 'react'
import { Hidden, Features as HiddenFeatures } from '../internal/hidden'
import { useComputed } from './use-computed'
import { useEvent } from './use-event'
import { useOwnerDocument } from './use-owner'
@@ -60,15 +59,10 @@ export function useRootContainers({
mainTreeNodeRef,
MainTreeNode: useMemo(() => {
return function MainTreeNode() {
let hasPassedInMainTreeNode = useComputed(
() => (_mainTreeNodeRef?.current ?? null) !== null,
[_mainTreeNodeRef]
)
if (hasPassedInMainTreeNode) return null
if (_mainTreeNodeRef != null) return null
return <Hidden features={HiddenFeatures.Hidden} ref={mainTreeNodeRef} />
}
}, [mainTreeNodeRef]),
}, [mainTreeNodeRef, _mainTreeNodeRef]),
}
}