From 47fd4d9663473f993e4d5df7c1f97ad7e31dcbcb Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Wed, 17 Apr 2024 09:58:11 +0800
Subject: [PATCH] save id, url, thumbnail and createdAt in the digest

---
 packages/api/src/jobs/ai/create_digest.ts |  6 ++++++
 packages/api/src/services/digest.ts       | 16 ++++++++++++----
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/packages/api/src/jobs/ai/create_digest.ts b/packages/api/src/jobs/ai/create_digest.ts
index 1c8bf442e..d264f08f1 100644
--- a/packages/api/src/jobs/ai/create_digest.ts
+++ b/packages/api/src/jobs/ai/create_digest.ts
@@ -365,6 +365,12 @@ export const createDigestJob = async (jobData: CreateDigestJobData) => {
     urlsToAudio: [],
     jobState: 'completed',
     speechFiles,
+    libraryItems: filteredSummaries.map((item) => ({
+      id: item.libraryItem.id,
+      url: item.libraryItem.originalUrl,
+      thumbnail: item.libraryItem.thumbnail ?? undefined,
+    })),
+    createdAt: new Date(),
   }
 
   await writeDigest(jobData.userId, digest)
diff --git a/packages/api/src/services/digest.ts b/packages/api/src/services/digest.ts
index e3b38d8b7..1c9d20a05 100644
--- a/packages/api/src/services/digest.ts
+++ b/packages/api/src/services/digest.ts
@@ -2,9 +2,20 @@ import { redisDataSource } from '../redis_data_source'
 import { SpeechFile } from '@omnivore/text-to-speech-handler'
 import { logger } from '../utils/logger'
 
+interface Chapter {
+  title: string
+}
+
+interface LibraryItem {
+  id: string
+  url: string
+  thumbnail?: string
+}
+
 export interface Digest {
   id: string
   jobState: string
+  createdAt: Date
 
   url?: string
   title?: string
@@ -13,10 +24,7 @@ export interface Digest {
 
   urlsToAudio?: string[]
   speechFiles?: SpeechFile[]
-}
-
-interface Chapter {
-  title: string
+  libraryItems?: LibraryItem[]
 }
 
 const digestKey = (userId: string) => `digest:${userId}`