fix: bug after pause seed

hydralauncher · Dec 23, 2024 · 56217fb · 56217fb
1 parent c9be6b6
commit 56217fb
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 14 deletions.
diff --git a/python_rpc/main.py b/python_rpc/main.py
@@ -59,12 +59,15 @@ def seed_status():
         return auth_error
 
     seed_status = []
+
     for game_id, downloader in downloads.items():
         if not downloader:
             continue
-
+        
         response = downloader.get_download_status()
-
+        if response is None:
+            continue
+
         if response.get('status') == 5:
             seed_status.append({
                 'gameId': game_id,

diff --git a/src/main/services/download/download-manager.ts b/src/main/services/download/download-manager.ts
@@ -158,13 +158,14 @@ export class DownloadManager {
   }
 
   public static async getSeedStatus() {
-    const seedStatus = await PythonRPC.rpc.get<LibtorrentPayload[] | []>(
-      "/seed-status"
-    );
+    const seedStatus = await PythonRPC.rpc
+      .get<LibtorrentPayload[] | []>("/seed-status")
+      .then((res) => res.data);
 
-    if (!seedStatus.data.length) return;
+    console.log(seedStatus);
+    if (!seedStatus.length) return;
 
-    seedStatus.data.forEach(async (status) => {
+    seedStatus.forEach(async (status) => {
       const game = await gameRepository.findOne({
         where: { id: status.gameId },
       });
@@ -188,10 +189,7 @@ export class DownloadManager {
       }
     });
 
-    WindowManager.mainWindow?.webContents.send(
-      "on-seeding-status",
-      JSON.parse(JSON.stringify(seedStatus.data))
-    );
+    WindowManager.mainWindow?.webContents.send("on-seeding-status", seedStatus);
   }
 
   static async pauseDownload() {

diff --git a/src/renderer/src/pages/downloads/download-group.tsx b/src/renderer/src/pages/downloads/download-group.tsx
@@ -77,13 +77,12 @@ export function DownloadGroup({
   };
 
   const seedingMap = useMemo(() => {
-    if (!Array.isArray(seedingStatus) || seedingStatus.length === 0) {
-      return new Map<number, SeedingStatus>();
-    }
     const map = new Map<number, SeedingStatus>();
+
     seedingStatus.forEach((seed) => {
       map.set(seed.gameId, seed);
     });
+
     return map;
   }, [seedingStatus]);