@flowgram.ai/free-layout-core 0.2.27 → 0.2.28

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) {
@@ -1185,7 +1213,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,
@@ -1625,7 +1653,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 {
@@ -1918,7 +1946,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);
@@ -2086,10 +2114,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);
   }
   /**
    * 获取画布中的非游离节点
@@ -2244,10 +2272,11 @@ var WorkflowDocument = class extends FlowDocument {
    */
   toJSON() {
     const rootJSON = this.toNodeJSON(this.root);
-    return {
+    const json = {
       nodes: rootJSON.blocks ?? [],
       edges: rootJSON.edges ?? []
     };
+    return json;
   }
   dispose() {
     super.dispose();
@@ -2259,10 +2288,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) {
@@ -2272,13 +2302,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) {
@@ -2410,7 +2446,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);
@@ -2580,7 +2616,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;
@@ -3502,6 +3538,7 @@ export {
   WorkflowSelectService,
   WorkflowSimpleLineContribution,
   bindConfigEntity,
+  buildGroupJSON,
   compose,
   composeAsync,
   delay,