PublicMirrors/hydra
1
0
Fork 0
mirror of https://github.com/hydralauncher/hydra.git synced 2025-01-23 13:34:54 +03:00
Code Issues Actions Packages Projects Releases 1 Wiki Activity

feat: implement pause and resume functionality for game seeding in DownloadManager

Browse Source
This commit is contained in:
Hachi-R 2024-12-23 13:50:57 -03:00
parent ea90b49604
commit 843301c8b6
4 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python_rpc/main.py
View File

@ -148,6 +148,14 @@ def action():
downloader = downloads.get(game_id) downloader = downloads.get(game_id)
if downloader: if downloader:
downloader.cancel_download() downloader.cancel_download()
elif action == 'resume_seeding':
torrent_downloader = TorrentDownloader(torrent_session)
downloads[game_id] = torrent_downloader
torrent_downloader.start_download(data['url'], data['save_path'], "")
elif action == 'pause_seeding':
downloader = downloads.get(game_id)
if downloader:
downloader.cancel_download()
else: else:
return jsonify({"error": "Invalid action"}), 400 return jsonify({"error": "Invalid action"}), 400

2
src/main/events/torrenting/pause-game-seed.ts
View File

@ -11,7 +11,7 @@ const pauseGameSeed = async (
shouldSeed: false, shouldSeed: false,
}); });
await DownloadManager.cancelDownload(gameId); await DownloadManager.pauseSeeding(gameId);
}; };
registerEvent("pauseGameSeed", pauseGameSeed); registerEvent("pauseGameSeed", pauseGameSeed);

2
src/main/events/torrenting/resume-game-seed.ts
View File

@ -22,7 +22,7 @@ const resumeGameSeed = async (
shouldSeed: true, shouldSeed: true,
}); });
await DownloadManager.startDownload(game); await DownloadManager.resumeSeeding(game);
}; };
registerEvent("resumeGameSeed", resumeGameSeed); registerEvent("resumeGameSeed", resumeGameSeed);

19
src/main/services/download/download-manager.ts
View File

@ -19,6 +19,7 @@ import { calculateETA, getDirSize } from "./helpers";
import { QueryDeepPartialEntity } from "typeorm/query-builder/QueryPartialEntity"; import { QueryDeepPartialEntity } from "typeorm/query-builder/QueryPartialEntity";
import { RealDebridClient } from "./real-debrid"; import { RealDebridClient } from "./real-debrid";
import path from "path"; import path from "path";
import { logger } from "../logger";
export class DownloadManager { export class DownloadManager {
private static downloadingGameId: number | null = null; private static downloadingGameId: number | null = null;
@ -164,6 +165,8 @@ export class DownloadManager {
if (!seedStatus.length) return; if (!seedStatus.length) return;
logger.log(seedStatus);
seedStatus.forEach(async (status) => { seedStatus.forEach(async (status) => {
const game = await gameRepository.findOne({ const game = await gameRepository.findOne({
where: { id: status.gameId }, where: { id: status.gameId },
@ -219,6 +222,22 @@ export class DownloadManager {
this.downloadingGameId = null; this.downloadingGameId = null;
} }
static async resumeSeeding(game: Game) {
await PythonRPC.rpc.post("/action", {
action: "resume_seeding",
game_id: game.id,
url: game.uri,
save_path: game.downloadPath,
});
}
static async pauseSeeding(gameId: number) {
await PythonRPC.rpc.post("/action", {
action: "pause_seeding",
game_id: gameId,
});
}
static async startDownload(game: Game) { static async startDownload(game: Game) {
switch (game.downloader) { switch (game.downloader) {
case Downloader.Gofile: { case Downloader.Gofile: {