dify/web/app/components/workflow/hooks/use-check-start-node-form.ts

import { useCallback } from 'react'
import { useTranslation } from 'react-i18next'
import { useStoreApi } from 'reactflow'
import { useWorkflowStore } from '@/app/components/workflow/store'
import { BlockEnum } from '@/app/components/workflow/types'
import { useToastContext } from '@/app/components/base/toast'
import type { InputVar } from '@/app/components/workflow/types'

export const useCheckStartNodeForm = () => {
  const { t } = useTranslation()
  const storeApi = useStoreApi()
  const workflowStore = useWorkflowStore()
  const { notify } = useToastContext()

  const checkStartNodeForm = useCallback(() => {
    const { getNodes } = storeApi.getState()
    const nodes = getNodes()
    const startNode = nodes.find(node => node.data.type === BlockEnum.Start)
    const variables: InputVar[] = startNode?.data.variables || []
    const inputs = workflowStore.getState().inputs

    let hasEmptyInput = ''
    const requiredVars = variables.filter(({ required }) => required)

    if (requiredVars?.length) {
      requiredVars.forEach(({ variable, label }) => {
        if (hasEmptyInput)
          return

        if (!inputs[variable])
          hasEmptyInput = label as string
      })
    }

    if (hasEmptyInput) {
      notify({ type: 'error', message: t('appDebug.errorMessage.valueOfVarRequired', { key: hasEmptyInput }) })
      return false
    }

    return true
  }, [storeApi, workflowStore, notify, t])

  return {
    checkStartNodeForm,
  }
}