@flowgram.ai/free-auto-layout-plugin 0.2.9 → 0.2.11

package/dist/esm/index.js

@@ -2092,13 +2092,17 @@ import {
 } from "@flowgram.ai/free-layout-core";
 import { FlowNodeBaseType, FlowNodeTransformData } from "@flowgram.ai/document";
 var LayoutStore = class {
-  constructor() {
+  constructor(config) {
+    this.config = config;
     this.init = false;
   }
   get initialized() {
     return this.init;
   }
   getNode(id) {
+    if (!id) {
+      return void 0;
+    }
     return this.store.nodes.get(id);
   }
   getNodeByIndex(index) {
@@ -2277,8 +2281,13 @@ var LayoutStore = class {
       });
       const { outputLines } = node.getData(WorkflowNodeLinesData);
       const sortedLines = outputLines.sort((a, b) => {
+        const aNode = this.getNode(a.to?.id);
+        const bNode = this.getNode(b.to?.id);
         const aPort = a.fromPort;
         const bPort = b.fromPort;
+        if (aPort === bPort && aNode && bNode) {
+          return aNode.position.y - bNode.position.y;
+        }
         if (aPort && bPort) {
           return aPort.point.y - bPort.point.y;
         }
@@ -2349,16 +2358,6 @@ var LayoutPosition = class {
 // src/layout/dagre.ts
 import { FlowNodeTransformData as FlowNodeTransformData2 } from "@flowgram.ai/document";
 import { Graph as DagreGraph } from "@dagrejs/graphlib";
-
-// src/layout/constant.ts
-var DagreLayoutOptions = {
-  rankdir: "LR",
-  nodesep: 100,
-  ranksep: 100,
-  ranker: "network-simplex"
-};
-
-// src/layout/dagre.ts
 var DagreLayout = class {
   constructor(store) {
     this.store = store;
@@ -2407,7 +2406,7 @@ var DagreLayout = class {
   createGraph() {
     const graph = new DagreGraph({ multigraph: true });
     graph.setDefaultEdgeLabel(() => ({}));
-    graph.setGraph(DagreLayoutOptions);
+    graph.setGraph(this.store.config);
     return graph;
   }
   graphSetData() {
@@ -2529,8 +2528,8 @@ var DagreLayout = class {
 
 // src/layout/layout.ts
 var Layout = class {
-  constructor() {
-    this._store = new LayoutStore();
+  constructor(config) {
+    this._store = new LayoutStore(config);
     this._layout = new DagreLayout(this._store);
     this._position = new LayoutPosition(this._store);
   }
@@ -2567,8 +2566,30 @@ var Layout = class {
   }
 };
 
+// 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
 var AutoLayoutService = class {
+  constructor() {
+    this.layoutConfig = DefaultLayoutConfig;
+  }
+  init(options) {
+    this.layoutConfig = {
+      ...this.layoutConfig,
+      ...options.layoutConfig
+    };
+  }
   async layout(options = {}) {
     await this.layoutNode(this.document.root, options);
   }
@@ -2579,7 +2600,7 @@ var AutoLayoutService = class {
     }
     const edges = this.getNodesAllLines(nodes);
     await Promise.all(nodes.map(async (child) => this.layoutNode(child, options)));
-    const layout2 = new Layout();
+    const layout2 = new Layout(this.layoutConfig);
     layout2.init({ nodes, edges, container: node }, options);
     layout2.layout();
     await layout2.position();
@@ -2605,7 +2626,11 @@ AutoLayoutService = __decorateClass([
 var createFreeAutoLayoutPlugin = definePluginCreator({
   onBind: ({ bind }) => {
     bind(AutoLayoutService).toSelf().inSingletonScope();
-  }
+  },
+  onInit: (ctx, opts) => {
+    ctx.get(AutoLayoutService).init(opts);
+  },
+  singleton: true
 });
 
 // src/index.ts
@@ -2613,6 +2638,7 @@ import { Graph as Graph8 } from "@dagrejs/graphlib";
 export {
   AutoLayoutService,
   Graph8 as DagreGraph,
+  DefaultLayoutConfig,
   Layout,
   createFreeAutoLayoutPlugin,
   dagreLib