feat: move initial seeding for DownloadManager

This commit is contained in:
Hachi-R 2024-12-23 14:07:03 -03:00
parent 843301c8b6
commit f853a2a39e
4 changed files with 26 additions and 7 deletions

View file

@ -1,6 +1,7 @@
import { DownloadManager, Ludusavi, startMainLoop } from "./services";
import {
downloadQueueRepository,
gameRepository,
userPreferencesRepository,
} from "./repository";
import { UserPreferences } from "./entity";
@ -35,14 +36,20 @@ const loadState = async (userPreferences: UserPreferences | null) => {
},
});
const seedList = await gameRepository.find({
where: {
shouldSeed: true,
downloader: 1,
progress: 1,
},
});
if (nextQueueItem?.game.status === "active") {
DownloadManager.startRPC(nextQueueItem.game);
DownloadManager.startRPC(nextQueueItem.game, seedList);
} else {
PythonRPC.spawn();
}
await startSeedProcess();
startMainLoop();
};

View file

@ -24,13 +24,17 @@ import { logger } from "../logger";
export class DownloadManager {
private static downloadingGameId: number | null = null;
public static startRPC(game: Game) {
public static startRPC(game: Game, initialSeeding?: Game[]) {
if (game && game.status === "active") {
PythonRPC.spawn({
game_id: game.id,
url: game.uri!,
save_path: game.downloadPath!,
});
}, initialSeeding?.map((game) => ({
game_id: game.id,
url: game.uri!,
save_path: game.downloadPath!,
})));
this.downloadingGameId = game.id;
}

View file

@ -9,7 +9,7 @@ import { logger } from "./logger";
import { Readable } from "node:stream";
import { app, dialog } from "electron";
interface StartDownloadPayload {
interface GamePayload {
game_id: number;
url: string;
save_path: string;
@ -42,12 +42,13 @@ export class PythonRPC {
readable.on("data", logger.log);
}
public static spawn(initialDownload?: StartDownloadPayload) {
public static spawn(initialDownload?: GamePayload, initialSeeding?: GamePayload[]) {
const commonArgs = [
this.BITTORRENT_PORT,
this.RPC_PORT,
this.RPC_PASSWORD,
initialDownload ? JSON.stringify(initialDownload) : "",
initialSeeding ? JSON.stringify(initialSeeding) : "",
];
if (app.isPackaged) {