youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

node selected

Browse Source
This commit is contained in:
StyleZhang
2024-04-01 20:32:59 +08:00
parent 5c3162cc33
commit e800109c02
4 changed files with 48 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
web/app/components/workflow/utils.ts
View File

@ -57,20 +57,38 @@ export const initialNodes = (nodes: Node[], edges: Edge[]) => {
}
export const initialEdges = (edges: Edge[], nodes: Node[]) => {
let selectedNode: Node | null = null
const nodesMap = nodes.reduce((acc, node) => {
acc[node.id] = node
if (node.data?.selected)
selectedNode = node
return acc
}, {} as Record<string, Node>)
return edges.map((edge) => {
edge.type = 'custom'
if (!edge.data?.sourceType)
edge.data = { ...edge.data, sourceType: nodesMap[edge.source].data.type! } as any
if (!edge.data?.sourceType) {
edge.data = {
...edge.data,
sourceType: nodesMap[edge.source].data.type!,
} as any
}
if (!edge.data?.targetType)
edge.data = { ...edge.data, targetType: nodesMap[edge.target].data.type! } as any
if (!edge.data?.targetType) {
edge.data = {
...edge.data,
targetType: nodesMap[edge.target].data.type!,
} as any
}
if (selectedNode) {
edge.data = {
...edge.data,
_connectedNodeIsSelected: edge.source === selectedNode.id || edge.target === selectedNode.id,
} as any
}
return edge
})
}