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

package/dist/esm/index.js

@@ -72,6 +72,34 @@ import { Rectangle as Rectangle6, SizeSchema } from "@flowgram.ai/utils";
 import { bindConfigEntity } from "@flowgram.ai/core";
 import { delay } from "@flowgram.ai/utils";
 
+// src/utils/build-group-json.ts
+import { FlowNodeBaseType } from "@flowgram.ai/document";
+var buildGroupJSON = (json) => {
+  const { nodes, edges } = json;
+  const groupJSONs = nodes.filter(
+    (nodeJSON) => nodeJSON.type === FlowNodeBaseType.GROUP
+  );
+  const nodeJSONMap = new Map(nodes.map((n) => [n.id, n]));
+  const groupNodeJSONs = groupJSONs.map((groupJSON) => {
+    const groupBlocks = (groupJSON.data.blockIDs ?? []).map((blockID) => nodeJSONMap.get(blockID)).filter(Boolean);
+    const groupEdges = edges?.filter(
+      (edge) => groupBlocks.some((block) => block.id === edge.sourceNodeID || block.id === edge.targetNodeID)
+    );
+    const groupNodeJSON = {
+      ...groupJSON,
+      blocks: groupBlocks,
+      edges: groupEdges
+    };
+    return groupNodeJSON;
+  });
+  const groupBlockSet = new Set(groupJSONs.map((groupJSON) => groupJSON.data.blockIDs).flat());
+  const processedNodes = nodes.filter((nodeJSON) => !groupBlockSet.has(nodeJSON.id)).concat(groupNodeJSONs);
+  return {
+    nodes: processedNodes,
+    edges
+  };
+};
+
 // src/utils/nanoid.ts
 import { nanoid as nanoidOrigin } from "nanoid";
 function nanoid(n) {
@@ -173,22 +201,23 @@ var WorkflowPortEntity = class extends Entity {
   }
   // 设置连线的错误态，外部应使用 validate 进行更新
   set hasError(hasError) {
-    this._hasError = hasError;
-    this._onErrorChangedEmitter.fire();
+    if (hasError !== this._hasError) {
+      this._hasError = hasError;
+      this._onErrorChangedEmitter.fire();
+    }
   }
   validate() {
     const anyLineHasError = this.allLines.some((line) => {
       if (line.disposed || line.isHidden) {
         return false;
       }
-      line.validateSelf();
       return line.hasError;
     });
     const isPortHasError = this.node.document.isErrorPort(this);
     this.hasError = anyLineHasError || isPortHasError;
   }
   isErrorPort() {
-    return this.node.document.isErrorPort(this);
+    return this.node.document.isErrorPort(this, this.hasError);
   }
   get point() {
     const { targetElement } = this;
@@ -375,7 +404,6 @@ var WorkflowNodePortsData = class extends EntityData {
       port.allLines.forEach((line) => {
         line.validate();
       });
-      port.validate();
     });
   }
   /**
@@ -506,6 +534,9 @@ var _WorkflowNodeLinesData = class _WorkflowNodeLinesData extends EntityData2 {
   get outputLines() {
     return this.data.outputLines;
   }
+  get allLines() {
+    return this.data.inputLines.concat(this.data.outputLines);
+  }
   /**
    * 输入节点
    */
