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

This commit is contained in:
Zamitto 2024-12-23 14:09:07 -03:00
commit 0f396b83f1
8 changed files with 35 additions and 8 deletions

View File

@ -148,6 +148,14 @@ def action():
downloader = downloads.get(game_id)
if downloader:
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:
return jsonify({"error": "Invalid action"}), 400

View File

@ -15,8 +15,8 @@ class ProfileImageProcessor:
mime_type = image.get_format_mimetype()
return image_path, mime_type
else:
newUUID = str(uuid.uuid4())
new_image_path = os.path.join(tempfile.gettempdir(), newUUID) + ".webp"
new_uuid = str(uuid.uuid4())
new_image_path = os.path.join(tempfile.gettempdir(), new_uuid) + ".webp"
image.save(new_image_path)
new_image = Image.open(new_image_path)

View File

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

View File

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

View File

@ -19,6 +19,7 @@ import { calculateETA, getDirSize } from "./helpers";
import { QueryDeepPartialEntity } from "typeorm/query-builder/QueryPartialEntity";
import { RealDebridClient } from "./real-debrid";
import path from "path";
import { logger } from "../logger";
export class DownloadManager {
private static downloadingGameId: number | null = null;
@ -164,6 +165,8 @@ export class DownloadManager {
if (!seedStatus.length) return;
logger.log(seedStatus);
seedStatus.forEach(async (status) => {
const game = await gameRepository.findOne({
where: { id: status.gameId },
@ -172,7 +175,7 @@ export class DownloadManager {
if (!game) return;
const totalSize = await getDirSize(
path.join(game.downloadPath!, status.folderName!)
path.join(game.downloadPath!, status.folderName)
);
if (totalSize < status.fileSize) {
@ -219,6 +222,22 @@ export class DownloadManager {
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) {
switch (game.downloader) {
case Downloader.Gofile: {

View File

@ -9,7 +9,7 @@ import type {
export class RealDebridClient {
private static instance: AxiosInstance;
private static baseURL = "https://api.real-debrid.com/rest/1.0";
private static readonly baseURL = "https://api.real-debrid.com/rest/1.0";
static authorize(apiToken: string) {
this.instance = axios.create({

View File

@ -10,7 +10,7 @@ import { logger } from "../logger";
export class TorBoxClient {
private static instance: AxiosInstance;
private static baseURL = "https://api.torbox.app/v1/api";
private static readonly baseURL = "https://api.torbox.app/v1/api";
public static apiToken: string;
static authorize(apiToken: string) {

View File

@ -28,7 +28,7 @@ export class PythonRPC {
private static pythonProcess: cp.ChildProcess | null = null;
public static rpc = axios.create({
public static readonly rpc = axios.create({
baseURL: `http://localhost:${this.RPC_PORT}`,
headers: {
"x-hydra-rpc-password": this.RPC_PASSWORD,