| 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, |
| } |
| } |
|
|