初始化

wwf

2025-05-20 938c3e5a587ce950a94964ea509b9e7f8834dfae

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import {
  createContext,
  useEffect,
  useRef,
} from 'react'
import { useStore } from 'reactflow'
import {
  createHooksStore,
} from './store'
import type { Shape } from './store'
 
type HooksStore = ReturnType<typeof createHooksStore>
export const HooksStoreContext = createContext<HooksStore | null | undefined>(null)
type HooksStoreContextProviderProps = Partial<Shape> & {
  children: React.ReactNode
}
export const HooksStoreContextProvider = ({ children, ...restProps }: HooksStoreContextProviderProps) => {
  const storeRef = useRef<HooksStore | undefined>(undefined)
  const d3Selection = useStore(s => s.d3Selection)
  const d3Zoom = useStore(s => s.d3Zoom)
 
  useEffect(() => {
    if (storeRef.current && d3Selection && d3Zoom)
      storeRef.current.getState().refreshAll(restProps)
  // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [d3Selection, d3Zoom])
 
  if (!storeRef.current)
    storeRef.current = createHooksStore(restProps)
 
  return (
    <HooksStoreContext.Provider value={storeRef.current}>
      {children}
    </HooksStoreContext.Provider>
  )
}