dify

dify/web/app/components/workflow/hooks/use-auto-generate-webhook-url.ts

@@ -0,0 +1,48 @@
+import { useCallback } from 'react'
+import { produce } from 'immer'
+import { useStoreApi } from 'reactflow'
+import { useStore as useAppStore } from '@/app/components/app/store'
+import { BlockEnum } from '@/app/components/workflow/types'
+import { fetchWebhookUrl } from '@/service/apps'
+
+export const useAutoGenerateWebhookUrl = () => {
+  const reactFlowStore = useStoreApi()
+
+  return useCallback(async (nodeId: string) => {
+    const appId = useAppStore.getState().appDetail?.id
+    if (!appId)
+      return
+
+    const { getNodes } = reactFlowStore.getState()
+    const node = getNodes().find(n => n.id === nodeId)
+    if (!node || node.data.type !== BlockEnum.TriggerWebhook)
+      return
+
+    if (node.data.webhook_url && node.data.webhook_url.length > 0)
+      return
+
+    try {
+      const response = await fetchWebhookUrl({ appId, nodeId })
+      const { getNodes: getLatestNodes, setNodes } = reactFlowStore.getState()
+      let hasUpdated = false
+      const updatedNodes = produce(getLatestNodes(), (draft) => {
+        const targetNode = draft.find(n => n.id === nodeId)
+        if (!targetNode || targetNode.data.type !== BlockEnum.TriggerWebhook)
+          return
+
+        targetNode.data = {
+          ...targetNode.data,
+          webhook_url: response.webhook_url,
+          webhook_debug_url: response.webhook_debug_url,
+        }
+        hasUpdated = true
+      })
+
+      if (hasUpdated)
+        setNodes(updatedNodes)
+    }
+    catch (error: unknown) {
+      console.error('Failed to auto-generate webhook URL:', error)
+    }
+  }, [reactFlowStore])
+}