feat: update unread count

RSSNext · May 29, 2024 · d177f9f · d177f9f
1 parent fb6b136
commit d177f9f
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 1 deletion.
diff --git a/src/renderer/src/components/entry-column/item-wrapper.tsx b/src/renderer/src/components/entry-column/item-wrapper.tsx
@@ -27,6 +27,7 @@ export function EntryItemWrapper({
 
   const updateEntry = useUpdateEntry({
     entryId: entry?.entries.id,
+    feedId: entry?.feeds.id,
   })
 
   const read = useMutation({

diff --git a/src/renderer/src/hooks/useEntryActions.tsx b/src/renderer/src/hooks/useEntryActions.tsx
@@ -32,6 +32,7 @@ export const useEntryActions = ({
 
   const updateEntry = useUpdateEntry({
     entryId: entry?.entries.id,
+    feedId: entry?.feeds.id,
   })
 
   const collect = useMutation({

diff --git a/src/renderer/src/hooks/useUpdateEntry.tsx b/src/renderer/src/hooks/useUpdateEntry.tsx
@@ -1,10 +1,17 @@
 import type { EntriesResponse, ListResponse } from "@renderer/lib/types"
 import { Queries } from "@renderer/queries"
+import type { Response as SubscriptionsResponse } from "@renderer/queries/subscriptions"
 import type { InfiniteData, QueryKey } from "@tanstack/react-query"
 import { useQueryClient } from "@tanstack/react-query"
 import { produce } from "immer"
 
-export const useUpdateEntry = ({ entryId }: { entryId?: string }) => {
+export const useUpdateEntry = ({
+  entryId,
+  feedId,
+}: {
+  entryId?: string
+  feedId?: string
+}) => {
   const queryClient = useQueryClient()
 
   const updateEntry = (changed: Partial<EntriesResponse[number]>) => {
@@ -34,6 +41,28 @@ export const useUpdateEntry = ({ entryId }: { entryId?: string }) => {
       })
       queryClient.setQueryData<typeof assertData>(key, finaldata)
     })
+
+    if (changed.read !== undefined) {
+      const entriesData = queryClient.getQueriesData({
+        queryKey: ["subscriptions"],
+      })
+      entriesData.forEach(([key, data]: [QueryKey, unknown]) => {
+        const chage = changed.read ? -1 : 1
+        const assertData = data as SubscriptionsResponse
+        const finaldata = produce(assertData, (assertData) => {
+          for (const list of assertData.list) {
+            for (const item of list.list) {
+              if (item.feeds.id === feedId) {
+                assertData.unread += chage
+                list.unread += chage
+                item.unread = (item.unread || 0) + chage
+              }
+            }
+          }
+        })
+        queryClient.setQueryData<typeof assertData>(key, finaldata)
+      })
+    }
   }
 
   return updateEntry