1

src/components/custom-world-home/creationWorkShelf.ts

@@ -364,6 +364,7 @@ function mapMatch3DWorkToShelfItem(
   const status = item.publicationStatus === 'published' ? 'published' : 'draft';
   const publicWorkCode =
     status === 'published' ? buildMatch3DPublicWorkCode(item.profileId) : null;
+  const coverImageSrc = resolveMatch3DWorkCoverImageSrc(item);
 
   return {
     id: item.workId,
@@ -372,7 +373,7 @@ function mapMatch3DWorkToShelfItem(
     title: item.gameName,
     summary: item.summary,
     updatedAt: item.updatedAt,
-    coverImageSrc: item.coverImageSrc ?? null,
+    coverImageSrc,
     coverRenderMode: 'image',
     coverCharacterImageSrcs: [],
     publicWorkCode,
@@ -408,6 +409,7 @@ function mapPuzzleWorkToShelfItem(
   const status = item.publicationStatus;
   const publicWorkCode =
     status === 'published' ? buildPuzzlePublicWorkCode(item.profileId) : null;
+  const coverImageSrc = resolvePuzzleWorkCoverImageSrc(item);
 
   return {
     id: item.workId,
@@ -419,7 +421,7 @@ function mapPuzzleWorkToShelfItem(
       item.summary.trim() ||
       (status === 'draft' ? '未填写作品描述' : ''),
     updatedAt: item.updatedAt,
-    coverImageSrc: item.coverImageSrc ?? null,
+    coverImageSrc,
     coverRenderMode: 'image',
     coverCharacterImageSrcs: [],
     publicWorkCode,
@@ -571,6 +573,7 @@ function mapSquareHoleWorkToShelfItem(
     status === 'published'
       ? buildSquareHolePublicWorkCode(item.profileId)
       : null;
+  const coverImageSrc = resolveSquareHoleWorkCoverImageSrc(item);
 
   return {
     id: item.workId,
@@ -579,7 +582,7 @@ function mapSquareHoleWorkToShelfItem(
     title: item.gameName,
     summary: item.summary,
     updatedAt: item.updatedAt,
-    coverImageSrc: item.coverImageSrc ?? null,
+    coverImageSrc,
     coverRenderMode: 'image',
     coverCharacterImageSrcs: [],
     publicWorkCode,
@@ -607,6 +610,90 @@ function mapSquareHoleWorkToShelfItem(
   };
 }
 
+function normalizeCoverImageSrc(value?: string | null) {
+  return value?.trim() || null;
+}
+
+function resolvePuzzleWorkCoverImageSrc(item: PuzzleWorkSummary) {
+  const directCoverImageSrc = normalizeCoverImageSrc(item.coverImageSrc);
+  if (directCoverImageSrc) {
+    return directCoverImageSrc;
+  }
+
+  for (const level of item.levels ?? []) {
+    const selectedCandidateImageSrc =
+      level.selectedCandidateId && level.candidates.length > 0
+        ? normalizeCoverImageSrc(
+            level.candidates.find(
+              (candidate) => candidate.candidateId === level.selectedCandidateId,
+            )?.imageSrc,
+          )
+        : null;
+    const fallbackCandidateImageSrc = normalizeCoverImageSrc(
+      level.candidates[level.candidates.length - 1]?.imageSrc,
+    );
+    const levelCoverImageSrc =
+      selectedCandidateImageSrc ||
+      normalizeCoverImageSrc(level.coverImageSrc) ||
+      fallbackCandidateImageSrc;
+
+    if (levelCoverImageSrc) {
+      return levelCoverImageSrc;
+    }
+  }
+
+  return null;
+}
+
+function resolveMatch3DWorkCoverImageSrc(item: Match3DWorkSummary) {
+  const directCoverImageSrc = normalizeCoverImageSrc(item.coverImageSrc);
+  if (directCoverImageSrc) {
+    return directCoverImageSrc;
+  }
+
+  const backgroundImageSrc =
+    normalizeCoverImageSrc(item.backgroundImageSrc) ||
+    normalizeCoverImageSrc(item.generatedBackgroundAsset?.imageSrc) ||
+    normalizeCoverImageSrc(item.generatedBackgroundAsset?.containerImageSrc);
+  if (backgroundImageSrc) {
+    return backgroundImageSrc;
+  }
+
+  for (const asset of item.generatedItemAssets ?? []) {
+    const imageViewSrc = normalizeCoverImageSrc(
+      asset.imageViews?.find((view) => normalizeCoverImageSrc(view.imageSrc))
+        ?.imageSrc,
+    );
+    const itemImageSrc = normalizeCoverImageSrc(asset.imageSrc);
+    if (imageViewSrc || itemImageSrc) {
+      return imageViewSrc || itemImageSrc;
+    }
+  }
+
+  return null;
+}
+
+function resolveSquareHoleWorkCoverImageSrc(item: SquareHoleWorkSummary) {
+  const directCoverImageSrc = normalizeCoverImageSrc(item.coverImageSrc);
+  if (directCoverImageSrc) {
+    return directCoverImageSrc;
+  }
+
+  const backgroundImageSrc = normalizeCoverImageSrc(item.backgroundImageSrc);
+  if (backgroundImageSrc) {
+    return backgroundImageSrc;
+  }
+
+  for (const option of [...item.shapeOptions, ...item.holeOptions]) {
+    const optionImageSrc = normalizeCoverImageSrc(option.imageSrc);
+    if (optionImageSrc) {
+      return optionImageSrc;
+    }
+  }
+
+  return null;
+}
+
 function buildWorkShelfActions<TItem>(
   item: TItem,
   adapter: WorkShelfAdapter<TItem>,