opensource
/
dify
mirror of https://github.com/langgenius/dify.git
2
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

hooks

This commit is contained in:
StyleZhang 2024-02-27 18:20:32 +08:00
parent a311f88c99
commit 236cc6f526
2 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
web/app/components/workflow/hooks.ts
View File

@ -28,9 +28,9 @@ export const useWorkflow = () => {
setEdges,
} = store.getState()
const newNodes = produce(getNodes(), (draft) => {
const currentNode = draft.find(n => n.id === node.id)
if (currentNode)
currentNode.data = { ...currentNode.data, hovering: true }
const currentNode = draft.find(n => n.id === node.id)!
currentNode.data.hovering = true
})
setNodes(newNodes)
const newEdges = produce(edges, (draft) => {
@ -53,9 +53,9 @@ export const useWorkflow = () => {
setEdges,
} = store.getState()
const newNodes = produce(getNodes(), (draft) => {
const currentNode = draft.find(n => n.id === node.id)
if (currentNode)
currentNode.data = { ...currentNode.data, hovering: false }
const currentNode = draft.find(n => n.id === node.id)!
currentNode.data.hovering = false
})
setNodes(newNodes)
const newEdges = produce(edges, (draft) => {
@ -105,7 +105,7 @@ export const useWorkflow = () => {
setSelectedNode(null)
const newNodes = produce(getNodes(), (draft) => {
draft.forEach((item) => {
item.data = { ...item.data, selected: false }
item.data.selected = false
})
})
setNodes(newNodes)
@ -114,7 +114,10 @@ export const useWorkflow = () => {
setSelectedNode(selectNode)
const newNodes = produce(getNodes(), (draft) => {
draft.forEach((item) => {
item.data = { ...item.data, selected: item.id === selectNode.id }
if (item.id === selectNode.id)
item.data.selected = true
else
item.data.selected = false
})
})
setNodes(newNodes)
@ -126,9 +129,9 @@ export const useWorkflow = () => {
setNodes,
} = store.getState()
const newNodes = produce(getNodes(), (draft) => {
const currentNode = draft.find(n => n.id === id)
if (currentNode)
currentNode.data = { ...currentNode.data, ...data }
const currentNode = draft.find(n => n.id === id)!
currentNode.data = { ...currentNode.data, ...data }
})
setNodes(newNodes)
setSelectedNode({ id, data })

1
web/app/components/workflow/index.tsx
View File

@ -86,6 +86,7 @@ const Workflow: FC<WorkflowProps> = memo(({
} = useWorkflow()
useEffect(() => {
console.log(nodesInitialized, '2')
if (nodesInitialized && initialData.needUpdatePosition)
handleInitialLayoutNodes()
}, [nodesInitialized])