refactor(fiber): Update getFiberDEV function logic

aidenybai · Nov 15, 2024 · c4c79cf · c4c79cf
1 parent ecb4041
commit c4c79cf
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/fiber.ts b/src/fiber.ts
@@ -145,6 +145,13 @@ export const registerDevtoolsHook = ({
 const REACT_MAJOR_VERSION = Number(React.version.split('.')[0]);
 const dispatcherRefs = new Set();
 
+export const getFiberDEV = (): Fiber | null => {
+  return (
+    ReactSharedInternals?.A?.getOwner() ??
+    ReactSharedInternals?.ReactCurrentOwner?.current
+  );
+};
+
 export const controlDispatcherRef = (currentDispatcherRef: any) => {
   const ref = currentDispatcherRef;
   if (ref && !dispatcherRefs.has(ref)) {