From 129a68bb06031f4db8cb37f659d3685abfb633cf Mon Sep 17 00:00:00 2001
From: StyleZhang <jasonapring2015@outlook.com>
Date: Fri, 15 Mar 2024 19:34:31 +0800
Subject: [PATCH] auto layout

---
 .../workflow/block-selector/index.tsx          |  2 +-
 web/app/components/workflow/constants.ts       |  6 +++++-
 .../components/workflow/hooks/use-workflow.ts  | 18 +++++++++++++++---
 web/app/components/workflow/utils.ts           |  4 +++-
 4 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/web/app/components/workflow/block-selector/index.tsx b/web/app/components/workflow/block-selector/index.tsx
index 6b948a86fa..b742f5f1c3 100644
--- a/web/app/components/workflow/block-selector/index.tsx
+++ b/web/app/components/workflow/block-selector/index.tsx
@@ -99,7 +99,7 @@ const NodeSelector: FC<NodeSelectorProps> = ({
         }
       </PortalToFollowElemTrigger>
       <PortalToFollowElemContent className='z-[1000]'>
-        <div className={`w-[256px] rounded-lg border-[0.5px] border-gray-200 bg-white shadow-lg ${popupClassName}`}>
+        <div className={`rounded-lg border-[0.5px] border-gray-200 bg-white shadow-lg ${popupClassName}`}>
           <div className='px-2 pt-2'>
             <div
               className='flex items-center px-2 rounded-lg bg-gray-100'
diff --git a/web/app/components/workflow/constants.ts b/web/app/components/workflow/constants.ts
index 40b855b81a..581ce2c8b7 100644
--- a/web/app/components/workflow/constants.ts
+++ b/web/app/components/workflow/constants.ts
@@ -230,8 +230,12 @@ export const NODE_WIDTH = 220
 export const X_OFFSET = 64
 export const NODE_WIDTH_X_OFFSET = NODE_WIDTH + X_OFFSET
 export const Y_OFFSET = 39
-export const TREE_DEEPTH = 20
+export const TREE_DEEPTH = 30
 export const START_INITIAL_POSITION = { x: 80, y: 282 }
+export const AUTO_LAYOUT_OFFSET = {
+  x: -42,
+  y: 243,
+}
 
 export const RETRIEVAL_OUTPUT_STRUCT = `{
   "content": "",
diff --git a/web/app/components/workflow/hooks/use-workflow.ts b/web/app/components/workflow/hooks/use-workflow.ts
index d1ddaff3cd..07233fc369 100644
--- a/web/app/components/workflow/hooks/use-workflow.ts
+++ b/web/app/components/workflow/hooks/use-workflow.ts
@@ -8,6 +8,7 @@ import produce from 'immer'
 import {
   getIncomers,
   getOutgoers,
+  useReactFlow,
   useStoreApi,
 } from 'reactflow'
 import type { Connection } from 'reactflow'
@@ -20,6 +21,7 @@ import type { Node } from '../types'
 import { BlockEnum } from '../types'
 import { useWorkflowStore } from '../store'
 import {
+  AUTO_LAYOUT_OFFSET,
   START_INITIAL_POSITION,
   SUPPORT_OUTPUT_VARS_NODE,
 } from '../constants'
@@ -27,6 +29,7 @@ import {
   useNodesExtraData,
   useNodesInitialData,
 } from './use-nodes-data'
+import { useNodesSyncDraft } from './use-nodes-sync-draft'
 import { useStore as useAppStore } from '@/app/components/app/store'
 import {
   fetchNodesDefaultConfigs,
@@ -43,7 +46,9 @@ export const useIsChatMode = () => {
 
 export const useWorkflow = () => {
   const store = useStoreApi()
+  const reactflow = useReactFlow()
   const nodesExtraData = useNodesExtraData()
+  const { handleSyncWorkflowDraft } = useNodesSyncDraft()
 
   const handleLayout = useCallback(async () => {
     const {
@@ -51,6 +56,7 @@ export const useWorkflow = () => {
       edges,
       setNodes,
     } = store.getState()
+    const { setViewport } = reactflow
     const nodes = getNodes()
     const layout = getLayoutByDagre(nodes, edges)
 
@@ -58,13 +64,19 @@ export const useWorkflow = () => {
       draft.forEach((node) => {
         const nodeWithPosition = layout.node(node.id)
         node.position = {
-          x: nodeWithPosition.x,
-          y: nodeWithPosition.y,
+          x: nodeWithPosition.x + AUTO_LAYOUT_OFFSET.x,
+          y: nodeWithPosition.y + AUTO_LAYOUT_OFFSET.y,
         }
       })
     })
     setNodes(newNodes)
-  }, [store])
+    setViewport({
+      x: 0,
+      y: 0,
+      zoom: 0.8,
+    })
+    setTimeout(() => handleSyncWorkflowDraft())
+  }, [store, reactflow, handleSyncWorkflowDraft])
 
   const getTreeLeafNodes = useCallback((nodeId: string) => {
     const {
diff --git a/web/app/components/workflow/utils.ts b/web/app/components/workflow/utils.ts
index 0c733a74f0..8281164f9c 100644
--- a/web/app/components/workflow/utils.ts
+++ b/web/app/components/workflow/utils.ts
@@ -146,10 +146,12 @@ export const getNodesPositionMap = (nodes: Node[], edges: Edge[]) => {
   return positionMap
 }
 
+const dagreGraph = new dagre.graphlib.Graph()
+dagreGraph.setDefaultEdgeLabel(() => ({}))
+
 export const getLayoutByDagre = (originNodes: Node[], originEdges: Edge[]) => {
   const nodes = cloneDeep(originNodes)
   const edges = cloneDeep(originEdges)
-  const dagreGraph = new dagre.graphlib.Graph()
   dagreGraph.setGraph({
     rankdir: 'LR',
     align: 'UL',