@@ -692,8 +723,16 @@ var POINT_RADIUS = 10;
 var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
   constructor(opts) {
     super(opts);
-    this._processing = false;
-    this._hasError = false;
+    this._uiState = {
+      hasError: false,
+      flowing: false,
+      disabled: false,
+      vertical: false,
+      hideArrow: false,
+      reverse: false,
+      highlightColor: "",
+      lockedColor: ""
+    };
     this.stackIndex = 0;
     /**
      * 线条数据
@@ -713,6 +752,14 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
     if (opts.drawingTo) {
       this.isDrawing = true;
     }
+    this.onEntityChange(() => {
+      this.fromPort?.validate();
+      this.toPort?.validate();
+    });
+    this.onDispose(() => {
+      this.fromPort?.validate();
+      this.toPort?.validate();
+    });
   }
   /**
    * 转成线条 id
@@ -722,6 +769,43 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
     const { from, to, fromPort, toPort } = info;
     return `${from}_${fromPort || ""}-${to || ""}_${toPort || ""}`;
   }
+  /**
+   * 线条的 UI 状态
+   */
+  get uiState() {
+    return this._uiState;
+  }
+  /**
+   * 更新线条的 ui 状态
+   * @param newState
+   */
+  updateUIState(newState) {
+    let changed = false;
+    Object.keys(newState).forEach((key) => {
+      const value = newState[key];
+      if (this._uiState[key] !== value) {
+        this._uiState[key] = value;
+        changed = true;
+      }
+    });
+    if (changed) {
+      this.fireChange();
+    }
+  }
+  /**
+   * 线条的扩展数据
+   */
+  get lineData() {
+    return this._lineData;
+  }
+  /**
+   * 更新线条扩展数据
+   * @param data
+   */
+  set lineData(data) {
+    this._lineData = data;
+    this.fireChange();
+  }
   /**
    * 获取线条的前置节点
    */
@@ -743,29 +827,30 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
   }
   /**
    * 获取是否 testrun processing
+   * @deprecated  use `uiState.flowing` instead
    */
   get processing() {
-    return this._processing;
+    return this._uiState.flowing;
   }
   /**
    * 设置 testrun processing 状态
+   * @deprecated  use `uiState.flowing` instead
    */
   set processing(status) {
-    if (this._processing !== status) {
-      this._processing = status;
+    if (this._uiState.flowing !== status) {
+      this._uiState.flowing = status;
       this.fireChange();
     }
   }
   // 获取连线是否为错误态
   get hasError() {
-    return this._hasError;
+    return this.uiState.hasError;
   }
   // 设置连线的错误态
   set hasError(hasError) {
-    if (this._hasError !== hasError) {
-      this._hasError = hasError;
-      this.fireChange();
-    }
+    this.updateUIState({
+      hasError
+    });
     if (this._node) {
       this._node.dataset.hasError = this.hasError ? "true" : "false";
     }
@@ -780,6 +865,7 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
     if (this.toPort === toPort) {
       return;
     }
+    const prePort = this.toPort;
     if (toPort && toPort.portType === "input" && this.linesManager.canAddLine(this.fromPort, toPort, true)) {
       const { node, portID } = toPort;
       this._to = node;
@@ -792,6 +878,9 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
       this.info.to = void 0;
       this.info.toPort = "";
     }
+    if (prePort) {
+      prePort.validate();
+    }
     this.fireChange();
   }
   /**
@@ -818,13 +907,20 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
     return this.info.drawingTo;
   }
   get highlightColor() {
-    return this.info.highlightColor || "";
+    return this.uiState.highlightColor || "";
   }
-  set highlightColor(color) {
-    if (this.info.highlightColor !== color) {
-      this.info.highlightColor = color;
-      this.fireChange();
-    }
+  set highlightColor(highlightColor) {
+    this.updateUIState({
+      highlightColor
+    });
+  }
+  get lockedColor() {
+    return this.uiState.lockedColor;
+  }
+  set lockedColor(lockedColor) {
+    this.updateUIState({
+      lockedColor
+    });
   }
   /**
    * 获取线条的边框位置大小
@@ -855,23 +951,23 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
   }
   /** 是否反转箭头 */
   get reverse() {
-    return this.linesManager.isReverseLine(this);
+    return this.linesManager.isReverseLine(this, this.uiState.reverse);
   }
   /** 是否隐藏箭头 */
   get hideArrow() {
-    return this.linesManager.isHideArrowLine(this);
+    return this.linesManager.isHideArrowLine(this, this.uiState.hideArrow);
   }
   /** 是否流动 */
   get flowing() {
-    return this.linesManager.isFlowingLine(this);
+    return this.linesManager.isFlowingLine(this, this.uiState.flowing);
   }
   /** 是否禁用 */
   get disabled() {
-    return this.linesManager.isDisabledLine(this);
+    return this.linesManager.isDisabledLine(this, this.uiState.disabled);
   }
   /** 是否竖向 */
   get vertical() {
-    return this.linesManager.isVerticalLine(this);
+    return this.linesManager.isVerticalLine(this, this.uiState.vertical);
   }
   /** 获取线条渲染器类型 */
   get renderType() {
@@ -898,15 +994,16 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
   }
   // 校验连线是否为错误态
   validate() {
-    const { fromPort, toPort } = this;
     this.validateSelf();
-    fromPort?.validate();
-    toPort?.validate();
   }
+  /**
+   * use `validate` instead
+   * @deprecated
+   */
   validateSelf() {
     const { fromPort, toPort } = this;
     if (fromPort) {
-      this.hasError = this.linesManager.isErrorLine(fromPort, toPort);
+      this.hasError = this.linesManager.isErrorLine(fromPort, toPort, this.uiState.hasError);
     }
   }
   is(line) {
@@ -937,6 +1034,9 @@ var _WorkflowLineEntity = class _WorkflowLineEntity extends Entity2 {
       sourcePortID: this.info.fromPort,
       targetPortID: this.info.toPort
     };
+    if (this._lineData !== void 0) {
+      json.data = this._lineData;
+    }
     if (!json.sourcePortID) {
       delete json.sourcePortID;
     }
@@ -1129,7 +1229,7 @@ import {
   FlowNodeTransformData as FlowNodeTransformData6,
   FlowOperationBaseService
 } from "@flowgram.ai/document";
-import { FlowNodeBaseType as FlowNodeBaseType2 } from "@flowgram.ai/document";
+import { FlowNodeBaseType as FlowNodeBaseType3 } from "@flowgram.ai/document";
 import {
   CommandService,
   MouseTouchEvent,
@@ -1339,7 +1439,6 @@ var WorkflowLinesManager = class {
       }
       fromNode.removeLine(line);
       toNode?.removeLine(line);
-      line.validate();
     });
     line.onDispose(() => {
       if (available) {
@@ -1383,41 +1482,41 @@ var WorkflowLinesManager = class {
   get disposed() {
     return this.toDispose.disposed;
   }
-  isErrorLine(fromPort, toPort) {
+  isErrorLine(fromPort, toPort, defaultValue) {
     if (this.options.isErrorLine) {
       return this.options.isErrorLine(fromPort, toPort, this);
     }
-    return false;
+    return !!defaultValue;
   }
-  isReverseLine(line) {
+  isReverseLine(line, defaultValue = false) {
     if (this.options.isReverseLine) {
       return this.options.isReverseLine(line);
     }
-    return false;
+    return defaultValue;
   }
-  isHideArrowLine(line) {
+  isHideArrowLine(line, defaultValue = false) {
     if (this.options.isHideArrowLine) {
       return this.options.isHideArrowLine(line);
     }
-    return false;
+    return defaultValue;
   }
-  isFlowingLine(line) {
+  isFlowingLine(line, defaultValue = false) {
     if (this.options.isFlowingLine) {
       return this.options.isFlowingLine(line);
     }
-    return false;
+    return defaultValue;
   }
-  isDisabledLine(line) {
+  isDisabledLine(line, defaultValue = false) {
     if (this.options.isDisabledLine) {
       return this.options.isDisabledLine(line);
     }
-    return false;
+    return defaultValue;
   }
-  isVerticalLine(line) {
+  isVerticalLine(line, defaultValue = false) {
     if (this.options.isVerticalLine) {
       return this.options.isVerticalLine(line);
     }
-    return false;
+    return defaultValue;
   }
   setLineRenderType(line) {
     if (this.options.setLineRenderType) {
@@ -1435,6 +1534,9 @@ var WorkflowLinesManager = class {
     if (line.isHidden) {
       return this.lineColor.hidden;
     }
+    if (line.lockedColor) {
+      return line.lockedColor;
+    }
     if (line.hasError) {
       return this.lineColor.error;
     }
@@ -1566,7 +1668,7 @@ import { Emitter as Emitter4 } from "@flowgram.ai/utils";
 import { NodeEngineContext } from "@flowgram.ai/form-core";
 import {
   FlowDocument,
-  FlowNodeBaseType,
+  FlowNodeBaseType as FlowNodeBaseType2,
   FlowNodeTransformData as FlowNodeTransformData5
 } from "@flowgram.ai/document";
 import {
@@ -1859,7 +1961,7 @@ var WorkflowDocument = class extends FlowDocument {
         toJSON: () => this.toNodeJSON(node)
       });
       node.onDispose(() => {
-        if (!node.parent || node.parent.flowNodeType === FlowNodeBaseType.ROOT) {
+        if (!node.parent || node.parent.flowNodeType === FlowNodeBaseType2.ROOT) {
           return;
         }
         const parentTransform = node.parent.getData(FlowNodeTransformData5);
@@ -2027,10 +2129,10 @@ var WorkflowDocument = class extends FlowDocument {
     );
   }
   getAllNodes() {
-    return this.entityManager.getEntities(WorkflowNodeEntity).filter((n) => n.id !== FlowNodeBaseType.ROOT);
+    return this.entityManager.getEntities(WorkflowNodeEntity).filter((n) => n.id !== FlowNodeBaseType2.ROOT);
   }
   getAllPorts() {
-    return this.entityManager.getEntities(WorkflowPortEntity).filter((p) => p.node.id !== FlowNodeBaseType.ROOT);
+    return this.entityManager.getEntities(WorkflowPortEntity).filter((p) => p.node.id !== FlowNodeBaseType2.ROOT);
   }
   /**
    * 获取画布中的非游离节点
@@ -2174,21 +2276,22 @@ var WorkflowDocument = class extends FlowDocument {
   /**
    * 判断端口是否为错误态
    */
-  isErrorPort(port) {
+  isErrorPort(port, defaultValue = false) {
     if (typeof this.options.isErrorPort === "function") {
       return this.options.isErrorPort(port);
     }
-    return false;
+    return defaultValue;
   }
   /**
    * 导出数据
    */
   toJSON() {
     const rootJSON = this.toNodeJSON(this.root);
-    return {
+    const json = {
       nodes: rootJSON.blocks ?? [],
       edges: rootJSON.edges ?? []
     };
+    return json;
   }
   dispose() {
     super.dispose();
@@ -2200,10 +2303,11 @@ var WorkflowDocument = class extends FlowDocument {
   renderJSON(json, options) {
     const { parent = this.root, isClone = false } = options ?? {};
     const containerID = this.getNodeSubCanvas(parent)?.canvasNode.id ?? parent.id;
-    const nodes = json.nodes.map(
+    const processedJSON = buildGroupJSON(json);
+    const nodes = processedJSON.nodes.map(
       (nodeJSON) => this.createWorkflowNode(nodeJSON, isClone, containerID)
     );
-    const edges = json.edges.map((edge) => this.createWorkflowLine(edge, containerID)).filter(Boolean);
+    const edges = processedJSON.edges.map((edge) => this.createWorkflowLine(edge, containerID)).filter(Boolean);
     return { nodes, edges };
   }
   getNodeSubCanvas(node) {
@@ -2213,13 +2317,19 @@ var WorkflowDocument = class extends FlowDocument {
     return subCanvas;
   }
   getNodeChildren(node) {
-    if (!node) return [];
+    if (!node || node.flowNodeType === FlowNodeBaseType2.GROUP) return [];
     const subCanvas = this.getNodeSubCanvas(node);
-    const childrenWithCanvas = subCanvas ? subCanvas.canvasNode.collapsedChildren : node.collapsedChildren;
-    const children = childrenWithCanvas.filter((child) => {
+    const realChildren = subCanvas ? subCanvas.canvasNode.blocks : node.blocks;
+    const childrenWithoutSubCanvas = realChildren.filter((child) => {
       const childMeta = child.getNodeMeta();
       return !childMeta.subCanvas?.(node)?.isCanvas;
     }).filter(Boolean);
+    const children = childrenWithoutSubCanvas.map((child) => {
+      if (child.flowNodeType === FlowNodeBaseType2.GROUP) {
+        return [child, ...child.blocks];
+      }
+      return child;
+    }).flat();
     return children;
   }
   toLineJSON(line) {
@@ -2351,7 +2461,7 @@ var WorkflowDragService = class {
     }
     this.isDragging = true;
     const sameParent = this.childrenOfContainer(selectedNodes);
-    if (sameParent && sameParent.flowNodeType !== FlowNodeBaseType2.ROOT) {
+    if (sameParent && sameParent.flowNodeType !== FlowNodeBaseType3.ROOT) {
       selectedNodes = [sameParent];
     }
     let startPosition = this.getNodesPosition(selectedNodes);
@@ -2521,7 +2631,7 @@ var WorkflowDragService = class {
     if (!mousePos) {
       return { x: 0, y: 0 };
     }
-    if (!subNodeType || !containerNode || containerNode.flowNodeType === FlowNodeBaseType2.ROOT) {
+    if (!subNodeType || !containerNode || containerNode.flowNodeType === FlowNodeBaseType3.ROOT) {
       return mousePos;
     }
     const isParentEmpty = !containerNode.children || containerNode.children.length === 0;
@@ -2732,7 +2842,7 @@ var WorkflowDragService = class {
             return;
           }
           config.updateCursor("grab");
-          line.highlightColor = this.linesManager.lineColor.drawing;
+          line.highlightColor = originLine?.lockedColor || this.linesManager.lineColor.drawing;
           this.hoverService.updateHoveredKey("");
         }
         if (!line) {
@@ -2754,7 +2864,7 @@ var WorkflowDragService = class {
         this._onDragLineEventEmitter.fire({
           type: "onDrag"
         });
-        this.setLineColor(line, this.linesManager.lineColor.drawing);
+        this.setLineColor(line, originLine?.lockedColor || this.linesManager.lineColor.drawing);
         if (toNode && this.canBuildContainerLine(toNode, dragPos)) {
           toPort = this.getNearestPort(toNode, dragPos);
           const { hasError } = this.handleDragOnNode(toNode, fromPort, line, toPort, originLine);
@@ -3443,6 +3553,7 @@ export {
   WorkflowSelectService,
   WorkflowSimpleLineContribution,
   bindConfigEntity,
+  buildGroupJSON,
   compose,
   composeAsync,
   delay,