Skip to content

Commit 6223572

Browse files
authored
fix(thumbnail): translate subflow block nodes (#114)
1 parent 0d849a1 commit 6223572

File tree

1 file changed

+16
-2
lines changed

1 file changed

+16
-2
lines changed

src/utils/thumbnail.ts

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ interface BlockThumbnail {
2929
task?: string;
3030
subflow?: string;
3131
executorName?: string;
32-
nodes?: Node[];
32+
nodes?: NodeThumbnail[];
3333
handleOutputsFrom?: HandleOutputFrom[];
3434
uiData?: any;
3535
}
@@ -311,7 +311,7 @@ export class Thumbnail implements ThumbnailProvider {
311311
icon: subflowData.icon,
312312
inputHandleDefs: subflowData.pkgData.localizeHandleDefs(subflowData.data.inputs_def),
313313
outputHandleDefs: subflowData.pkgData.localizeHandleDefs(subflowData.data.outputs_def),
314-
nodes: subflowData.data.nodes,
314+
nodes: await this._getNodesThumbnail(subflowData.data.nodes, subflowData),
315315
handleOutputsFrom: subflowData.data.outputs_from,
316316
subflow: subflowData.blockResourceName,
317317
uiData: subflowData.uiData,
@@ -377,6 +377,20 @@ export class Thumbnail implements ThumbnailProvider {
377377
};
378378
}
379379

380+
private async _getNodesThumbnail(rawNodes: Node[] | undefined, flowData: FlowLikeData): Promise<NodeThumbnail[]> {
381+
if (!rawNodes) {
382+
return [];
383+
}
384+
const nodes: NodeThumbnail[] = [];
385+
for (const rawNode of rawNodes) {
386+
const node = await this._getNodeThumbnail(rawNode, flowData);
387+
if (node) {
388+
nodes.push(node);
389+
}
390+
}
391+
return nodes;
392+
}
393+
380394
private async _getNodeThumbnail(raw: Node, flowData: FlowLikeData): Promise<NodeThumbnail | undefined> {
381395
const node: NodeThumbnail = {
382396
nodeId: raw.node_id,

0 commit comments

Comments
 (0)