Merge branch 'feature/seed-completed-downloads' into feat/achievements-points

src/main/entity/game.entity.ts

@@ -85,6 +85,9 @@ export class Game {
   @Column("boolean", { default: false })
   isDeleted: boolean;
 
+  @Column("boolean", { default: false })
+  shouldSeed: boolean;
+
   @CreateDateColumn()
   createdAt: Date;
 

src/main/entity/user-preferences.entity.ts

@@ -41,6 +41,9 @@ export class UserPreferences {
   @Column("boolean", { default: false })
   disableNsfwAlert: boolean;
 
+  @Column("boolean", { default: true })
+  seedAfterDownloadComplete: boolean;
+
   @Column("boolean", { default: false })
   showHiddenAchievementsDescription: boolean;
 

src/main/events/auth/sign-out.ts

@@ -1,10 +1,5 @@
 import { registerEvent } from "../register-event";
-import {
-  DownloadManager,
-  HydraApi,
-  PythonInstance,
-  gamesPlaytime,
-} from "@main/services";
+import { DownloadManager, HydraApi, gamesPlaytime } from "@main/services";
 import { dataSource } from "@main/data-source";
 import { DownloadQueue, Game, UserAuth, UserSubscription } from "@main/entity";
 
@@ -32,7 +27,8 @@ const signOut = async (_event: Electron.IpcMainInvokeEvent) => {
   DownloadManager.cancelDownload();
 
   /* Disconnects libtorrent */
-  PythonInstance.killTorrent();
+  // TODO
+  // TorrentDownloader.killTorrent();
 
   HydraApi.handleSignOut();
 

src/main/events/index.ts

@@ -32,6 +32,8 @@ import "./torrenting/cancel-game-download";
 import "./torrenting/pause-game-download";
 import "./torrenting/resume-game-download";
 import "./torrenting/start-game-download";
+import "./torrenting/pause-game-seed";
+import "./torrenting/resume-game-seed";
 import "./user-preferences/get-user-preferences";
 import "./user-preferences/update-user-preferences";
 import "./user-preferences/auto-launch";

src/main/events/library/close-game.ts

@@ -1,8 +1,10 @@
 import { gameRepository } from "@main/repository";
 import { registerEvent } from "../register-event";
-import { PythonInstance, logger } from "@main/services";
+import { logger } from "@main/services";
 import sudo from "sudo-prompt";
 import { app } from "electron";
+import { PythonRPC } from "@main/services/python-rpc";
+import { ProcessPayload } from "@main/services/download/types";
 
 const getKillCommand = (pid: number) => {
   if (process.platform == "win32") {
@@ -16,7 +18,10 @@ const closeGame = async (
   _event: Electron.IpcMainInvokeEvent,
   gameId: number
 ) => {
-  const processes = await PythonInstance.getProcessList();
+  const processes =
+    (await PythonRPC.rpc.get<ProcessPayload[] | null>("/process-list")).data ||
+    [];
+
   const game = await gameRepository.findOne({
     where: { id: gameId, isDeleted: false },
   });

src/main/events/profile/process-profile-image.ts

@@ -1,11 +1,11 @@
 import { registerEvent } from "../register-event";
-import { PythonInstance } from "@main/services";
 
 const processProfileImage = async (
   _event: Electron.IpcMainInvokeEvent,
   path: string
 ) => {
-  return PythonInstance.processProfileImage(path);
+  return path;
+  // return PythonInstance.processProfileImage(path);
 };
 
 registerEvent("processProfileImage", processProfileImage);

src/main/events/torrenting/pause-game-seed.ts

@@ -0,0 +1,20 @@
+import { registerEvent } from "../register-event";
+
+import { DownloadManager } from "@main/services";
+import { dataSource } from "@main/data-source";
+import { Game } from "@main/entity";
+
+const pauseGameSeed = async (
+  _event: Electron.IpcMainInvokeEvent,
+  gameId: number
+) => {
+  await dataSource.transaction(async (transactionalEntityManager) => {
+    await transactionalEntityManager
+      .getRepository(Game)
+      .update({ id: gameId }, { status: "complete", shouldSeed: false });
+  });
+
+  await DownloadManager.cancelDownload(gameId);
+};
+
+registerEvent("pauseGameSeed", pauseGameSeed);

src/main/events/torrenting/resume-game-seed.ts

@@ -0,0 +1,32 @@
+import { registerEvent } from "../register-event";
+import { gameRepository } from "../../repository";
+
+import { DownloadManager } from "@main/services";
+import { dataSource } from "@main/data-source";
+import { Game } from "@main/entity";
+
+const resumeGameSeed = async (
+  _event: Electron.IpcMainInvokeEvent,
+  gameId: number
+) => {
+  const game = await gameRepository.findOne({
+    where: {
+      id: gameId,
+      isDeleted: false,
+      downloader: 1,
+      progress: 1,
+    },
+  });
+
+  if (!game) return;
+
+  await dataSource.transaction(async (transactionalEntityManager) => {
+    await transactionalEntityManager
+      .getRepository(Game)
+      .update({ id: gameId }, { status: "seeding", shouldSeed: true });
+  });
+
+  await DownloadManager.startDownload(game);
+};
+
+registerEvent("resumeGameSeed", resumeGameSeed);

src/main/events/user-preferences/authenticate-real-debrid.ts

@@ -1,4 +1,4 @@
-import { RealDebridClient } from "@main/services/real-debrid";
+import { RealDebridClient } from "@main/services/download/real-debrid";
 import { registerEvent } from "../register-event";
 
 const authenticateRealDebrid = async (

src/main/index.ts

@@ -5,12 +5,14 @@ import path from "node:path";
 import url from "node:url";
 import fs from "node:fs";
 import { electronApp, optimizer } from "@electron-toolkit/utils";
-import { logger, PythonInstance, WindowManager } from "@main/services";
+import { logger, WindowManager } from "@main/services";
 import { dataSource } from "@main/data-source";
 import resources from "@locales";
 import { userPreferencesRepository } from "@main/repository";
 import { knexClient, migrationConfig } from "./knex-client";
 import { databaseDirectory } from "./constants";
+import { PythonRPC } from "./services/python-rpc";
+import { Aria2 } from "./services/aria2";
 
 const { autoUpdater } = updater;
 
@@ -146,7 +148,8 @@ app.on("window-all-closed", () => {
 
 app.on("before-quit", () => {
   /* Disconnects libtorrent */
-  PythonInstance.kill();
+  PythonRPC.kill();
+  Aria2.kill();
 });
 
 app.on("activate", () => {

src/main/knex-client.ts

@@ -13,6 +13,8 @@ import { AddBackgroundImageUrl } from "./migrations/20241016100249_add_backgroun
 import { AddWinePrefixToGame } from "./migrations/20241019081648_add_wine_prefix_to_game";
 import { AddStartMinimizedColumn } from "./migrations/20241030171454_add_start_minimized_column";
 import { AddDisableNsfwAlertColumn } from "./migrations/20241106053733_add_disable_nsfw_alert_column";
+import { AddShouldSeedColumn } from "./migrations/20241108200154_add_should_seed_colum";
+import { AddSeedAfterDownloadColumn } from "./migrations/20241108201806_add_seed_after_download";
 import { AddHiddenAchievementDescriptionColumn } from "./migrations/20241216140633_add_hidden_achievement_description_column ";
 
 export type HydraMigration = Knex.Migration & { name: string };
@@ -32,6 +34,8 @@ class MigrationSource implements Knex.MigrationSource<HydraMigration> {
       AddWinePrefixToGame,
       AddStartMinimizedColumn,
       AddDisableNsfwAlertColumn,
+      AddShouldSeedColumn,
+      AddSeedAfterDownloadColumn,
       AddHiddenAchievementDescriptionColumn,
     ]);
   }

src/main/main.ts

@@ -1,21 +1,20 @@
+import { Ludusavi, startMainLoop } from "./services";
 import {
-  DownloadManager,
-  Ludusavi,
-  PythonInstance,
-  startMainLoop,
-} from "./services";
-import {
-  downloadQueueRepository,
+  // downloadQueueRepository,
   userPreferencesRepository,
 } from "./repository";
 import { UserPreferences } from "./entity";
-import { RealDebridClient } from "./services/real-debrid";
+import { RealDebridClient } from "./services/download/real-debrid";
 import { HydraApi } from "./services/hydra-api";
 import { uploadGamesBatch } from "./services/library-sync";
+import { PythonRPC } from "./services/python-rpc";
+import { Aria2 } from "./services/aria2";
 
 const loadState = async (userPreferences: UserPreferences | null) => {
   import("./events");
 
+  Aria2.spawn();
+
   if (userPreferences?.realDebridApiToken) {
     RealDebridClient.authorize(userPreferences?.realDebridApiToken);
   }
@@ -26,20 +25,23 @@ const loadState = async (userPreferences: UserPreferences | null) => {
     uploadGamesBatch();
   });
 
-  const [nextQueueItem] = await downloadQueueRepository.find({
-    order: {
-      id: "DESC",
-    },
-    relations: {
-      game: true,
-    },
-  });
+  // const [nextQueueItem] = await downloadQueueRepository.find({
+  //   order: {
+  //     id: "DESC",
+  //   },
+  //   relations: {
+  //     game: true,
+  //   },
+  // });
 
-  if (nextQueueItem?.game.status === "active") {
-    DownloadManager.startDownload(nextQueueItem.game);
-  } else {
-    PythonInstance.spawn();
-  }
+  PythonRPC.spawn();
+  // start download
+
+  // if (nextQueueItem?.game.status === "active") {
+  //   DownloadManager.startDownload(nextQueueItem.game);
+  // } else {
+  //   PythonInstance.spawn();
+  // }
 
   startMainLoop();
 };

src/main/migrations/20241108200154_add_should_seed_colum.ts

@@ -0,0 +1,17 @@
+import type { HydraMigration } from "@main/knex-client";
+import type { Knex } from "knex";
+
+export const AddShouldSeedColumn: HydraMigration = {
+  name: "AddShouldSeedColumn",
+  up: (knex: Knex) => {
+    return knex.schema.alterTable("game", (table) => {
+      return table.boolean("shouldSeed").notNullable().defaultTo(true);
+    });
+  },
+
+  down: async (knex: Knex) => {
+    return knex.schema.alterTable("game", (table) => {
+      return table.dropColumn("shouldSeed");
+    });
+  },
+};

src/main/migrations/20241108201806_add_seed_after_download.ts

@@ -0,0 +1,20 @@
+import type { HydraMigration } from "@main/knex-client";
+import type { Knex } from "knex";
+
+export const AddSeedAfterDownloadColumn: HydraMigration = {
+  name: "AddSeedAfterDownloadColumn",
+  up: (knex: Knex) => {
+    return knex.schema.alterTable("user_preferences", (table) => {
+      return table
+        .boolean("seedAfterDownloadComplete")
+        .notNullable()
+        .defaultTo(true);
+    });
+  },
+
+  down: async (knex: Knex) => {
+    return knex.schema.alterTable("user_preferences", (table) => {
+      return table.dropColumn("seedAfterDownloadComplete");
+    });
+  },
+};

src/main/services/aria2.ts

@@ -0,0 +1,32 @@
+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;
+
+  public static spawn() {
+    const binaryPath = app.isPackaged
+      ? path.join(process.resourcesPath, "aria2", "aria2c")
+      : path.join(__dirname, "..", "..", "aria2", "aria2c");
+
+    this.process = cp.spawn(
+      binaryPath,
+      [
+        "--enable-rpc",
+        "--rpc-listen-all",
+        "--file-allocation=none",
+        "--allow-overwrite=true",
+      ],
+      { stdio: "inherit", windowsHide: true }
+    );
+
+    console.log(this.process);
+  }
+
+  public static kill() {
+    this.process?.kill();
+  }
+}

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

@@ -1,39 +1,102 @@
 import { Game } from "@main/entity";
 import { Downloader } from "@shared";
-import { PythonInstance } from "./python-instance";
 import { WindowManager } from "../window-manager";
-import { downloadQueueRepository, gameRepository } from "@main/repository";
+import {
+  downloadQueueRepository,
+  gameRepository,
+  userPreferencesRepository,
+} from "@main/repository";
 import { publishDownloadCompleteNotification } from "../notifications";
-import { RealDebridDownloader } from "./real-debrid-downloader";
 import type { DownloadProgress } from "@types";
 import { GofileApi, QiwiApi } from "../hosters";
-import { GenericHttpDownloader } from "./generic-http-downloader";
+import { PythonRPC } from "../python-rpc";
+import {
+  LibtorrentPayload,
+  LibtorrentStatus,
+  PauseDownloadPayload,
+} from "./types";
+import { calculateETA, getDirSize } from "./helpers";
+import { QueryDeepPartialEntity } from "typeorm/query-builder/QueryPartialEntity";
+import { RealDebridClient } from "./real-debrid";
+import path from "path";
 
 export class DownloadManager {
-  private static currentDownloader: Downloader | null = null;
   private static downloadingGameId: number | null = null;
 
-  public static async watchDownloads() {
-    let status: DownloadProgress | null = null;
+  private static async getDownloadStatus() {
+    const response = await PythonRPC.rpc.get<LibtorrentPayload | null>(
+      "/status"
+    );
 
-    if (this.currentDownloader === Downloader.Torrent) {
-      status = await PythonInstance.getStatus();
-    } else if (this.currentDownloader === Downloader.RealDebrid) {
-      status = await RealDebridDownloader.getStatus();
-    } else {
-      status = await GenericHttpDownloader.getStatus();
+    if (response.data === null || !this.downloadingGameId) return null;
+
+    const gameId = this.downloadingGameId;
+
+    try {
+      const {
+        progress,
+        numPeers,
+        numSeeds,
+        downloadSpeed,
+        bytesDownloaded,
+        fileSize,
+        folderName,
+        status,
+      } = response.data;
+
+      const isDownloadingMetadata =
+        status === LibtorrentStatus.DownloadingMetadata;
+
+      const isCheckingFiles = status === LibtorrentStatus.CheckingFiles;
+
+      if (!isDownloadingMetadata && !isCheckingFiles) {
+        const update: QueryDeepPartialEntity<Game> = {
+          bytesDownloaded,
+          fileSize,
+          progress,
+          status: "active",
+        };
+
+        await gameRepository.update(
+          { id: gameId },
+          {
+            ...update,
+            folderName,
+          }
+        );
+      }
+
+      return {
+        numPeers,
+        numSeeds,
+        downloadSpeed,
+        timeRemaining: calculateETA(fileSize, bytesDownloaded, downloadSpeed),
+        isDownloadingMetadata,
+        isCheckingFiles,
+        progress,
+        gameId,
+      } as DownloadProgress;
+    } catch (err) {
+      return null;
     }
+  }
+
+  public static async watchDownloads() {
+    const status = await this.getDownloadStatus();
+
+    //   // status = await RealDebridDownloader.getStatus();
 
     if (status) {
       const { gameId, progress } = status;
-
       const game = await gameRepository.findOne({
         where: { id: gameId, isDeleted: false },
       });
+      const userPreferences = await userPreferencesRepository.findOneBy({
+        id: 1,
+      });
 
       if (WindowManager.mainWindow && game) {
         WindowManager.mainWindow.setProgressBar(progress === 1 ? -1 : progress);
-
         WindowManager.mainWindow.webContents.send(
           "on-download-progress",
           JSON.parse(
@@ -44,12 +107,27 @@ export class DownloadManager {
           )
         );
       }
-
       if (progress === 1 && game) {
         publishDownloadCompleteNotification(game);
 
-        await downloadQueueRepository.delete({ game });
+        if (
+          userPreferences?.seedAfterDownloadComplete &&
+          game.downloader === Downloader.Torrent
+        ) {
+          gameRepository.update(
+            { id: gameId },
+            { status: "seeding", shouldSeed: true }
+          );
+        } else {
+          gameRepository.update(
+            { id: gameId },
+            { status: "complete", shouldSeed: false }
+          );
 
+          this.cancelDownload(gameId);
+        }
+
+        await downloadQueueRepository.delete({ game });
         const [nextQueueItem] = await downloadQueueRepository.find({
           order: {
             id: "DESC",
@@ -58,25 +136,117 @@ export class DownloadManager {
             game: true,
           },
         });
-
         if (nextQueueItem) {
           this.resumeDownload(nextQueueItem.game);
+        } else {
+          this.downloadingGameId = -1;
         }
       }
     }
   }
 
+  public static async getSeedStatus() {
+    const seedStatus = await PythonRPC.rpc.get<LibtorrentPayload[] | []>(
+      "/seed-status"
+    );
+
+    if (!seedStatus.data.length) return;
+
+    seedStatus.data.forEach(async (status) => {
+      const game = await gameRepository.findOne({
+        where: { id: status.gameId },
+      });
+
+      if (!game) return;
+
+      const totalSize = await getDirSize(
+        path.join(game.downloadPath!, status.folderName!)
+      );
+
+      if (totalSize < status.fileSize) {
+        await this.cancelDownload(game.id);
+
+        await gameRepository.update(game.id, {
+          status: "paused",
+          shouldSeed: false,
+          progress: totalSize / status.fileSize,
+        });
+
+        WindowManager.mainWindow?.webContents.send("on-hard-delete");
+      }
+    });
+
+    WindowManager.mainWindow?.webContents.send(
+      "on-seeding-status",
+      JSON.parse(JSON.stringify(seedStatus.data))
+    );
+
+    // const gamesToSeed = await gameRepository.find({
+    //   where: { shouldSeed: true, isDeleted: false },
+    // });
+    // if (gamesToSeed.length === 0) return;
+    // const seedStatus = await PythonRPC.rpc
+    //   .get<LibtorrentPayload[] | null>("/seed-status")
+    //   .then((results) => {
+    //     if (results === null) return [];
+    //     return results.data;
+    //   });
+    // if (!seedStatus.length === 0) {
+    //   for (const game of gamesToSeed) {
+    //     if (game.uri && game.downloadPath) {
+    //       await this.resumeSeeding(game.id, game.uri, game.downloadPath);
+    //     }
+    //   }
+    // }
+    // const gameIds = seedStatus.map((status) => status.gameId);
+    // for (const gameId of gameIds) {
+    //   const game = await gameRepository.findOne({
+    //     where: { id: gameId },
+    //   });
+    //   if (game) {
+    //     const isNotDeleted = fs.existsSync(
+    //       path.join(game.downloadPath!, game.folderName!)
+    //     );
+    //     if (!isNotDeleted) {
+    //       await this.pauseSeeding(game.id);
+    //       await gameRepository.update(game.id, {
+    //         status: "complete",
+    //         shouldSeed: false,
+    //       });
+    //       WindowManager.mainWindow?.webContents.send("on-hard-delete");
+    //     }
+    //   }
+    // }
+    // const updateList = await gameRepository.find({
+    //   where: {
+    //     id: In(gameIds),
+    //     status: Not(In(["complete", "seeding"])),
+    //     shouldSeed: true,
+    //     isDeleted: false,
+    //   },
+    // });
+    // if (updateList.length > 0) {
+    //   await gameRepository.update(
+    //     { id: In(updateList.map((game) => game.id)) },
+    //     { status: "seeding" }
+    //   );
+    // }
+    // WindowManager.mainWindow?.webContents.send(
+    //   "on-seeding-status",
+    //   JSON.parse(JSON.stringify(seedStatus))
+    // );
+  }
+
   static async pauseDownload() {
-    if (this.currentDownloader === Downloader.Torrent) {
-      await PythonInstance.pauseDownload();
-    } else if (this.currentDownloader === Downloader.RealDebrid) {
-      await RealDebridDownloader.pauseDownload();
-    } else {
-      await GenericHttpDownloader.pauseDownload();
-    }
+    await PythonRPC.rpc
+      .post("/action", {
+        action: "pause",
+        game_id: this.downloadingGameId,
+      } as PauseDownloadPayload)
+      .catch(() => {});
 
     WindowManager.mainWindow?.setProgressBar(-1);
-    this.currentDownloader = null;
+
     this.downloadingGameId = null;
   }
 
@@ -85,16 +255,13 @@ export class DownloadManager {
   }
 
   static async cancelDownload(gameId = this.downloadingGameId!) {
-    if (this.currentDownloader === Downloader.Torrent) {
-      PythonInstance.cancelDownload(gameId);
-    } else if (this.currentDownloader === Downloader.RealDebrid) {
-      RealDebridDownloader.cancelDownload(gameId);
-    } else {
-      GenericHttpDownloader.cancelDownload(gameId);
-    }
+    await PythonRPC.rpc.post("/action", {
+      action: "cancel",
+      game_id: gameId,
+    });
 
     WindowManager.mainWindow?.setProgressBar(-1);
-    this.currentDownloader = null;
+
     this.downloadingGameId = null;
   }
 
@@ -106,34 +273,57 @@ export class DownloadManager {
         const token = await GofileApi.authorize();
         const downloadLink = await GofileApi.getDownloadLink(id!);
 
-        GenericHttpDownloader.startDownload(game, downloadLink, {
-          Cookie: `accountToken=${token}`,
+        await PythonRPC.rpc.post("/action", {
+          action: "start",
+          game_id: game.id,
+          url: downloadLink,
+          save_path: game.downloadPath,
+          header: `Cookie: accountToken=${token}`,
         });
         break;
       }
       case Downloader.PixelDrain: {
         const id = game!.uri!.split("/").pop();
 
-        await GenericHttpDownloader.startDownload(
-          game,
-          `https://pixeldrain.com/api/file/${id}?download`
-        );
+        await PythonRPC.rpc.post("/action", {
+          action: "start",
+          game_id: game.id,
+          url: `https://pixeldrain.com/api/file/${id}?download`,
+          save_path: game.downloadPath,
+        });
         break;
       }
       case Downloader.Qiwi: {
         const downloadUrl = await QiwiApi.getDownloadUrl(game.uri!);
 
-        await GenericHttpDownloader.startDownload(game, downloadUrl);
+        await PythonRPC.rpc.post("/action", {
+          action: "start",
+          game_id: game.id,
+          url: downloadUrl,
+          save_path: game.downloadPath,
+        });
         break;
       }
       case Downloader.Torrent:
-        PythonInstance.startDownload(game);
+        await PythonRPC.rpc.post("/action", {
+          action: "start",
+          game_id: game.id,
+          url: game.uri,
+          save_path: game.downloadPath,
+        });
         break;
-      case Downloader.RealDebrid:
-        RealDebridDownloader.startDownload(game);
+      case Downloader.RealDebrid: {
+        const downloadUrl = await RealDebridClient.getDownloadUrl(game.uri!);
+
+        await PythonRPC.rpc.post("/action", {
+          action: "start",
+          game_id: game.id,
+          url: downloadUrl,
+          save_path: game.downloadPath,
+        });
+      }
     }
 
-    this.currentDownloader = game.downloader;
     this.downloadingGameId = game.id;
   }
 }

src/main/services/download/generic-http-downloader.ts

@@ -1,109 +0,0 @@
-import { Game } from "@main/entity";
-import { gameRepository } from "@main/repository";
-import { calculateETA } from "./helpers";
-import { DownloadProgress } from "@types";
-import { HttpDownload } from "./http-download";
-
-export class GenericHttpDownloader {
-  public static downloads = new Map<number, HttpDownload>();
-  public static downloadingGame: Game | null = null;
-
-  public static async getStatus() {
-    if (this.downloadingGame) {
-      const download = this.downloads.get(this.downloadingGame.id)!;
-      const status = download.getStatus();
-
-      if (status) {
-        const progress =
-          Number(status.completedLength) / Number(status.totalLength);
-
-        await gameRepository.update(
-          { id: this.downloadingGame!.id },
-          {
-            bytesDownloaded: Number(status.completedLength),
-            fileSize: Number(status.totalLength),
-            progress,
-            status: "active",
-            folderName: status.folderName,
-          }
-        );
-
-        const result = {
-          numPeers: 0,
-          numSeeds: 0,
-          downloadSpeed: status.downloadSpeed,
-          timeRemaining: calculateETA(
-            status.totalLength,
-            status.completedLength,
-            status.downloadSpeed
-          ),
-          isDownloadingMetadata: false,
-          isCheckingFiles: false,
-          progress,
-          gameId: this.downloadingGame!.id,
-        } as DownloadProgress;
-
-        if (progress === 1) {
-          this.downloads.delete(this.downloadingGame.id);
-          this.downloadingGame = null;
-        }
-
-        return result;
-      }
-    }
-
-    return null;
-  }
-
-  static async pauseDownload() {
-    if (this.downloadingGame) {
-      const httpDownload = this.downloads.get(this.downloadingGame!.id!);
-
-      if (httpDownload) {
-        await httpDownload.pauseDownload();
-      }
-
-      this.downloadingGame = null;
-    }
-  }
-
-  static async startDownload(
-    game: Game,
-    downloadUrl: string,
-    headers?: Record<string, string>
-  ) {
-    this.downloadingGame = game;
-
-    if (this.downloads.has(game.id)) {
-      await this.resumeDownload(game.id!);
-      return;
-    }
-
-    const httpDownload = new HttpDownload(
-      game.downloadPath!,
-      downloadUrl,
-      headers
-    );
-
-    httpDownload.startDownload();
-
-    this.downloads.set(game.id!, httpDownload);
-  }
-
-  static async cancelDownload(gameId: number) {
-    const httpDownload = this.downloads.get(gameId);
-
-    if (httpDownload) {
-      await httpDownload.cancelDownload();
-      this.downloads.delete(gameId);
-    }
-  }
-
-  static async resumeDownload(gameId: number) {
-    const httpDownload = this.downloads.get(gameId);
-
-    if (httpDownload) {
-      await httpDownload.resumeDownload();
-    }
-  }
-}

src/main/services/download/helpers.ts

@@ -1,3 +1,6 @@
+import path from "node:path";
+import fs from "node:fs";
+
 export const calculateETA = (
   totalLength: number,
   completedLength: number,
@@ -11,3 +14,26 @@ export const calculateETA = (
 
   return -1;
 };
+
+export const getDirSize = async (dir: string): Promise<number> => {
+  const getItemSize = async (filePath: string): Promise<number> => {
+    const stat = await fs.promises.stat(filePath);
+
+    if (stat.isDirectory()) {
+      return getDirSize(filePath);
+    }
+
+    return stat.size;
+  };
+
+  try {
+    const files = await fs.promises.readdir(dir);
+    const filePaths = files.map((file) => path.join(dir, file));
+    const sizes = await Promise.all(filePaths.map(getItemSize));
+
+    return sizes.reduce((total, size) => total + size, 0);
+  } catch (error) {
+    console.error(error);
+    return 0;
+  }
+};

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

@@ -1,54 +0,0 @@
-import { WindowManager } from "../window-manager";
-import path from "node:path";
-
-export class HttpDownload {
-  private downloadItem: Electron.DownloadItem;
-
-  constructor(
-    private downloadPath: string,
-    private downloadUrl: string,
-    private headers?: Record<string, string>
-  ) {}
-
-  public getStatus() {
-    return {
-      completedLength: this.downloadItem.getReceivedBytes(),
-      totalLength: this.downloadItem.getTotalBytes(),
-      downloadSpeed: this.downloadItem.getCurrentBytesPerSecond(),
-      folderName: this.downloadItem.getFilename(),
-    };
-  }
-
-  async cancelDownload() {
-    this.downloadItem.cancel();
-  }
-
-  async pauseDownload() {
-    this.downloadItem.pause();
-  }
-
-  async resumeDownload() {
-    this.downloadItem.resume();
-  }
-
-  async startDownload() {
-    return new Promise((resolve) => {
-      const options = this.headers ? { headers: this.headers } : {};
-      WindowManager.mainWindow?.webContents.downloadURL(
-        this.downloadUrl,
-        options
-      );
-
-      WindowManager.mainWindow?.webContents.session.once(
-        "will-download",
-        (_event, item, _webContents) => {
-          this.downloadItem = item;
-
-          item.setSavePath(path.join(this.downloadPath, item.getFilename()));
-
-          resolve(null);
-        }
-      );
-    });
-  }
-}

src/main/services/download/index.ts

@@ -1,2 +1 @@
 export * from "./download-manager";
-export * from "./python-instance";

src/main/services/download/python-instance.ts

@@ -1,188 +0,0 @@
-import cp from "node:child_process";
-
-import { Game } from "@main/entity";
-import {
-  RPC_PASSWORD,
-  RPC_PORT,
-  startTorrentClient as startRPCClient,
-} from "./torrent-client";
-import { gameRepository } from "@main/repository";
-import type { DownloadProgress } from "@types";
-import { QueryDeepPartialEntity } from "typeorm/query-builder/QueryPartialEntity";
-import { calculateETA } from "./helpers";
-import axios from "axios";
-import {
-  CancelDownloadPayload,
-  StartDownloadPayload,
-  PauseDownloadPayload,
-  LibtorrentStatus,
-  LibtorrentPayload,
-  ProcessPayload,
-} from "./types";
-import { pythonInstanceLogger as logger } from "../logger";
-
-export class PythonInstance {
-  private static pythonProcess: cp.ChildProcess | null = null;
-  private static downloadingGameId = -1;
-
-  private static rpc = axios.create({
-    baseURL: `http://localhost:${RPC_PORT}`,
-    headers: {
-      "x-hydra-rpc-password": RPC_PASSWORD,
-    },
-  });
-
-  public static spawn(args?: StartDownloadPayload) {
-    logger.log("spawning python process with args:", args);
-    this.pythonProcess = startRPCClient(args);
-  }
-
-  public static kill() {
-    if (this.pythonProcess) {
-      logger.log("killing python process");
-      this.pythonProcess.kill();
-      this.pythonProcess = null;
-      this.downloadingGameId = -1;
-    }
-  }
-
-  public static killTorrent() {
-    if (this.pythonProcess) {
-      logger.log("killing torrent in python process");
-      this.rpc.post("/action", { action: "kill-torrent" });
-      this.downloadingGameId = -1;
-    }
-  }
-
-  public static async getProcessList() {
-    return (
-      (await this.rpc.get<ProcessPayload[] | null>("/process-list")).data || []
-    );
-  }
-
-  public static async getStatus() {
-    if (this.downloadingGameId === -1) return null;
-
-    const response = await this.rpc.get<LibtorrentPayload | null>("/status");
-
-    if (response.data === null) return null;
-
-    try {
-      const {
-        progress,
-        numPeers,
-        numSeeds,
-        downloadSpeed,
-        bytesDownloaded,
-        fileSize,
-        folderName,
-        status,
-        gameId,
-      } = response.data;
-
-      this.downloadingGameId = gameId;
-
-      const isDownloadingMetadata =
-        status === LibtorrentStatus.DownloadingMetadata;
-
-      const isCheckingFiles = status === LibtorrentStatus.CheckingFiles;
-
-      if (!isDownloadingMetadata && !isCheckingFiles) {
-        const update: QueryDeepPartialEntity<Game> = {
-          bytesDownloaded,
-          fileSize,
-          progress,
-          status: "active",
-        };
-
-        await gameRepository.update(
-          { id: gameId },
-          {
-            ...update,
-            folderName,
-          }
-        );
-      }
-
-      if (progress === 1 && !isCheckingFiles) {
-        this.downloadingGameId = -1;
-      }
-
-      return {
-        numPeers,
-        numSeeds,
-        downloadSpeed,
-        timeRemaining: calculateETA(fileSize, bytesDownloaded, downloadSpeed),
-        isDownloadingMetadata,
-        isCheckingFiles,
-        progress,
-        gameId,
-      } as DownloadProgress;
-    } catch (err) {
-      return null;
-    }
-  }
-
-  static async pauseDownload() {
-    await this.rpc
-      .post("/action", {
-        action: "pause",
-        game_id: this.downloadingGameId,
-      } as PauseDownloadPayload)
-      .catch(() => {});
-
-    this.downloadingGameId = -1;
-  }
-
-  static async startDownload(game: Game) {
-    if (!this.pythonProcess) {
-      this.spawn({
-        game_id: game.id,
-        magnet: game.uri!,
-        save_path: game.downloadPath!,
-      });
-    } else {
-      await this.rpc
-        .post("/action", {
-          action: "start",
-          game_id: game.id,
-          magnet: game.uri,
-          save_path: game.downloadPath,
-        } as StartDownloadPayload)
-        .catch(this.handleRpcError);
-    }
-
-    this.downloadingGameId = game.id;
-  }
-
-  static async cancelDownload(gameId: number) {
-    await this.rpc
-      .post("/action", {
-        action: "cancel",
-        game_id: gameId,
-      } as CancelDownloadPayload)
-      .catch(() => {});
-
-    this.downloadingGameId = -1;
-  }
-
-  static async processProfileImage(imagePath: string) {
-    return this.rpc
-      .post<{ imagePath: string; mimeType: string }>("/profile-image", {
-        image_path: imagePath,
-      })
-      .then((response) => response.data);
-  }
-
-  private static async handleRpcError(error: unknown) {
-    logger.error(error);
-
-    return this.rpc.get("/healthcheck").catch(() => {
-      logger.error(
-        "RPC healthcheck failed. Killing process and starting again"
-      );
-      this.kill();
-      this.spawn();
-    });
-  }
-}

src/main/services/download/real-debrid-downloader.ts

@@ -1,72 +0,0 @@
-import { Game } from "@main/entity";
-import { RealDebridClient } from "../real-debrid";
-import { HttpDownload } from "./http-download";
-import { GenericHttpDownloader } from "./generic-http-downloader";
-
-export class RealDebridDownloader extends GenericHttpDownloader {
-  private static realDebridTorrentId: string | null = null;
-
-  private static async getRealDebridDownloadUrl() {
-    if (this.realDebridTorrentId) {
-      let torrentInfo = await RealDebridClient.getTorrentInfo(
-        this.realDebridTorrentId
-      );
-
-      if (torrentInfo.status === "waiting_files_selection") {
-        await RealDebridClient.selectAllFiles(this.realDebridTorrentId);
-
-        torrentInfo = await RealDebridClient.getTorrentInfo(
-          this.realDebridTorrentId
-        );
-      }
-
-      const { links, status } = torrentInfo;
-
-      if (status === "downloaded") {
-        const [link] = links;
-
-        const { download } = await RealDebridClient.unrestrictLink(link);
-        return decodeURIComponent(download);
-      }
-
-      return null;
-    }
-
-    if (this.downloadingGame?.uri) {
-      const { download } = await RealDebridClient.unrestrictLink(
-        this.downloadingGame?.uri
-      );
-
-      return decodeURIComponent(download);
-    }
-
-    return null;
-  }
-
-  static async startDownload(game: Game) {
-    if (this.downloads.has(game.id)) {
-      await this.resumeDownload(game.id!);
-      this.downloadingGame = game;
-      return;
-    }
-
-    if (game.uri?.startsWith("magnet:")) {
-      this.realDebridTorrentId = await RealDebridClient.getTorrentId(
-        game!.uri!
-      );
-    }
-
-    this.downloadingGame = game;
-
-    const downloadUrl = await this.getRealDebridDownloadUrl();
-
-    if (downloadUrl) {
-      this.realDebridTorrentId = null;
-
-      const httpDownload = new HttpDownload(game.downloadPath!, downloadUrl);
-      httpDownload.startDownload();
-
-      this.downloads.set(game.id!, httpDownload);
-    }
-  }
-}

src/main/services/download/real-debrid.ts

@@ -83,4 +83,37 @@ export class RealDebridClient {
     const torrent = await RealDebridClient.addMagnet(magnetUri);
     return torrent.id;
   }
+
+  public static async getDownloadUrl(uri: string) {
+    let realDebridTorrentId: string | null = null;
+
+    if (uri.startsWith("magnet:")) {
+      realDebridTorrentId = await this.getTorrentId(uri);
+    }
+
+    if (realDebridTorrentId) {
+      let torrentInfo = await this.getTorrentInfo(realDebridTorrentId);
+
+      if (torrentInfo.status === "waiting_files_selection") {
+        await this.selectAllFiles(realDebridTorrentId);
+
+        torrentInfo = await this.getTorrentInfo(realDebridTorrentId);
+      }
+
+      const { links, status } = torrentInfo;
+
+      if (status === "downloaded") {
+        const [link] = links;
+
+        const { download } = await this.unrestrictLink(link);
+        return decodeURIComponent(download);
+      }
+
+      return null;
+    }
+
+    const { download } = await this.unrestrictLink(uri);
+
+    return decodeURIComponent(download);
+  }
 }

src/main/services/download/torbox.ts

@@ -0,0 +1,96 @@
+import axios, { AxiosInstance } from "axios";
+import parseTorrent from "parse-torrent";
+import type {
+  TorBoxUserRequest,
+  TorBoxTorrentInfoRequest,
+  TorBoxAddTorrentRequest,
+  TorBoxRequestLinkRequest,
+} from "@types";
+
+export class TorBoxClient {
+  private static instance: AxiosInstance;
+  private static baseURL = "https://api.torbox.app/v1/api";
+  public static apiToken: string;
+
+  static authorize(apiToken: string) {
+    this.instance = axios.create({
+      baseURL: this.baseURL,
+      headers: {
+        Authorization: `Bearer ${apiToken}`,
+      },
+    });
+    this.apiToken = apiToken;
+  }
+
+  static async addMagnet(magnet: string) {
+    const form = new FormData();
+    form.append("magnet", magnet);
+
+    const response = await this.instance.post<TorBoxAddTorrentRequest>(
+      "/torrents/createtorrent",
+      form
+    );
+
+    return response.data.data;
+  }
+
+  static async getTorrentInfo(id: number) {
+    const response =
+      await this.instance.get<TorBoxTorrentInfoRequest>("/torrents/mylist");
+    const data = response.data.data;
+
+    const info = data.find((item) => item.id === id);
+
+    if (!info) {
+      return null;
+    }
+
+    return info;
+  }
+
+  static async getUser() {
+    const response = await this.instance.get<TorBoxUserRequest>(`/user/me`);
+    return response.data.data;
+  }
+
+  static async requestLink(id: number) {
+    const searchParams = new URLSearchParams({});
+
+    searchParams.set("token", this.apiToken);
+    searchParams.set("torrent_id", id.toString());
+    searchParams.set("zip_link", "true");
+
+    const response = await this.instance.get<TorBoxRequestLinkRequest>(
+      "/torrents/requestdl?" + searchParams.toString()
+    );
+
+    if (response.status !== 200) {
+      console.error(response.data.error);
+      console.error(response.data.detail);
+      return null;
+    }
+
+    return response.data.data;
+  }
+
+  private static async getAllTorrentsFromUser() {
+    const response =
+      await this.instance.get<TorBoxTorrentInfoRequest>("/torrents/mylist");
+
+    return response.data.data;
+  }
+
+  static async getTorrentId(magnetUri: string) {
+    const userTorrents = await this.getAllTorrentsFromUser();
+
+    const { infoHash } = await parseTorrent(magnetUri);
+    const userTorrent = userTorrents.find(
+      (userTorrent) => userTorrent.hash === infoHash
+    );
+
+    if (userTorrent) return userTorrent.id;
+
+    const torrent = await this.addMagnet(magnetUri);
+    return torrent.torrent_id;
+  }
+}

src/main/services/download/torrent-client.ts

@@ -1,77 +0,0 @@
-import path from "node:path";
-import cp from "node:child_process";
-import crypto from "node:crypto";
-import fs from "node:fs";
-import { app, dialog } from "electron";
-import type { StartDownloadPayload } from "./types";
-import { Readable } from "node:stream";
-import { pythonInstanceLogger as logger } from "../logger";
-
-const binaryNameByPlatform: Partial<Record<NodeJS.Platform, string>> = {
-  darwin: "hydra-download-manager",
-  linux: "hydra-download-manager",
-  win32: "hydra-download-manager.exe",
-};
-
-export const BITTORRENT_PORT = "5881";
-export const RPC_PORT = "8084";
-export const RPC_PASSWORD = crypto.randomBytes(32).toString("hex");
-
-const logStderr = (readable: Readable | null) => {
-  if (!readable) return;
-
-  readable.setEncoding("utf-8");
-  readable.on("data", logger.log);
-};
-
-export const startTorrentClient = (args?: StartDownloadPayload) => {
-  const commonArgs = [
-    BITTORRENT_PORT,
-    RPC_PORT,
-    RPC_PASSWORD,
-    args ? encodeURIComponent(JSON.stringify(args)) : "",
-  ];
-
-  if (app.isPackaged) {
-    const binaryName = binaryNameByPlatform[process.platform]!;
-    const binaryPath = path.join(
-      process.resourcesPath,
-      "hydra-download-manager",
-      binaryName
-    );
-
-    if (!fs.existsSync(binaryPath)) {
-      dialog.showErrorBox(
-        "Fatal",
-        "Hydra Download Manager binary not found. Please check if it has been removed by Windows Defender."
-      );
-
-      app.quit();
-    }
-
-    const childProcess = cp.spawn(binaryPath, commonArgs, {
-      windowsHide: true,
-      stdio: ["inherit", "inherit"],
-    });
-
-    logStderr(childProcess.stderr);
-
-    return childProcess;
-  } else {
-    const scriptPath = path.join(
-      __dirname,
-      "..",
-      "..",
-      "torrent-client",
-      "main.py"
-    );
-
-    const childProcess = cp.spawn("python3", [scriptPath, ...commonArgs], {
-      stdio: ["inherit", "inherit"],
-    });
-
-    logStderr(childProcess.stderr);
-
-    return childProcess;
-  }
-};

src/main/services/download/types.ts

@@ -1,9 +1,3 @@
-export interface StartDownloadPayload {
-  game_id: number;
-  magnet: string;
-  save_path: string;
-}
-
 export interface PauseDownloadPayload {
   game_id: number;
 }
@@ -25,6 +19,7 @@ export interface LibtorrentPayload {
   numPeers: number;
   numSeeds: number;
   downloadSpeed: number;
+  uploadSpeed: number;
   bytesDownloaded: number;
   fileSize: number;
   folderName: string;
@@ -33,7 +28,15 @@ export interface LibtorrentPayload {
 }
 
 export interface ProcessPayload {
-  exe: string;
+  exe: string | null;
   pid: number;
   name: string;
 }
+
+export interface PauseSeedingPayload {
+  game_id: number;
+}
+
+export interface ResumeSeedingPayload {
+  game_id: number;
+}

src/main/services/hydra-api.ts

@@ -30,7 +30,7 @@ export class HydraApi {
   private static instance: AxiosInstance;
 
   private static readonly EXPIRATION_OFFSET_IN_MS = 1000 * 60 * 5; // 5 minutes
-  private static readonly ADD_LOG_INTERCEPTOR = true;
+  private static readonly ADD_LOG_INTERCEPTOR = false;
 
   private static secondsToMilliseconds = (seconds: number) => seconds * 1000;
 

src/main/services/main-loop.ts

@@ -10,6 +10,7 @@ export const startMainLoop = async () => {
       watchProcesses(),
       DownloadManager.watchDownloads(),
       AchievementWatcherManager.watchAchievements(),
+      DownloadManager.getSeedStatus(),
     ]);
 
     await sleep(1500);

src/main/services/process-watcher.ts

@@ -2,10 +2,11 @@ import { gameRepository } from "@main/repository";
 import { WindowManager } from "./window-manager";
 import { createGame, updateGamePlaytime } from "./library-sync";
 import type { GameRunning } from "@types";
-import { PythonInstance } from "./download";
+import { PythonRPC } from "./python-rpc";
 import { Game } from "@main/entity";
 import axios from "axios";
 import { exec } from "child_process";
+import { ProcessPayload } from "./download/types";
 
 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 ""}'`,
@@ -88,12 +89,14 @@ const findGamePathByProcess = (
 };
 
 const getSystemProcessMap = async () => {
-  const processes = await PythonInstance.getProcessList();
+  const processes =
+    (await PythonRPC.rpc.get<ProcessPayload[] | null>("/process-list")).data ||
+    [];
 
   const map = new Map<string, Set<string>>();
 
   processes.forEach((process) => {
-    const key = process.name.toLowerCase();
+    const key = process.name?.toLowerCase();
     const value = process.exe;
 
     if (!key || !value) return;

src/main/services/python-rpc.ts

@@ -0,0 +1,99 @@
+import axios from "axios";
+
+import cp from "node:child_process";
+import fs from "node:fs";
+import path from "node:path";
+import crypto from "node:crypto";
+
+import { logger } from "./logger";
+import { Readable } from "node:stream";
+import { app, dialog } from "electron";
+import { startSeedProcess } from "./seed";
+
+const binaryNameByPlatform: Partial<Record<NodeJS.Platform, string>> = {
+  darwin: "hydra-python-rpc",
+  linux: "hydra-python-rpc",
+  win32: "hydra-python-rpc.exe",
+};
+
+export class PythonRPC {
+  public static readonly BITTORRENT_PORT = "5881";
+  public static readonly RPC_PORT = "8084";
+  private static readonly RPC_PASSWORD = crypto.randomBytes(32).toString("hex");
+
+  private static pythonProcess: cp.ChildProcess | null = null;
+
+  public static rpc = axios.create({
+    baseURL: `http://localhost:${this.RPC_PORT}`,
+    headers: {
+      "x-hydra-rpc-password": this.RPC_PASSWORD,
+    },
+  });
+
+  private static logStderr(readable: Readable | null) {
+    if (!readable) return;
+
+    readable.setEncoding("utf-8");
+    readable.on("data", logger.log);
+  }
+
+  public static spawn() {
+    console.log([this.BITTORRENT_PORT, this.RPC_PORT, this.RPC_PASSWORD]);
+    const commonArgs = [this.BITTORRENT_PORT, this.RPC_PORT, this.RPC_PASSWORD];
+
+    if (app.isPackaged) {
+      const binaryName = binaryNameByPlatform[process.platform]!;
+      const binaryPath = path.join(
+        process.resourcesPath,
+        "hydra-python-rpc",
+        binaryName
+      );
+
+      if (!fs.existsSync(binaryPath)) {
+        dialog.showErrorBox(
+          "Fatal",
+          "Hydra Python Instance binary not found. Please check if it has been removed by Windows Defender."
+        );
+
+        app.quit();
+      }
+
+      const childProcess = cp.spawn(binaryPath, commonArgs, {
+        windowsHide: true,
+        stdio: ["inherit", "inherit"],
+      });
+
+      this.logStderr(childProcess.stderr);
+
+      this.pythonProcess = childProcess;
+    } else {
+      const scriptPath = path.join(
+        __dirname,
+        "..",
+        "..",
+        "python_rpc",
+        "main.py"
+      );
+
+      console.log(scriptPath);
+
+      const childProcess = cp.spawn("python3", [scriptPath, ...commonArgs], {
+        stdio: ["inherit", "inherit"],
+      });
+
+      this.logStderr(childProcess.stderr);
+
+      this.pythonProcess = childProcess;
+
+      startSeedProcess();
+    }
+  }
+
+  public static kill() {
+    if (this.pythonProcess) {
+      logger.log("Killing python process");
+      this.pythonProcess.kill();
+      this.pythonProcess = null;
+    }
+  }
+}

src/main/services/seed.ts

@@ -0,0 +1,23 @@
+import { gameRepository } from "@main/repository";
+import { DownloadManager } from "./download/download-manager";
+import { sleep } from "@main/helpers";
+
+export const startSeedProcess = async () => {
+  const seedList = await gameRepository.find({
+    where: {
+      shouldSeed: true,
+      downloader: 1,
+      progress: 1,
+    },
+  });
+
+  if (seedList.length === 0) return;
+
+  await sleep(1000);
+  // wait for python process to start
+
+  seedList.map(async (game) => {
+    await DownloadManager.startDownload(game);
+    await sleep(100);
+  });
+};

src/main/vite-env.d.ts

@@ -1,7 +1,6 @@
 /// <reference types="vite/client" />
 
 interface ImportMetaEnv {
-  readonly MAIN_VITE_STEAMGRIDDB_API_KEY: string;
   readonly MAIN_VITE_API_URL: string;
   readonly MAIN_VITE_ANALYTICS_API_URL: string;
   readonly MAIN_VITE_AUTH_URL: string;