@flowgram.ai/free-layout-core 1.0.10 → 1.0.11

package/dist/esm/index.js

@@ -155,7 +155,8 @@ function getAntiOverlapPosition(doc, position, containerNode) {
 
 // src/utils/statics.ts
 import { Rectangle as Rectangle2 } from "@flowgram.ai/utils";
-var getPortEntityId = (node, portType, portID = "") => `port_${portType}_${node.id}_${portID}`;
+var getPortEntityIdByNodeId = (nodeId, portType, portID = "") => `port_${portType}_${nodeId}_${portID}`;
+var getPortEntityId = (node, portType, portID = "") => getPortEntityIdByNodeId(node.id, portType, portID);
 var WORKFLOW_LINE_ENTITY = "WorkflowLineEntity";
 function domReactToBounds(react) {
   return new Rectangle2(react.x, react.y, react.width, react.height);
@@ -167,7 +168,7 @@ var WorkflowNodeEntity = FlowNodeEntity;
 
 // src/entities/workflow-line-entity.ts
 import { isEqual as isEqual2 } from "lodash-es";
-import { domUtils, Emitter as Emitter2 } from "@flowgram.ai/utils";
+import { domUtils, Emitter as Emitter2, Disposable as Disposable2 } from "@flowgram.ai/utils";
 import { Entity as Entity2 } from "@flowgram.ai/core";
 
 // src/entity-datas/workflow-node-ports-data.ts
@@ -178,11 +179,7 @@ import { EntityData, SizeData } from "@flowgram.ai/core";
 // src/entities/workflow-port-entity.ts
 import { Rectangle as Rectangle3, Emitter, Compare } from "@flowgram.ai/utils";
 import { FlowNodeTransformData } from "@flowgram.ai/document";
-import {
-  Entity,
-  PlaygroundConfigEntity,
-  TransformData as TransformData3
-} from "@flowgram.ai/core";
+import { Entity, PlaygroundConfigEntity, TransformData as TransformData3 } from "@flowgram.ai/core";
 
 // src/utils/location-config-to-point.ts
 function locationConfigToPoint(bounds, config, _offset = { x: 0, y: 0 }) {
@@ -374,16 +371,8 @@ var WorkflowPortEntity = class extends Entity {
    * 当前点位上连接的线条（包含 isDrawing === true 的线条）
    */
   get allLines() {
-    const lines = [];
-    const allLines = this.entityManager.getEntities({
-      type: WORKFLOW_LINE_ENTITY
-    });
-    allLines.forEach((line) => {
-      if (line.toPort === this || line.fromPort === this) {
-        lines.push(line);
-      }
-    });
-    return lines;
+    const document2 = this.node.document;
+    return document2.linesManager.getLinesByPortId(this.id);
   }
   update(data) {
     let changed = false;
@@ -893,6 +882,9 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
       this.toPort?.validate();
     });
     this.toDispose.push(this._onLineDataChangeEmitter);
+    this.preDispose.push(
+      Disposable2.create(() => this.linesManager.rebindLinePorts(this, this.info, void 0))
+    );
   }
   /**
    * 转成线条 id
@@ -1002,19 +994,20 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
     const prePort = this.toPort;
     if (toPort && toPort.portType === "input" && this.linesManager.canAddLine(this.fromPort, toPort, true)) {
       const { node, portID } = toPort;
-      this._to = node;
-      this.info.drawingTo = void 0;
-      this.info.to = node.id;
-      this.info.toPort = portID;
+      this.updateInfo((nextInfo) => {
+        nextInfo.drawingTo = void 0;
+        nextInfo.to = node.id;
+        nextInfo.toPort = portID;
+      });
     } else {
-      this._to = void 0;
-      this.info.to = void 0;
-      this.info.toPort = "";
+      this.updateInfo((nextInfo) => {
+        nextInfo.to = void 0;
+        nextInfo.toPort = "";
+      });
     }
     if (prePort) {
       prePort.validate();
     }
-    this.fireChange();
   }
   setFromPort(fromPort) {
     if (!this.isDrawing) {
@@ -1026,19 +1019,20 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
     const prePort = this.fromPort;
     if (fromPort && fromPort.portType === "output" && this.linesManager.canAddLine(fromPort, this.toPort, true)) {
       const { node, portID } = fromPort;
-      this._from = node;
-      this.info.drawingFrom = void 0;
-      this.info.from = node.id;
-      this.info.fromPort = portID;
+      this.updateInfo((nextInfo) => {
+        nextInfo.drawingFrom = void 0;
+        nextInfo.from = node.id;
+        nextInfo.fromPort = portID;
+      });
     } else {
-      this._from = void 0;
-      this.info.from = void 0;
-      this.info.fromPort = "";
+      this.updateInfo((nextInfo) => {
+        nextInfo.from = void 0;
+        nextInfo.fromPort = "";
+      });
     }
     if (prePort) {
       prePort.validate();
     }
-    this.fireChange();
   }
   /**
    * 设置线条画线时的目标位置
@@ -1046,27 +1040,41 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
   set drawingTo(pos) {
     const oldDrawingTo = this.info.drawingTo;
     if (!pos) {
-      this.info.drawingTo = void 0;
-      this.fireChange();
+      if (oldDrawingTo === void 0) {
+        this.fireChange();
+        return;
+      }
+      this.updateInfo((nextInfo) => {
+        nextInfo.drawingTo = void 0;
+      });
       return;
     }
     if (!oldDrawingTo || pos.x !== oldDrawingTo.x || pos.y !== oldDrawingTo.y) {
-      this.info.to = void 0;
-      this.info.drawingTo = pos;
-      this.fireChange();
+      this.updateInfo((nextInfo) => {
+        nextInfo.to = void 0;
+        nextInfo.toPort = "";
+        nextInfo.drawingTo = pos;
+      });
     }
   }
   set drawingFrom(pos) {
     const oldDrawingFrom = this.info.drawingFrom;
     if (!pos) {
-      this.info.drawingFrom = void 0;
-      this.fireChange();
+      if (oldDrawingFrom === void 0) {
+        this.fireChange();
+        return;
+      }
+      this.updateInfo((nextInfo) => {
+        nextInfo.drawingFrom = void 0;
+      });
       return;
     }
     if (!oldDrawingFrom || pos.x !== oldDrawingFrom.x || pos.y !== oldDrawingFrom.y) {
-      this.info.from = void 0;
-      this.info.drawingFrom = pos;
-      this.fireChange();
+      this.updateInfo((nextInfo) => {
+        nextInfo.from = void 0;
+        nextInfo.fromPort = "";
+        nextInfo.drawingFrom = pos;
+      });
     }
   }
   get drawingFrom() {
@@ -1177,13 +1185,26 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
    * @param info 线条信息
    */
   initInfo(info) {
-    if (!isEqual2(info, this.info)) {
-      this.info = info;
-      this._from = info.from ? this.document.getNode(info.from) : void 0;
-      this._to = info.to ? this.document.getNode(info.to) : void 0;
-      this._lineData = info.data;
-      this.fireChange();
+    this.applyInfo(info);
+  }
+  updateInfo(mutator) {
+    const nextInfo = {
+      ...this.info
+    };
+    mutator(nextInfo);
+    this.applyInfo(nextInfo);
+  }
+  applyInfo(nextInfo) {
+    if (isEqual2(nextInfo, this.info)) {
+      return;
     }
+    const prevInfo = this.info;
+    this.info = nextInfo;
+    this._from = nextInfo.from ? this.document.getNode(nextInfo.from) : void 0;
+    this._to = nextInfo.to ? this.document.getNode(nextInfo.to) : void 0;
+    this._lineData = nextInfo.data;
+    this.linesManager.rebindLinePorts(this, prevInfo, nextInfo);
+    this.fireChange();
   }
   // 校验连线是否为错误态
   validate() {
@@ -1522,6 +1543,7 @@ var WorkflowDocumentOptionsDefault = {
 var WorkflowLinesManager = class {
   constructor() {
     this.toDispose = new DisposableCollection();
+    this.portLineMap = /* @__PURE__ */ new Map();
     // 线条类型
     this._lineType = 0 /* BEZIER */;
     this.onAvailableLinesChangeEmitter = new Emitter4();
@@ -1586,6 +1608,58 @@ var WorkflowLinesManager = class {
   getAllLines() {
     return this.entityManager.getEntities(WorkflowLineEntity);
   }
+  getLinesByPortId(portId) {
+    return Array.from(this.portLineMap.get(portId) || []);
+  }
+  rebindLinePorts(line, prevInfo, nextInfo) {
+    const prevFromPortId = this.getLinePortId(prevInfo, "output");
+    const prevToPortId = this.getLinePortId(prevInfo, "input");
+    const nextFromPortId = this.getLinePortId(nextInfo, "output");
+    const nextToPortId = this.getLinePortId(nextInfo, "input");
+    if (prevFromPortId !== nextFromPortId) {
+      this.detachLineFromPortId(line, prevFromPortId);
+      this.attachLineToPortId(line, nextFromPortId);
+    }
+    if (prevToPortId !== nextToPortId) {
+      this.detachLineFromPortId(line, prevToPortId);
+      this.attachLineToPortId(line, nextToPortId);
+    }
+  }
+  attachLineToPortId(line, portId) {
+    if (!portId) {
+      return;
+    }
+    let lines = this.portLineMap.get(portId);
+    if (!lines) {
+      lines = /* @__PURE__ */ new Set();
+      this.portLineMap.set(portId, lines);
+    }
+    lines.add(line);
+  }
+  detachLineFromPortId(line, portId) {
+    if (!portId) {
+      return;
+    }
+    const lines = this.portLineMap.get(portId);
+    if (!lines) {
+      return;
+    }
+    lines.delete(line);
+    if (!lines.size) {
+      this.portLineMap.delete(portId);
+    }
+  }
+  getLinePortId(info, portType) {
+    if (!info) {
+      return void 0;
+    }
+    const nodeId = portType === "output" ? info.from : info.to;
+    if (!nodeId) {
+      return void 0;
+    }
+    const portId = portType === "output" ? info.fromPort : info.toPort;
+    return getPortEntityIdByNodeId(nodeId, portType, portId);
+  }
   getAllAvailableLines() {
     return this.getAllLines().filter((l) => !l.isDrawing && !l.isHidden);
   }
@@ -1690,6 +1764,7 @@ var WorkflowLinesManager = class {
     return targetLine;
   }
   dispose() {
+    this.portLineMap.clear();
     this.toDispose.dispose();
   }
   get disposed() {
@@ -3991,6 +4066,7 @@ export {
   getAntiOverlapPosition,
   getLineCenter,
   getPortEntityId,
+  getPortEntityIdByNodeId,
   nanoid,
   useConfigEntity,
   useCurrentDomNode,