F95API/app/scripts/classes/game-download.js

/* istanbul ignore file */

'use strict';

// Core modules
const fs = require('fs');

// Public modules from npm
// const { File } = require('megajs');

// Modules from file
const { prepareBrowser, preparePage } = require('../puppeteer-helper.js');
const shared = require('../shared.js');

class GameDownload {
  constructor() {
    /**
     * @public
     * Platform that hosts game files
     * @type String
     */
    this.hosting = '';
    /**
     * @public
     * Link to game files
     * @type String
     */
    this.link = null;
    /**
     * @public
     * Operating systems supported by the game version indicated in this class.
     * Can be *WINDOWS/LINUX/MACOS*
     * @type String[]
     */
    this.supportedOS = [];
  }

  /**
   * @public
   * Download the game data in the indicated path.
   * Supported hosting platforms: MEGA, NOPY
   * @param {String} path Save path
   * @return {Promise<Boolean>} Result of the operation
   */
  async download(path) {
    if (this.link.includes('mega.nz'))
      return await downloadMEGA(this.link, path);
    else if (this.link.includes('nopy.to'))
      return await downloadNOPY(this.link, path);
  }
}
module.exports = GameDownload;

async function downloadMEGA(url, savepath) {
  // The URL is masked
  const browser = await prepareBrowser();
  const page = await preparePage(browser);
  await page.setCookie(...shared.cookies); // Set cookies to avoid login
  await page.goto(url);
  await page.waitForSelector('a.host_link');

  // Obtain the link for the unmasked page and click it
  const link = await page.$('a.host_link');
  await link.click();
  await page.goto(url, {
    waitUntil: shared.WAIT_STATEMENT
  }); // Go to the game page and wait until it loads

  // Obtain the URL after the redirect
  const downloadURL = page.url();

  // Close browser and page
  await page.close();
  await browser.close();

  const stream = fs.createWriteStream(savepath);
  const file = File.fromURL(downloadURL);
  file.download().pipe(stream);
  return fs.existsSync(savepath);
}

async function downloadNOPY(url, savepath) {
  // Prepare browser
  const browser = await prepareBrowser();
  const page = await preparePage(browser);
  await page.goto(url);
  await page.waitForSelector('#download');

  // Set the save path
  await page._client.send('Page.setDownloadBehavior', {
    behavior: 'allow',
    downloadPath: path.basename(path.dirname(savepath)) // It's a directory
  });

  // Obtain the download button and click it
  const downloadButton = await page.$('#download');
  await downloadButton.click();

  // Await for all the connections to close
  await page.waitForNavigation({
    waitUntil: 'networkidle0',
    timeout: 0 // Disable timeout
  });

  // Close browser and page
  await page.close();
  await browser.close();

  return fs.existsSync(savepath);
}