@flowgram.ai/free-auto-layout-plugin 0.1.26 → 0.1.28

package/dist/esm/index.js

@@ -2085,10 +2085,9 @@ import {
   WorkflowDocument,
   WorkflowNodeLinesData
 } from "@flowgram.ai/free-layout-core";
-import { FlowNodeBaseType } from "@flowgram.ai/document";
 
 // src/layout/store.ts
-import { FlowNodeTransformData } from "@flowgram.ai/document";
+import { FlowNodeBaseType, FlowNodeTransformData } from "@flowgram.ai/document";
 var LayoutStore = class {
   constructor() {
     this.init = false;
@@ -2120,11 +2119,20 @@ var LayoutStore = class {
   /** 创建布局数据 */
   createStore(params) {
     const { nodes, edges } = params;
+    const layoutNodes = this.createLayoutNodes(nodes);
+    const layoutEdges = this.createEdgesStore(edges);
+    const virtualEdges = this.createVirtualEdges(params);
     const store = {
       nodes: /* @__PURE__ */ new Map(),
       edges: /* @__PURE__ */ new Map()
     };
-    nodes.forEach((node, index) => {
+    layoutNodes.forEach((node) => store.nodes.set(node.id, node));
+    layoutEdges.concat(virtualEdges).forEach((edge) => store.edges.set(edge.id, edge));
+    return store;
+  }
+  /** 创建节点布局数据 */
+  createLayoutNodes(nodes) {
+    const layoutNodes = nodes.map((node, index) => {
       const { bounds } = node.getData(FlowNodeTransformData);
       const layoutNode = {
         id: node.id,
@@ -2140,9 +2148,13 @@ var LayoutStore = class {
         size: { width: bounds.width, height: bounds.height },
         hasChildren: node.collapsedChildren?.length > 0
       };
-      store.nodes.set(layoutNode.id, layoutNode);
+      return layoutNode;
     });
-    edges.forEach((edge) => {
+    return layoutNodes;
+  }
+  /** 创建线条布局数据 */
+  createEdgesStore(edges) {
+    const layoutEdges = edges.map((edge) => {
       const { from, to } = edge.info;
       if (!from || !to || edge.vertical) {
         return;
@@ -2158,9 +2170,58 @@ var LayoutStore = class {
         // 初始化时，index 未计算
         name: edge.id
       };
-      store.edges.set(layoutEdge.id, layoutEdge);
-    });
-    return store;
+      return layoutEdge;
+    }).filter(Boolean);
+    return layoutEdges;
+  }
+  /** 创建虚拟线条数据 */
+  createVirtualEdges(params) {
+    const { nodes, edges } = params;
+    const groupNodes = nodes.filter((n) => n.flowNodeType === FlowNodeBaseType.GROUP);
+    const virtualEdges = groupNodes.map((group) => {
+      const { id: groupId, blocks = [] } = group;
+      const blockIdSet = new Set(blocks.map((b) => b.id));
+      const groupFromEdges = edges.filter((edge) => blockIdSet.has(edge.to?.id ?? "")).map((edge) => {
+        const { from, to } = edge.info;
+        if (!from || !to || edge.vertical) {
+          return;
+        }
+        const id = `virtual_${groupId}_${to}`;
+        const layoutEdge = {
+          id,
+          entity: edge,
+          from,
+          to: groupId,
+          fromIndex: "",
+          // 初始化时，index 未计算
+          toIndex: "",
+          // 初始化时，index 未计算
+          name: id
+        };
+        return layoutEdge;
+      }).filter(Boolean);
+      const groupToEdges = edges.filter((edge) => blockIdSet.has(edge.from.id ?? "")).map((edge) => {
+        const { from, to } = edge.info;
+        if (!from || !to || edge.vertical) {
+          return;
+        }
+        const id = `virtual_${groupId}_${from}`;
+        const layoutEdge = {
+          id,
+          entity: edge,
+          from: groupId,
+          to,
+          fromIndex: "",
+          // 初始化时，index 未计算
+          toIndex: "",
+          // 初始化时，index 未计算
+          name: id
+        };
+        return layoutEdge;
+      }).filter(Boolean);
+      return [...groupFromEdges, ...groupToEdges];
+    }).flat();
+    return virtualEdges;
   }
   /** 创建节点索引映射 */
   createIndexMap() {
@@ -2498,27 +2559,25 @@ var AutoLayoutService = class {
     await this.layoutNode(this.document.root, options);
   }
   async layoutNode(node, options) {
-    const nodes = this.getAvailableBlocks(node);
+    const nodes = node.blocks;
     if (!nodes || !Array.isArray(nodes) || !nodes.length) {
       return;
     }
-    const edges = node.blocks.map((child) => {
-      const childLinesData = child.getData(WorkflowNodeLinesData);
-      return childLinesData.outputLines.filter(Boolean);
-    }).flat();
+    const edges = this.getNodesAllLines(nodes);
     await Promise.all(nodes.map(async (child) => this.layoutNode(child, options)));
     const layout2 = new Layout();
     layout2.init({ nodes, edges }, options);
     layout2.layout();
     await layout2.position();
   }
-  getAvailableBlocks(node) {
-    const commonNodes = node.blocks.filter((n) => !this.shouldFlatNode(n));
-    const flatNodes = node.blocks.filter((n) => this.shouldFlatNode(n)).map((flatNode) => flatNode.blocks).flat();
-    return [...commonNodes, ...flatNodes];
-  }
-  shouldFlatNode(node) {
-    return node.flowNodeType === FlowNodeBaseType.GROUP;
+  getNodesAllLines(nodes) {
+    const lines = nodes.map((node) => {
+      const linesData = node.getData(WorkflowNodeLinesData);
+      const outputLines = linesData.outputLines.filter(Boolean);
+      const inputLines = linesData.inputLines.filter(Boolean);
+      return [...outputLines, ...inputLines];
+    }).flat();
+    return lines;
   }
 };
 __decorateClass([