mirror of
https://github.com/langgenius/dify.git
synced 2025-08-01 22:06:17 +00:00
48 lines
1.2 KiB
TypeScript
48 lines
1.2 KiB
TypeScript
![]() |
import { useCallback } from 'react'
|
||
|
import { useStoreApi } from 'reactflow'
|
||
|
import { produce } from 'immer'
|
||
|
import type { OnSelectBlock } from '@/app/components/workflow/types'
|
||
|
import { generateNewNode } from '@/app/components/workflow/utils'
|
||
|
import { useNodesMetaData } from '@/app/components/workflow/hooks'
|
||
|
|
||
|
export const useReplaceDataSourceNode = (id: string) => {
|
||
|
const store = useStoreApi()
|
||
|
const { nodesMap: nodesMetaDataMap } = useNodesMetaData()
|
||
|
|
||
|
const handleReplaceNode = useCallback<OnSelectBlock>((
|
||
|
type,
|
||
|
toolDefaultValue,
|
||
|
) => {
|
||
|
const {
|
||
|
getNodes,
|
||
|
setNodes,
|
||
|
} = store.getState()
|
||
|
const nodes = getNodes()
|
||
|
const emptyNodeIndex = nodes.findIndex(node => node.id === id)
|
||
|
|
||
|
if (emptyNodeIndex < 0) return
|
||
|
const {
|
||
|
defaultValue,
|
||
|
} = nodesMetaDataMap![type]
|
||
|
const emptyNode = nodes[emptyNodeIndex]
|
||
|
const { newNode } = generateNewNode({
|
||
|
data: {
|
||
|
...(defaultValue as any),
|
||
|
...(toolDefaultValue || {}),
|
||
|
},
|
||
|
position: {
|
||
|
x: emptyNode.position.x,
|
||
|
y: emptyNode.position.y,
|
||
|
},
|
||
|
})
|
||
|
const newNodes = produce(nodes, (draft) => {
|
||
|
draft[emptyNodeIndex] = newNode
|
||
|
})
|
||
|
setNodes(newNodes)
|
||
|
}, [])
|
||
|
|
||
|
return {
|
||
|
handleReplaceNode,
|
||
|
}
|
||
|
}
|