CMVR-IOT-UI/src/utils/flow.js

export const getInput = (nodeId, isStrict = true) => {
    const data = lf.getGraphData();
    let flag = false
    if (isStrict) {
      data.edges.forEach(item => {
        if (item.targetNodeId === nodeId) {
          flag = true
        }
      });
    } else {
      flag = true
    }
    
    if (flag) {
      const arr = []
      data.nodes.forEach(item => {
        if (item.id === nodeId) {
          return
        }

        if (item.properties.inputParams) {
          const obj = {
              value: item.id,
              label: item.properties.name,
              type: 'input',
              children: []
          }
          const optionData = transformTree(item.properties.inputParams, obj.children, 'input')
          obj.children.push(optionData)
          arr.push(obj)
        }
        if (item.properties.nodeParams) {
          const obj = {
            value: item.id,
            label: item.properties.name + '（输入）',
            type: 'input',
            children: []
          }
          const optionData = transformTree(item.properties.nodeParams, obj.children, 'input')
          obj.children.push(optionData)
          arr.push(obj)
        }

        if (item.properties.outputParams) {
          const obj = {
            value: item.id,
            label: item.properties.name + '（输出）',
            type: 'output',
            children: []
          }
          const optionData = transformTree(item.properties.outputParams, obj.children, 'output')
          obj.children.push(optionData)
          arr.push(obj)
        }
      })
      return arr
    }
    return flag
}

const transformTree = (arr, parent = [], type) => {
  arr.forEach(item => {
    const currentNode = {
      value: item.name,
      label: item.name,
      type,
      children: []
    };
    parent.push(currentNode);
    
    if (item.children?.length > 0) {
      transformTree(item.children, currentNode.children);
    } else {
      delete currentNode.children
    }
  });
  return parent;
};

export const getStartNodeFormData = () => {
  const flowData = lf.getGraphData();
  const startNode = flowData.nodes.find(item =>  item.type === 'start')
  const data = addValueProperty(startNode.properties?.inputParams || [])
  return data
}

export const addValueProperty = (data) => {
  // 处理单个对象或对象数组
  const processItem = (item, index, path) => {
    // 根据类型设置默认值
    item.value = '';
    item.propPath = `${path}.${index}.`
    if (item.type.includes('array') && !item.type.includes('object')) {
      const type = item.type.match(/<([^>]+)>/)[1]
      item.children = [{ name: 'Array Item', type, value: '', propPath:  item.propPath + '.0'}]
    }
    // 递归处理子项
    if (item.children && item.children.length > 0) {
      item.children = item.children.map((cItem, cIndex) => {
        return processItem(cItem, cIndex , item.propPath + 'children')
      });
    }

    return item;
  };

  // 如果传入的是数组，处理每个元素；否则处理单个对象
  return Array.isArray(data) ? data.map((item, index) => processItem(item, index, '')) : processItem(data, 0, '')
}

export const recursiveFilter = (data, id, type = ['loop', 'customGroup'], result = []) => {
    data.forEach((item) => {
        if (type.includes(item.type) && item.children && item.children.includes(id)) {
            result.push(item)
            recursiveFilter(data, item.id, result)
        }
    });
  return result
}

export const formatTableData = (arr, data = {}) => {
  arr.forEach(item => {
    data[item.name] = item.value
    if (item?.children && item.children.length > 0) {
      data[item.name] = {}
      formatTableData(item.children, data[item.name])
    }
  })
  return data
}

export const addNewEdge = (nodeId) => {
  const { edges } = lf.getGraphData();
  const arr = JSON.parse(JSON.stringify(edges))
  arr.forEach(_edge => {
    const { sourceNodeId, targetNodeId, sourceAnchorId, targetAnchorId } = _edge
    if (sourceNodeId === nodeId || targetNodeId === nodeId) {
      lf.deleteEdge(_edge.id);
      lf.addEdge({
        type: "bezier",
        sourceNodeId: sourceNodeId,
        targetNodeId: targetNodeId,
        sourceAnchorId: sourceAnchorId,
        targetAnchorId: targetAnchorId
      });
    }
  })
}

export const getInputNumber = (nodeId) => {
    const data = lf.getGraphData();
    let flag = false
    data.edges.forEach(item => {
      if (item.targetNodeId === nodeId) {
          flag = true
      }
    });
    if (flag) {
      const arr = []
      data.nodes.forEach(item => {
        if (item.properties.inputParams) {
          item.properties.inputParams.forEach(item => {
            if (item.type === 'number') {
              arr.push(item)
            }
          })
        }
      })
      return arr
    }
    return flag
}

export const removeSwitchEdge = (nodeId, sourceAnchorId) => {
  const { edges } = lf.getGraphData();
  const edge = edges.find(item => {
    return item.sourceNodeId === nodeId && item.sourceAnchorId === sourceAnchorId
  })

  if (edge) {
    lf.deleteEdge(edge.id);
  }
}

export const convertToTree = (data) => {
  // 创建 id 到节点的映射
  const nodeMap = new Map();
  data.forEach(node => {
    nodeMap.set(node.id, { ...node, name: node.properties.name || '' }); // 浅拷贝节点
  });

  // 标记非根节点（被其他节点引用的节点）
  const nonRootIds = new Set();
  data.forEach(node => {
    if (node.children) {
      node.children.forEach(childId => {
        nonRootIds.add(childId);
      });
    }
  });

  // 构建树形结构
  const result = [];
  data.forEach(node => {
    // 只处理根节点（未被引用的节点）
    if (!nonRootIds.has(node.id)) {
      const rootNode = nodeMap.get(node.id);
      // 递归处理子节点
      const processNode = (currentNode) => {
        if (currentNode.children) {
          currentNode.children = currentNode.children.map(childId => {
            const childNode = nodeMap.get(childId);
            processNode(childNode); // 递归处理子节点
            return childNode;
          });
        }
        return currentNode;
      };
      
      result.push(processNode(rootNode));
    }
  });

  return result;
}