@flowgram.ai/free-layout-core 0.1.0-alpha.34 → 0.1.0-alpha.36

package/dist/esm/index.js

@@ -318,7 +318,14 @@ var WorkflowPortEntity = class extends Entity {
     );
   }
   isHovered(x, y) {
-    return this.bounds.contains(x, y);
+    const { point } = this;
+    const size = this._size || { width: PORT_SIZE, height: PORT_SIZE };
+    const halfWidth = size.width / 2;
+    const halfHeight = size.height / 2;
+    if (size.width <= 0 || size.height <= 0) {
+      return false;
+    }
+    return x >= point.x - halfWidth && x <= point.x + halfWidth && y >= point.y - halfHeight && y <= point.y + halfHeight;
   }
   /**
    * 相对节点左上角的位置
@@ -419,6 +426,7 @@ var WorkflowNodePortsData = class extends EntityData {
     this._staticPorts = [];
     /** 存储 port 实体的 id，用于判断 port 是否存在 */
     this._portIDSet = /* @__PURE__ */ new Set();
+    this._portDisposeIDSet = /* @__PURE__ */ new Set();
     this.entity = entity;
     const meta = entity.getNodeMeta();
     const defaultPorts = meta.useDynamicPort ? [] : [{ type: "input" }, { type: "output" }];
@@ -505,6 +513,7 @@ var WorkflowNodePortsData = class extends EntityData {
       });
       ports.forEach((port) => this.updatePortEntity(port));
       this._prePorts = ports;
