hydra/src/main/services/real-debrid.ts

import { Game } from "@main/entity";
import axios, { AxiosInstance } from "axios";
import type {
  RealDebridAddMagnet,
  RealDebridTorrentInfo,
  RealDebridUnrestrictLink,
  RealDebridUser,
} from "@types";

const base = "https://api.real-debrid.com/rest/1.0";

export class RealDebridClient {
  private static instance: AxiosInstance;

  static async addMagnet(magnet: string) {
    const searchParams = new URLSearchParams({ magnet });

    const response = await this.instance.post<RealDebridAddMagnet>(
      "/torrents/addMagnet",
      searchParams.toString()
    );

    return response.data;
  }

  static async getInfo(id: string) {
    const response = await this.instance.get<RealDebridTorrentInfo>(
      `/torrents/info/${id}`
    );
    return response.data;
  }

  static async getUser() {
    const response = await this.instance.get<RealDebridUser>(`/user`);
    return response.data;
  }

  static async selectAllFiles(id: string) {
    const searchParams = new URLSearchParams({ files: "all" });

    await this.instance.post(
      `/torrents/selectFiles/${id}`,
      searchParams.toString()
    );
  }

  static async unrestrictLink(link: string) {
    const searchParams = new URLSearchParams({ link });

    const response = await this.instance.post<RealDebridUnrestrictLink>(
      "/unrestrict/link",
      searchParams.toString()
    );

    return response.data;
  }

  static async getAllTorrentsFromUser() {
    const response =
      await this.instance.get<RealDebridTorrentInfo[]>("/torrents");

    return response.data;
  }

  static extractSHA1FromMagnet(magnet: string) {
    return magnet.match(/btih:([0-9a-fA-F]*)/)?.[1].toLowerCase();
  }

  static async getDownloadUrl(game: Game) {
    const torrents = await RealDebridClient.getAllTorrentsFromUser();
    const hash = RealDebridClient.extractSHA1FromMagnet(game!.repack.magnet);
    let torrent = torrents.find((t) => t.hash === hash);

    // User haven't downloaded this torrent yet
    if (!torrent) {
      const magnet = await RealDebridClient.addMagnet(game!.repack.magnet);

      if (magnet) {
        await RealDebridClient.selectAllFiles(magnet.id);
        torrent = await RealDebridClient.getInfo(magnet.id);

        const { links } = torrent;
        const { download } = await RealDebridClient.unrestrictLink(links[0]);

        if (!download) {
          throw new Error("Torrent not cached on Real Debrid");
        }

        return download;
      }
    }

    throw new Error();
  }

  static authorize(apiToken: string) {
    this.instance = axios.create({
      baseURL: base,
      headers: {
        Authorization: `Bearer ${apiToken}`,
      },
    });
  }
}