dify/web/app/components/workflow/nodes/data-source-empty/hooks.ts

import type { OnSelectBlock } from '@/app/components/workflow/types'
import { produce } from 'immer'
import { useCallback } from 'react'
import { useNodesMetaData } from '@/app/components/workflow/hooks'
import { useCollaborativeWorkflow } from '@/app/components/workflow/hooks/use-collaborative-workflow'
import { generateNewNode } from '@/app/components/workflow/utils'

export const useReplaceDataSourceNode = (id: string) => {
  const collaborativeWorkflow = useCollaborativeWorkflow()
  const { nodesMap: nodesMetaDataMap } = useNodesMetaData()

  const handleReplaceNode = useCallback<OnSelectBlock>((
    type,
    pluginDefaultValue,
  ) => {
    const {
      nodes,
      setNodes,
    } = collaborativeWorkflow.getState()
    const emptyNodeIndex = nodes.findIndex(node => node.id === id)

    if (emptyNodeIndex < 0)
      return
    const nodeMetaData = nodesMetaDataMap?.[type]
    if (!nodeMetaData)
      return
    const { defaultValue } = nodeMetaData
    const emptyNode = nodes[emptyNodeIndex]
    const { newNode } = generateNewNode({
      data: {
        ...(defaultValue as any),
        ...pluginDefaultValue,
      },
      position: {
        x: emptyNode!.position.x,
        y: emptyNode!.position.y,
      },
    })
    const newNodes = produce(nodes, (draft) => {
      draft[emptyNodeIndex] = newNode
    })
    const newNodesWithoutTempNodes = produce(newNodes, (draft) => {
      return draft.filter(node => !node.data._isTempNode)
    })
    setNodes(newNodesWithoutTempNodes)
  }, [collaborativeWorkflow, id, nodesMetaDataMap])

  return {
    handleReplaceNode,
  }
}