dify/web/app/components/workflow/hooks.ts

import {
  useCallback,
  useMemo,
  useState,
} from 'react'
import type { Node } from './types'

export const useWorkflow = (nodes: Node[]) => {
  const [selectedNodeId, setSelectedNodeId] = useState('')

  const handleSelectedNodeIdChange = useCallback((nodeId: string) => setSelectedNodeId(nodeId), [])

  const selectedNode = useMemo(() => {
    return nodes.find(node => node.id === selectedNodeId)
  }, [nodes, selectedNodeId])

  return {
    selectedNodeId,
    selectedNode,
    handleSelectedNodeIdChange,
  }
}