dify/web/app/components/workflow/plugin-dependency/hooks.ts

import { useCallback } from 'react'
import { useCheckPipelineDependencies } from '@/service/use-pipeline'
import { useMutationCheckDependencies } from '@/service/use-plugins'
import { useStore as usePluginDependenciesStore } from './store'

export const usePluginDependencies = () => {
  const { mutateAsync: checkWorkflowDependencies } = useMutationCheckDependencies()
  const { mutateAsync: checkPipelineDependencies } = useCheckPipelineDependencies()

  const handleCheckPluginDependencies = useCallback(
    async (id: string, isPipeline = false) => {
      const checkDependencies = isPipeline ? checkPipelineDependencies : checkWorkflowDependencies
      const { leaked_dependencies } = await checkDependencies(id)
      const { setDependencies } = usePluginDependenciesStore.getState()
      setDependencies(leaked_dependencies)
    },
    [checkWorkflowDependencies, checkPipelineDependencies],
  )

  return {
    handleCheckPluginDependencies,
  }
}