|
import { |
|
Position, |
|
getConnectedEdges, |
|
getIncomers, |
|
getOutgoers, |
|
} from 'reactflow' |
|
import dagre from '@dagrejs/dagre' |
|
import { v4 as uuid4 } from 'uuid' |
|
import { |
|
cloneDeep, |
|
groupBy, |
|
isEqual, |
|
uniqBy, |
|
} from 'lodash-es' |
|
import type { |
|
Edge, |
|
InputVar, |
|
Node, |
|
ToolWithProvider, |
|
ValueSelector, |
|
} from './types' |
|
import { BlockEnum, ErrorHandleMode } from './types' |
|
import { |
|
CUSTOM_NODE, |
|
ITERATION_CHILDREN_Z_INDEX, |
|
ITERATION_NODE_Z_INDEX, |
|
NODE_WIDTH_X_OFFSET, |
|
START_INITIAL_POSITION, |
|
} from './constants' |
|
import { CUSTOM_ITERATION_START_NODE } from './nodes/iteration-start/constants' |
|
import type { QuestionClassifierNodeType } from './nodes/question-classifier/types' |
|
import type { IfElseNodeType } from './nodes/if-else/types' |
|
import { branchNameCorrect } from './nodes/if-else/utils' |
|
import type { ToolNodeType } from './nodes/tool/types' |
|
import type { IterationNodeType } from './nodes/iteration/types' |
|
import { CollectionType } from '@/app/components/tools/types' |
|
import { toolParametersToFormSchemas } from '@/app/components/tools/utils/to-form-schema' |
|
|
|
const WHITE = 'WHITE' |
|
const GRAY = 'GRAY' |
|
const BLACK = 'BLACK' |
|
|
|
const isCyclicUtil = (nodeId: string, color: Record<string, string>, adjList: Record<string, string[]>, stack: string[]) => { |
|
color[nodeId] = GRAY |
|
stack.push(nodeId) |
|
|
|
for (let i = 0; i < adjList[nodeId].length; ++i) { |
|
const childId = adjList[nodeId][i] |
|
|
|
if (color[childId] === GRAY) { |
|
stack.push(childId) |
|
return true |
|
} |
|
if (color[childId] === WHITE && isCyclicUtil(childId, color, adjList, stack)) |
|
return true |
|
} |
|
color[nodeId] = BLACK |
|
if (stack.length > 0 && stack[stack.length - 1] === nodeId) |
|
stack.pop() |
|
return false |
|
} |
|
|
|
const getCycleEdges = (nodes: Node[], edges: Edge[]) => { |
|
const adjList: Record<string, string[]> = {} |
|
const color: Record<string, string> = {} |
|
const stack: string[] = [] |
|
|
|
for (const node of nodes) { |
|
color[node.id] = WHITE |
|
adjList[node.id] = [] |
|
} |
|
|
|
for (const edge of edges) |
|
adjList[edge.source]?.push(edge.target) |
|
|
|
for (let i = 0; i < nodes.length; i++) { |
|
if (color[nodes[i].id] === WHITE) |
|
isCyclicUtil(nodes[i].id, color, adjList, stack) |
|
} |
|
|
|
const cycleEdges = [] |
|
if (stack.length > 0) { |
|
const cycleNodes = new Set(stack) |
|
for (const edge of edges) { |
|
if (cycleNodes.has(edge.source) && cycleNodes.has(edge.target)) |
|
cycleEdges.push(edge) |
|
} |
|
} |
|
|
|
return cycleEdges |
|
} |
|
|
|
export function getIterationStartNode(iterationId: string): Node { |
|
return generateNewNode({ |
|
id: `${iterationId}start`, |
|
type: CUSTOM_ITERATION_START_NODE, |
|
data: { |
|
title: '', |
|
desc: '', |
|
type: BlockEnum.IterationStart, |
|
isInIteration: true, |
|
}, |
|
position: { |
|
x: 24, |
|
y: 68, |
|
}, |
|
zIndex: ITERATION_CHILDREN_Z_INDEX, |
|
parentId: iterationId, |
|
selectable: false, |
|
draggable: false, |
|
}).newNode |
|
} |
|
|
|
export function generateNewNode({ data, position, id, zIndex, type, ...rest }: Omit<Node, 'id'> & { id?: string }): { |
|
newNode: Node |
|
newIterationStartNode?: Node |
|
} { |
|
const newNode = { |
|
id: id || `${Date.now()}`, |
|
type: type || CUSTOM_NODE, |
|
data, |
|
position, |
|
targetPosition: Position.Left, |
|
sourcePosition: Position.Right, |
|
zIndex: data.type === BlockEnum.Iteration ? ITERATION_NODE_Z_INDEX : zIndex, |
|
...rest, |
|
} as Node |
|
|
|
if (data.type === BlockEnum.Iteration) { |
|
const newIterationStartNode = getIterationStartNode(newNode.id); |
|
(newNode.data as IterationNodeType).start_node_id = newIterationStartNode.id; |
|
(newNode.data as IterationNodeType)._children = [newIterationStartNode.id] |
|
return { |
|
newNode, |
|
newIterationStartNode, |
|
} |
|
} |
|
|
|
return { |
|
newNode, |
|
} |
|
} |
|
|
|
export const preprocessNodesAndEdges = (nodes: Node[], edges: Edge[]) => { |
|
const hasIterationNode = nodes.some(node => node.data.type === BlockEnum.Iteration) |
|
|
|
if (!hasIterationNode) { |
|
return { |
|
nodes, |
|
edges, |
|
} |
|
} |
|
const nodesMap = nodes.reduce((prev, next) => { |
|
prev[next.id] = next |
|
return prev |
|
}, {} as Record<string, Node>) |
|
const iterationNodesWithStartNode = [] |
|
const iterationNodesWithoutStartNode = [] |
|
|
|
for (let i = 0; i < nodes.length; i++) { |
|
const currentNode = nodes[i] as Node<IterationNodeType> |
|
|
|
if (currentNode.data.type === BlockEnum.Iteration) { |
|
if (currentNode.data.start_node_id) { |
|
if (nodesMap[currentNode.data.start_node_id]?.type !== CUSTOM_ITERATION_START_NODE) |
|
iterationNodesWithStartNode.push(currentNode) |
|
} |
|
else { |
|
iterationNodesWithoutStartNode.push(currentNode) |
|
} |
|
} |
|
} |
|
const newIterationStartNodesMap = {} as Record<string, Node> |
|
const newIterationStartNodes = [...iterationNodesWithStartNode, ...iterationNodesWithoutStartNode].map((iterationNode, index) => { |
|
const newNode = getIterationStartNode(iterationNode.id) |
|
newNode.id = newNode.id + index |
|
newIterationStartNodesMap[iterationNode.id] = newNode |
|
return newNode |
|
}) |
|
const newEdges = iterationNodesWithStartNode.map((iterationNode) => { |
|
const newNode = newIterationStartNodesMap[iterationNode.id] |
|
const startNode = nodesMap[iterationNode.data.start_node_id] |
|
const source = newNode.id |
|
const sourceHandle = 'source' |
|
const target = startNode.id |
|
const targetHandle = 'target' |
|
return { |
|
id: `${source}-${sourceHandle}-${target}-${targetHandle}`, |
|
type: 'custom', |
|
source, |
|
sourceHandle, |
|
target, |
|
targetHandle, |
|
data: { |
|
sourceType: newNode.data.type, |
|
targetType: startNode.data.type, |
|
isInIteration: true, |
|
iteration_id: startNode.parentId, |
|
_connectedNodeIsSelected: true, |
|
}, |
|
zIndex: ITERATION_CHILDREN_Z_INDEX, |
|
} |
|
}) |
|
nodes.forEach((node) => { |
|
if (node.data.type === BlockEnum.Iteration && newIterationStartNodesMap[node.id]) |
|
(node.data as IterationNodeType).start_node_id = newIterationStartNodesMap[node.id].id |
|
}) |
|
|
|
return { |
|
nodes: [...nodes, ...newIterationStartNodes], |
|
edges: [...edges, ...newEdges], |
|
} |
|
} |
|
|
|
export const initialNodes = (originNodes: Node[], originEdges: Edge[]) => { |
|
const { nodes, edges } = preprocessNodesAndEdges(cloneDeep(originNodes), cloneDeep(originEdges)) |
|
const firstNode = nodes[0] |
|
|
|
if (!firstNode?.position) { |
|
nodes.forEach((node, index) => { |
|
node.position = { |
|
x: START_INITIAL_POSITION.x + index * NODE_WIDTH_X_OFFSET, |
|
y: START_INITIAL_POSITION.y, |
|
} |
|
}) |
|
} |
|
|
|
const iterationNodeMap = nodes.reduce((acc, node) => { |
|
if (node.parentId) { |
|
if (acc[node.parentId]) |
|
acc[node.parentId].push(node.id) |
|
else |
|
acc[node.parentId] = [node.id] |
|
} |
|
return acc |
|
}, {} as Record<string, string[]>) |
|
|
|
return nodes.map((node) => { |
|
if (!node.type) |
|
node.type = CUSTOM_NODE |
|
|
|
const connectedEdges = getConnectedEdges([node], edges) |
|
node.data._connectedSourceHandleIds = connectedEdges.filter(edge => edge.source === node.id).map(edge => edge.sourceHandle || 'source') |
|
node.data._connectedTargetHandleIds = connectedEdges.filter(edge => edge.target === node.id).map(edge => edge.targetHandle || 'target') |
|
|
|
if (node.data.type === BlockEnum.IfElse) { |
|
const nodeData = node.data as IfElseNodeType |
|
|
|
if (!nodeData.cases && nodeData.logical_operator && nodeData.conditions) { |
|
(node.data as IfElseNodeType).cases = [ |
|
{ |
|
case_id: 'true', |
|
logical_operator: nodeData.logical_operator, |
|
conditions: nodeData.conditions, |
|
}, |
|
] |
|
} |
|
node.data._targetBranches = branchNameCorrect([ |
|
...(node.data as IfElseNodeType).cases.map(item => ({ id: item.case_id, name: '' })), |
|
{ id: 'false', name: '' }, |
|
]) |
|
} |
|
|
|
if (node.data.type === BlockEnum.QuestionClassifier) { |
|
node.data._targetBranches = (node.data as QuestionClassifierNodeType).classes.map((topic) => { |
|
return topic |
|
}) |
|
} |
|
|
|
if (node.data.type === BlockEnum.Iteration) { |
|
const iterationNodeData = node.data as IterationNodeType |
|
iterationNodeData._children = iterationNodeMap[node.id] || [] |
|
iterationNodeData.is_parallel = iterationNodeData.is_parallel || false |
|
iterationNodeData.parallel_nums = iterationNodeData.parallel_nums || 10 |
|
iterationNodeData.error_handle_mode = iterationNodeData.error_handle_mode || ErrorHandleMode.Terminated |
|
} |
|
|
|
return node |
|
}) |
|
} |
|
|
|
export const initialEdges = (originEdges: Edge[], originNodes: Node[]) => { |
|
const { nodes, edges } = preprocessNodesAndEdges(cloneDeep(originNodes), cloneDeep(originEdges)) |
|
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>) |
|
|
|
const cycleEdges = getCycleEdges(nodes, edges) |
|
return edges.filter((edge) => { |
|
return !cycleEdges.find(cycEdge => cycEdge.source === edge.source && cycEdge.target === edge.target) |
|
}).map((edge) => { |
|
edge.type = 'custom' |
|
|
|
if (!edge.sourceHandle) |
|
edge.sourceHandle = 'source' |
|
|
|
if (!edge.targetHandle) |
|
edge.targetHandle = 'target' |
|
|
|
if (!edge.data?.sourceType && edge.source && nodesMap[edge.source]) { |
|
edge.data = { |
|
...edge.data, |
|
sourceType: nodesMap[edge.source].data.type!, |
|
} as any |
|
} |
|
|
|
if (!edge.data?.targetType && edge.target && nodesMap[edge.target]) { |
|
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 |
|
}) |
|
} |
|
|
|
export const getLayoutByDagre = (originNodes: Node[], originEdges: Edge[]) => { |
|
const dagreGraph = new dagre.graphlib.Graph() |
|
dagreGraph.setDefaultEdgeLabel(() => ({})) |
|
const nodes = cloneDeep(originNodes).filter(node => !node.parentId && node.type === CUSTOM_NODE) |
|
const edges = cloneDeep(originEdges).filter(edge => !edge.data?.isInIteration) |
|
dagreGraph.setGraph({ |
|
rankdir: 'LR', |
|
align: 'UL', |
|
nodesep: 40, |
|
ranksep: 60, |
|
ranker: 'tight-tree', |
|
marginx: 30, |
|
marginy: 200, |
|
}) |
|
nodes.forEach((node) => { |
|
dagreGraph.setNode(node.id, { |
|
width: node.width!, |
|
height: node.height!, |
|
}) |
|
}) |
|
|
|
edges.forEach((edge) => { |
|
dagreGraph.setEdge(edge.source, edge.target) |
|
}) |
|
|
|
dagre.layout(dagreGraph) |
|
|
|
return dagreGraph |
|
} |
|
|
|
export const canRunBySingle = (nodeType: BlockEnum) => { |
|
return nodeType === BlockEnum.LLM |
|
|| nodeType === BlockEnum.KnowledgeRetrieval |
|
|| nodeType === BlockEnum.Code |
|
|| nodeType === BlockEnum.TemplateTransform |
|
|| nodeType === BlockEnum.QuestionClassifier |
|
|| nodeType === BlockEnum.HttpRequest |
|
|| nodeType === BlockEnum.Tool |
|
|| nodeType === BlockEnum.ParameterExtractor |
|
|| nodeType === BlockEnum.Iteration |
|
} |
|
|
|
type ConnectedSourceOrTargetNodesChange = { |
|
type: string |
|
edge: Edge |
|
}[] |
|
export const getNodesConnectedSourceOrTargetHandleIdsMap = (changes: ConnectedSourceOrTargetNodesChange, nodes: Node[]) => { |
|
const nodesConnectedSourceOrTargetHandleIdsMap = {} as Record<string, any> |
|
|
|
changes.forEach((change) => { |
|
const { |
|
edge, |
|
type, |
|
} = change |
|
const sourceNode = nodes.find(node => node.id === edge.source)! |
|
if (sourceNode) { |
|
nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id] = nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id] || { |
|
_connectedSourceHandleIds: [...(sourceNode?.data._connectedSourceHandleIds || [])], |
|
_connectedTargetHandleIds: [...(sourceNode?.data._connectedTargetHandleIds || [])], |
|
} |
|
} |
|
|
|
const targetNode = nodes.find(node => node.id === edge.target)! |
|
if (targetNode) { |
|
nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] = nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id] || { |
|
_connectedSourceHandleIds: [...(targetNode?.data._connectedSourceHandleIds || [])], |
|
_connectedTargetHandleIds: [...(targetNode?.data._connectedTargetHandleIds || [])], |
|
} |
|
} |
|
|
|
if (sourceNode) { |
|
if (type === 'remove') { |
|
const index = nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id]._connectedSourceHandleIds.findIndex((handleId: string) => handleId === edge.sourceHandle) |
|
nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id]._connectedSourceHandleIds.splice(index, 1) |
|
} |
|
|
|
if (type === 'add') |
|
nodesConnectedSourceOrTargetHandleIdsMap[sourceNode.id]._connectedSourceHandleIds.push(edge.sourceHandle || 'source') |
|
} |
|
|
|
if (targetNode) { |
|
if (type === 'remove') { |
|
const index = nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id]._connectedTargetHandleIds.findIndex((handleId: string) => handleId === edge.targetHandle) |
|
nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id]._connectedTargetHandleIds.splice(index, 1) |
|
} |
|
|
|
if (type === 'add') |
|
nodesConnectedSourceOrTargetHandleIdsMap[targetNode.id]._connectedTargetHandleIds.push(edge.targetHandle || 'target') |
|
} |
|
}) |
|
|
|
return nodesConnectedSourceOrTargetHandleIdsMap |
|
} |
|
|
|
export const genNewNodeTitleFromOld = (oldTitle: string) => { |
|
const regex = /^(.+?)\s*\((\d+)\)\s*$/ |
|
const match = oldTitle.match(regex) |
|
|
|
if (match) { |
|
const title = match[1] |
|
const num = parseInt(match[2], 10) |
|
return `${title} (${num + 1})` |
|
} |
|
else { |
|
return `${oldTitle} (1)` |
|
} |
|
} |
|
|
|
export const getValidTreeNodes = (nodes: Node[], edges: Edge[]) => { |
|
const startNode = nodes.find(node => node.data.type === BlockEnum.Start) |
|
|
|
if (!startNode) { |
|
return { |
|
validNodes: [], |
|
maxDepth: 0, |
|
} |
|
} |
|
|
|
const list: Node[] = [startNode] |
|
let maxDepth = 1 |
|
|
|
const traverse = (root: Node, depth: number) => { |
|
if (depth > maxDepth) |
|
maxDepth = depth |
|
|
|
const outgoers = getOutgoers(root, nodes, edges) |
|
|
|
if (outgoers.length) { |
|
outgoers.forEach((outgoer) => { |
|
list.push(outgoer) |
|
if (outgoer.data.type === BlockEnum.Iteration) |
|
list.push(...nodes.filter(node => node.parentId === outgoer.id)) |
|
traverse(outgoer, depth + 1) |
|
}) |
|
} |
|
else { |
|
list.push(root) |
|
if (root.data.type === BlockEnum.Iteration) |
|
list.push(...nodes.filter(node => node.parentId === root.id)) |
|
} |
|
} |
|
|
|
traverse(startNode, maxDepth) |
|
|
|
return { |
|
validNodes: uniqBy(list, 'id'), |
|
maxDepth, |
|
} |
|
} |
|
|
|
export const getToolCheckParams = ( |
|
toolData: ToolNodeType, |
|
buildInTools: ToolWithProvider[], |
|
customTools: ToolWithProvider[], |
|
workflowTools: ToolWithProvider[], |
|
language: string, |
|
) => { |
|
const { provider_id, provider_type, tool_name } = toolData |
|
const isBuiltIn = provider_type === CollectionType.builtIn |
|
const currentTools = provider_type === CollectionType.builtIn ? buildInTools : provider_type === CollectionType.custom ? customTools : workflowTools |
|
const currCollection = currentTools.find(item => item.id === provider_id) |
|
const currTool = currCollection?.tools.find(tool => tool.name === tool_name) |
|
const formSchemas = currTool ? toolParametersToFormSchemas(currTool.parameters) : [] |
|
const toolInputVarSchema = formSchemas.filter((item: any) => item.form === 'llm') |
|
const toolSettingSchema = formSchemas.filter((item: any) => item.form !== 'llm') |
|
|
|
return { |
|
toolInputsSchema: (() => { |
|
const formInputs: InputVar[] = [] |
|
toolInputVarSchema.forEach((item: any) => { |
|
formInputs.push({ |
|
label: item.label[language] || item.label.en_US, |
|
variable: item.variable, |
|
type: item.type, |
|
required: item.required, |
|
}) |
|
}) |
|
return formInputs |
|
})(), |
|
notAuthed: isBuiltIn && !!currCollection?.allow_delete && !currCollection?.is_team_authorization, |
|
toolSettingSchema, |
|
language, |
|
} |
|
} |
|
|
|
export const changeNodesAndEdgesId = (nodes: Node[], edges: Edge[]) => { |
|
const idMap = nodes.reduce((acc, node) => { |
|
acc[node.id] = uuid4() |
|
|
|
return acc |
|
}, {} as Record<string, string>) |
|
|
|
const newNodes = nodes.map((node) => { |
|
return { |
|
...node, |
|
id: idMap[node.id], |
|
} |
|
}) |
|
|
|
const newEdges = edges.map((edge) => { |
|
return { |
|
...edge, |
|
source: idMap[edge.source], |
|
target: idMap[edge.target], |
|
} |
|
}) |
|
|
|
return [newNodes, newEdges] as [Node[], Edge[]] |
|
} |
|
|
|
export const isMac = () => { |
|
return navigator.userAgent.toUpperCase().includes('MAC') |
|
} |
|
|
|
const specialKeysNameMap: Record<string, string | undefined> = { |
|
ctrl: '⌘', |
|
alt: '⌥', |
|
} |
|
|
|
export const getKeyboardKeyNameBySystem = (key: string) => { |
|
if (isMac()) |
|
return specialKeysNameMap[key] || key |
|
|
|
return key |
|
} |
|
|
|
const specialKeysCodeMap: Record<string, string | undefined> = { |
|
ctrl: 'meta', |
|
} |
|
|
|
export const getKeyboardKeyCodeBySystem = (key: string) => { |
|
if (isMac()) |
|
return specialKeysCodeMap[key] || key |
|
|
|
return key |
|
} |
|
|
|
export const getTopLeftNodePosition = (nodes: Node[]) => { |
|
let minX = Infinity |
|
let minY = Infinity |
|
|
|
nodes.forEach((node) => { |
|
if (node.position.x < minX) |
|
minX = node.position.x |
|
|
|
if (node.position.y < minY) |
|
minY = node.position.y |
|
}) |
|
|
|
return { |
|
x: minX, |
|
y: minY, |
|
} |
|
} |
|
|
|
export const isEventTargetInputArea = (target: HTMLElement) => { |
|
if (target.tagName === 'INPUT' || target.tagName === 'TEXTAREA') |
|
return true |
|
|
|
if (target.contentEditable === 'true') |
|
return true |
|
} |
|
|
|
export const variableTransformer = (v: ValueSelector | string) => { |
|
if (typeof v === 'string') |
|
return v.replace(/^{{#|#}}$/g, '').split('.') |
|
|
|
return `{{#${v.join('.')}#}}` |
|
} |
|
|
|
type ParallelInfoItem = { |
|
parallelNodeId: string |
|
depth: number |
|
isBranch?: boolean |
|
} |
|
type NodeParallelInfo = { |
|
parallelNodeId: string |
|
edgeHandleId: string |
|
depth: number |
|
} |
|
type NodeHandle = { |
|
node: Node |
|
handle: string |
|
} |
|
type NodeStreamInfo = { |
|
upstreamNodes: Set<string> |
|
downstreamEdges: Set<string> |
|
} |
|
export const getParallelInfo = (nodes: Node[], edges: Edge[], parentNodeId?: string) => { |
|
let startNode |
|
|
|
if (parentNodeId) { |
|
const parentNode = nodes.find(node => node.id === parentNodeId) |
|
if (!parentNode) |
|
throw new Error('Parent node not found') |
|
|
|
startNode = nodes.find(node => node.id === (parentNode.data as IterationNodeType).start_node_id) |
|
} |
|
else { |
|
startNode = nodes.find(node => node.data.type === BlockEnum.Start) |
|
} |
|
if (!startNode) |
|
throw new Error('Start node not found') |
|
|
|
const parallelList = [] as ParallelInfoItem[] |
|
const nextNodeHandles = [{ node: startNode, handle: 'source' }] |
|
let hasAbnormalEdges = false |
|
|
|
const traverse = (firstNodeHandle: NodeHandle) => { |
|
const nodeEdgesSet = {} as Record<string, Set<string>> |
|
const totalEdgesSet = new Set<string>() |
|
const nextHandles = [firstNodeHandle] |
|
const streamInfo = {} as Record<string, NodeStreamInfo> |
|
const parallelListItem = { |
|
parallelNodeId: '', |
|
depth: 0, |
|
} as ParallelInfoItem |
|
const nodeParallelInfoMap = {} as Record<string, NodeParallelInfo> |
|
nodeParallelInfoMap[firstNodeHandle.node.id] = { |
|
parallelNodeId: '', |
|
edgeHandleId: '', |
|
depth: 0, |
|
} |
|
|
|
while (nextHandles.length) { |
|
const currentNodeHandle = nextHandles.shift()! |
|
const { node: currentNode, handle: currentHandle = 'source' } = currentNodeHandle |
|
const currentNodeHandleKey = currentNode.id |
|
const connectedEdges = edges.filter(edge => edge.source === currentNode.id && edge.sourceHandle === currentHandle) |
|
const connectedEdgesLength = connectedEdges.length |
|
const outgoers = nodes.filter(node => connectedEdges.some(edge => edge.target === node.id)) |
|
const incomers = getIncomers(currentNode, nodes, edges) |
|
|
|
if (!streamInfo[currentNodeHandleKey]) { |
|
streamInfo[currentNodeHandleKey] = { |
|
upstreamNodes: new Set<string>(), |
|
downstreamEdges: new Set<string>(), |
|
} |
|
} |
|
|
|
if (nodeEdgesSet[currentNodeHandleKey]?.size > 0 && incomers.length > 1) { |
|
const newSet = new Set<string>() |
|
for (const item of totalEdgesSet) { |
|
if (!streamInfo[currentNodeHandleKey].downstreamEdges.has(item)) |
|
newSet.add(item) |
|
} |
|
if (isEqual(nodeEdgesSet[currentNodeHandleKey], newSet)) { |
|
parallelListItem.depth = nodeParallelInfoMap[currentNode.id].depth |
|
nextNodeHandles.push({ node: currentNode, handle: currentHandle }) |
|
break |
|
} |
|
} |
|
|
|
if (nodeParallelInfoMap[currentNode.id].depth > parallelListItem.depth) |
|
parallelListItem.depth = nodeParallelInfoMap[currentNode.id].depth |
|
|
|
outgoers.forEach((outgoer) => { |
|
const outgoerConnectedEdges = getConnectedEdges([outgoer], edges).filter(edge => edge.source === outgoer.id) |
|
const sourceEdgesGroup = groupBy(outgoerConnectedEdges, 'sourceHandle') |
|
const incomers = getIncomers(outgoer, nodes, edges) |
|
|
|
if (outgoers.length > 1 && incomers.length > 1) |
|
hasAbnormalEdges = true |
|
|
|
Object.keys(sourceEdgesGroup).forEach((sourceHandle) => { |
|
nextHandles.push({ node: outgoer, handle: sourceHandle }) |
|
}) |
|
if (!outgoerConnectedEdges.length) |
|
nextHandles.push({ node: outgoer, handle: 'source' }) |
|
|
|
const outgoerKey = outgoer.id |
|
if (!nodeEdgesSet[outgoerKey]) |
|
nodeEdgesSet[outgoerKey] = new Set<string>() |
|
|
|
if (nodeEdgesSet[currentNodeHandleKey]) { |
|
for (const item of nodeEdgesSet[currentNodeHandleKey]) |
|
nodeEdgesSet[outgoerKey].add(item) |
|
} |
|
|
|
if (!streamInfo[outgoerKey]) { |
|
streamInfo[outgoerKey] = { |
|
upstreamNodes: new Set<string>(), |
|
downstreamEdges: new Set<string>(), |
|
} |
|
} |
|
|
|
if (!nodeParallelInfoMap[outgoer.id]) { |
|
nodeParallelInfoMap[outgoer.id] = { |
|
...nodeParallelInfoMap[currentNode.id], |
|
} |
|
} |
|
|
|
if (connectedEdgesLength > 1) { |
|
const edge = connectedEdges.find(edge => edge.target === outgoer.id)! |
|
nodeEdgesSet[outgoerKey].add(edge.id) |
|
totalEdgesSet.add(edge.id) |
|
|
|
streamInfo[currentNodeHandleKey].downstreamEdges.add(edge.id) |
|
streamInfo[outgoerKey].upstreamNodes.add(currentNodeHandleKey) |
|
|
|
for (const item of streamInfo[currentNodeHandleKey].upstreamNodes) |
|
streamInfo[item].downstreamEdges.add(edge.id) |
|
|
|
if (!parallelListItem.parallelNodeId) |
|
parallelListItem.parallelNodeId = currentNode.id |
|
|
|
const prevDepth = nodeParallelInfoMap[currentNode.id].depth + 1 |
|
const currentDepth = nodeParallelInfoMap[outgoer.id].depth |
|
|
|
nodeParallelInfoMap[outgoer.id].depth = Math.max(prevDepth, currentDepth) |
|
} |
|
else { |
|
for (const item of streamInfo[currentNodeHandleKey].upstreamNodes) |
|
streamInfo[outgoerKey].upstreamNodes.add(item) |
|
|
|
nodeParallelInfoMap[outgoer.id].depth = nodeParallelInfoMap[currentNode.id].depth |
|
} |
|
}) |
|
} |
|
|
|
parallelList.push(parallelListItem) |
|
} |
|
|
|
while (nextNodeHandles.length) { |
|
const nodeHandle = nextNodeHandles.shift()! |
|
traverse(nodeHandle) |
|
} |
|
|
|
return { |
|
parallelList, |
|
hasAbnormalEdges, |
|
} |
|
} |
|
|