dify

dify/web/app/components/rag-pipeline/hooks/use-pipeline-start-run.tsx

@@ -0,0 +1,64 @@
+import { useCallback } from 'react'
+import { useWorkflowStore } from '@/app/components/workflow/store'
+import {
+  WorkflowRunningStatus,
+} from '@/app/components/workflow/types'
+import { useWorkflowInteractions } from '@/app/components/workflow/hooks'
+import {
+  useInputFieldPanel,
+  useNodesSyncDraft,
+} from '.'
+
+export const usePipelineStartRun = () => {
+  const workflowStore = useWorkflowStore()
+  const { handleCancelDebugAndPreviewPanel } = useWorkflowInteractions()
+  const { doSyncWorkflowDraft } = useNodesSyncDraft()
+  const { closeAllInputFieldPanels } = useInputFieldPanel()
+
+  const handleWorkflowStartRunInWorkflow = useCallback(async () => {
+    const {
+      workflowRunningData,
+    } = workflowStore.getState()
+
+    if (workflowRunningData?.result.status === WorkflowRunningStatus.Running)
+      return
+
+    const {
+      isPreparingDataSource,
+      setIsPreparingDataSource,
+      showDebugAndPreviewPanel,
+      setShowEnvPanel,
+      setShowDebugAndPreviewPanel,
+    } = workflowStore.getState()
+
+    if (!isPreparingDataSource && workflowRunningData) {
+      workflowStore.setState({
+        isPreparingDataSource: true,
+        workflowRunningData: undefined,
+      })
+      return
+    }
+
+    setShowEnvPanel(false)
+    closeAllInputFieldPanels()
+
+    if (showDebugAndPreviewPanel) {
+      setIsPreparingDataSource?.(false)
+      handleCancelDebugAndPreviewPanel()
+      return
+    }
+
+    await doSyncWorkflowDraft()
+    setIsPreparingDataSource?.(true)
+    setShowDebugAndPreviewPanel(true)
+  }, [workflowStore, handleCancelDebugAndPreviewPanel, doSyncWorkflowDraft])
+
+  const handleStartWorkflowRun = useCallback(() => {
+    handleWorkflowStartRunInWorkflow()
+  }, [handleWorkflowStartRunInWorkflow])
+
+  return {
+    handleStartWorkflowRun,
+    handleWorkflowStartRunInWorkflow,
+  }
+}