From 255488ec7decf76ca8cd69aa329d6f195b6d1694 Mon Sep 17 00:00:00 2001
From: Jackson Harper <jacksonh@gmail.com>
Date: Mon, 12 Aug 2024 22:27:24 +0800
Subject: [PATCH] Better handling of pending items

---
 .../templates/library/LibraryContainer.tsx    | 12 +++-
 .../library_items/useLibraryItems.tsx         | 70 +++++++++++++++----
 2 files changed, 68 insertions(+), 14 deletions(-)

diff --git a/packages/web/components/templates/library/LibraryContainer.tsx b/packages/web/components/templates/library/LibraryContainer.tsx
index b07d1678b..74fd822d8 100644
--- a/packages/web/components/templates/library/LibraryContainer.tsx
+++ b/packages/web/components/templates/library/LibraryContainer.tsx
@@ -23,6 +23,7 @@ import {
   useDeleteItem,
   useGetLibraryItems,
   useMoveItemToFolder,
+  useRefreshProcessingItems,
   useUpdateItemReadStatus,
 } from '../../../lib/networking/library_items/useLibraryItems'
 import {
@@ -181,7 +182,16 @@ export function LibraryContainer(props: LibraryContainerProps): JSX.Element {
       .map((li) => li.node.id)
   }, [libraryItems])
 
-  console.log('processingItems: ', processingItems)
+  const refreshProcessingItems = useRefreshProcessingItems()
+
+  useEffect(() => {
+    if (processingItems.length) {
+      refreshProcessingItems.mutateAsync({
+        attempt: 0,
+        itemIds: processingItems,
+      })
+    }
+  }, [processingItems])
 
   const focusFirstItem = useCallback(() => {
     if (libraryItems.length < 1) {
diff --git a/packages/web/lib/networking/library_items/useLibraryItems.tsx b/packages/web/lib/networking/library_items/useLibraryItems.tsx
index a522238fc..7f79f22f3 100644
--- a/packages/web/lib/networking/library_items/useLibraryItems.tsx
+++ b/packages/web/lib/networking/library_items/useLibraryItems.tsx
@@ -24,6 +24,7 @@ import {
   GQL_UPDATE_LIBRARY_ITEM,
 } from './gql'
 import { gqlEndpoint } from '../../appConfig'
+import { useState } from 'react'
 
 function gqlFetcher(
   query: string,
@@ -518,22 +519,65 @@ export const useUpdateItemReadStatus = () => {
   })
 }
 
-export function useRefreshProcessingItems(itemIds: string[]) {
-  const fullQuery = `in:all includes:${itemIds.join(',')}`
+export function useRefreshProcessingItems() {
+  const maxAttempts = 3
 
-  return useQuery({
-    queryKey: ['libraryItems', fullQuery],
-    queryFn: async () => {
-      const response = (await gqlFetcher(GQL_SEARCH_QUERY, {
-        after: 0,
-        first: 10,
-        query: fullQuery,
-        includeContent: false,
-      })) as LibraryItemsData
-      return response.search
+  const delay = (ms: number) =>
+    new Promise((resolve) => setTimeout(resolve, ms))
+
+  const queryClient = useQueryClient()
+  const refreshItems = async (variables: {
+    attempt: number
+    itemIds: string[]
+  }) => {
+    const fullQuery = `in:all includes:${variables.itemIds.join(',')}`
+    const result = (await gqlFetcher(GQL_SEARCH_QUERY, {
+      first: 10,
+      query: fullQuery,
+      includeContent: false,
+    })) as LibraryItemsData
+    if (result.search.errorCodes?.length) {
+      throw new Error(result.search.errorCodes[0])
+    }
+    return result.search
+  }
+  const mutation = useMutation({
+    mutationFn: refreshItems,
+    retry: 3,
+    retryDelay: 10,
+    onSuccess: async (
+      data: LibraryItems,
+      variables: {
+        attempt: number
+        itemIds: string[]
+      }
+    ) => {
+      let shouldRefetch = false
+      console.log('got processing items: ', data.edges)
+      for (const item of data.edges) {
+        if (item.node.state !== State.PROCESSING) {
+          overwriteItemPropertiesInCache(
+            queryClient,
+            item.node.id,
+            undefined,
+            item.node
+          )
+        } else {
+          shouldRefetch = true
+        }
+      }
+      if (shouldRefetch && variables.attempt < maxAttempts) {
+        await delay(3500 * variables.attempt + 1)
+        mutation.mutate({
+          attempt: variables.attempt + 1,
+          itemIds: data.edges
+            .filter((item) => item.node.state == State.PROCESSING)
+            .map((it) => it.node.id),
+        })
+      }
     },
-    enabled: itemIds.length > 0,
   })
+  return mutation
 }
 
 export const useGetLibraryItemContent = (username: string, slug: string) => {