+      this.clearPortsCache();
       this.fireChange();
     }
     this.allPorts.forEach((port) => {
@@ -517,19 +526,28 @@ var WorkflowNodePortsData = class extends EntityData {
    * 获取所有 port entities
    */
   get allPorts() {
-    return Array.from(this._portIDSet).map((portId) => this.getPortEntity(portId)).filter(Boolean);
+    if (!this._allPortsCache) {
+      this._allPortsCache = Array.from(this._portIDSet).map((portId) => this.getPortEntity(portId)).filter(Boolean);
+    }
+    return this._allPortsCache;
   }
   /**
    * 获取输入点位
    */
   get inputPorts() {
-    return this.allPorts.filter((port) => port.portType === "input");
+    if (!this._inputPortsCache) {
+      this._inputPortsCache = this.allPorts.filter((port) => port.portType === "input");
+    }
+    return this._inputPortsCache;
   }
   /**
    * 获取输出点位
    */
   get outputPorts() {
-    return this.allPorts.filter((port) => port.portType === "output");
+    if (!this._outputPortsCache) {
+      this._outputPortsCache = this.allPorts.filter((port) => port.portType === "output");
+    }
+    return this._outputPortsCache;
   }
   /**
    * 获取输入点位置
@@ -570,6 +588,11 @@ var WorkflowNodePortsData = class extends EntityData {
   getPortId(portType, portKey = "") {
     return getPortEntityId(this.entity, portType, portKey);
   }
+  clearPortsCache() {
+    this._allPortsCache = void 0;
+    this._inputPortsCache = void 0;
+    this._outputPortsCache = void 0;
+  }
   /**
    * 创建 port 实体
    */
@@ -583,10 +606,18 @@ var WorkflowNodePortsData = class extends EntityData {
         ...portInfo
       });
     }
-    portEntity.onDispose(() => {
-      this._portIDSet.delete(id);
-    });
-    this._portIDSet.add(id);
+    if (!this._portDisposeIDSet.has(id)) {
+      this._portDisposeIDSet.add(id);
+      portEntity.onDispose(() => {
+        this._portIDSet.delete(id);
+        this._portDisposeIDSet.delete(id);
+        this.clearPortsCache();
+      });
+    }
+    if (!this._portIDSet.has(id)) {
+      this._portIDSet.add(id);
+      this.clearPortsCache();
+    }
     return portEntity;
   }
   /**
@@ -1448,7 +1479,7 @@ import {
   Point
 } from "@flowgram.ai/utils";
 import { FlowNodeTransformData as FlowNodeTransformData6 } from "@flowgram.ai/document";
-import { FlowNodeBaseType as FlowNodeBaseType3 } from "@flowgram.ai/document";
+import { FlowNodeBaseType as FlowNodeBaseType4 } from "@flowgram.ai/document";
 import {
   CommandService,
   MouseTouchEvent,
@@ -1458,10 +1489,9 @@ import {
 } from "@flowgram.ai/core";
 
 // src/workflow-lines-manager.ts
-import { last } from "lodash-es";
 import { inject as inject3, injectable as injectable3 } from "inversify";
 import { DisposableCollection, Emitter as Emitter4 } from "@flowgram.ai/utils";
-import { FlowNodeRenderData as FlowNodeRenderData2, FlowNodeTransformData as FlowNodeTransformData3 } from "@flowgram.ai/document";
+import { FlowNodeBaseType as FlowNodeBaseType2, FlowNodeRenderData as FlowNodeRenderData2, FlowNodeTransformData as FlowNodeTransformData3 } from "@flowgram.ai/document";
 import { EntityManager as EntityManager2, PlaygroundConfigEntity as PlaygroundConfigEntity2 } from "@flowgram.ai/core";
 
 // src/workflow-document-option.ts
@@ -1568,6 +1598,10 @@ var WorkflowLinesManager = class {
   forceUpdate() {
     this.onForceUpdateEmitter.fire();
   }
+  invalidateSortedNodesCache() {
+    this.sortedNodesCache = void 0;
+    this.sortedNodesCacheVersion = void 0;
+  }
   get lineType() {
     return this._lineType;
   }
@@ -1754,13 +1788,18 @@ var WorkflowLinesManager = class {
    */
   getCloseInLineFromMousePos(mousePos, minDistance = LINE_HOVER_DISTANCE) {
     let targetLine, targetLineDist;
-    this.getAllLines().forEach((line) => {
+    const lines = this.getAllLines();
+    for (let i = 0; i < lines.length; i++) {
+      const line = lines[i];
+      if (!this.isPointInBounds(mousePos, line.bounds, minDistance)) {
+        continue;
+      }
       const dist = line.getHoverDist(mousePos);
-      if (dist <= minDistance && (!targetLineDist || targetLineDist >= dist)) {
+      if (dist <= minDistance && (targetLineDist === void 0 || targetLineDist >= dist)) {
         targetLineDist = dist;
         targetLine = line;
       }
-    });
+    }
     return targetLine;
   }
   dispose() {
@@ -1879,39 +1918,44 @@ var WorkflowLinesManager = class {
    * @param pos
    */
   getPortFromMousePos(pos, portType) {
-    const allNodes = this.getSortedNodes().reverse();
-    const allPorts = allNodes.map((node) => {
-      if (!portType) {
-        return node.ports.allPorts;
-      }
-      return portType === "input" ? node.ports.inputPorts : node.ports.outputPorts;
-    }).flat();
-    const targetPort = allPorts.find((port) => port.isHovered(pos.x, pos.y));
-    if (targetPort) {
-      const containNodes = this.getContainNodesFromMousePos(pos);
-      const targetNode = last(containNodes);
-      if (targetNode && targetNode !== targetPort.node) {
-        return;
+    return this.getHoveredPortFromSortedNodes(pos, this.getSortedNodes(), portType);
+  }
+  getPortsFromMousePos(pos) {
+    return this.getHoveredPortsFromSortedNodes(pos, this.getSortedNodes(), void 0, {
+      collectBoth: true
+    });
+  }
+  getNodeAndPortFromMousePos(pos, portType) {
+    const sortedNodes = this.getSortedNodes();
+    const nodeHitInfo = this.getNodeHitInfoFromSortedNodes(
+      pos,
+      sortedNodes,
+      this.selectService.selection
+    );
+    const ports = this.getHoveredPortsFromSortedNodes(pos, sortedNodes, portType, {
+      topCoverNode: nodeHitInfo.topCoverNode
+    });
+    return {
+      node: nodeHitInfo.topNode,
+      port: ports.port
+    };
+  }
+  getHoverNodeFromMousePos(pos) {
+    const nodes = this.getHoverNodes();
+    for (let i = 0; i < nodes.length; i++) {
+      const node = nodes[i];
+      const { bounds } = node.getData(FlowNodeTransformData3);
+      if (this.isPointInBounds(pos, bounds)) {
+        return node;
       }
     }
-    return targetPort;
   }
   /**
    * 根据鼠标位置找到 node
    * @param pos - 鼠标位置
    */
   getNodeFromMousePos(pos) {
-    const { selection } = this.selectService;
-    const containNodes = this.getContainNodesFromMousePos(pos);
-    if (selection?.length) {
-      const filteredNodes = containNodes.filter(
-        (node) => selection.some((_node) => node.id === _node.id)
-      );
-      if (filteredNodes?.length) {
-        return last(filteredNodes);
-      }
-    }
-    return last(containNodes);
+    return this.getTopNodeFromSortedNodes(pos, this.getSortedNodes(), this.selectService.selection);
   }
   registerContribution(factory) {
     this.contributionFactories.push(factory);
@@ -1921,19 +1965,125 @@ var WorkflowLinesManager = class {
     line.addData(WorkflowLineRenderData);
   }
   getSortedNodes() {
-    return this.document.getAllNodes().sort((a, b) => this.getNodeIndex(a) - this.getNodeIndex(b));
+    const nodeVersion = this.entityManager.getEntityVersion(WorkflowNodeEntity);
+    if (this.sortedNodesCache && this.sortedNodesCacheVersion === nodeVersion) {
+      return this.sortedNodesCache;
+    }
+    this.sortedNodesCache = [...this.document.getAllNodes()].sort(
+      (a, b) => this.getNodeIndex(a) - this.getNodeIndex(b)
+    );
+    this.sortedNodesCacheVersion = nodeVersion;
+    return this.sortedNodesCache;
+  }
+  getHoveredPortFromSortedNodes(pos, sortedNodes, portType) {
+    return this.getHoveredPortsFromSortedNodes(pos, sortedNodes, portType).port;
+  }
+  getHoveredPortsFromSortedNodes(pos, sortedNodes, portType, options = {}) {
+    let inputPort;
+    let outputPort;
+    let anyPort;
+    const { collectBoth = false } = options;
+    for (let i = sortedNodes.length - 1; i >= 0; i--) {
+      const node = sortedNodes[i];
+      if (!portType && !anyPort) {
+        anyPort = this.findHoveredPort(node.ports.allPorts, pos);
+      }
+      if ((!portType || portType === "output") && !outputPort) {
+        outputPort = this.findHoveredPort(node.ports.outputPorts, pos);
+      }
+      if ((!portType || portType === "input") && !inputPort) {
+        inputPort = this.findHoveredPort(node.ports.inputPorts, pos);
+      }
+      if (portType === "input" && inputPort || portType === "output" && outputPort || !portType && (collectBoth ? inputPort && outputPort : anyPort)) {
+        break;
+      }
+    }
+    const needCoverCheck = inputPort || outputPort || anyPort;
+    const topCoverNode = needCoverCheck ? options.topCoverNode || this.getNodeHitInfoFromSortedNodes(pos, sortedNodes).topCoverNode : void 0;
+    inputPort = this.filterCoveredPort(inputPort, topCoverNode);
+    outputPort = this.filterCoveredPort(outputPort, topCoverNode);
+    anyPort = this.filterCoveredPort(anyPort, topCoverNode);
+    return {
+      input: inputPort,
+      output: outputPort,
+      port: portType === "input" ? inputPort : portType === "output" ? outputPort : anyPort
+    };
   }
-  /** 获取鼠标坐标位置的所有节点（stackIndex 从小到大排序） */
-  getContainNodesFromMousePos(pos) {
-    const allNodes = this.getSortedNodes();
+  findHoveredPort(ports, pos) {
+    for (let i = 0; i < ports.length; i++) {
+      const port = ports[i];
+      if (port.isHovered(pos.x, pos.y)) {
+        return port;
+      }
+    }
+  }
+  filterCoveredPort(port, topCoverNode) {
+    if (port && topCoverNode && topCoverNode !== port.node) {
+      return void 0;
+    }
+    return port;
+  }
+  getTopNodeFromSortedNodes(pos, sortedNodes, selection) {
+    return this.getNodeHitInfoFromSortedNodes(pos, sortedNodes, selection).topNode;
+  }
+  getNodeHitInfoFromSortedNodes(pos, sortedNodes, selection) {
     const zoom = this.entityManager.getEntity(PlaygroundConfigEntity2)?.config?.zoom || 1;
-    const containNodes = allNodes.map((node) => {
+    const padding = 4 / zoom;
+    const selectedIDs = selection?.length ? new Set(selection.map((node) => node.id)) : void 0;
+    let topCoverNode;
+    for (let i = sortedNodes.length - 1; i >= 0; i--) {
+      const node = sortedNodes[i];
       const { bounds } = node.getData(FlowNodeTransformData3);
-      if (bounds.clone().pad(4 / zoom).contains(pos.x, pos.y)) {
-        return node;
+      if (!this.isPointInBounds(pos, bounds, padding)) {
+        continue;
       }
-    }).filter(Boolean);
-    return containNodes;
+      if (!topCoverNode) {
+        topCoverNode = node;
+      }
+      if (!selectedIDs) {
+        return {
+          topCoverNode,
+          topNode: node
+        };
+      }
+      if (selectedIDs.has(node.id)) {
+        return {
+          topCoverNode,
+          topNode: node
+        };
+      }
+    }
+    return {
+      topCoverNode,
+      topNode: topCoverNode
+    };
+  }
+  getHoverNodes() {
+    const nodeVersion = this.entityManager.getEntityVersion(WorkflowNodeEntity);
+    const activatedID = this.selectService.activatedNode?.id;
+    if (this.hoverNodesCache && this.hoverNodesCacheVersion === nodeVersion && this.hoverNodesCacheActivatedID === activatedID) {
+      return this.hoverNodesCache;
+    }
+    const nodes = this.document.getAllNodes().filter(
+      (node) => node.id !== "root" && node.flowNodeType !== FlowNodeBaseType2.ROOT && node.flowNodeType !== FlowNodeBaseType2.GROUP
+    ).reverse();
+    if (activatedID) {
+      const activatedIndex = nodes.findIndex((node) => node.id === activatedID);
+      if (activatedIndex > 0) {
+        const [activatedNode] = nodes.splice(activatedIndex, 1);
+        nodes.unshift(activatedNode);
+      }
+    }
+    this.hoverNodesCache = nodes;
+    this.hoverNodesCacheVersion = nodeVersion;
+    this.hoverNodesCacheActivatedID = activatedID;
+    return this.hoverNodesCache;
+  }
+  isPointInBounds(pos, bounds, padding = 0) {
+    if (bounds.width + padding * 2 <= 0 || bounds.height + padding * 2 <= 0) {
+      return false;
+    }
+    return pos.x >= bounds.x - padding && pos.x <= bounds.right + padding && pos.y >= bounds.y - padding && pos.y <= bounds.bottom + padding;
   }
   getNodeIndex(node) {
     const nodeRenderData = node.getData(FlowNodeRenderData2);
@@ -1963,7 +2113,7 @@ import { Emitter as Emitter5 } from "@flowgram.ai/utils";
 import { NodeEngineContext } from "@flowgram.ai/form-core";
 import {
   FlowDocument,
-  FlowNodeBaseType as FlowNodeBaseType2,
+  FlowNodeBaseType as FlowNodeBaseType3,
   FlowNodeTransformData as FlowNodeTransformData5
 } from "@flowgram.ai/document";
 import {
@@ -2277,7 +2427,7 @@ var WorkflowDocument = class extends FlowDocument {
         toJSON: () => this.toNodeJSON(node)
       });
       node.onDispose(() => {
-        if (!node.parent || node.parent.flowNodeType === FlowNodeBaseType2.ROOT) {
+        if (!node.parent || node.parent.flowNodeType === FlowNodeBaseType3.ROOT) {
           return;
         }
         const parentTransform = node.parent.getData(FlowNodeTransformData5);
@@ -2448,13 +2598,13 @@ var WorkflowDocument = class extends FlowDocument {
     );
   }
   getAllNodes() {
-    return this.entityManager.getEntities(WorkflowNodeEntity).filter((n) => n.id !== FlowNodeBaseType2.ROOT);
+    return this.entityManager.getEntities(WorkflowNodeEntity).filter((n) => n.id !== FlowNodeBaseType3.ROOT);
   }
   getAllEdges() {
     return this.entityManager.getEntities(WorkflowLineEntity);
   }
   getAllPorts() {
-    return this.entityManager.getEntities(WorkflowPortEntity).filter((p) => p.node.id !== FlowNodeBaseType2.ROOT);
+    return this.entityManager.getEntities(WorkflowPortEntity).filter((p) => p.node.id !== FlowNodeBaseType3.ROOT);
   }
   /**
    * 获取画布中的非游离节点
@@ -2666,7 +2816,7 @@ var WorkflowDocument = class extends FlowDocument {
     return subCanvas;
   }
   getNodeChildren(node) {
-    if (!node || node.flowNodeType === FlowNodeBaseType2.GROUP) return [];
+    if (!node || node.flowNodeType === FlowNodeBaseType3.GROUP) return [];
     const subCanvas = this.getNodeSubCanvas(node);
     const realChildren = subCanvas ? subCanvas.canvasNode.blocks : node.blocks;
     const childrenWithoutSubCanvas = realChildren.filter((child) => {
@@ -2674,7 +2824,7 @@ var WorkflowDocument = class extends FlowDocument {
       return !childMeta.subCanvas?.(node)?.isCanvas;
     }).filter(Boolean);
     const children = childrenWithoutSubCanvas.map((child) => {
-      if (child.flowNodeType === FlowNodeBaseType2.GROUP) {
+      if (child.flowNodeType === FlowNodeBaseType3.GROUP) {
         return [child, ...child.blocks];
       }
       return child;
@@ -2993,7 +3143,7 @@ var WorkflowDragService = class {
     if (!mousePos) {
       return { x: 0, y: 0 };
     }
-    if (!subNodeType || !containerNode || containerNode.flowNodeType === FlowNodeBaseType3.ROOT) {
+    if (!subNodeType || !containerNode || containerNode.flowNodeType === FlowNodeBaseType4.ROOT) {
       return mousePos;
     }
     const isParentEmpty = !containerNode.children || containerNode.children.length === 0;
@@ -3206,7 +3356,7 @@ var WorkflowDragService = class {
       return;
     }
     const sourceContainer = nodes[0]?.parent;
-    if (!sourceContainer || sourceContainer.flowNodeType === FlowNodeBaseType3.ROOT) {
+    if (!sourceContainer || sourceContainer.flowNodeType === FlowNodeBaseType4.ROOT) {
       return;
     }
     const valid = nodes.every((node) => node?.parent === sourceContainer);
@@ -3351,15 +3501,15 @@ var WorkflowDragService = class {
   }
   updateDrawingLine(isDrawingTo, line, dragPos, originLine) {
     let hasError = false;
-    const mouseNode = this.linesManager.getNodeFromMousePos(dragPos);
     let toNode;
     let toPort;
     let fromPort;
     let fromNode;
     if (isDrawingTo) {
+      const mouseHitInfo = this.linesManager.getNodeAndPortFromMousePos(dragPos, "input");
       fromPort = line.fromPort;
-      toNode = mouseNode;
-      toPort = this.linesManager.getPortFromMousePos(dragPos, "input");
+      toNode = mouseHitInfo.node;
+      toPort = mouseHitInfo.port;
       if (toNode && this.canBuildContainerLine(toNode, dragPos)) {
         toPort = this.getNearestPort(toNode, dragPos, "input");
         hasError = this.checkDraggingPort(isDrawingTo, line, toNode, toPort, originLine).hasError;
@@ -3386,9 +3536,10 @@ var WorkflowDragService = class {
         };
       }
     } else {
+      const mouseHitInfo = this.linesManager.getNodeAndPortFromMousePos(dragPos, "output");
       toPort = line.toPort;
-      fromNode = mouseNode;
-      fromPort = this.linesManager.getPortFromMousePos(dragPos, "output");
+      fromNode = mouseHitInfo.node;
+      fromPort = mouseHitInfo.port;
       if (fromNode && this.canBuildContainerLine(fromNode, dragPos)) {
         fromPort = this.getNearestPort(fromNode, dragPos, "output");
         hasError = this.checkDraggingPort(