1

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

@@ -2,9 +2,9 @@ import type { BigFishWorkSummary } from '../../../packages/shared/src/contracts/
 import type { CustomWorldWorkSummary } from '../../../packages/shared/src/contracts/customWorldAgent';
 import type { Match3DWorkSummary } from '../../../packages/shared/src/contracts/match3dWorks';
 import type { PuzzleWorkSummary } from '../../../packages/shared/src/contracts/puzzleWorkSummary';
+import type { CustomWorldLibraryEntry } from '../../../packages/shared/src/contracts/runtime';
 import type { SquareHoleWorkSummary } from '../../../packages/shared/src/contracts/squareHoleWorks';
 import type { VisualNovelWorkSummary } from '../../../packages/shared/src/contracts/visualNovel';
-import type { CustomWorldLibraryEntry } from '../../../packages/shared/src/contracts/runtime';
 import { buildPublicWorkStagePath } from '../../routing/appPageRoutes';
 import {
   buildBigFishPublicWorkCode,
@@ -83,6 +83,8 @@ export type CreationWorkShelfItem = {
   id: string;
   kind: CreationWorkShelfKind;
   status: CreationWorkShelfStatus;
+  isGenerating?: boolean;
+  hasUnreadUpdate?: boolean;
   title: string;
   summary: string;
   updatedAt: string;
@@ -114,6 +116,9 @@ export function buildCreationWorkShelfItems(params: {
   canDeleteSquareHole?: boolean;
   canDeletePuzzle?: boolean;
   canDeleteVisualNovel?: boolean;
+  getItemState?: (
+    item: CreationWorkShelfItem,
+  ) => { isGenerating?: boolean; hasUnreadUpdate?: boolean } | null;
 }) {
   const {
     rpgItems,
@@ -129,6 +134,7 @@ export function buildCreationWorkShelfItems(params: {
     canDeleteSquareHole = false,
     canDeletePuzzle = false,
     canDeleteVisualNovel = false,
+    getItemState,
   } = params;
 
   return [
@@ -150,10 +156,21 @@ export function buildCreationWorkShelfItems(params: {
     ...visualNovelItems.map((item) =>
       mapVisualNovelWorkToShelfItem(item, canDeleteVisualNovel),
     ),
-  ].sort(
-    (left, right) =>
-      getShelfItemTime(right.updatedAt) - getShelfItemTime(left.updatedAt),
-  );
+  ]
+    .map((item) => {
+      const state = getItemState?.(item);
+      return state
+        ? {
+            ...item,
+            isGenerating: state.isGenerating,
+            hasUnreadUpdate: state.hasUnreadUpdate,
+          }
+        : item;
+    })
+    .sort(
+      (left, right) =>
+        getShelfItemTime(right.updatedAt) - getShelfItemTime(left.updatedAt),
+    );
 }
 
 function mapRpgWorkToShelfItem(
@@ -355,14 +372,14 @@ function mapVisualNovelWorkToShelfItem(
   item: VisualNovelWorkSummary,
   canDelete: boolean,
 ): CreationWorkShelfItem {
-  const status =
-    item.publishStatus === 'published' ? 'published' : 'draft';
+  const status = item.publishStatus === 'published' ? 'published' : 'draft';
   const publicWorkCode =
-    status === 'published' ? buildVisualNovelPublicWorkCode(item.profileId) : null;
+    status === 'published'
+      ? buildVisualNovelPublicWorkCode(item.profileId)
+      : null;
   const title = item.title?.trim() || '未命名视觉小说';
   const summary =
-    item.description?.trim() ||
-    (status === 'draft' ? '未填写作品描述' : '');
+    item.description?.trim() || (status === 'draft' ? '未填写作品描述' : '');
 
   return {
     id: item.profileId,