@flowgram.ai/free-layout-core 0.3.2 → 0.3.4

package/dist/esm/index.js

@@ -167,7 +167,7 @@ import { FlowNodeRenderData } from "@flowgram.ai/document";
 import { EntityData, SizeData } from "@flowgram.ai/core";
 
 // src/entities/workflow-port-entity.ts
-import { Rectangle as Rectangle3, Emitter } from "@flowgram.ai/utils";
+import { Rectangle as Rectangle3, Emitter, Compare } from "@flowgram.ai/utils";
 import { FlowNodeTransformData } from "@flowgram.ai/document";
 import {
   Entity,
@@ -176,17 +176,18 @@ import {
 } from "@flowgram.ai/core";
 var PORT_SIZE = 24;
 var WorkflowPortEntity = class extends Entity {
-  // relativePosition
   constructor(opts) {
     super(opts);
     this.portID = "";
-    this._disabled = false;
     this._hasError = false;
     this._onErrorChangedEmitter = new Emitter();
     this.onErrorChanged = this._onErrorChangedEmitter.event;
     this.portID = opts.portID || "";
     this.portType = opts.type;
-    this._disabled = opts.disabled ?? false;
+    this._disabled = opts.disabled;
+    this._offset = opts.offset;
+    this._location = opts.location;
+    this._size = opts.size;
     this.node = opts.node;
     this.updateTargetElement(opts.targetElement);
     this.toDispose.push(this.node.getData(TransformData3).onDataChange(() => this.fireChange()));
@@ -195,6 +196,9 @@ var WorkflowPortEntity = class extends Entity {
   static getPortEntityId(node, portType, portID = "") {
     return getPortEntityId(node, portType, portID);
   }
+  get position() {
+    return this._location;
+  }
   // 获取连线是否为错误态
   get hasError() {
     return this._hasError;
@@ -229,18 +233,47 @@ var WorkflowPortEntity = class extends Entity {
         clientY: pos.y
       });
     }
-    if (this.portType === "input") {
-      return bounds.leftCenter;
+    let point = { x: 0, y: 0 };
+    const offset = this._offset || { x: 0, y: 0 };
+    if (this._location) {
+      switch (this._location) {
+        case "left":
+          point = bounds.leftCenter;
+          break;
+        case "top":
+          point = bounds.topCenter;
+          break;
+        case "right":
+          point = bounds.rightCenter;
+          break;
+        case "bottom":
+          point = bounds.bottomCenter;
+          break;
+      }
+    } else {
+      if (this.portType === "input") {
+        point = bounds.leftCenter;
+      } else {
+        point = bounds.rightCenter;
+      }
     }
-    return bounds.rightCenter;
+    return {
+      x: point.x + offset.x,
+      y: point.y + offset.y
+    };
   }
   /**
-   * 点的区域
+   * 端口热区
    */
   get bounds() {
     const { point } = this;
-    const halfSize = PORT_SIZE / 2;
-    return new Rectangle3(point.x - halfSize, point.y - halfSize, PORT_SIZE, PORT_SIZE);
+    const size = this._size || { width: PORT_SIZE, height: PORT_SIZE };
+    return new Rectangle3(
+      point.x - size.width / 2,
+      point.y - size.height / 2,
+      size.width,
+      size.height
+    );
   }
   isHovered(x, y) {
     return this.bounds.contains(x, y);
@@ -307,6 +340,32 @@ var WorkflowPortEntity = class extends Entity {
     });
     return lines;
   }
+  update(data) {
+    let changed = false;
+    if (data.targetElement !== this.targetElement) {
+      this.targetElement = data.targetElement;
+      changed = true;
+    }
+    if (data.location !== this._location) {
+      this._location = data.location;
+      changed = true;
+    }
+    if (Compare.isChanged(data.offset, this._offset)) {
+      this._offset = data.offset;
+      changed = true;
+    }
+    if (Compare.isChanged(data.size, this._size)) {
+      this._size = data.size;
+      changed = true;
+    }
+    if (data.disabled !== this._disabled) {
+      this._disabled = data.disabled;
+      changed = true;
+    }
+    if (changed) {
+      this.fireChange();
+    }
+  }
   dispose() {
     this.lines.forEach((l) => l.dispose());
     super.dispose();
@@ -345,17 +404,26 @@ var WorkflowNodePortsData = class extends EntityData {
     return {};
   }
   /**
-   * 更新静态的 ports 数据
+   * Update all ports data, includes static ports and dynamic ports
+   * @param ports
    */
-  updateStaticPorts(ports) {
+  updateAllPorts(ports) {
     const meta = this.entity.getNodeMeta();
-    this._staticPorts = ports;
+    if (ports) {
+      this._staticPorts = ports;
+    }
     if (meta.useDynamicPort) {
       this.updateDynamicPorts();
     } else {
       this.updatePorts(this._staticPorts);
     }
   }
+  /**
+   * @deprecated use `updateAllPorts` instead
+   */
+  updateStaticPorts(ports) {
+    this.updateAllPorts(ports);
+  }
   /**
    * 动态计算点位，通过 dom 的 data-port-key
    */
@@ -494,9 +562,7 @@ var WorkflowNodePortsData = class extends EntityData {
    */
   updatePortEntity(portInfo) {
     const portEntity = this.getOrCreatePortEntity(portInfo);
-    if (portInfo.targetElement) {
-      portEntity.updateTargetElement(portInfo.targetElement);
-    }
+    portEntity.update(portInfo);
     return portEntity;
   }
 };
@@ -1561,6 +1627,14 @@ var WorkflowLinesManager = class {
     if (fromPort === toPort || fromPort.node === toPort.node || fromPort.portType !== "output" || toPort.portType !== "input" || toPort.disabled) {
       return false;
     }
+    const fromCanAdd = fromPort.node.getNodeRegistry().canAddLine;
+    const toCanAdd = toPort.node.getNodeRegistry().canAddLine;
+    if (fromCanAdd && !fromCanAdd(fromPort, toPort, this, silent)) {
+      return false;
+    }
+    if (toCanAdd && !toCanAdd(fromPort, toPort, this, silent)) {
+      return false;
+    }
     if (this.options.canAddLine) {
       return this.options.canAddLine(fromPort, toPort, this, silent);
     }
@@ -2147,10 +2221,13 @@ var WorkflowDocument = class extends FlowDocument {
       from: line.from.id,
       to: line.to.id
     }));
-    const startNodeId = allNode.find((node) => node.isStart).id;
-    const endNodeId = allNode.find((node) => node.isNodeEnd).id;
+    const startNodeId = allNode.find((node) => node.isStart)?.id;
+    const endNodeId = allNode.find((node) => node.isNodeEnd)?.id;
     const nodeInContainer = allNode.filter((node) => node.parent?.getNodeMeta().isContainer).map((node) => node.id);
-    const associatedCache = /* @__PURE__ */ new Set([endNodeId, ...nodeInContainer]);
+    const associatedCache = new Set(nodeInContainer);
+    if (endNodeId) {
+      associatedCache.add(endNodeId);
+    }
     const bfs = (nodeId) => {
       if (associatedCache.has(nodeId)) {
         return;
@@ -2164,7 +2241,9 @@ var WorkflowDocument = class extends FlowDocument {
       }, []);
       nextNodes.forEach(bfs);
     };
-    bfs(startNodeId);
+    if (startNodeId) {
+      bfs(startNodeId);
+    }
     const associatedNodes = allNode.filter((node) => associatedCache.has(node.id));
     return associatedNodes;
   }
@@ -2286,6 +2365,11 @@ var WorkflowDocument = class extends FlowDocument {
    * 导出数据
    */
   toJSON() {
+    if (this.disposed) {
+      throw new Error(
+        "The WorkflowDocument has been disposed and it is no longer possible to call toJSON."
+      );
+    }
     const rootJSON = this.toNodeJSON(this.root);
     const json = {
       nodes: rootJSON.blocks ?? [],