111 lines
2.8 KiB
JavaScript
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);
|
|
}
|