dify/web/app/components/workflow/hooks-store/provider.tsx

import type { Shape } from './store'
import { createContext, useEffect, useRef } from 'react'
import { useStore } from 'reactflow'
import { createHooksStore } 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)
  const { accessControl } = restProps

  useEffect(() => {
    if (storeRef.current && d3Selection && d3Zoom) storeRef.current.getState().refreshAll(restProps)
  }, [d3Selection, d3Zoom])

  useEffect(() => {
    if (storeRef.current && accessControl) storeRef.current.getState().refreshAll({ accessControl })
  }, [accessControl])

  if (!storeRef.current) storeRef.current = createHooksStore(restProps)

  return (
    <HooksStoreContext.Provider value={storeRef.current}>{children}</HooksStoreContext.Provider>
  )
}