mirror of https://github.com/langgenius/dify.git
30 lines
911 B
TypeScript
30 lines
911 B
TypeScript
import {
|
|
createContext,
|
|
useRef,
|
|
} from 'react'
|
|
import {
|
|
createWorkflowStore,
|
|
} from './store'
|
|
import type { StateCreator } from 'zustand'
|
|
import type { WorkflowSliceShape } from '@/app/components/workflow-app/store/workflow/workflow-slice'
|
|
|
|
type WorkflowStore = ReturnType<typeof createWorkflowStore>
|
|
export const WorkflowContext = createContext<WorkflowStore | null>(null)
|
|
|
|
export type WorkflowProviderProps = {
|
|
children: React.ReactNode
|
|
injectWorkflowStoreSliceFn?: StateCreator<WorkflowSliceShape>
|
|
}
|
|
export const WorkflowContextProvider = ({ children, injectWorkflowStoreSliceFn }: WorkflowProviderProps) => {
|
|
const storeRef = useRef<WorkflowStore | undefined>(undefined)
|
|
|
|
if (!storeRef.current)
|
|
storeRef.current = createWorkflowStore({ injectWorkflowStoreSliceFn })
|
|
|
|
return (
|
|
<WorkflowContext.Provider value={storeRef.current}>
|
|
{children}
|
|
</WorkflowContext.Provider>
|
|
)
|
|
}
|