SingleFile/cli/back-ends/jsdom.js

/*
 * Copyright 2010-2020 Gildas Lormeau
 * contact : gildas.lormeau <at> gmail.com
 * 
 * This file is part of SingleFile.
 *
 *   The code in this file is free software: you can redistribute it and/or 
 *   modify it under the terms of the GNU Affero General Public License 
 *   (GNU AGPL) as published by the Free Software Foundation, either version 3
 *   of the License, or (at your option) any later version.
 * 
 *   The code in this file is distributed in the hope that it will be useful, 
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of 
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero 
 *   General Public License for more details.
 *
 *   As additional permission under GNU AGPL version 3 section 7, you may 
 *   distribute UNMODIFIED VERSIONS OF THIS file without the copy of the GNU 
 *   AGPL normally required by section 4, provided you include this license 
 *   notice and a URL through which recipients can access the Corresponding 
 *   Source.
 */

/* global require, exports, Buffer */

const crypto = require("crypto");

const jsdom = require("jsdom");
const { JSDOM, VirtualConsole } = jsdom;
const iconv = require("iconv-lite");

exports.initialize = async () => { };

exports.getPageData = async options => {
	let win;
	try {
		const dom = await JSDOM.fromURL(options.url, await getBrowserOptions(options));
		win = dom.window;
		return await getPageData(win, options);
	} finally {
		if (win) {
			win.close();
		}
	}
};

exports.closeBrowser = () => { };

async function getPageData(win, options) {
	const doc = win.document;
	const scripts = await require("./common/scripts.js").get(options);
	win.TextDecoder = class {
		constructor(utfLabel) {
			this.utfLabel = utfLabel;
		}
		decode(buffer) {
			return iconv.decode(Buffer.from(buffer), this.utfLabel);
		}
	};
	win.crypto = {
		subtle: {
			digest: async function digestText(algo, text) {
				const hash = crypto.createHash(algo.replace("-", "").toLowerCase());
				hash.update(text, "utf-8");
				return hash.digest();
			}
		}
	};
	win.Element.prototype.getBoundingClientRect = undefined;
	win.getComputedStyle = () => { };
	win.eval(scripts);
	if (win.document.readyState == "loading" || win.document.readyState == "interactive") {
		await new Promise(resolve => win.onload = resolve);
	}
	executeFrameScripts(doc, scripts);
	options.removeHiddenElements = false;
	options.loadDeferredImages = false;
	const pageData = await win.singlefile.lib.getPageData(options, { fetch: url => fetchResource(url, options) }, doc, win);
	if (options.includeInfobar) {
		await win.singlefile.common.ui.content.infobar.includeScript(pageData);
	}
	return pageData;

	async function fetchResource(resourceURL) {
		return new Promise((resolve, reject) => {
			const xhrRequest = new win.XMLHttpRequest();
			xhrRequest.withCredentials = true;
			xhrRequest.responseType = "arraybuffer";
			xhrRequest.onerror = event => reject(new Error(event.detail));
			xhrRequest.onreadystatechange = () => {
				if (xhrRequest.readyState == win.XMLHttpRequest.DONE) {
					resolve({
						arrayBuffer: async () => new Uint8Array(xhrRequest.response).buffer,
						headers: {
							get: headerName => xhrRequest.getResponseHeader(headerName)
						},
						status: xhrRequest.status
					});
				}
			};
			xhrRequest.open("GET", resourceURL, true);
			xhrRequest.send();
		});
	}
}

async function getBrowserOptions(options) {
	class ResourceLoader extends jsdom.ResourceLoader {
		_getRequestOptions(fetchOptions) {
			const requestOptions = super._getRequestOptions(fetchOptions);
			if (options.httpHeaders) {
				requestOptions.headers = Object.assign(requestOptions.headers, options.httpHeaders);
			}
			return requestOptions;
		}
	}
	const resourceLoader = new ResourceLoader({
		userAgent: options.userAgent
	});
	const jsdomOptions = {
		virtualConsole: new VirtualConsole(),
		userAgent: options.userAgent,
		pretendToBeVisual: true,
		runScripts: "outside-only",
		resources: resourceLoader
	};
	if (options.browserWidth && options.browserHeight) {
		jsdomOptions.beforeParse = function (window) {
			window.outerWidth = window.innerWidth = options.browserWidth;
			window.outerHeight = window.innerHeight = options.browserHeight;
		};
	}
	if (options.browserCookies && options.browserCookies.length) {
		jsdomOptions.cookieJar = new jsdom.CookieJar();
		await Promise.all(options.browserCookies.map(cookie => {
			let cookieString = cookie.name + "=" + cookie.value;
			if (cookie.path) {
				cookieString += ";path=" + cookie.path;
			}
			if (cookie.domain) {
				cookieString += ";domain=" + cookie.domain;
			}
			if (cookie.expires) {
				cookieString += ";max-age=" + cookie.expires;
			}
			if (cookie.secure) {
				cookieString += ";secure";
			}
			if (cookie.sameSite) {
				cookieString += ";samesite=" + options.sameSite;
			}
			const cookieOptions = {
				http: Boolean(cookie.httpOnly)
			};
			return new Promise((resolve, reject) => jsdomOptions.cookieJar.setCookie(cookieString, options.url, cookieOptions, error => error ? reject(error) : resolve()));
		}));
	}
	return jsdomOptions;
}

function executeFrameScripts(doc, scripts) {
	const frameElements = doc.querySelectorAll("iframe, frame");
	frameElements.forEach(frameElement => {
		try {
			frameElement.contentWindow.Element.prototype.getBoundingClientRect = undefined;
			frameElement.contentWindow.eval(scripts);
			executeFrameScripts(frameElement.contentDocument, scripts);
		} catch (error) {
			// ignored
		}
	});
}