@flowgram.ai/free-auto-layout-plugin 0.2.2 → 0.2.3

package/dist/index.d.mts

@@ -146,6 +146,7 @@ declare class LayoutStore {
     private indexMap;
     private init;
     private store;
+    private container;
     get initialized(): boolean;
     getNode(id: string): LayoutNode | undefined;
     getNodeByIndex(index: string): LayoutNode | undefined;
@@ -217,6 +218,7 @@ interface LayoutEdge {
 interface LayoutParams {
     nodes: WorkflowNodeEntity[];
     edges: WorkflowLineEntity[];
+    container: WorkflowNodeEntity;
 }
 interface LayoutOptions {
     getFollowNode?: GetFollowNode;

package/dist/index.d.ts

@@ -146,6 +146,7 @@ declare class LayoutStore {
     private indexMap;
     private init;
     private store;
+    private container;
     get initialized(): boolean;
     getNode(id: string): LayoutNode | undefined;
     getNodeByIndex(index: string): LayoutNode | undefined;
@@ -217,6 +218,7 @@ interface LayoutEdge {
 interface LayoutParams {
     nodes: WorkflowNodeEntity[];
     edges: WorkflowLineEntity[];
+    container: WorkflowNodeEntity;
 }
 interface LayoutOptions {
     getFollowNode?: GetFollowNode;

package/dist/index.js

@@ -2108,9 +2108,10 @@ var import_core2 = require("@flowgram.ai/core");
 
 // src/services.ts
 var import_inversify = require("inversify");
-var import_free_layout_core = require("@flowgram.ai/free-layout-core");
+var import_free_layout_core2 = require("@flowgram.ai/free-layout-core");
 
 // src/layout/store.ts
+var import_free_layout_core = require("@flowgram.ai/free-layout-core");
 var import_document = require("@flowgram.ai/document");
 var LayoutStore = class {
   constructor() {
@@ -2142,7 +2143,8 @@ var LayoutStore = class {
   }
   /** 创建布局数据 */
   createStore(params) {
-    const { nodes, edges } = params;
+    const { nodes, edges, container } = params;
+    this.container = container;
     const layoutNodes = this.createLayoutNodes(nodes);
     const layoutEdges = this.createEdgesStore(edges);
     const virtualEdges = this.createVirtualEdges(params);
@@ -2282,27 +2284,37 @@ var LayoutStore = class {
     this.nodes.forEach((node) => {
       nodeIdList.push(node.id);
     });
-    const sameFromEdges = /* @__PURE__ */ new Map();
     this.edges.forEach((edge) => {
       nodeIdList.push(edge.to);
-      if (edge.entity.info.fromPort) {
-        const edgesForFrom = sameFromEdges.get(edge.from) || [];
-        sameFromEdges.set(edge.from, [...edgesForFrom, edge]);
-      }
     });
-    sameFromEdges.forEach((edges, from) => {
-      const sortedEdges = edges.sort((a, b) => {
-        const aPort = a.entity.fromPort;
-        const bPort = b.entity.fromPort;
+    const visited = /* @__PURE__ */ new Set();
+    const visit = (node) => {
+      if (visited.has(node.id)) {
+        return;
+      }
+      visited.add(node.id);
+      nodeIdList.push(node.id);
+      node.blocks.forEach((child) => {
+        visit(child);
+      });
+      const { outputLines } = node.getData(import_free_layout_core.WorkflowNodeLinesData);
+      const sortedLines = outputLines.sort((a, b) => {
+        const aPort = a.fromPort;
+        const bPort = b.fromPort;
         if (aPort && bPort) {
           return aPort.point.y - bPort.point.y;
         }
         return 0;
       });
-      sortedEdges.forEach((edge) => {
-        nodeIdList.push(edge.to);
+      sortedLines.forEach((line) => {
+        const { to } = line;
+        if (!to) {
+          return;
+        }
+        visit(to);
       });
-    });
+    };
+    visit(this.container);
     const uniqueNodeIds = nodeIdList.reduceRight((acc, nodeId) => {
       if (!acc.includes(nodeId)) {
         acc.unshift(nodeId);
@@ -2590,13 +2602,13 @@ var AutoLayoutService = class {
     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.init({ nodes, edges, container: node }, options);
     layout2.layout();
     await layout2.position();
   }
   getNodesAllLines(nodes) {
     const lines = nodes.map((node) => {
-      const linesData = node.getData(import_free_layout_core.WorkflowNodeLinesData);
+      const linesData = node.getData(import_free_layout_core2.WorkflowNodeLinesData);
       const outputLines = linesData.outputLines.filter(Boolean);
       const inputLines = linesData.inputLines.filter(Boolean);
       return [...outputLines, ...inputLines];
@@ -2605,7 +2617,7 @@ var AutoLayoutService = class {
   }
 };
 __decorateClass([
-  (0, import_inversify.inject)(import_free_layout_core.WorkflowDocument)
+  (0, import_inversify.inject)(import_free_layout_core2.WorkflowDocument)
 ], AutoLayoutService.prototype, "document", 2);
 AutoLayoutService = __decorateClass([
   (0, import_inversify.injectable)()