fix(canvas): 节点复制 - 修复内部粘贴和批量状态残留 (#60)

Co-authored-by: Danborad <Danborad@users.noreply.github.com>

web/src/pages/canvas/use-canvas-generation.ts

@@ -160,8 +160,11 @@ export function useCanvasGeneration({ projectId, domainProjectId, projectLoaded,
     }, [nodesRef, setNodes]);
 
     const recoverInterruptedGenerationTasks = useCallback(async () => {
-        const recoveryNodes = nodesRef.current.filter((node) => node.metadata?.status === NODE_STATUS_LOADING || node.metadata?.errorDetails === "页面刷新后生成已中断，请重新生成。" || Boolean(node.metadata?.taskId && node.metadata.status !== NODE_STATUS_SUCCESS));
-        if (!recoveryNodes.length) return;
+        const recoveryNodes = nodesRef.current.filter((node) => {
+            const aggregateBatchRoot = node.metadata?.isBatchRoot && node.metadata.batchChildIds?.length && !node.metadata.taskId;
+            if (aggregateBatchRoot) return false;
+            return node.metadata?.status === NODE_STATUS_LOADING || node.metadata?.errorDetails === "页面刷新后生成已中断，请重新生成。" || Boolean(node.metadata?.taskId && node.metadata.status !== NODE_STATUS_SUCCESS);
+        });
         const taskIds = Array.from(new Set(recoveryNodes.map((node) => node.metadata?.taskId).filter((id): id is string => Boolean(id))));
         const tasks = (await Promise.all(taskIds.map((id) => queryGenerationTask(id).catch(() => undefined)))).filter((task): task is GenerationTask => Boolean(task));
         if (recoveryNodes.some((node) => !node.metadata?.taskId)) {
@@ -194,6 +197,30 @@ export function useCanvasGeneration({ projectId, domainProjectId, projectLoaded,
                 recoveringTaskIdsRef.current.delete(task.id);
             }
         }));
+        setNodes((current) => current.map((node) => {
+            if (!node.metadata?.isBatchRoot || !node.metadata.batchChildIds?.length || node.metadata.taskId) return node;
+            const children = node.metadata.batchChildIds.map((id) => current.find((item) => item.id === id)).filter(Boolean) as CanvasNodeData[];
+            const primary = children.find((item) => item.id === node.metadata?.primaryImageId && item.metadata?.content) || children.find((item) => item.metadata?.content);
+            const loading = children.some((item) => item.metadata?.status === NODE_STATUS_LOADING);
+            const failed = children.find((item) => item.metadata?.status === NODE_STATUS_ERROR);
+            return {
+                ...node,
+                metadata: {
+                    ...node.metadata,
+                    ...(primary ? {
+                        content: primary.metadata?.content,
+                        storageKey: primary.metadata?.storageKey,
+                        mimeType: primary.metadata?.mimeType,
+                        bytes: primary.metadata?.bytes,
+                        naturalWidth: primary.metadata?.naturalWidth,
+                        naturalHeight: primary.metadata?.naturalHeight,
+                        primaryImageId: primary.id,
+                    } : {}),
+                    status: primary ? NODE_STATUS_SUCCESS : loading ? NODE_STATUS_LOADING : NODE_STATUS_ERROR,
+                    errorDetails: primary ? undefined : failed?.metadata?.errorDetails || "全部图片生成失败",
+                },
+            };
+        }));
     }, [applyGenerationTaskResult, bindGenerationTask, nodesRef, projectId, setNodes]);
 
     useEffect(() => {