@flowgram.ai/free-auto-layout-plugin 0.1.25 → 0.1.27

package/dist/esm/index.js

@@ -2085,6 +2085,7 @@ 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";
@@ -2228,7 +2229,6 @@ var LayoutStore = class {
 };
 
 // src/layout/position.ts
-import { FlowNodeTransformData as FlowNodeTransformData2 } from "@flowgram.ai/document";
 import { startTween, TransformData } from "@flowgram.ai/core";
 var LayoutPosition = class {
   constructor(store) {
@@ -2260,23 +2260,19 @@ var LayoutPosition = class {
     };
     const deltaX = (position2.x - transform.position.x) * step / 100;
     const deltaY = (position2.y - transform.bounds.height / 2 - transform.position.y) * step / 100;
-    if (layoutNode.hasChildren) {
-      layoutNode.entity.collapsedChildren.forEach((childNode) => {
-        const childNodeTransformData = childNode.getData(FlowNodeTransformData2);
-        childNodeTransformData.fireChange();
-      });
-    }
     transform.update({
       position: {
         x: transform.position.x + deltaX,
         y: transform.position.y + deltaY
       }
     });
+    const document = layoutNode.entity.document;
+    document.layout.updateAffectedTransform(layoutNode.entity);
   }
 };
 
 // src/layout/dagre.ts
-import { FlowNodeTransformData as FlowNodeTransformData3 } from "@flowgram.ai/document";
+import { FlowNodeTransformData as FlowNodeTransformData2 } from "@flowgram.ai/document";
 import { Graph as DagreGraph } from "@dagrejs/graphlib";
 
 // src/layout/constant.ts
@@ -2388,7 +2384,7 @@ var DagreLayout = class {
     if (!layoutNode.hasChildren) {
       return 0;
     }
-    const nodeTransform = layoutNode.entity.getData(FlowNodeTransformData3);
+    const nodeTransform = layoutNode.entity.getData(FlowNodeTransformData2);
     const { bounds, padding } = nodeTransform;
     const leftOffset = -bounds.width / 2 + padding.left;
     return leftOffset;
@@ -2502,11 +2498,11 @@ var AutoLayoutService = class {
     await this.layoutNode(this.document.root, options);
   }
   async layoutNode(node, options) {
-    const nodes = node.collapsedChildren;
+    const nodes = this.getAvailableBlocks(node);
     if (!nodes || !Array.isArray(nodes) || !nodes.length) {
       return;
     }
-    const edges = node.collapsedChildren.map((child) => {
+    const edges = node.blocks.map((child) => {
       const childLinesData = child.getData(WorkflowNodeLinesData);
       return childLinesData.outputLines.filter(Boolean);
     }).flat();
@@ -2516,6 +2512,14 @@ var AutoLayoutService = class {
     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;
+  }
 };
 __decorateClass([
   inject(WorkflowDocument)