feat: adding initial download sources

src/main/events/cloud-save/upload-save-game.ts

@@ -1,8 +1,8 @@
 import { CloudSync } from "@main/services";
 import { registerEvent } from "../register-event";
 import type { GameShop } from "@types";
-import { t } from "i18next";
-import { format } from "date-fns";
+import i18next, { t } from "i18next";
+import { formatDate } from "date-fns";
 
 const uploadSaveGame = async (
   _event: Electron.IpcMainInvokeEvent,
@@ -10,13 +10,15 @@ const uploadSaveGame = async (
   shop: GameShop,
   downloadOptionTitle: string | null
 ) => {
+  const { language } = i18next;
+
   return CloudSync.uploadSaveGame(
     objectId,
     shop,
     downloadOptionTitle,
     t("backup_from", {
       ns: "game_details",
-      date: format(new Date(), "dd/MM/yyyy"),
+      date: formatDate(new Date(), language),
     })
   );
 };

src/main/events/download-sources/create-download-sources.ts

@@ -1,13 +1,13 @@
 import { HydraApi } from "@main/services";
 import { registerEvent } from "../register-event";
 
-const createDownloadSource = async (
+const createDownloadSources = async (
   _event: Electron.IpcMainInvokeEvent,
-  url: string
+  urls: string[]
 ) => {
   await HydraApi.post("/profile/download-sources", {
-    url,
+    urls,
   });
 };
 
-registerEvent("createDownloadSource", createDownloadSource);
+registerEvent("createDownloadSources", createDownloadSources);

src/main/events/index.ts

@@ -90,7 +90,7 @@ import "./themes/get-custom-theme-by-id";
 import "./themes/get-active-custom-theme";
 import "./themes/close-editor-window";
 import "./themes/toggle-custom-theme";
-import "./download-sources/create-download-source";
+import "./download-sources/create-download-sources";
 import "./download-sources/remove-download-source";
 import "./download-sources/get-download-sources";
 import { isPortableVersion } from "@main/helpers";

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

@@ -12,7 +12,15 @@ const startGameDownload = async (
   _event: Electron.IpcMainInvokeEvent,
   payload: StartGameDownloadPayload
 ) => {
-  const { objectId, title, shop, downloadPath, downloader, uri } = payload;
+  const {
+    objectId,
+    title,
+    shop,
+    downloadPath,
+    downloader,
+    uri,
+    automaticallyExtract,
+  } = payload;
 
   const gameKey = levelKeys.game(shop, objectId);
 
@@ -74,6 +82,8 @@ const startGameDownload = async (
     shouldSeed: false,
     timestamp: Date.now(),
     queued: true,
+    extracting: false,
+    automaticallyExtract,
   };
 
   try {

src/main/events/user-preferences/update-user-preferences.ts

@@ -23,10 +23,6 @@ const updateUserPreferences = async (
     patchUserProfile({ language: preferences.language }).catch(() => {});
   }
 
-  if (!preferences.downloadsPath) {
-    preferences.downloadsPath = null;
-  }
-
   await db.put<string, UserPreferences>(
     levelKeys.userPreferences,
     {

src/main/knex-client.ts

@@ -1,11 +0,0 @@
-import knex from "knex";
-import { databasePath } from "./constants";
-import { app } from "electron";
-
-export const knexClient = knex({
-  debug: !app.isPackaged,
-  client: "better-sqlite3",
-  connection: {
-    filename: databasePath,
-  },
-});

src/main/knexfile.ts

@@ -1,10 +0,0 @@
-const config = {
-  development: {
-    migrations: {
-      extension: "ts",
-      stub: "migrations/migration.stub",
-    },
-  },
-};
-
-export default config;

src/main/level/sublevels/keys.ts

@@ -13,6 +13,5 @@ export const levelKeys = {
   downloads: "downloads",
   userPreferences: "userPreferences",
   language: "language",
-  sqliteMigrationDone: "sqliteMigrationDone",
   screenState: "screenState",
 };

src/main/main.ts

@@ -1,4 +1,4 @@
-import { DownloadManager, logger, Ludusavi, startMainLoop } from "./services";
+import { DownloadManager, Ludusavi, startMainLoop } from "./services";
 import { RealDebridClient } from "./services/download/real-debrid";
 import { HydraApi } from "./services/hydra-api";
 import { uploadGamesBatch } from "./services/library-sync";
@@ -6,26 +6,17 @@ import { Aria2 } from "./services/aria2";
 import { downloadsSublevel } from "./level/sublevels/downloads";
 import { sortBy } from "lodash-es";
 import { Downloader } from "@shared";
-import {
-  gameAchievementsSublevel,
-  gamesSublevel,
-  levelKeys,
-  db,
-} from "./level";
-import { Auth, User, type UserPreferences } from "@types";
-import { knexClient } from "./knex-client";
+import { levelKeys, db } from "./level";
+import type { UserPreferences } from "@types";
 import { TorBoxClient } from "./services/download/torbox";
 
 export const loadState = async () => {
-  const userPreferences = await migrateFromSqlite().then(async () => {
-    await db.put<string, boolean>(levelKeys.sqliteMigrationDone, true, {
+  const userPreferences = await db.get<string, UserPreferences | null>(
+    levelKeys.userPreferences,
+    {
       valueEncoding: "json",
-    });
-
-    return db.get<string, UserPreferences | null>(levelKeys.userPreferences, {
-      valueEncoding: "json",
-    });
-  });
+    }
+  );
 
   await import("./events");
 
@@ -52,6 +43,15 @@ export const loadState = async () => {
       return sortBy(games, "timestamp", "DESC");
     });
 
+  downloads.forEach((download) => {
+    if (download.extracting) {
+      downloadsSublevel.put(levelKeys.game(download.shop, download.objectId), {
+        ...download,
+        extracting: false,
+      });
+    }
+  });
+
   const [nextItemOnQueue] = downloads.filter((game) => game.queued);
 
   const downloadsToSeed = downloads.filter(
@@ -66,137 +66,3 @@ export const loadState = async () => {
 
   startMainLoop();
 };
-
-const migrateFromSqlite = async () => {
-  const sqliteMigrationDone = await db.get(levelKeys.sqliteMigrationDone);
-
-  if (sqliteMigrationDone) {
-    return;
-  }
-
-  const migrateGames = knexClient("game")
-    .select("*")
-    .then((games) => {
-      return gamesSublevel.batch(
-        games.map((game) => ({
-          type: "put",
-          key: levelKeys.game(game.shop, game.objectID),
-          value: {
-            objectId: game.objectID,
-            shop: game.shop,
-            title: game.title,
-            iconUrl: game.iconUrl,
-            playTimeInMilliseconds: game.playTimeInMilliseconds,
-            lastTimePlayed: game.lastTimePlayed,
-            remoteId: game.remoteId,
-            winePrefixPath: game.winePrefixPath,
-            launchOptions: game.launchOptions,
-            executablePath: game.executablePath,
-            isDeleted: game.isDeleted === 1,
-          },
-        }))
-      );
-    })
-    .then(() => {
-      logger.info("Games migrated successfully");
-    });
-
-  const migrateUserPreferences = knexClient("user_preferences")
-    .select("*")
-    .then(async (userPreferences) => {
-      if (userPreferences.length > 0) {
-        const { realDebridApiToken, ...rest } = userPreferences[0];
-
-        await db.put<string, UserPreferences>(
-          levelKeys.userPreferences,
-          {
-            ...rest,
-            realDebridApiToken,
-            preferQuitInsteadOfHiding: rest.preferQuitInsteadOfHiding === 1,
-            runAtStartup: rest.runAtStartup === 1,
-            startMinimized: rest.startMinimized === 1,
-            disableNsfwAlert: rest.disableNsfwAlert === 1,
-            seedAfterDownloadComplete: rest.seedAfterDownloadComplete === 1,
-            showHiddenAchievementsDescription:
-              rest.showHiddenAchievementsDescription === 1,
-            downloadNotificationsEnabled:
-              rest.downloadNotificationsEnabled === 1,
-            repackUpdatesNotificationsEnabled:
-              rest.repackUpdatesNotificationsEnabled === 1,
-            achievementNotificationsEnabled:
-              rest.achievementNotificationsEnabled === 1,
-          },
-          { valueEncoding: "json" }
-        );
-
-        if (rest.language) {
-          await db.put<string, string>(levelKeys.language, rest.language, {
-            valueEncoding: "utf-8",
-          });
-        }
-      }
-    })
-    .then(() => {
-      logger.info("User preferences migrated successfully");
-    });
-
-  const migrateAchievements = knexClient("game_achievement")
-    .select("*")
-    .then((achievements) => {
-      return gameAchievementsSublevel.batch(
-        achievements.map((achievement) => ({
-          type: "put",
-          key: levelKeys.game(achievement.shop, achievement.objectId),
-          value: {
-            achievements: JSON.parse(achievement.achievements),
-            unlockedAchievements: JSON.parse(achievement.unlockedAchievements),
-          },
-        }))
-      );
-    })
-    .then(() => {
-      logger.info("Achievements migrated successfully");
-    });
-
-  const migrateUser = knexClient("user_auth")
-    .select("*")
-    .then(async (users) => {
-      if (users.length > 0) {
-        await db.put<string, User>(
-          levelKeys.user,
-          {
-            id: users[0].userId,
-            displayName: users[0].displayName,
-            profileImageUrl: users[0].profileImageUrl,
-            backgroundImageUrl: users[0].backgroundImageUrl,
-            subscription: users[0].subscription,
-          },
-          {
-            valueEncoding: "json",
-          }
-        );
-
-        await db.put<string, Auth>(
-          levelKeys.auth,
-          {
-            accessToken: users[0].accessToken,
-            refreshToken: users[0].refreshToken,
-            tokenExpirationTimestamp: users[0].tokenExpirationTimestamp,
-          },
-          {
-            valueEncoding: "json",
-          }
-        );
-      }
-    })
-    .then(() => {
-      logger.info("User data migrated successfully");
-    });
-
-  return Promise.allSettled([
-    migrateGames,
-    migrateUserPreferences,
-    migrateAchievements,
-    migrateUser,
-  ]);
-};

src/main/services/7zip.ts

@@ -0,0 +1,38 @@
+import { app } from "electron";
+import cp from "node:child_process";
+import path from "node:path";
+
+export const binaryName = {
+  linux: "7zzs",
+  darwin: "7zz",
+  win32: "7zr.exe",
+};
+
+export class _7Zip {
+  private static readonly binaryPath = app.isPackaged
+    ? path.join(process.resourcesPath, binaryName[process.platform])
+    : path.join(
+        __dirname,
+        "..",
+        "..",
+        "binaries",
+        binaryName[process.platform]
+      );
+
+  public static extractFile(
+    filePath: string,
+    outputPath: string,
+    cb: () => void
+  ) {
+    const child = cp.spawn(this.binaryPath, [
+      "x",
+      filePath,
+      "-o" + outputPath,
+      "-y",
+    ]);
+
+    child.on("exit", () => {
+      cb();
+    });
+  }
+}

src/main/services/aria2.ts

@@ -2,18 +2,15 @@ import path from "node:path";
 import cp from "node:child_process";
 import { app } from "electron";
 
-export const startAria2 = () => {};
-
 export class Aria2 {
   private static process: cp.ChildProcess | null = null;
+  private static readonly binaryPath = app.isPackaged
+    ? path.join(process.resourcesPath, "aria2", "aria2c")
+    : path.join(__dirname, "..", "..", "aria2", "aria2c");
 
   public static spawn() {
-    const binaryPath = app.isPackaged
-      ? path.join(process.resourcesPath, "aria2", "aria2c")
-      : path.join(__dirname, "..", "..", "aria2", "aria2c");
-
     this.process = cp.spawn(
-      binaryPath,
+      this.binaryPath,
       [
         "--enable-rpc",
         "--rpc-listen-all",

src/main/services/download/download-manager.ts

@@ -1,6 +1,9 @@
 import { Downloader, DownloadError } from "@shared";
 import { WindowManager } from "../window-manager";
-import { publishDownloadCompleteNotification } from "../notifications";
+import {
+  publishDownloadCompleteNotification,
+  publishExtractionCompleteNotification,
+} from "../notifications";
 import type { Download, DownloadProgress, UserPreferences } from "@types";
 import {
   GofileApi,
@@ -22,9 +25,11 @@ import { logger } from "../logger";
 import { db, downloadsSublevel, gamesSublevel, levelKeys } from "@main/level";
 import { sortBy } from "lodash-es";
 import { TorBoxClient } from "./torbox";
+import { _7Zip } from "../7zip";
 
 export class DownloadManager {
   private static downloadingGameId: string | null = null;
+  private static readonly extensionsToExtract = [".rar", ".zip", ".7z"];
 
   public static async startRPC(
     download?: Download,
@@ -150,13 +155,47 @@ export class DownloadManager {
             queued: false,
           });
         } else {
+          const shouldExtract =
+            download.downloader !== Downloader.Torrent &&
+            this.extensionsToExtract.some((ext) =>
+              download.folderName?.endsWith(ext)
+            ) &&
+            download.automaticallyExtract;
+
           downloadsSublevel.put(gameId, {
             ...download,
             status: "complete",
             shouldSeed: false,
             queued: false,
+            extracting: shouldExtract,
           });
 
+          if (shouldExtract) {
+            _7Zip.extractFile(
+              path.join(download.downloadPath, download.folderName!),
+              path.join(
+                download.downloadPath,
+                path.parse(download.folderName!).name
+              ),
+              async () => {
+                const download = await downloadsSublevel.get(gameId);
+
+                downloadsSublevel.put(gameId, {
+                  ...download!,
+                  extracting: false,
+                });
+
+                WindowManager.mainWindow?.webContents.send(
+                  "on-extraction-complete",
+                  game.shop,
+                  game.objectId
+                );
+
+                publishExtractionCompleteNotification(game);
+              }
+            );
+          }
+
           this.cancelDownload(gameId);
         }
 

src/main/services/index.ts

@@ -8,3 +8,4 @@ export * from "./main-loop";
 export * from "./hydra-api";
 export * from "./ludusavi";
 export * from "./cloud-sync";
+export * from "./7zip";

src/main/services/notifications/index.ts

@@ -128,6 +128,17 @@ export const publishCombinedNewAchievementNotification = async (
   }
 };
 
+export const publishExtractionCompleteNotification = async (game: Game) => {
+  new Notification({
+    title: t("extraction_complete", { ns: "notifications" }),
+    body: t("game_extracted", {
+      ns: "notifications",
+      title: game.title,
+    }),
+    icon: trayIcon,
+  }).show();
+};
+
 export const publishNewAchievementNotification = async (info: {
   achievements: { displayName: string; iconUrl: string }[];
   unlockedAchievementCount: number;

src/main/services/process-watcher.ts

@@ -6,9 +6,9 @@ import axios from "axios";
 import { exec } from "child_process";
 import { ProcessPayload } from "./download/types";
 import { gamesSublevel, levelKeys } from "@main/level";
-import { t } from "i18next";
+import i18next, { t } from "i18next";
 import { CloudSync } from "./cloud-sync";
-import { format } from "date-fns";
+import { formatDate } from "date-fns";
 
 const commands = {
   findWineDir: `lsof -c wine 2>/dev/null | grep '/drive_c/windows$' | head -n 1 | awk '{for(i=9;i<=NF;i++) printf "%s ", $i; print ""}'`,
@@ -229,6 +229,8 @@ function onOpenGame(game: Game) {
   if (game.remoteId) {
     updateGamePlaytime(game, 0, new Date()).catch(() => {});
 
+    const { language } = i18next;
+
     if (game.automaticCloudSync) {
       CloudSync.uploadSaveGame(
         game.objectId,
@@ -236,7 +238,7 @@ function onOpenGame(game: Game) {
         null,
         t("automatic_backup_from", {
           ns: "game_details",
-          date: format(new Date(), "dd/MM/yyyy"),
+          date: formatDate(new Date(), language),
         })
       );
     }
@@ -296,6 +298,8 @@ const onCloseGame = (game: Game) => {
   )!;
   gamesPlaytime.delete(levelKeys.game(game.shop, game.objectId));
 
+  const { language } = i18next;
+
   if (game.remoteId) {
     updateGamePlaytime(
       game,
@@ -310,7 +314,7 @@ const onCloseGame = (game: Game) => {
         null,
         t("automatic_backup_from", {
           ns: "game_details",
-          date: format(new Date(), "dd/MM/yyyy"),
+          date: formatDate(new Date(), language),
         })
       );
     }