From c4c79cf4160e67ef355f69788ce2a668f599df6c Mon Sep 17 00:00:00 2001
From: Aiden Bai <aiden.bai05@gmail.com>
Date: Thu, 14 Nov 2024 17:58:07 -0800
Subject: [PATCH] refactor(fiber): Update getFiberDEV function logic

---
 src/fiber.ts | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/src/fiber.ts b/src/fiber.ts
index fa8859ae..b37672eb 100644
--- 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)) {