refactor: simplify game entry preparation in download processes by consolidating logic into a helper function

src/main/events/torrenting/add-game-to-queue.ts

@@ -2,14 +2,9 @@ import { registerEvent } from "../register-event";
 import type { Download, StartGameDownloadPayload } from "@types";
 import { DownloadManager, HydraApi, logger } from "@main/services";
 import { createGame } from "@main/services/library-sync";
-import {
-  downloadsSublevel,
-  gamesShopAssetsSublevel,
-  gamesSublevel,
-  levelKeys,
-} from "@main/level";
+import { downloadsSublevel, gamesSublevel, levelKeys } from "@main/level";
 import { parseBytes } from "@shared";
-import { handleDownloadError } from "@main/helpers";
+import { handleDownloadError, prepareGameEntry } from "@main/helpers";
 
 const addGameToQueue = async (
   _event: Electron.IpcMainInvokeEvent,
@@ -54,30 +49,7 @@ const addGameToQueue = async (
     return handleDownloadError(err, downloader);
   }
 
-  const game = await gamesSublevel.get(gameKey);
-  const gameAssets = await gamesShopAssetsSublevel.get(gameKey);
-
-  await downloadsSublevel.del(gameKey);
-
-  if (game) {
-    await gamesSublevel.put(gameKey, {
-      ...game,
-      isDeleted: false,
-    });
-  } else {
-    await gamesSublevel.put(gameKey, {
-      title,
-      iconUrl: gameAssets?.iconUrl ?? null,
-      libraryHeroImageUrl: gameAssets?.libraryHeroImageUrl ?? null,
-      logoImageUrl: gameAssets?.logoImageUrl ?? null,
-      objectId,
-      shop,
-      remoteId: null,
-      playTimeInMilliseconds: 0,
-      lastTimePlayed: null,
-      isDeleted: false,
-    });
-  }
+  await prepareGameEntry({ gameKey, title, objectId, shop });
 
   try {
     await downloadsSublevel.put(gameKey, download);

src/main/events/torrenting/start-game-download.ts

@@ -3,13 +3,8 @@ import type { Download, StartGameDownloadPayload } from "@types";
 import { DownloadManager, HydraApi, logger } from "@main/services";
 import { createGame } from "@main/services/library-sync";
 import { parseBytes } from "@shared";
-import {
-  downloadsSublevel,
-  gamesShopAssetsSublevel,
-  gamesSublevel,
-  levelKeys,
-} from "@main/level";
-import { handleDownloadError } from "@main/helpers";
+import { downloadsSublevel, gamesSublevel, levelKeys } from "@main/level";
+import { handleDownloadError, prepareGameEntry } from "@main/helpers";
 
 const startGameDownload = async (
   _event: Electron.IpcMainInvokeEvent,
@@ -39,30 +34,7 @@ const startGameDownload = async (
     }
   }
 
-  const game = await gamesSublevel.get(gameKey);
-  const gameAssets = await gamesShopAssetsSublevel.get(gameKey);
-
-  await downloadsSublevel.del(gameKey);
-
-  if (game) {
-    await gamesSublevel.put(gameKey, {
-      ...game,
-      isDeleted: false,
-    });
-  } else {
-    await gamesSublevel.put(gameKey, {
-      title,
-      iconUrl: gameAssets?.iconUrl ?? null,
-      libraryHeroImageUrl: gameAssets?.libraryHeroImageUrl ?? null,
-      logoImageUrl: gameAssets?.logoImageUrl ?? null,
-      objectId,
-      shop,
-      remoteId: null,
-      playTimeInMilliseconds: 0,
-      lastTimePlayed: null,
-      isDeleted: false,
-    });
-  }
+  await prepareGameEntry({ gameKey, title, objectId, shop });
 
   await DownloadManager.cancelDownload(gameKey);
 

src/main/helpers/download-game-helper.ts

@@ -0,0 +1,45 @@
+import {
+  downloadsSublevel,
+  gamesShopAssetsSublevel,
+  gamesSublevel,
+} from "@main/level";
+import type { GameShop } from "@types";
+
+interface PrepareGameEntryParams {
+  gameKey: string;
+  title: string;
+  objectId: string;
+  shop: GameShop;
+}
+
+export const prepareGameEntry = async ({
+  gameKey,
+  title,
+  objectId,
+  shop,
+}: PrepareGameEntryParams): Promise<void> => {
+  const game = await gamesSublevel.get(gameKey);
+  const gameAssets = await gamesShopAssetsSublevel.get(gameKey);
+
+  await downloadsSublevel.del(gameKey);
+
+  if (game) {
+    await gamesSublevel.put(gameKey, {
+      ...game,
+      isDeleted: false,
+    });
+  } else {
+    await gamesSublevel.put(gameKey, {
+      title,
+      iconUrl: gameAssets?.iconUrl ?? null,
+      libraryHeroImageUrl: gameAssets?.libraryHeroImageUrl ?? null,
+      logoImageUrl: gameAssets?.logoImageUrl ?? null,
+      objectId,
+      shop,
+      remoteId: null,
+      playTimeInMilliseconds: 0,
+      lastTimePlayed: null,
+      isDeleted: false,
+    });
+  }
+};

src/main/helpers/index.ts

@@ -95,3 +95,4 @@ export const getThemeSoundPath = (
 
 export * from "./reg-parser";
 export * from "./download-error-handler";
+export * from "./download-game-helper";