From 70958c9ef33af2703a2a7ad5be3d1572b810948a Mon Sep 17 00:00:00 2001
From: Hongbo Wu <hongbo@omnivore.app>
Date: Thu, 16 May 2024 18:00:09 +0800
Subject: [PATCH] fix: use new get content api

---
 src/api.ts   | 102 ++++++++++++++++++++++++++++++++++++++++++++++++---
 src/index.ts |   2 +-
 2 files changed, 98 insertions(+), 6 deletions(-)

diff --git a/src/api.ts b/src/api.ts
index 970c443..47f47ff 100644
--- a/src/api.ts
+++ b/src/api.ts
@@ -1,5 +1,89 @@
 import { Item, ItemFormat, Omnivore } from '@omnivore-app/api'
 
+interface GetContentResponse {
+  data: {
+    libraryItemId: string
+    downloadUrl: string
+    error?: string
+  }[]
+}
+
+const baseUrl = (endpoint: string) => endpoint.replace(/\/api\/graphql$/, '')
+
+const getContent = async (
+  endpoint: string,
+  apiKey: string,
+  libraryItemIds: string[]
+): Promise<GetContentResponse> => {
+  const response = await fetch(`${baseUrl(endpoint)}/api/content`, {
+    method: 'POST',
+    headers: {
+      'content-type': 'application/json',
+      Authorization: apiKey,
+    },
+    body: JSON.stringify({ libraryItemIds, format: 'highlightedMarkdown' }),
+  })
+
+  if (!response.ok) {
+    console.error('Failed to fetch content', response.statusText)
+    throw new Error('Failed to fetch content')
+  }
+
+  return (await response.json()) as GetContentResponse
+}
+
+const downloadFromUrl = async (url: string): Promise<string> => {
+  // polling until download is ready or failed
+  const response = await fetch(url)
+  if (!response.ok) {
+    if (response.status === 404) {
+      // retry after 1 second if download returns 404
+      await new Promise((resolve) => setTimeout(resolve, 1000))
+      return downloadFromUrl(url)
+    }
+
+    console.error('Failed to download content', response.statusText)
+    throw new Error('Failed to download content')
+  }
+
+  return await response.text()
+}
+
+const fetchContentForItems = async (
+  endpoint: string,
+  apiKey: string,
+  items: Item[]
+) => {
+  const content = await getContent(
+    endpoint,
+    apiKey,
+    items.map((a) => a.id)
+  )
+
+  await Promise.allSettled(
+    content.data.map(async (c) => {
+      if (c.error) {
+        console.error('Error fetching content', c.error)
+        return
+      }
+
+      const item = items.find((i) => i.id === c.libraryItemId)
+      if (!item) {
+        console.error('Item not found', c.libraryItemId)
+        return
+      }
+
+      // timeout if download takes too long
+      item.content = await Promise.race([
+        downloadFromUrl(c.downloadUrl),
+        new Promise<string>(
+          (_, reject) => setTimeout(() => reject('Timeout'), 60_000) // 60 seconds
+        ),
+      ])
+    })
+  )
+}
+
 export const getOmnivoreItems = async (
   apiKey: string,
   after = 0,
@@ -8,11 +92,11 @@ export const getOmnivoreItems = async (
   query = '',
   includeContent = false,
   format: ItemFormat = 'html',
-  baseUrl?: string
+  endpoint: string
 ): Promise<[Item[], boolean]> => {
   const omnivore = new Omnivore({
     apiKey,
-    baseUrl,
+    baseUrl: baseUrl(endpoint),
     timeoutMs: 10000,
   })
 
@@ -20,12 +104,20 @@ export const getOmnivoreItems = async (
     after,
     first,
     query: `${updatedAt ? 'updated:' + updatedAt : ''} sort:saved-asc ${query}`,
-    includeContent,
+    includeContent: false,
     format,
   })
 
   const items = result.edges.map((e) => e.node)
 
+  if (includeContent && items.length > 0) {
+    try {
+      await fetchContentForItems(endpoint, apiKey, items)
+    } catch (error) {
+      console.error('Error fetching content', error)
+    }
+  }
+
   return [items, result.pageInfo.hasNextPage]
 }
 
@@ -34,11 +126,11 @@ export const getDeletedOmnivoreItems = async (
   after = 0,
   first = 10,
   updatedAt = '',
-  baseUrl: string
+  endpoint: string
 ): Promise<[Item[], boolean]> => {
   const omnivore = new Omnivore({
     apiKey,
-    baseUrl,
+    baseUrl: baseUrl(endpoint),
     timeoutMs: 10000,
   })
 
diff --git a/src/index.ts b/src/index.ts
index 5819e1e..69b0197 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -227,7 +227,7 @@ const fetchOmnivore = async (inBackground = false) => {
     preParseTemplate(articleTemplate)
     preParseTemplate(highlightTemplate)
 
-    const size = 50
+    const size = 15
     for (let after = 0; ; after += size) {
       const [items, hasNextPage] = await getOmnivoreItems(
         apiKey,