dify

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

@@ -0,0 +1,20 @@
+import { useCallback } from 'react'
+import { useStore as usePluginDependenciesStore } from './store'
+import { useMutationCheckDependencies } from '@/service/use-plugins'
+import { useCheckPipelineDependencies } from '@/service/use-pipeline'
+
+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,
+  }
+}