admin/flowgram.ai
1
0
Fork 0
mirror of https://gitee.com/ByteDance/flowgram.ai.git synced 2025-07-07 17:43:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: node.toJSON use document.toNodeJSON (#418)

Browse Source
This commit is contained in:
xiamidaxia 2025-06-27 14:26:53 +08:00 committed by GitHub
parent bf69e6cb89
commit 080d28ba1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 35 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
packages/canvas-engine/document/src/entities/flow-node-entity.ts
View File

@ -337,37 +337,7 @@ export class FlowNodeEntity extends Entity<FlowNodeEntityConfig> {
* @param newId
*/
toJSON(): FlowNodeJSON {
if (this.document.options.toNodeJSON) {
return this.document.options.toNodeJSON(this);
}
const nodesMap: Record<string, FlowNodeJSON> = {};
let startNodeJSON: FlowNodeJSON;
this.document.traverse((node) => {
const isSystemNode = node.id.startsWith('$');
if (isSystemNode) return;
const nodeJSONData = this.getJSONData();
const nodeJSON: FlowNodeJSON = {
id: node.id,
type: node.flowNodeType,
};
if (nodeJSONData !== undefined) {
nodeJSON.data = nodeJSONData;
}
if (!startNodeJSON) startNodeJSON = nodeJSON;
let { parent } = node;
if (parent && parent.id.startsWith('$')) {
parent = parent.originParent;
}
const parentJSON = parent ? nodesMap[parent.id] : undefined;
if (parentJSON) {
if (!parentJSON.blocks) {
parentJSON.blocks = [];
}
parentJSON.blocks.push(nodeJSON);
}
nodesMap[node.id] = nodeJSON;
}, this);
return startNodeJSON!;
return this.document.toNodeJSON(this);
}
get isVertical(): boolean {

34
packages/canvas-engine/document/src/flow-document.ts
View File

@ -593,6 +593,40 @@ export class FlowDocument<T = FlowDocumentJSON> implements Disposable {
return result;
}
toNodeJSON(node: FlowNodeEntity): FlowNodeJSON {
if (this.options.toNodeJSON) {
return this.options.toNodeJSON(node);
}
const nodesMap: Record<string, FlowNodeJSON> = {};
let startNodeJSON: FlowNodeJSON;
this.traverse((node) => {
const isSystemNode = node.id.startsWith('$');
if (isSystemNode) return;
const nodeJSONData = node.getJSONData();
const nodeJSON: FlowNodeJSON = {
id: node.id,
type: node.flowNodeType,
};
if (nodeJSONData !== undefined) {
nodeJSON.data = nodeJSONData;
}
if (!startNodeJSON) startNodeJSON = nodeJSON;
let { parent } = node;
if (parent && parent.id.startsWith('$')) {
parent = parent.originParent;
}
const parentJSON = parent ? nodesMap[parent.id] : undefined;
if (parentJSON) {
if (!parentJSON.blocks) {
parentJSON.blocks = [];
}
parentJSON.blocks.push(nodeJSON);
}
nodesMap[node.id] = nodeJSON;
}, node);
return startNodeJSON!;
}
/**
* 移动节点
* @param param0