external/hydra
1
0
Fork 0
mirror of https://github.com/hydralauncher/hydra.git synced 2025-03-09 15:40:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use webContents.downloadURL to download http

Browse source
This commit is contained in:
Zamitto 2024-07-15 17:48:10 -03:00
parent 6c6fff71fe
commit b5b7fe31ae
1 changed files with 60 additions and 47 deletions
Download patch file Download diff file Expand all files Collapse all files

107
src/main/services/download/http-download.ts
View file

@ -1,59 +1,42 @@
import type { ChildProcess } from "node:child_process"; import { DownloadItem } from "electron";
import { logger } from "../logger"; import { WindowManager } from "../window-manager";
import { sleep } from "@main/helpers";
import { startAria2 } from "../aria2c";
import Aria2 from "aria2";
export class HTTPDownload { export class HTTPDownload {
private static connected = false; private static id = 0;
private static aria2c: ChildProcess | null = null;
private static aria2 = new Aria2({}); private static downloads: Record<string, DownloadItem> = {};
private static async connect() { public static getStatus(gid: string): {
this.aria2c = startAria2(); completedLength: number;
totalLength: number;
let retries = 0; downloadSpeed: number;
} | null {
while (retries < 4 && !this.connected) { const downloadItem = this.downloads[gid];
try { if (downloadItem) {
await this.aria2.open(); return {
logger.log("Connected to aria2"); completedLength: downloadItem.getReceivedBytes(),
totalLength: downloadItem.getTotalBytes(),
this.connected = true; downloadSpeed: 0,
} catch (err) { };
await sleep(100);
logger.log("Failed to connect to aria2, retrying...");
retries++;
}
}
}
public static getStatus(gid: string) {
if (this.connected) {
return this.aria2.call("tellStatus", gid);
} }
return null; return null;
} }
public static disconnect() {
if (this.aria2c) {
this.aria2c.kill();
this.connected = false;
}
}
static async cancelDownload(gid: string) { static async cancelDownload(gid: string) {
await this.aria2.call("forceRemove", gid); const downloadItem: DownloadItem = this.downloads[gid];
downloadItem?.cancel();
this.downloads;
} }
static async pauseDownload(gid: string) { static async pauseDownload(gid: string) {
await this.aria2.call("forcePause", gid); const downloadItem = this.downloads[gid];
downloadItem?.pause();
} }
static async resumeDownload(gid: string) { static async resumeDownload(gid: string) {
await this.aria2.call("unpause", gid); const downloadItem = this.downloads[gid];
downloadItem?.resume();
} }
static async startDownload( static async startDownload(
@ -61,14 +44,44 @@ export class HTTPDownload {
downloadUrl: string, downloadUrl: string,
header: string[] = [] header: string[] = []
) { ) {
console.log(header); return new Promise<string>((resolve) => {
if (!this.connected) await this.connect(); WindowManager.mainWindow?.webContents.downloadURL(downloadUrl, {
headers: { Cookie: header[0].split(": ")[1] },
});
const options = { WindowManager.mainWindow?.webContents.session.on(
dir: downloadPath, "will-download",
header, (_event, item, _webContents) => {
}; const gid = ++this.id;
return this.aria2.call("addUri", [downloadUrl], options); this.downloads[gid.toString()] = item;
// Set the save path, making Electron not to prompt a save dialog.
item.setSavePath(downloadPath);
item.on("updated", (_event, state) => {
if (state === "interrupted") {
console.log("Download is interrupted but can be resumed");
} else if (state === "progressing") {
if (item.isPaused()) {
console.log("Download is paused");
} else {
console.log(`Received bytes: ${item.getReceivedBytes()}`);
}
}
});
item.once("done", (_event, state) => {
if (state === "completed") {
console.log("Download successfully");
} else {
console.log(`Download failed: ${state}`);
}
});
resolve(gid.toString());
}
);
});
} }
} }