mirror of
https://github.com/langgenius/dify.git
synced 2026-07-21 10:38:32 +08:00
23 lines
841 B
TypeScript
23 lines
841 B
TypeScript
import type { StateCreator } from 'zustand'
|
|
import type { SliceFromInjection } from './store/workflow'
|
|
import { createContext, useRef } from 'react'
|
|
import { createWorkflowStore } from './store/workflow'
|
|
|
|
type WorkflowStore = ReturnType<typeof createWorkflowStore>
|
|
export const WorkflowContext = createContext<WorkflowStore | null>(null)
|
|
|
|
type WorkflowProviderProps = {
|
|
children: React.ReactNode
|
|
injectWorkflowStoreSliceFn?: StateCreator<SliceFromInjection>
|
|
}
|
|
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>
|
|
}
|