Skip to content

Commit

Permalink
chore: merge with main
Browse files Browse the repository at this point in the history
  • Loading branch information
thegrannychaseroperation committed Feb 1, 2025
2 parents 12eb76f + 260fc46 commit 0c8bccc
Show file tree
Hide file tree
Showing 6 changed files with 55 additions and 2 deletions.
14 changes: 12 additions & 2 deletions src/main/services/download/download-manager.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ import { Downloader } from "@shared";
import { WindowManager } from "../window-manager";
import { publishDownloadCompleteNotification } from "../notifications";
import type { Download, DownloadProgress, UserPreferences } from "@types";
import { GofileApi, QiwiApi, DatanodesApi } from "../hosters";
import { GofileApi, QiwiApi, DatanodesApi, MediafireApi } from "../hosters";
import { PythonRPC } from "../python-rpc";
import {
LibtorrentPayload,
Expand Down Expand Up @@ -109,7 +109,7 @@ export class DownloadManager {

if (!download || !game) return;

const userPreferences = await db.get<string, UserPreferences>(
const userPreferences = await db.get<string, UserPreferences | null>(
levelKeys.userPreferences,
{
valueEncoding: "json",
Expand Down Expand Up @@ -298,6 +298,16 @@ export class DownloadManager {
save_path: download.downloadPath,
};
}
case Downloader.Mediafire: {
const downloadUrl = await MediafireApi.getDownloadUrl(download.uri);

return {
action: "start",
game_id: downloadId,
url: downloadUrl,
save_path: download.downloadPath,
};
}
case Downloader.Torrent:
return {
action: "start",
Expand Down
1 change: 1 addition & 0 deletions src/main/services/hosters/index.ts
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
export * from "./gofile";
export * from "./qiwi";
export * from "./datanodes";
export * from "./mediafire";
39 changes: 39 additions & 0 deletions src/main/services/hosters/mediafire.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,39 @@
import axios, { AxiosResponse } from "axios";
import { JSDOM } from "jsdom";

export class MediafireApi {
private static readonly session = axios.create();

public static async getDownloadUrl(mediafireUrl: string): Promise<string> {
const response: AxiosResponse<string> = await this.session.get(
mediafireUrl,
{
headers: {
"User-Agent":
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/125.0.0.0 Safari/537.36",
},
maxRedirects: 0,
validateStatus: (status: number) => status === 200 || status === 302,
}
);

if (response.status === 302) {
const location = response.headers["location"];
if (!location) {
throw new Error("Missing location header in 302 redirect response");
}
return location;
}

const dom = new JSDOM(response.data);
const downloadButton = dom.window.document.querySelector(
"a#downloadButton"
) as HTMLAnchorElement;

if (!downloadButton?.href) {
throw new Error("Download button URL not found in page content");
}

return downloadButton.href;
}
}
1 change: 1 addition & 0 deletions src/renderer/src/constants.ts
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ export const DOWNLOADER_NAME = {
[Downloader.PixelDrain]: "PixelDrain",
[Downloader.Qiwi]: "Qiwi",
[Downloader.Datanodes]: "Datanodes",
[Downloader.Mediafire]: "Mediafire",
};

export const MAX_MINUTES_TO_SHOW_IN_PLAYTIME = 120;
1 change: 1 addition & 0 deletions src/shared/constants.ts
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ export enum Downloader {
PixelDrain,
Qiwi,
Datanodes,
Mediafire,
}

export enum DownloadSourceStatus {
Expand Down
1 change: 1 addition & 0 deletions src/shared/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,7 @@ export const getDownloadersForUri = (uri: string) => {
if (uri.startsWith("https://pixeldrain.com")) return [Downloader.PixelDrain];
if (uri.startsWith("https://qiwi.gg")) return [Downloader.Qiwi];
if (uri.startsWith("https://datanodes.to")) return [Downloader.Datanodes];
if (uri.startsWith("https://www.mediafire.com")) return [Downloader.Mediafire];

if (realDebridHosts.some((host) => uri.startsWith(host)))
return [Downloader.RealDebrid];
Expand Down

0 comments on commit 0c8bccc

Please sign in to comment.