SingleFile/cli/single-file

#!/usr/bin/env node

/*
 * 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, URL */

const fileUrl = require("file-url");
const args = require("./args");
const fs = require("fs");

const backEnds = {
	jsdom: "./back-ends/jsdom.js",
	puppeteer: "./back-ends/puppeteer.js",
	"puppeteer-firefox": "./back-ends/puppeteer-firefox.js",
	"webdriver-chromium": "./back-ends/webdriver-chromium.js",
	"webdriver-gecko": "./back-ends/webdriver-gecko.js"
};
args.compressCSS = args.compressCss;
args.compressHTML = args.compressHtml;
args.includeBOM = args.includeBom;
if (args.url && !/^(https?|file):\/\//.test(args.url)) {
	args.url = fileUrl(args.url);
}
args.retrieveLinks = true;
args.browserScripts = args.browserScripts.map(path => require.resolve(path));
const backend = require(backEnds[args.backEnd]);
backend.initialize(args).then(() => {
	let tasks;
	if (args.urlsFile) {
		tasks = fs.readFileSync(args.urlsFile).toString().split("\n")
			.map(url => ({ url: rewriteURL(url, args.urlRewriteRules), depth: 0 }))
			.filter(task => task.url);
	} else {
		tasks = [{ url: rewriteURL(args.url, args.urlRewriteRules), depth: 0 }];
	}
	return runTasks(tasks, args);
}).then(() => {
	if (!args.browserDebug) {
		return backend.closeBrowser();
	}
});

async function runTasks(tasks, options) {
	const availableTasks = tasks.filter(task => !task.status).length;
	const processingTasks = tasks.filter(task => task.status == "processing").length;
	const promisesTasks = [];
	for (let workerIndex = 0; workerIndex < Math.min(availableTasks, options.maxParallelWorkers - processingTasks); workerIndex++) {
		promisesTasks.push(runNextTask(tasks, options));
	}
	await Promise.all(promisesTasks);
}

async function runNextTask(tasks, options) {
	const task = tasks.find(task => !task.status);
	if (task) {
		options = JSON.parse(JSON.stringify(options));
		options.url = task.url;
		options.output = null;
		task.status = "processing";
		const pageData = await capturePage(options);
		task.status = "processed";
		if (pageData && options.crawlLinks) {
			pageData.links = pageData.links
				.map(urlLink => rewriteURL(urlLink, options.urlRewriteRules))
				.filter(urlLink => !tasks.find(task => task.url == urlLink));
			if (options.crawlInnerLinksOnly) {
				const urlHost = getHostURL(options.url);
				pageData.links = pageData.links.filter(urlLink => urlLink.startsWith(urlHost));
			}
			if (task.depth < options.crawlMaxDepth) {
				tasks.splice(tasks.length, 0, ...pageData.links.map(url => ({ url, depth: task.depth + 1 })));
			}
		}
		await runTasks(tasks, options);
	}
}

function rewriteURL(url, rewriteRules) {
	url = url.trim();
	rewriteRules.forEach(rewriteRule => {
		const parts = rewriteRule.split(/ +/);
		if (parts.length == 2) {
			url = url.replace(new RegExp(parts[0]), parts[1]).trim();
		}
	});
	return url;
}

function getHostURL(url) {
	url = new URL(url);
	return url.protocol + "//" + (url.username ? url.username + (url.password || "") + "@" : "") + url.hostname;
}

async function capturePage(options) {
	try {
		const pageData = await backend.getPageData(options);
		if (options.output) {
			fs.writeFileSync(getFilename(options.output), pageData.content);
		} else {
			if (options.filenameTemplate && pageData.filename) {
				fs.writeFileSync(getFilename(pageData.filename), pageData.content);
			} else {
				console.log(pageData.content); // eslint-disable-line no-console
			}
		}
		return pageData;
	} catch (error) {
		const message = "URL: " + options.url + "\nStack: " + error.stack + "\n";
		if (options.errorFile) {
			fs.writeFileSync(options.errorFile, message, { flag: "a" });
		} else {
			console.error(message); // eslint-disable-line no-console
		}
	}
}

function getFilename(filename, index = 1) {
	let newFilename = filename;
	if (index > 1) {
		const regExpMatchExtension = /(\.[^.]+)$/;
		const matchExtension = newFilename.match(regExpMatchExtension);
		if (matchExtension && matchExtension[1]) {
			newFilename = newFilename.replace(regExpMatchExtension, " - " + index + matchExtension[1]);
		} else {
			newFilename += " - " + index;
		}
	}
	if (fs.existsSync(newFilename)) {
		return getFilename(filename, index + 1);
	} else {
		return newFilename;
	}
}