@hanzogui/use-store 2.0.0

package/src/useStoreDebug.tsx

@@ -0,0 +1,41 @@
+import React from 'react'
+
+import type { StoreInfo } from './interfaces'
+
+// const { ReactCurrentOwner } = (React as any)
+//   .__SECRET_INTERNALS_DO_NOT_USE_OR_YOU_WILL_BE_FIRED
+export const useCurrentComponent = () => {
+  return {}
+  // return ReactCurrentOwner &&
+  //   ReactCurrentOwner.current &&
+  //   ReactCurrentOwner.current.elementType
+  //   ? ReactCurrentOwner.current.elementType
+  //   : {}
+}
+
+export function useDebugStoreComponent(StoreCons: any) {
+  const cmp = useCurrentComponent()
+
+  // add in outer loop to pickup immediately
+  DebugStores.add(StoreCons)
+  if (!DebugComponents.has(cmp)) {
+    DebugComponents.set(cmp, new Set())
+  }
+  const stores = DebugComponents.get(cmp)!
+  stores.add(StoreCons)
+
+  React.useLayoutEffect(() => {
+    return () => {
+      DebugStores.delete(StoreCons)
+      stores.delete(StoreCons)
+    }
+  }, [])
+}
+
+export const shouldDebug = (component: any, info: Pick<StoreInfo, 'storeInstance'>) => {
+  const StoreCons = info.storeInstance?.constructor
+  return DebugComponents.get(component)?.has(StoreCons)
+}
+
+export const DebugComponents = new Map<any, Set<any>>()
+export const DebugStores = new Set<any>()