add dev field

MrWangJustToDo · May 14, 2024 · 455ccb2 · 455ccb2
1 parent 74596b4
commit 455ccb2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/packages/myreact-dom/src/client/index.ts b/packages/myreact-dom/src/client/index.ts
@@ -11,6 +11,7 @@ import {
   enableEventSystem,
   enableEventTrack,
   enableHighlight,
+  injectDevField,
 } from "../shared";
 
 import { render, hydrate, hydrateRoot, createRoot } from "./mount";
@@ -34,6 +35,8 @@ initGlobalRenderPlatform();
 // not need for new refresh runtime package
 // initGlobalHMR();
 
+injectDevField();
+
 checkReconcilerVersion();
 
 checkMyReactVersion();

diff --git a/packages/myreact-dom/src/shared/dev.ts b/packages/myreact-dom/src/shared/dev.ts
@@ -7,6 +7,7 @@ export type DevToolRuntime = (dispatch: CustomRenderDispatch, platform: CustomRe
 export const setDevTools = (devToolRuntime: DevToolRuntime, dispatch: CustomRenderDispatch, platform: CustomRenderPlatform) => {
   try {
     devToolRuntime(dispatch, platform, initHMR);
+    delete globalThis[DEV_INJECT_FIELD];
   } catch (e) {
     if (__DEV__) {
       console.error("devToolRuntime failed:", e);
@@ -19,8 +20,16 @@ export const DEV_TOOL_RUNTIME_FIELD = "__MY_REACT_DEVTOOL_RUNTIME__";
 
 export const DISPATCH_FIELD = "__@my-react/dispatch__";
 
+const DEV_INJECT_FIELD = "__@my-react/react-inject__";
+
 const DEV_REFRESH_FIELD = "__@my-react/react-refresh__";
 
+export const injectDevField = () => {
+  if (__DEV__ && typeof globalThis !== "undefined") {
+    globalThis[DEV_INJECT_FIELD] = setDevTools;
+  }
+};
+
 export const autoSetDevTools = (dispatch: CustomRenderDispatch, platform: CustomRenderPlatform) => {
   const runtime = globalThis[DEV_TOOL_RUNTIME_FIELD];