@flowgram.ai/free-auto-layout-plugin 0.2.16 → 0.2.18

package/dist/esm/index.js

@@ -2086,6 +2086,23 @@ import {
   WorkflowNodeLinesData as WorkflowNodeLinesData2
 } from "@flowgram.ai/free-layout-core";
 
+// src/layout/constant.ts
+var DefaultLayoutConfig = {
+  rankdir: "LR",
+  align: void 0,
+  nodesep: 100,
+  edgesep: 10,
+  ranksep: 100,
+  marginx: 0,
+  marginy: 0,
+  acyclicer: void 0,
+  ranker: "network-simplex"
+};
+var DefaultLayoutOptions = {
+  getFollowNode: void 0,
+  enableAnimation: false
+};
+
 // src/layout/store.ts
 import {
   WorkflowNodeLinesData
@@ -2118,9 +2135,10 @@ var LayoutStore = class {
   get edges() {
     return Array.from(this.store.edges.values());
   }
-  create(params) {
+  create(params, options) {
     this.store = this.createStore(params);
     this.indexMap = this.createIndexMap();
+    this.setOptions(options);
     this.init = true;
   }
   /** 创建布局数据 */
@@ -2310,6 +2328,27 @@ var LayoutStore = class {
     }, []);
     return uniqueNodeIds;
   }
+  /** 记录运行选项 */
+  setOptions(options) {
+    this.options = options;
+    this.setFollowNode(options.getFollowNode);
+  }
+  /** 设置跟随节点配置 */
+  setFollowNode(getFollowNode) {
+    if (!getFollowNode) return;
+    const context = { store: this };
+    this.nodes.forEach((node) => {
+      const followTo = getFollowNode(node, context)?.followTo;
+      if (!followTo) return;
+      const followToNode = this.getNode(followTo);
+      if (!followToNode) return;
+      if (!followToNode.followedBy) {
+        followToNode.followedBy = [];
+      }
+      followToNode.followedBy.push(node.id);
+      node.followTo = followTo;
+    });
+  }
 };
 
 // src/layout/position.ts
@@ -2319,6 +2358,17 @@ var LayoutPosition = class {
     this.store = store;
   }
   async position() {
+    if (this.store.options.enableAnimation) {
+      return this.positionWithAnimation();
+    }
+    return this.positionDirectly();
+  }
+  positionDirectly() {
+    this.store.nodes.forEach((layoutNode) => {
+      this.updateNodePosition({ layoutNode, step: 100 });
+    });
+  }
+  async positionWithAnimation() {
     return new Promise((resolve) => {
       startTween({
         from: { d: 0 },
@@ -2533,9 +2583,8 @@ var Layout = class {
     this._layout = new DagreLayout(this._store);
     this._position = new LayoutPosition(this._store);
   }
-  init(params, options = {}) {
-    this._store.create(params);
-    this.setFollowNode(options.getFollowNode);
+  init(params, options) {
+    this._store.create(params, options);
   }
   layout() {
     if (!this._store.initialized) {
@@ -2549,34 +2598,6 @@ var Layout = class {
     }
     return await this._position.position();
   }
-  setFollowNode(getFollowNode) {
-    if (!getFollowNode) return;
-    const context = { store: this._store };
-    this._store.nodes.forEach((node) => {
-      const followTo = getFollowNode(node, context)?.followTo;
-      if (!followTo) return;
-      const followToNode = this._store.getNode(followTo);
-      if (!followToNode) return;
-      if (!followToNode.followedBy) {
-        followToNode.followedBy = [];
-      }
-      followToNode.followedBy.push(node.id);
-      node.followTo = followTo;
-    });
-  }
-};
-
-// src/layout/constant.ts
-var DefaultLayoutConfig = {
-  rankdir: "LR",
-  align: void 0,
-  nodesep: 100,
-  edgesep: 10,
-  ranksep: 100,
-  marginx: 0,
-  marginy: 0,
-  acyclicer: void 0,
-  ranker: "network-simplex"
 };
 
 // src/services.ts
@@ -2591,7 +2612,10 @@ var AutoLayoutService = class {
     };
   }
   async layout(options = {}) {
-    await this.layoutNode(this.document.root, options);
+    await this.layoutNode(this.document.root, {
+      ...DefaultLayoutOptions,
+      ...options
+    });
   }
   async layoutNode(node, options) {
     const nodes = node.blocks;