mirror of
https://github.com/webrecorder/browsertrix-crawler.git
synced 2025-10-19 14:33:17 +00:00

* profiles: support loading profiles via a URL. * add 'request' dependency * README: mention profile URLs
988 lines
28 KiB
JavaScript
988 lines
28 KiB
JavaScript
const child_process = require("child_process");
|
|
const path = require("path");
|
|
const fs = require("fs");
|
|
const os = require("os");
|
|
const fsp = require("fs/promises");
|
|
|
|
// to ignore HTTPS error for HEAD check
|
|
const HTTPS_AGENT = require("https").Agent({
|
|
rejectUnauthorized: false,
|
|
});
|
|
|
|
const HTTP_AGENT = require("http").Agent();
|
|
|
|
const fetch = require("node-fetch");
|
|
const puppeteer = require("puppeteer-core");
|
|
const { Cluster } = require("puppeteer-cluster");
|
|
const { RedisCrawlState, MemoryCrawlState } = require("./util/state");
|
|
const AbortController = require("abort-controller");
|
|
const Sitemapper = require("sitemapper");
|
|
const { v4: uuidv4 } = require("uuid");
|
|
const yaml = require("js-yaml");
|
|
|
|
const warcio = require("warcio");
|
|
|
|
const behaviors = fs.readFileSync(path.join(__dirname, "node_modules", "browsertrix-behaviors", "dist", "behaviors.js"), {encoding: "utf8"});
|
|
|
|
const TextExtract = require("./util/textextract");
|
|
const { S3StorageSync, getFileSize } = require("./util/storage");
|
|
const { ScreenCaster, WSTransport, RedisPubSubTransport } = require("./util/screencaster");
|
|
const { parseArgs } = require("./util/argParser");
|
|
const { initRedis } = require("./util/redis");
|
|
|
|
const { getBrowserExe, loadProfile, evaluateWithCLI } = require("./util/browser");
|
|
|
|
const { BEHAVIOR_LOG_FUNC, HTML_TYPES, DEFAULT_SELECTORS } = require("./util/constants");
|
|
|
|
const { BlockRules } = require("./util/blockrules");
|
|
|
|
|
|
// ============================================================================
|
|
class Crawler {
|
|
constructor() {
|
|
this.headers = {};
|
|
this.crawlState = null;
|
|
|
|
this.emulateDevice = null;
|
|
|
|
// pages file
|
|
this.pagesFH = null;
|
|
|
|
// was the limit hit?
|
|
this.limitHit = false;
|
|
|
|
this.userAgent = "";
|
|
|
|
const res = parseArgs();
|
|
this.params = res.parsed;
|
|
this.origConfig = res.origConfig;
|
|
|
|
this.saveStateFiles = [];
|
|
this.lastSaveTime = 0;
|
|
this.saveStateInterval = this.params.saveStateInterval * 1000;
|
|
|
|
this.debugLogging = this.params.logging.includes("debug");
|
|
|
|
if (this.params.profile) {
|
|
this.statusLog("With Browser Profile: " + this.params.profile);
|
|
}
|
|
|
|
this.emulateDevice = this.params.emulateDevice;
|
|
|
|
this.debugLog("Seeds", this.params.scopedSeeds);
|
|
|
|
this.captureBasePrefix = `http://${process.env.PROXY_HOST}:${process.env.PROXY_PORT}/${this.params.collection}/record`;
|
|
this.capturePrefix = this.captureBasePrefix + "/id_/";
|
|
|
|
this.gotoOpts = {
|
|
waitUntil: this.params.waitUntil,
|
|
timeout: this.params.timeout
|
|
};
|
|
|
|
// root collections dir
|
|
this.collDir = path.join(this.params.cwd, "collections", this.params.collection);
|
|
|
|
// pages directory
|
|
this.pagesDir = path.join(this.collDir, "pages");
|
|
|
|
// pages file
|
|
this.pagesFile = path.join(this.pagesDir, "pages.jsonl");
|
|
|
|
this.blockRules = null;
|
|
}
|
|
|
|
statusLog(...args) {
|
|
console.log(...args);
|
|
}
|
|
|
|
debugLog(...args) {
|
|
if (this.debugLogging) {
|
|
console.log(...args);
|
|
}
|
|
}
|
|
|
|
configureUA() {
|
|
// override userAgent
|
|
if (this.params.userAgent) {
|
|
|
|
if (this.emulateDevice) {
|
|
this.emulateDevice.userAgent = this.params.userAgent;
|
|
}
|
|
|
|
this.userAgent = this.params.userAgent;
|
|
return;
|
|
}
|
|
|
|
this.browserExe = getBrowserExe();
|
|
|
|
// if device set, it overrides the default Chrome UA
|
|
if (this.emulateDevice) {
|
|
this.userAgent = this.emulateDevice.userAgent;
|
|
} else {
|
|
let version = process.env.BROWSER_VERSION;
|
|
|
|
try {
|
|
version = child_process.execFileSync(this.browserExe, ["--version"], {encoding: "utf8"});
|
|
version = version.match(/[\d.]+/)[0];
|
|
} catch(e) {
|
|
console.error(e);
|
|
}
|
|
|
|
this.userAgent = `Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/${version} Safari/537.36`;
|
|
}
|
|
|
|
// suffix to append to default userAgent
|
|
if (this.params.userAgentSuffix) {
|
|
this.userAgent += " " + this.params.userAgentSuffix;
|
|
|
|
if (this.emulateDevice) {
|
|
this.emulateDevice.userAgent += " " + this.params.userAgentSuffix;
|
|
}
|
|
}
|
|
}
|
|
|
|
async initCrawlState() {
|
|
const redisUrl = this.params.redisStoreUrl;
|
|
|
|
if (redisUrl) {
|
|
if (!redisUrl.startsWith("redis://")) {
|
|
throw new Error("stateStoreUrl must start with redis:// -- Only redis-based store currently supported");
|
|
}
|
|
|
|
let redis;
|
|
|
|
try {
|
|
redis = await initRedis(redisUrl);
|
|
} catch (e) {
|
|
throw new Error("Unable to connect to state store Redis: " + redisUrl);
|
|
}
|
|
|
|
this.statusLog(`Storing state via Redis ${redisUrl} @ key prefix "${this.params.crawlId}"`);
|
|
|
|
this.crawlState = new RedisCrawlState(redis, this.params.crawlId, this.params.timeout);
|
|
|
|
} else {
|
|
this.statusLog("Storing state in memory");
|
|
|
|
this.crawlState = new MemoryCrawlState();
|
|
}
|
|
|
|
if (this.params.saveState === "always" && this.params.saveStateInterval) {
|
|
this.statusLog(`Saving crawl state every ${this.params.saveStateInterval} seconds, keeping last ${this.params.saveStateHistory} states`);
|
|
}
|
|
|
|
return this.crawlState;
|
|
}
|
|
|
|
initScreenCaster() {
|
|
let transport;
|
|
|
|
if (this.params.screencastPort) {
|
|
transport = new WSTransport(this.params.screencastPort);
|
|
this.debugLog(`Screencast server started on: ${this.params.screencastPort}`);
|
|
} else if (this.params.redisStoreUrl && this.params.screencastRedis) {
|
|
const crawlId = process.env.CRAWL_ID || os.hostname();
|
|
transport = new RedisPubSubTransport(this.params.redisStoreUrl, crawlId);
|
|
this.debugLog("Screencast enabled via redis pubsub");
|
|
}
|
|
|
|
if (!transport) {
|
|
return null;
|
|
}
|
|
|
|
return new ScreenCaster(transport, this.params.workers);
|
|
}
|
|
|
|
bootstrap() {
|
|
let opts = {};
|
|
if (this.params.logging.includes("pywb")) {
|
|
opts = {stdio: "inherit", cwd: this.params.cwd};
|
|
}
|
|
else{
|
|
opts = {stdio: "ignore", cwd: this.params.cwd};
|
|
}
|
|
|
|
this.configureUA();
|
|
|
|
this.headers = {"User-Agent": this.userAgent};
|
|
|
|
const subprocesses = [];
|
|
|
|
subprocesses.push(child_process.spawn("redis-server", {...opts, cwd: "/tmp/"}));
|
|
|
|
child_process.spawnSync("wb-manager", ["init", this.params.collection], opts);
|
|
|
|
opts.env = {...process.env, COLL: this.params.collection, ROLLOVER_SIZE: this.params.rolloverSize};
|
|
|
|
subprocesses.push(child_process.spawn("uwsgi", [path.join(__dirname, "uwsgi.ini")], opts));
|
|
|
|
process.on("exit", () => {
|
|
for (const proc of subprocesses) {
|
|
proc.kill();
|
|
}
|
|
});
|
|
|
|
if (!this.params.headless && !process.env.NO_XVFB) {
|
|
child_process.spawn("Xvfb", [
|
|
process.env.DISPLAY,
|
|
"-listen",
|
|
"tcp",
|
|
"-screen",
|
|
"0",
|
|
process.env.GEOMETRY,
|
|
"-ac",
|
|
"+extension",
|
|
"RANDR"
|
|
]);
|
|
}
|
|
}
|
|
|
|
get chromeArgs() {
|
|
// Chrome Flags, including proxy server
|
|
return [
|
|
...(process.env.CHROME_FLAGS ?? "").split(" ").filter(Boolean),
|
|
"--no-xshm", // needed for Chrome >80 (check if puppeteer adds automatically)
|
|
`--proxy-server=http://${process.env.PROXY_HOST}:${process.env.PROXY_PORT}`,
|
|
"--no-sandbox",
|
|
"--disable-background-media-suspend",
|
|
"--autoplay-policy=no-user-gesture-required",
|
|
"--disable-features=IsolateOrigins,site-per-process",
|
|
"--disable-popup-blocking",
|
|
"--disable-backgrounding-occluded-windows",
|
|
];
|
|
}
|
|
|
|
get puppeteerArgs() {
|
|
// Puppeter Options
|
|
return {
|
|
headless: this.params.headless,
|
|
executablePath: this.browserExe,
|
|
handleSIGINT: false,
|
|
handleSIGTERM: false,
|
|
handleSIGHUP: false,
|
|
ignoreHTTPSErrors: true,
|
|
args: this.chromeArgs,
|
|
userDataDir: this.profileDir,
|
|
defaultViewport: null,
|
|
};
|
|
}
|
|
|
|
async run() {
|
|
await fsp.mkdir(this.params.cwd, {recursive: true});
|
|
|
|
this.bootstrap();
|
|
|
|
try {
|
|
await this.crawl();
|
|
process.exit(0);
|
|
} catch(e) {
|
|
console.error("Crawl failed");
|
|
console.error(e);
|
|
process.exit(1);
|
|
}
|
|
}
|
|
|
|
_behaviorLog({data, type}) {
|
|
switch (type) {
|
|
case "info":
|
|
console.log(JSON.stringify(data));
|
|
break;
|
|
|
|
case "debug":
|
|
default:
|
|
if (this.params.behaviorsLogDebug) {
|
|
console.log("behavior debug: " + JSON.stringify(data));
|
|
}
|
|
}
|
|
}
|
|
|
|
async crawlPage({page, data}) {
|
|
try {
|
|
if (this.screencaster) {
|
|
await this.screencaster.newTarget(page.target());
|
|
}
|
|
|
|
if (this.emulateDevice) {
|
|
await page.emulate(this.emulateDevice);
|
|
}
|
|
|
|
if (this.params.profile) {
|
|
await page._client.send("Network.setBypassServiceWorker", {bypass: true});
|
|
}
|
|
|
|
if (this.params.behaviorOpts && !page.__bx_inited) {
|
|
await page.exposeFunction(BEHAVIOR_LOG_FUNC, (logdata) => this._behaviorLog(logdata));
|
|
await page.evaluateOnNewDocument(behaviors + `;\nself.__bx_behaviors.init(${this.params.behaviorOpts});`);
|
|
page.__bx_inited = true;
|
|
}
|
|
|
|
// run custom driver here
|
|
await this.driver({page, data, crawler: this});
|
|
|
|
const title = await page.title();
|
|
let text = "";
|
|
if (this.params.text) {
|
|
const client = await page.target().createCDPSession();
|
|
const result = await client.send("DOM.getDocument", {"depth": -1, "pierce": true});
|
|
text = await new TextExtract(result).parseTextFromDom();
|
|
}
|
|
|
|
await this.writePage(data, title, this.params.text ? text : null);
|
|
|
|
if (this.params.behaviorOpts) {
|
|
await Promise.allSettled(page.frames().map(frame => evaluateWithCLI(frame, "self.__bx_behaviors.run();")));
|
|
}
|
|
|
|
await this.writeStats();
|
|
|
|
await this.serializeConfig();
|
|
|
|
} catch (e) {
|
|
console.warn(e);
|
|
} finally {
|
|
|
|
try {
|
|
if (this.screencaster) {
|
|
await this.screencaster.endTarget(page.target());
|
|
}
|
|
} catch (e) {
|
|
console.warn(e);
|
|
}
|
|
}
|
|
}
|
|
|
|
async createWARCInfo(filename) {
|
|
const warcVersion = "WARC/1.0";
|
|
const type = "warcinfo";
|
|
const packageFileJSON = JSON.parse(await fsp.readFile("../app/package.json"));
|
|
const warcioPackageJSON = JSON.parse(await fsp.readFile("/app/node_modules/warcio/package.json"));
|
|
const pywbVersion = child_process.execSync("pywb -V", {encoding: "utf8"}).trim().split(" ")[1];
|
|
|
|
const info = {
|
|
"software": `Browsertrix-Crawler ${packageFileJSON.version} (with warcio.js ${warcioPackageJSON.version} pywb ${pywbVersion})`,
|
|
"format": "WARC File Format 1.0"
|
|
};
|
|
|
|
const warcInfo = {...info, ...this.params.warcInfo, };
|
|
const record = await warcio.WARCRecord.createWARCInfo({filename, type, warcVersion}, warcInfo);
|
|
const buffer = await warcio.WARCSerializer.serialize(record, {gzip: true});
|
|
return buffer;
|
|
}
|
|
|
|
async crawl() {
|
|
this.profileDir = await loadProfile(this.params.profile);
|
|
|
|
try {
|
|
this.driver = require(this.params.driver);
|
|
} catch(e) {
|
|
console.warn(e);
|
|
return;
|
|
}
|
|
|
|
if (this.params.generateWACZ && process.env.STORE_ENDPOINT_URL) {
|
|
const endpointUrl = process.env.STORE_ENDPOINT_URL + (process.env.STORE_PATH || "");
|
|
const storeInfo = {
|
|
endpointUrl,
|
|
accessKey: process.env.STORE_ACCESS_KEY,
|
|
secretKey: process.env.STORE_SECRET_KEY,
|
|
};
|
|
|
|
const opts = {
|
|
crawlId: process.env.CRAWL_ID || os.hostname(),
|
|
webhookUrl: process.env.WEBHOOK_URL,
|
|
userId: process.env.STORE_USER,
|
|
filename: process.env.STORE_FILENAME || "@ts-@id.wacz",
|
|
};
|
|
|
|
console.log("Initing Storage...");
|
|
this.storage = new S3StorageSync(storeInfo, opts);
|
|
}
|
|
|
|
// Puppeteer Cluster init and options
|
|
this.cluster = await Cluster.launch({
|
|
concurrency: this.params.newContext,
|
|
maxConcurrency: this.params.workers,
|
|
skipDuplicateUrls: false,
|
|
timeout: this.params.timeout * 2,
|
|
puppeteerOptions: this.puppeteerArgs,
|
|
puppeteer,
|
|
monitor: this.params.logging.includes("stats")
|
|
});
|
|
|
|
|
|
this.cluster.jobQueue = await this.initCrawlState();
|
|
|
|
if (this.params.state) {
|
|
await this.crawlState.load(this.params.state, this.params.scopedSeeds, true);
|
|
}
|
|
|
|
this.cluster.task((opts) => this.crawlPage(opts));
|
|
|
|
await this.initPages();
|
|
|
|
if (this.params.blockRules && this.params.blockRules.length) {
|
|
this.blockRules = new BlockRules(this.params.blockRules, this.captureBasePrefix, this.params.blockMessage, (text) => this.debugLog(text));
|
|
}
|
|
|
|
this.screencaster = this.initScreenCaster();
|
|
|
|
for (let i = 0; i < this.params.scopedSeeds.length; i++) {
|
|
const seed = this.params.scopedSeeds[i];
|
|
if (!await this.queueUrl(i, seed.url, 0, 0)) {
|
|
if (this.limitHit) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (seed.sitemap) {
|
|
await this.parseSitemap(seed.sitemap, i);
|
|
}
|
|
}
|
|
|
|
await this.cluster.idle();
|
|
await this.cluster.close();
|
|
|
|
await this.serializeConfig(true);
|
|
|
|
this.writeStats();
|
|
|
|
if (this.pagesFH) {
|
|
await this.pagesFH.sync();
|
|
await this.pagesFH.close();
|
|
}
|
|
|
|
// extra wait for all resources to land into WARCs
|
|
await this.awaitPendingClear();
|
|
|
|
if (this.params.combineWARC) {
|
|
await this.combineWARC();
|
|
}
|
|
|
|
if (this.params.generateCDX) {
|
|
this.statusLog("Generating CDX");
|
|
|
|
child_process.spawnSync("wb-manager", ["reindex", this.params.collection], {stdio: "inherit", cwd: this.params.cwd});
|
|
}
|
|
|
|
if (this.params.generateWACZ) {
|
|
await this.generateWACZ();
|
|
}
|
|
}
|
|
|
|
async generateWACZ() {
|
|
this.statusLog("Generating WACZ");
|
|
|
|
const archiveDir = path.join(this.collDir, "archive");
|
|
|
|
// Get a list of the warcs inside
|
|
const warcFileList = await fsp.readdir(archiveDir);
|
|
|
|
console.log(`Num WARC Files: ${warcFileList.length}`);
|
|
if (!warcFileList.length) {
|
|
throw new Error("No WARC Files, assuming crawl failed");
|
|
}
|
|
|
|
// Build the argument list to pass to the wacz create command
|
|
const waczFilename = this.params.collection.concat(".wacz");
|
|
const waczPath = path.join(this.collDir, waczFilename);
|
|
|
|
const createArgs = ["create", "--split-seeds", "-o", waczPath, "--pages", this.pagesFile];
|
|
const validateArgs = ["validate"];
|
|
|
|
if (process.env.WACZ_SIGN_URL) {
|
|
createArgs.push("--signing-url");
|
|
createArgs.push(process.env.WACZ_SIGN_URL);
|
|
if (process.env.WACZ_SIGN_TOKEN) {
|
|
createArgs.push("--signing-token");
|
|
createArgs.push(process.env.WACZ_SIGN_TOKEN);
|
|
}
|
|
}
|
|
|
|
createArgs.push("-f");
|
|
validateArgs.push("-f");
|
|
|
|
warcFileList.forEach((val, index) => createArgs.push(path.join(archiveDir, val))); // eslint-disable-line no-unused-vars
|
|
|
|
// create WACZ
|
|
const waczResult = child_process.spawnSync("wacz" , createArgs, {stdio: "inherit"});
|
|
|
|
if (waczResult.status !== 0) {
|
|
console.log("create result", waczResult);
|
|
throw new Error("Unable to write WACZ successfully");
|
|
}
|
|
|
|
this.debugLog(`WACZ successfully generated and saved to: ${waczPath}`);
|
|
|
|
// Verify WACZ
|
|
validateArgs.push(waczPath);
|
|
|
|
const waczVerifyResult = child_process.spawnSync("wacz", validateArgs, {stdio: "inherit"});
|
|
|
|
if (waczVerifyResult.status !== 0) {
|
|
console.log("validate", waczVerifyResult);
|
|
throw new Error("Unable to verify WACZ created successfully");
|
|
}
|
|
|
|
if (this.storage) {
|
|
const finished = await this.crawlState.finished();
|
|
await this.storage.uploadCollWACZ(waczPath, finished);
|
|
}
|
|
}
|
|
|
|
async writeStats() {
|
|
if (this.params.statsFilename) {
|
|
const total = this.cluster.allTargetCount;
|
|
const workersRunning = this.cluster.workersBusy.length;
|
|
const numCrawled = total - (await this.cluster.jobQueue.size()) - workersRunning;
|
|
const limit = {max: this.params.limit || 0, hit: this.limitHit};
|
|
const stats = {numCrawled, workersRunning, total, limit};
|
|
|
|
try {
|
|
await fsp.writeFile(this.params.statsFilename, JSON.stringify(stats, null, 2));
|
|
} catch (err) {
|
|
console.warn("Stats output failed", err);
|
|
}
|
|
}
|
|
}
|
|
|
|
async loadPage(page, urlData, selectorOptsList = DEFAULT_SELECTORS) {
|
|
const {url, seedId, depth, extraHops = 0} = urlData;
|
|
|
|
if (!await this.isHTML(url)) {
|
|
try {
|
|
if (await this.directFetchCapture(url)) {
|
|
return;
|
|
}
|
|
} catch (e) {
|
|
// ignore failed direct fetch attempt, do browser-based capture
|
|
}
|
|
}
|
|
|
|
if (this.blockRules) {
|
|
await this.blockRules.initPage(page);
|
|
}
|
|
|
|
let ignoreAbort = false;
|
|
|
|
// Detect if ERR_ABORTED is actually caused by trying to load a non-page (eg. downloadable PDF),
|
|
// if so, don't report as an error
|
|
page.on("requestfailed", (req) => {
|
|
ignoreAbort = shouldIgnoreAbort(req);
|
|
});
|
|
|
|
try {
|
|
await page.goto(url, this.gotoOpts);
|
|
} catch (e) {
|
|
let msg = e.message || "";
|
|
if (!msg.startsWith("net::ERR_ABORTED") || !ignoreAbort) {
|
|
this.statusLog(`ERROR: ${url}: ${msg}`);
|
|
}
|
|
}
|
|
|
|
const seed = this.params.scopedSeeds[seedId];
|
|
|
|
// skip extraction if at max depth
|
|
if (seed.isAtMaxDepth(depth) || !selectorOptsList) {
|
|
return;
|
|
}
|
|
|
|
for (const opts of selectorOptsList) {
|
|
const links = await this.extractLinks(page, opts);
|
|
await this.queueInScopeUrls(seedId, links, depth, extraHops);
|
|
}
|
|
}
|
|
|
|
async extractLinks(page, {selector = "a[href]", extract = "href", isAttribute = false} = {}) {
|
|
const results = [];
|
|
|
|
const loadProp = (selector, extract) => {
|
|
return [...document.querySelectorAll(selector)].map(elem => elem[extract]);
|
|
};
|
|
|
|
const loadAttr = (selector, extract) => {
|
|
return [...document.querySelectorAll(selector)].map(elem => elem.getAttribute(extract));
|
|
};
|
|
|
|
const loadFunc = isAttribute ? loadAttr : loadProp;
|
|
|
|
try {
|
|
const linkResults = await Promise.allSettled(page.frames().map(frame => frame.evaluate(loadFunc, selector, extract)));
|
|
|
|
if (linkResults) {
|
|
for (const linkResult of linkResults) {
|
|
if (!linkResult.value) continue;
|
|
for (const link of linkResult.value) {
|
|
results.push(link);
|
|
}
|
|
}
|
|
}
|
|
|
|
} catch (e) {
|
|
console.warn("Link Extraction failed", e);
|
|
}
|
|
return results;
|
|
}
|
|
|
|
async queueInScopeUrls(seedId, urls, depth, extraHops = 0) {
|
|
try {
|
|
depth += 1;
|
|
const seed = this.params.scopedSeeds[seedId];
|
|
|
|
// new number of extra hops, set if this hop is out-of-scope (oos)
|
|
const newExtraHops = extraHops + 1;
|
|
|
|
for (const possibleUrl of urls) {
|
|
const res = seed.isIncluded(possibleUrl, depth, newExtraHops);
|
|
|
|
if (!res) {
|
|
continue;
|
|
}
|
|
|
|
const {url, isOOS} = res;
|
|
|
|
if (url) {
|
|
await this.queueUrl(seedId, url, depth, isOOS ? newExtraHops : extraHops);
|
|
}
|
|
}
|
|
} catch (e) {
|
|
console.error("Queuing Error: ", e);
|
|
}
|
|
}
|
|
|
|
async queueUrl(seedId, url, depth, extraHops = 0) {
|
|
if (this.limitHit) {
|
|
return false;
|
|
}
|
|
|
|
if (this.params.limit > 0 && (await this.crawlState.numRealSeen() >= this.params.limit)) {
|
|
this.limitHit = true;
|
|
return false;
|
|
}
|
|
|
|
if (await this.crawlState.has(url)) {
|
|
return false;
|
|
}
|
|
|
|
await this.crawlState.add(url);
|
|
const urlData = {url, seedId, depth};
|
|
if (extraHops) {
|
|
urlData.extraHops = extraHops;
|
|
}
|
|
this.cluster.queue(urlData);
|
|
return true;
|
|
}
|
|
|
|
async initPages() {
|
|
try {
|
|
let createNew = false;
|
|
|
|
// create pages dir if doesn't exist and write pages.jsonl header
|
|
if (fs.existsSync(this.pagesDir) != true){
|
|
await fsp.mkdir(this.pagesDir);
|
|
createNew = true;
|
|
}
|
|
|
|
this.pagesFH = await fsp.open(this.pagesFile, "a");
|
|
|
|
if (createNew) {
|
|
const header = {"format": "json-pages-1.0", "id": "pages", "title": "All Pages"};
|
|
if (this.params.text) {
|
|
header["hasText"] = true;
|
|
this.statusLog("Text Extraction: Enabled");
|
|
} else {
|
|
header["hasText"] = false;
|
|
this.statusLog("Text Extraction: Disabled");
|
|
}
|
|
const header_formatted = JSON.stringify(header).concat("\n");
|
|
await this.pagesFH.writeFile(header_formatted);
|
|
}
|
|
|
|
} catch(err) {
|
|
console.error("pages/pages.jsonl creation failed", err);
|
|
}
|
|
}
|
|
|
|
async writePage({url, depth}, title, text) {
|
|
const id = uuidv4();
|
|
const row = {"id": id, "url": url, "title": title};
|
|
|
|
if (depth === 0) {
|
|
row.seed = true;
|
|
}
|
|
|
|
if (text !== null) {
|
|
row.text = text;
|
|
}
|
|
|
|
const processedRow = JSON.stringify(row) + "\n";
|
|
try {
|
|
await this.pagesFH.writeFile(processedRow);
|
|
} catch (err) {
|
|
console.warn("pages/pages.jsonl append failed", err);
|
|
}
|
|
}
|
|
|
|
resolveAgent(urlParsed) {
|
|
return urlParsed.protocol === "https:" ? HTTPS_AGENT : HTTP_AGENT;
|
|
}
|
|
|
|
async isHTML(url) {
|
|
try {
|
|
const resp = await fetch(url, {
|
|
method: "HEAD",
|
|
headers: this.headers,
|
|
agent: this.resolveAgent
|
|
});
|
|
if (resp.status !== 200) {
|
|
this.debugLog(`Skipping HEAD check ${url}, invalid status ${resp.status}`);
|
|
return true;
|
|
}
|
|
|
|
const contentType = resp.headers.get("Content-Type");
|
|
|
|
// just load if no content-type
|
|
if (!contentType) {
|
|
return true;
|
|
}
|
|
|
|
const mime = contentType.split(";")[0];
|
|
|
|
if (HTML_TYPES.includes(mime)) {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
} catch(e) {
|
|
// can't confirm not html, so try in browser
|
|
return true;
|
|
}
|
|
}
|
|
|
|
async directFetchCapture(url) {
|
|
//console.log(`Direct capture: ${this.capturePrefix}${url}`);
|
|
const abort = new AbortController();
|
|
const signal = abort.signal;
|
|
const resp = await fetch(this.capturePrefix + url, {signal, headers: this.headers, redirect: "manual"});
|
|
abort.abort();
|
|
return resp.status === 200 && !resp.headers.get("set-cookie");
|
|
}
|
|
|
|
async awaitPendingClear() {
|
|
this.statusLog("Waiting to ensure pending data is written to WARCs...");
|
|
|
|
const redis = await initRedis("redis://localhost/0");
|
|
|
|
while (true) {
|
|
const res = await redis.get(`pywb:${this.params.collection}:pending`);
|
|
if (res === "0" || !res) {
|
|
break;
|
|
}
|
|
|
|
this.debugLog(`Still waiting for ${res} pending requests to finish...`);
|
|
|
|
await this.sleep(1000);
|
|
}
|
|
}
|
|
|
|
sleep(time) {
|
|
return new Promise(resolve => setTimeout(resolve, time));
|
|
}
|
|
|
|
async parseSitemap(url, seedId) {
|
|
const sitemapper = new Sitemapper({
|
|
url,
|
|
timeout: 15000,
|
|
requestHeaders: this.headers
|
|
});
|
|
|
|
try {
|
|
const { sites } = await sitemapper.fetch();
|
|
await this.queueInScopeUrls(seedId, sites, 0);
|
|
} catch(e) {
|
|
console.warn(e);
|
|
}
|
|
}
|
|
|
|
async combineWARC() {
|
|
this.statusLog("Generating Combined WARCs");
|
|
|
|
// Get the list of created Warcs
|
|
const warcLists = await fsp.readdir(path.join(this.collDir, "archive"));
|
|
|
|
this.debugLog(`Combining ${warcLists.length} WARCs...`);
|
|
|
|
const fileSizeObjects = []; // Used to sort the created warc by fileSize
|
|
|
|
// Go through a list of the created works and create an array sorted by their filesize with the largest file first.
|
|
for (let i = 0; i < warcLists.length; i++) {
|
|
const fileName = path.join(this.collDir, "archive", warcLists[i]);
|
|
const fileSize = await getFileSize(fileName);
|
|
fileSizeObjects.push({"fileSize": fileSize, "fileName": fileName});
|
|
fileSizeObjects.sort((a, b) => b.fileSize - a.fileSize);
|
|
}
|
|
|
|
const generatedCombinedWarcs = [];
|
|
|
|
// Used to name combined warcs, default to -1 for first increment
|
|
let combinedWarcNumber = -1;
|
|
|
|
// write combine WARC to collection root
|
|
let combinedWarcFullPath = "";
|
|
|
|
// fileHandler
|
|
let fh = null;
|
|
|
|
// Iterate through the sorted file size array.
|
|
for (let j = 0; j < fileSizeObjects.length; j++) {
|
|
|
|
// if need to rollover to new warc
|
|
let doRollover = false;
|
|
|
|
// set to true for first warc
|
|
if (combinedWarcNumber < 0) {
|
|
doRollover = true;
|
|
} else {
|
|
// Check the size of the existing combined warc.
|
|
const currentCombinedWarcSize = await getFileSize(combinedWarcFullPath);
|
|
|
|
// If adding the current warc to the existing combined file creates a file smaller than the rollover size add the data to the combinedWarc
|
|
const proposedWarcSize = fileSizeObjects[j].fileSize + currentCombinedWarcSize;
|
|
|
|
doRollover = (proposedWarcSize >= this.params.rolloverSize);
|
|
}
|
|
|
|
if (doRollover) {
|
|
// If adding the current warc to the existing combined file creates a file larger than the rollover size do the following:
|
|
// 1. increment the combinedWarcNumber
|
|
// 2. create the name of the new combinedWarcFile
|
|
// 3. Write the header out to the new file
|
|
// 4. Write out the current warc data to the combinedFile
|
|
combinedWarcNumber = combinedWarcNumber + 1;
|
|
|
|
const combinedWarcName = `${this.params.collection}_${combinedWarcNumber}.warc.gz`;
|
|
|
|
// write combined warcs to root collection dir as they're output of a collection (like wacz)
|
|
combinedWarcFullPath = path.join(this.collDir, combinedWarcName);
|
|
|
|
if (fh) {
|
|
fh.end();
|
|
}
|
|
|
|
fh = fs.createWriteStream(combinedWarcFullPath, {flags: "a"});
|
|
|
|
generatedCombinedWarcs.push(combinedWarcName);
|
|
|
|
const warcBuffer = await this.createWARCInfo(combinedWarcName);
|
|
fh.write(warcBuffer);
|
|
}
|
|
|
|
this.debugLog(`Appending WARC ${fileSizeObjects[j].fileName}`);
|
|
|
|
const reader = fs.createReadStream(fileSizeObjects[j].fileName);
|
|
|
|
const p = new Promise((resolve) => {
|
|
reader.on("end", () => resolve());
|
|
});
|
|
|
|
reader.pipe(fh, {end: false});
|
|
|
|
await p;
|
|
}
|
|
|
|
if (fh) {
|
|
await fh.end();
|
|
}
|
|
|
|
this.debugLog(`Combined WARCs saved as: ${generatedCombinedWarcs}`);
|
|
}
|
|
|
|
async serializeConfig(done = false) {
|
|
switch (this.params.saveState) {
|
|
case "never":
|
|
return;
|
|
|
|
case "partial":
|
|
if (!done) {
|
|
return;
|
|
}
|
|
if (await this.crawlState.finished()) {
|
|
return;
|
|
}
|
|
break;
|
|
|
|
case "always":
|
|
default:
|
|
break;
|
|
}
|
|
|
|
const now = new Date();
|
|
|
|
if (!done) {
|
|
// if not done, save state only after specified interval has elapsed
|
|
if ((now.getTime() - this.lastSaveTime) < this.saveStateInterval) {
|
|
return;
|
|
}
|
|
}
|
|
|
|
this.lastSaveTime = now.getTime();
|
|
|
|
const ts = now.toISOString().slice(0,19).replace(/[T:-]/g, "");
|
|
|
|
const crawlDir = path.join(this.collDir, "crawls");
|
|
|
|
await fsp.mkdir(crawlDir, {recursive: true});
|
|
|
|
const filename = path.join(crawlDir, `crawl-${ts}-${this.params.crawlId}.yaml`);
|
|
|
|
const state = await this.crawlState.serialize();
|
|
|
|
if (this.origConfig) {
|
|
this.origConfig.state = state;
|
|
}
|
|
const res = yaml.dump(this.origConfig, {lineWidth: -1});
|
|
try {
|
|
this.statusLog("Saving crawl state to: " + filename);
|
|
await fsp.writeFile(filename, res);
|
|
} catch (e) {
|
|
console.error(`Failed to write save state file: ${filename}`, e);
|
|
return;
|
|
}
|
|
|
|
this.saveStateFiles.push(filename);
|
|
|
|
if (this.saveStateFiles.length > this.params.saveStateHistory) {
|
|
const oldFilename = this.saveStateFiles.shift();
|
|
this.statusLog(`Removing old save-state: ${oldFilename}`);
|
|
try {
|
|
await fsp.unlink(oldFilename);
|
|
} catch (e) {
|
|
console.error(`Failed to delete old save state file: ${oldFilename}`);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
function shouldIgnoreAbort(req) {
|
|
try {
|
|
const failure = req.failure() && req.failure().errorText;
|
|
if (failure !== "net::ERR_ABORTED" || req.resourceType() !== "document") {
|
|
return false;
|
|
}
|
|
|
|
const resp = req.response();
|
|
const headers = resp && resp.headers();
|
|
|
|
if (!headers) {
|
|
return false;
|
|
}
|
|
|
|
if (headers["content-disposition"] ||
|
|
(headers["content-type"] && !headers["content-type"].startsWith("text/"))) {
|
|
return true;
|
|
}
|
|
} catch (e) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
module.exports.Crawler = Crawler;
|