@flowgram.ai/free-layout-core 0.1.0-alpha.12 → 0.1.0-alpha.13

package/dist/index.js

@@ -156,8 +156,8 @@ function nanoid(n) {
 var import_utils = require("@flowgram.ai/utils");
 
 // src/utils/fit-view.ts
-var import_core = require("@flowgram.ai/core");
 var import_utils2 = require("@flowgram.ai/utils");
+var import_core = require("@flowgram.ai/core");
 var fitView = (doc, playgroundConfig, easing = true) => {
   const bounds = import_utils2.Rectangle.enlarge(
     doc.getAllNodes().map((node) => node.getData(import_core.TransformData).bounds)
@@ -218,17 +218,18 @@ var import_document3 = require("@flowgram.ai/document");
 var import_core4 = require("@flowgram.ai/core");
 var PORT_SIZE = 24;
 var WorkflowPortEntity = class extends import_core4.Entity {
-  // relativePosition
   constructor(opts) {
     super(opts);
     this.portID = "";
-    this._disabled = false;
     this._hasError = false;
     this._onErrorChangedEmitter = new import_utils5.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(import_core4.TransformData).onDataChange(() => this.fireChange()));
@@ -237,6 +238,9 @@ var WorkflowPortEntity = class extends import_core4.Entity {
   static getPortEntityId(node, portType, portID = "") {
     return getPortEntityId(node, portType, portID);
   }
+  get position() {
+    return this._location;
+  }
   // 获取连线是否为错误态
   get hasError() {
     return this._hasError;
@@ -261,28 +265,65 @@ var WorkflowPortEntity = class extends import_core4.Entity {
   isErrorPort() {
     return this.node.document.isErrorPort(this, this.hasError);
   }
+  get location() {
+    if (this._location) {
+      return this._location;
+    }
+    if (this.portType === "input") {
+      return "left";
+    }
+    return "right";
+  }
   get point() {
     const { targetElement } = this;
     const { bounds } = this.node.getData(import_document3.FlowNodeTransformData);
+    const location2 = this.location;
     if (targetElement) {
       const pos = domReactToBounds(targetElement.getBoundingClientRect()).center;
-      return this.entityManager.getEntity(import_core4.PlaygroundConfigEntity).getPosFromMouseEvent({
+      const point2 = this.entityManager.getEntity(import_core4.PlaygroundConfigEntity).getPosFromMouseEvent({
         clientX: pos.x,
         clientY: pos.y
       });
+      return {
+        x: point2.x,
+        y: point2.y,
+        location: location2
+      };
     }
-    if (this.portType === "input") {
-      return bounds.leftCenter;
+    let point = { x: 0, y: 0 };
+    const offset = this._offset || { x: 0, y: 0 };
+    switch (location2) {
+      case "left":
+        point = bounds.leftCenter;
+        break;
+      case "top":
+        point = bounds.topCenter;
+        break;
+      case "right":
+        point = bounds.rightCenter;
+        break;
+      case "bottom":
+        point = bounds.bottomCenter;
+        break;
     }
-    return bounds.rightCenter;
+    return {
+      x: point.x + offset.x,
+      y: point.y + offset.y,
+      location: location2
+    };
   }
   /**
-   * 点的区域
+   * 端口热区
    */
   get bounds() {
     const { point } = this;
-    const halfSize = PORT_SIZE / 2;
-    return new import_utils5.Rectangle(point.x - halfSize, point.y - halfSize, PORT_SIZE, PORT_SIZE);
+    const size = this._size || { width: PORT_SIZE, height: PORT_SIZE };
+    return new import_utils5.Rectangle(
+      point.x - size.width / 2,
+      point.y - size.height / 2,
+      size.width,
+      size.height
+    );
   }
   isHovered(x, y) {
     return this.bounds.contains(x, y);
@@ -349,6 +390,32 @@ var WorkflowPortEntity = class extends import_core4.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 (import_utils5.Compare.isChanged(data.offset, this._offset)) {
+      this._offset = data.offset;
+      changed = true;
+    }
+    if (import_utils5.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();
@@ -387,17 +454,26 @@ var WorkflowNodePortsData = class extends import_core5.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
    */
@@ -536,9 +612,7 @@ var WorkflowNodePortsData = class extends import_core5.EntityData {
    */
   updatePortEntity(portInfo) {
     const portEntity = this.getOrCreatePortEntity(portInfo);
-    if (portInfo.targetElement) {
-      portEntity.updateTargetElement(portInfo.targetElement);
-    }
+    portEntity.update(portInfo);
     return portEntity;
   }
 };
@@ -679,8 +753,8 @@ var WorkflowLineRenderData = class extends import_core7.EntityData {
       version: "",
       contributions: /* @__PURE__ */ new Map(),
       position: {
-        from: { x: 0, y: 0 },
-        to: { x: 0, y: 0 }
+        from: { x: 0, y: 0, location: "right" },
+        to: { x: 0, y: 0, location: "left" }
       }
     };
   }
@@ -726,16 +800,23 @@ var WorkflowLineRenderData = class extends import_core7.EntityData {
    */
   updatePosition() {
     this.data.position.from = this.entity.from.getData(WorkflowNodePortsData).getOutputPoint(this.entity.info.fromPort);
-    this.data.position.to = this.entity.info.drawingTo ?? this.entity.to?.getData(WorkflowNodePortsData)?.getInputPoint(this.entity.info.toPort) ?? {
-      x: this.data.position.from.x,
-      y: this.data.position.from.y
-    };
+    if (this.entity.info.drawingTo) {
+      this.data.position.to = this.entity.info.drawingTo;
+    } else {
+      this.data.position.to = this.entity.to?.getData(WorkflowNodePortsData)?.getInputPoint(this.entity.info.toPort) ?? {
+        x: this.data.position.from.x,
+        y: this.data.position.from.y,
+        location: this.data.position.from.location === "right" ? "left" : "top"
+      };
+    }
     this.data.version = [
       this.lineType,
       this.data.position.from.x,
       this.data.position.from.y,
+      this.data.position.from.location,
       this.data.position.to.x,
-      this.data.position.to.y
+      this.data.position.to.y,
+      this.data.position.to.location
     ].join("-");
   }
   get currentLine() {
@@ -765,13 +846,16 @@ var POINT_RADIUS = 10;
 var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity {
   constructor(opts) {
     super(opts);
+    this._onLineDataChangeEmitter = new import_utils8.Emitter();
+    this.onLineDataChange = this._onLineDataChangeEmitter.event;
     this._uiState = {
       hasError: false,
       flowing: false,
       disabled: false,
-      vertical: false,
       hideArrow: false,
       reverse: false,
+      shrink: 10,
+      curvature: 0.25,
       highlightColor: "",
       lockedColor: ""
     };
@@ -789,7 +873,8 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
       to: opts.to,
       drawingTo: opts.drawingTo,
       fromPort: opts.fromPort,
-      toPort: opts.toPort
+      toPort: opts.toPort,
+      data: opts.data
     });
     if (opts.drawingTo) {
       this.isDrawing = true;
@@ -802,6 +887,7 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
       this.fromPort?.validate();
       this.toPort?.validate();
     });
+    this.toDispose.push(this._onLineDataChangeEmitter);
   }
   /**
    * 转成线条 id
@@ -844,9 +930,13 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
    * 更新线条扩展数据
    * @param data
    */
-  set lineData(data) {
-    this._lineData = data;
-    this.fireChange();
+  set lineData(newValue) {
+    const oldValue = this._lineData;
+    if (!(0, import_lodash_es2.isEqual)(oldValue, newValue)) {
+      this._lineData = newValue;
+      this._onLineDataChangeEmitter.fire({ oldValue, newValue });
+      this.fireChange();
+    }
   }
   /**
    * 获取线条的前置节点
@@ -869,20 +959,17 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
   }
   /**
    * 获取是否 testrun processing
-   * @deprecated  use `uiState.flowing` instead
+   * @deprecated  use `flowing` instead
    */
   get processing() {
     return this._uiState.flowing;
   }
   /**
    * 设置 testrun processing 状态
-   * @deprecated  use `uiState.flowing` instead
+   * @deprecated  use `flowing` instead
    */
   set processing(status) {
-    if (this._uiState.flowing !== status) {
-      this._uiState.flowing = status;
-      this.fireChange();
-    }
+    this.flowing = status;
   }
   // 获取连线是否为错误态
   get hasError() {
@@ -912,7 +999,6 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
       const { node, portID } = toPort;
       this._to = node;
       this.info.drawingTo = void 0;
-      this.info.isDefaultLine = false;
       this.info.to = node.id;
       this.info.toPort = portID;
     } else {
@@ -937,7 +1023,6 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
     }
     if (!oldDrawingTo || pos.x !== oldDrawingTo.x || pos.y !== oldDrawingTo.y) {
       this.info.to = void 0;
-      this.info.isDefaultLine = false;
       this.info.drawingTo = pos;
       this.fireChange();
     }
@@ -1003,13 +1088,25 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
   get flowing() {
     return this.linesManager.isFlowingLine(this, this.uiState.flowing);
   }
+  set flowing(flowing) {
+    if (this._uiState.flowing !== flowing) {
+      this._uiState.flowing = flowing;
+      this.fireChange();
+    }
+  }
   /** 是否禁用 */
   get disabled() {
     return this.linesManager.isDisabledLine(this, this.uiState.disabled);
   }
   /** 是否竖向 */
   get vertical() {
-    return this.linesManager.isVerticalLine(this, this.uiState.vertical);
+    const fromLocation = this.fromPort.location;
+    const toLocation = this.toPort?.location;
+    if (toLocation) {
+      return toLocation === "top";
+    } else {
+      return fromLocation === "bottom";
+    }
   }
   /** 获取线条渲染器类型 */
   get renderType() {
@@ -1017,7 +1114,7 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
   }
   /** 获取线条样式 */
   get className() {
-    return this.linesManager.setLineClassName(this) ?? "";
+    return [this.linesManager.setLineClassName(this), this._uiState.className].filter((s) => !!s).join(" ");
   }
   get color() {
     return this.linesManager.getLineColor(this);
@@ -1031,6 +1128,7 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends import_core8.Entity
       this.info = info;
       this._from = this.document.getNode(info.from);
       this._to = info.to ? this.document.getNode(info.to) : void 0;
+      this._lineData = info.data;
       this.fireChange();
     }
   }
@@ -1287,6 +1385,7 @@ var WorkflowContentChangeType = /* @__PURE__ */ ((WorkflowContentChangeType2) =>
   WorkflowContentChangeType2["NODE_DATA_CHANGE"] = "NODE_DATA_CHANGE";
   WorkflowContentChangeType2["ADD_LINE"] = "ADD_LINE";
   WorkflowContentChangeType2["DELETE_LINE"] = "DELETE_LINE";
+  WorkflowContentChangeType2["LINE_DATA_CHANGE"] = "LINE_DATA_CHANGE";
   WorkflowContentChangeType2["META_CHANGE"] = "META_CHANGE";
   return WorkflowContentChangeType2;
 })(WorkflowContentChangeType || {});
@@ -1460,6 +1559,9 @@ var WorkflowLinesManager = class {
       WorkflowLineEntity.portInfoToLineId(portInfo)
     );
   }
+  getLineById(id) {
+    return this.entityManager.getEntityById(id);
+  }
   replaceLine(oldPortInfo, newPortInfo) {
     const oldLine = this.getLine(oldPortInfo);
     if (oldLine) {
@@ -1468,7 +1570,7 @@ var WorkflowLinesManager = class {
     return this.createLine(newPortInfo);
   }
   createLine(options) {
-    const { from, to, drawingTo, fromPort, toPort } = options;
+    const { from, to, drawingTo, fromPort, toPort, data } = options;
     const available = Boolean(from && to);
     const key = options.key || WorkflowLineEntity.portInfoToLineId(options);
     let line = this.entityManager.getEntityById(key);
@@ -1491,7 +1593,8 @@ var WorkflowLinesManager = class {
       fromPort,
       toPort,
       to,
-      drawingTo
+      drawingTo,
+      data
     });
     this.registerData(line);
     fromNode.addLine(line);
@@ -1512,6 +1615,14 @@ var WorkflowLinesManager = class {
         });
       }
     });
+    line.onLineDataChange(({ oldValue }) => {
+      this.onAvailableLinesChangeEmitter.fire({
+        type: "LINE_DATA_CHANGE" /* LINE_DATA_CHANGE */,
+        toJSON: () => line.toJSON(),
+        oldValue,
+        entity: line
+      });
+    });
     if (available) {
       this.onAvailableLinesChangeEmitter.fire({
         type: "ADD_LINE" /* ADD_LINE */,
@@ -1575,12 +1686,6 @@ var WorkflowLinesManager = class {
     }
     return defaultValue;
   }
-  isVerticalLine(line, defaultValue = false) {
-    if (this.options.isVerticalLine) {
-      return this.options.isVerticalLine(line);
-    }
-    return defaultValue;
-  }
   setLineRenderType(line) {
     if (this.options.setLineRenderType) {
       return this.options.setLineRenderType(line);
@@ -1624,6 +1729,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);
     }
@@ -1887,6 +2000,11 @@ var WorkflowDocument = class extends import_document8.FlowDocument {
   get loading() {
     return this._loading;
   }
+  /**
+   * use `ctx.tools.fitView()` instead
+   * @deprecated
+   * @param easing
+   */
   async fitView(easing) {
     return fitView(this, this.playgroundConfig, easing).then(() => {
       this.linesManager.forceUpdate();
@@ -2194,10 +2312,13 @@ var WorkflowDocument = class extends import_document8.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;
@@ -2211,7 +2332,9 @@ var WorkflowDocument = class extends import_document8.FlowDocument {
       }, []);
       nextNodes.forEach(bfs);
     };
-    bfs(startNodeId);
+    if (startNodeId) {
+      bfs(startNodeId);
+    }
     const associatedNodes = allNode.filter((node) => associatedCache.has(node.id));
     return associatedNodes;
   }
@@ -2333,6 +2456,11 @@ var WorkflowDocument = class extends import_document8.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 ?? [],
@@ -2413,7 +2541,8 @@ var WorkflowDocument = class extends import_document8.FlowDocument {
       from: json.sourceNodeID,
       fromPort: json.sourcePortID,
       to: json.targetNodeID,
-      toPort: json.targetPortID
+      toPort: json.targetPortID,
+      data: json.data
     };
     if (!parentId) {
       return this.linesManager.createLine(lineInfo);
@@ -2880,10 +3009,16 @@ var WorkflowDragService = class {
             originLine.highlightColor = this.linesManager.lineColor.hidden;
           }
           dragSuccess = true;
+          const pos = config.getPosFromMouseEvent(event);
           line = this.linesManager.createLine({
             from: fromPort.node.id,
             fromPort: fromPort.portID,
-            drawingTo: config.getPosFromMouseEvent(event)
+            data: originLine?.lineData,
+            drawingTo: {
+              x: pos.x,
+              y: pos.y,
+              location: fromPort.location === "right" ? "left" : "top"
+            }
           });
           if (!line) {
             return;
@@ -2918,9 +3053,17 @@ var WorkflowDragService = class {
           lineErrorReset = hasError;
         }
         if (line.toPort) {
-          line.drawingTo = { x: line.toPort.point.x, y: line.toPort.point.y };
+          line.drawingTo = {
+            x: line.toPort.point.x,
+            y: line.toPort.point.y,
+            location: line.toPort.location
+          };
         } else {
-          line.drawingTo = { x: dragPos.x, y: dragPos.y };
+          line.drawingTo = {
+            x: dragPos.x,
+            y: dragPos.y,
+            location: line.fromPort.location === "right" ? "left" : "top"
+          };
         }
         originLine?.validate();
         line.validate();
@@ -2964,7 +3107,8 @@ var WorkflowDragService = class {
             from: fromPort.node.id,
             fromPort: fromPort.portID,
             to: toPort.node.id,
-            toPort: toPort.portID
+            toPort: toPort.portID,
+            data: originLine?.lineData
           } : void 0;
           const isReset = originLine && toPort;
           if (isReset && !this.linesManager.canReset(
@@ -3234,6 +3378,7 @@ function usePlaygroundReadonlyState(listenChange) {
 function checkTargetDraggable(el) {
   return el && el.tagName !== "INPUT" && el.tagName !== "TEXTAREA" && !el.closest(".flow-canvas-not-draggable");
 }
+var isFirefox = navigator?.userAgent?.includes?.("Firefox");
 function useNodeRender(nodeFromProps) {
   const node = nodeFromProps || (0, import_react2.useContext)(import_core21.PlaygroundEntityContext);
   const renderData = node.getData(import_document13.FlowNodeRenderData);
@@ -3297,7 +3442,6 @@ function useNodeRender(nodeFromProps) {
   );
   const deleteNode = (0, import_react2.useCallback)(() => node.dispose(), [node]);
   (0, import_core21.useListenEvents)(portsData.onDataChange);
-  const isFirefox = navigator?.userAgent?.includes?.("Firefox");
   const onFocus = (0, import_react2.useCallback)(() => {
     if (isFirefox) {
       nodeRef.current?.setAttribute("draggable", "false");