From 4d64231abf02f1fca25af7aefcc963c4f7395025 Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 12 Oct 2023 12:59:36 +0800
Subject: [PATCH] save content reader in the item

---
 packages/api/src/routers/svc/email_attachment.ts    | 10 +++++++++-
 packages/api/test/routers/email_attachments.test.ts |  1 +
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/packages/api/src/routers/svc/email_attachment.ts b/packages/api/src/routers/svc/email_attachment.ts
index fde4be36d..b9bf3c77c 100644
--- a/packages/api/src/routers/svc/email_attachment.ts
+++ b/packages/api/src/routers/svc/email_attachment.ts
@@ -1,6 +1,10 @@
 import express from 'express'
 import { DeepPartial } from 'typeorm'
-import { LibraryItem, LibraryItemState } from '../../entity/library_item'
+import {
+  ContentReaderType,
+  LibraryItem,
+  LibraryItemState,
+} from '../../entity/library_item'
 import { UploadFile } from '../../entity/upload_file'
 import { env } from '../../env'
 import { PageType, UploadFileStatus } from '../../generated/graphql'
@@ -160,6 +164,10 @@ export function emailAttachmentRouter() {
         slug: generateSlug(title),
         state: LibraryItemState.Succeeded,
         user: { id: user.id },
+        contentReader:
+          itemType === PageType.File
+            ? ContentReaderType.PDF
+            : ContentReaderType.EPUB,
       }
 
       const item = await createLibraryItem(itemToCreate, user.id)
diff --git a/packages/api/test/routers/email_attachments.test.ts b/packages/api/test/routers/email_attachments.test.ts
index f699b6a55..79c0f6fd0 100644
--- a/packages/api/test/routers/email_attachments.test.ts
+++ b/packages/api/test/routers/email_attachments.test.ts
@@ -94,6 +94,7 @@ describe('Email attachments Router', () => {
       const item = await findLibraryItemById(res2.body.id, user.id)
 
       expect(item).to.exist
+      expect(item?.contentReader).to.eq('PDF')
     })
   })
 })