dify/web/app/components/workflow/nodes/start/use-config.ts

import type { StartNodeType } from './types'
import type { InputVar, MoreInfo, ValueSelector } from '@/app/components/workflow/types'
import { toast } from '@langgenius/dify-ui/toast'
import { useBoolean } from 'ahooks'
import { produce } from 'immer'
import { useCallback, useState } from 'react'
import { useTranslation } from 'react-i18next'
import { useIsChatMode, useNodesReadOnly, useWorkflow } from '@/app/components/workflow/hooks'
import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
import { ChangeType } from '@/app/components/workflow/types'
import { hasDuplicateStr } from '@/utils/var'
import useInspectVarsCrud from '../../hooks/use-inspect-vars-crud'

const useConfig = (id: string, payload: StartNodeType) => {
  const { t } = useTranslation()
  const { nodesReadOnly: readOnly } = useNodesReadOnly()
  const { handleOutVarRenameChange, isVarUsedInNodes, removeUsedVarInNodes } = useWorkflow()
  const isChatMode = useIsChatMode()

  const { inputs, setInputs } = useNodeCrud<StartNodeType>(id, payload)

  const { deleteNodeInspectorVars, renameInspectVarName, nodesWithInspectVars, deleteInspectVar } =
    useInspectVarsCrud()

  const [isShowAddVarModal, { setTrue: showAddVarModal, setFalse: hideAddVarModal }] =
    useBoolean(false)

  const [
    isShowRemoveVarConfirm,
    { setTrue: showRemoveVarConfirm, setFalse: hideRemoveVarConfirm },
  ] = useBoolean(false)
  const [removedVar, setRemovedVar] = useState<ValueSelector>([])
  const [removedIndex, setRemoveIndex] = useState(0)
  const handleVarListChange = useCallback(
    (newList: InputVar[], moreInfo?: { index: number; payload: MoreInfo }) => {
      if (moreInfo?.payload?.type === ChangeType.remove) {
        const varId = nodesWithInspectVars
          .find((node) => node.nodeId === id)
          ?.vars.find((varItem) => {
            return varItem.name === moreInfo?.payload?.payload?.beforeKey
          })?.id
        if (varId) deleteInspectVar(id, varId)

        if (isVarUsedInNodes([id, moreInfo?.payload?.payload?.beforeKey || ''])) {
          showRemoveVarConfirm()
          setRemovedVar([id, moreInfo?.payload?.payload?.beforeKey || ''])
          setRemoveIndex(moreInfo?.index as number)
          return
        }
      }

      const newInputs = produce(inputs, (draft: any) => {
        draft.variables = newList
      })
      setInputs(newInputs)
      if (moreInfo?.payload?.type === ChangeType.changeVarName) {
        const changedVar = newList[moreInfo.index]
        handleOutVarRenameChange(
          id,
          [id, inputs.variables[moreInfo.index]!.variable],
          [id, changedVar!.variable],
        )
        renameInspectVarName(id, inputs.variables[moreInfo.index]!.variable, changedVar!.variable)
      } else if (moreInfo?.payload?.type !== ChangeType.remove) {
        // edit var type
        deleteNodeInspectorVars(id)
      }
    },
    [
      deleteInspectVar,
      deleteNodeInspectorVars,
      handleOutVarRenameChange,
      id,
      inputs,
      isVarUsedInNodes,
      nodesWithInspectVars,
      renameInspectVarName,
      setInputs,
      showRemoveVarConfirm,
    ],
  )

  const removeVarInNode = useCallback(() => {
    const newInputs = produce(inputs, (draft) => {
      draft.variables.splice(removedIndex, 1)
    })
    setInputs(newInputs)
    removeUsedVarInNodes(removedVar)
    hideRemoveVarConfirm()
  }, [hideRemoveVarConfirm, inputs, removeUsedVarInNodes, removedIndex, removedVar, setInputs])

  const handleAddVariable = useCallback(
    (payload: InputVar) => {
      const newInputs = produce(inputs, (draft: StartNodeType) => {
        draft.variables.push(payload)
      })
      const newList = newInputs.variables
      let errorMsgKey: 'varKeyError.keyAlreadyExists' | '' = ''
      let typeName: 'variableConfig.varName' | 'variableConfig.labelName' | '' = ''
      if (hasDuplicateStr(newList.map((item) => item.variable))) {
        errorMsgKey = 'varKeyError.keyAlreadyExists'
        typeName = 'variableConfig.varName'
      } else if (hasDuplicateStr(newList.map((item) => item.label as string))) {
        errorMsgKey = 'varKeyError.keyAlreadyExists'
        typeName = 'variableConfig.labelName'
      }

      if (errorMsgKey && typeName) {
        toast.error(
          t(($) => $[errorMsgKey], {
            ns: 'appDebug',
            key: t(($) => $[typeName], { ns: 'appDebug' }),
          }),
        )
        return false
      }
      setInputs(newInputs)
      return true
    },
    [inputs, setInputs],
  )
  return {
    readOnly,
    isChatMode,
    inputs,
    isShowAddVarModal,
    showAddVarModal,
    hideAddVarModal,
    handleVarListChange,
    handleAddVariable,
    isShowRemoveVarConfirm,
    hideRemoveVarConfirm,
    onRemoveVarConfirm: removeVarInNode,
  }
}

export default useConfig