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

111 lines
2.8 KiB
JavaScript

/* 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);
}