single-file-core.js 62 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473
  1. /*
  2. * Copyright 2018 Gildas Lormeau
  3. * contact : gildas.lormeau <at> gmail.com
  4. *
  5. * This file is part of SingleFile.
  6. *
  7. * SingleFile is free software: you can redistribute it and/or modify
  8. * it under the terms of the GNU Lesser General Public License as published by
  9. * the Free Software Foundation, either version 3 of the License, or
  10. * (at your option) any later version.
  11. *
  12. * SingleFile is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public License
  18. * along with SingleFile. If not, see <http://www.gnu.org/licenses/>.
  19. */
  20. /* global CSSRule */
  21. this.SingleFileCore = this.SingleFileCore || (() => {
  22. const SELECTED_CONTENT_ATTRIBUTE_NAME = "data-single-file-selected-content";
  23. const SELECTED_CONTENT_ROOT_ATTRIBUTE_NAME = "data-single-file-selected-content-root";
  24. const DEBUG = false;
  25. let Download, DOM, URL, sessionId = 0;
  26. function getClass(...args) {
  27. [Download, DOM, URL] = args;
  28. return SingleFileClass;
  29. }
  30. class SingleFileClass {
  31. constructor(options) {
  32. this.options = options;
  33. if (options.sessionId === undefined) {
  34. options.sessionId = sessionId;
  35. sessionId++;
  36. }
  37. }
  38. async initialize() {
  39. this.processor = new PageProcessor(this.options);
  40. await this.processor.loadPage();
  41. await this.processor.initialize();
  42. }
  43. async run() {
  44. await this.processor.run();
  45. }
  46. async getPageData() {
  47. return this.processor.getPageData();
  48. }
  49. }
  50. SingleFileClass.SELECTED_CONTENT_ATTRIBUTE_NAME = SELECTED_CONTENT_ATTRIBUTE_NAME;
  51. SingleFileClass.SELECTED_CONTENT_ROOT_ATTRIBUTE_NAME = SELECTED_CONTENT_ROOT_ATTRIBUTE_NAME;
  52. // -------------
  53. // ProgressEvent
  54. // -------------
  55. const PAGE_LOADING = "page-loading";
  56. const PAGE_LOADED = "page-loaded";
  57. const RESOURCES_INITIALIZING = "resource-initializing";
  58. const RESOURCES_INITIALIZED = "resources-initialized";
  59. const RESOURCE_LOADED = "resource-loaded";
  60. const PAGE_ENDED = "page-ended";
  61. class ProgressEvent {
  62. constructor(type, details) {
  63. return { type, details, PAGE_LOADING, PAGE_LOADED, RESOURCES_INITIALIZING, RESOURCES_INITIALIZED, RESOURCE_LOADED, PAGE_ENDED };
  64. }
  65. }
  66. // -------------
  67. // PageProcessor
  68. // -------------
  69. const RESOLVE_URLS_STAGE = 0;
  70. const REPLACE_DATA_STAGE = 1;
  71. const REPLACE_DOCS_STAGE = 2;
  72. const CLEANUP_STAGE = 3;
  73. const STAGES = [{
  74. sequential: [
  75. { action: "preProcessPage" },
  76. { action: "replaceStyleContents" },
  77. { option: "selected", action: "removeUnselectedElements" },
  78. { option: "removeVideoSrc", action: "insertVideoPosters" },
  79. { option: "removeFrames", action: "removeFrames" },
  80. { option: "removeImports", action: "removeImports" },
  81. { option: "removeScripts", action: "removeScripts" },
  82. { action: "removeDiscardedResources" },
  83. { action: "resetCharsetMeta" },
  84. { action: "setInputValues" },
  85. { option: "insertFaviconLink", action: "insertFaviconLink" },
  86. { action: "resolveHrefs" },
  87. { action: "replaceCanvasElements" },
  88. { action: "insertFonts" },
  89. { option: "removeHiddenElements", action: "removeHiddenElements" },
  90. { action: "resolveStyleAttributeURLs" }
  91. ],
  92. parallel: [
  93. { action: "resolveStylesheetURLs" },
  94. { action: "resolveLinkedStylesheetURLs" },
  95. { option: "!removeFrames", action: "resolveFrameURLs" },
  96. { option: "!removeImports", action: "resolveHtmlImportURLs" }
  97. ]
  98. }, {
  99. sequential: [
  100. { option: "removeUnusedStyles", action: "removeUnusedStyles" },
  101. { option: "removeAlternativeMedias", action: "removeAlternativeMedias" },
  102. { option: "removeUnusedStyles", action: "removeUnusedFonts" }
  103. ],
  104. parallel: [
  105. { action: "processStylesheets" },
  106. { action: "processStyleAttributes" },
  107. { action: "processPageResources" },
  108. { option: "!removeScripts", action: "processScripts" }
  109. ]
  110. }, {
  111. sequential: [
  112. { option: "removeAlternativeImages", action: "removeAlternativeImages" },
  113. { option: "removeAlternativeFonts", action: "removeAlternativeFonts" },
  114. { option: "compressCSS", action: "compressCSS" }
  115. ],
  116. parallel: [
  117. { option: "!removeFrames", action: "processFrames" },
  118. { option: "!removeImports", action: "processHtmlImports" },
  119. ]
  120. }, {
  121. sequential: [
  122. { option: "compressHTML", action: "compressHTML" },
  123. { option: "insertSingleFileComment", action: "insertSingleFileComment" },
  124. { action: "cleanup" }
  125. ]
  126. }];
  127. class PageProcessor {
  128. constructor(options) {
  129. this.options = options;
  130. this.options.url = this.options.url || this.options.doc.location.href;
  131. this.options.baseURI = this.options.doc && this.options.doc.baseURI;
  132. this.batchRequest = new BatchRequest();
  133. this.processor = new DOMProcessor(options, this.batchRequest);
  134. if (this.options.doc) {
  135. const docData = DOM.preProcessDoc(this.options.doc, this.options.win, this.options);
  136. this.options.canvasData = docData.canvasData;
  137. this.options.fontsData = docData.fontsData;
  138. this.options.stylesheetContents = docData.stylesheetContents;
  139. this.options.imageData = docData.imageData;
  140. this.options.postersData = docData.postersData;
  141. this.options.usedFonts = docData.usedFonts;
  142. }
  143. this.options.content = this.options.content || (this.options.doc ? DOM.serialize(this.options.doc, false) : null);
  144. this.onprogress = options.onprogress || (() => { });
  145. }
  146. async loadPage() {
  147. this.onprogress(new ProgressEvent(PAGE_LOADING, { pageURL: this.options.url }));
  148. await this.processor.loadPage(this.options.content);
  149. this.onprogress(new ProgressEvent(PAGE_LOADED, { pageURL: this.options.url }));
  150. }
  151. async initialize() {
  152. this.onprogress(new ProgressEvent(RESOURCES_INITIALIZING, { pageURL: this.options.url }));
  153. await this.executeStage(RESOLVE_URLS_STAGE);
  154. this.pendingPromises = this.executeStage(REPLACE_DATA_STAGE);
  155. if (this.options.doc) {
  156. DOM.postProcessDoc(this.options.doc, this.options);
  157. this.options.doc = null;
  158. this.options.win = null;
  159. }
  160. }
  161. async run() {
  162. if (!this.options.windowId) {
  163. this.processor.initialize(this.batchRequest);
  164. this.onprogress(new ProgressEvent(RESOURCES_INITIALIZED, { pageURL: this.options.url, index: 0, max: this.processor.maxResources }));
  165. }
  166. await this.batchRequest.run(details => {
  167. details.pageURL = this.options.url;
  168. this.onprogress(new ProgressEvent(RESOURCE_LOADED, details));
  169. }, this.options);
  170. await this.pendingPromises;
  171. await this.executeStage(REPLACE_DOCS_STAGE);
  172. await this.executeStage(CLEANUP_STAGE);
  173. }
  174. async getPageData() {
  175. if (!this.options.windowId) {
  176. this.onprogress(new ProgressEvent(PAGE_ENDED, { pageURL: this.options.url }));
  177. }
  178. return this.processor.getPageData();
  179. }
  180. async executeStage(step) {
  181. if (DEBUG) {
  182. log("**** STARTED STAGE", step, "****");
  183. }
  184. STAGES[step].sequential.forEach(task => {
  185. let startTime;
  186. if (DEBUG) {
  187. startTime = Date.now();
  188. log(" -- STARTED task =", task.action);
  189. }
  190. this.executeTask(task);
  191. if (DEBUG) {
  192. log(" -- ENDED task =", task.action, "delay =", Date.now() - startTime);
  193. }
  194. });
  195. if (STAGES[step].parallel) {
  196. return await Promise.all(STAGES[step].parallel.map(task => {
  197. let startTime;
  198. if (DEBUG) {
  199. startTime = Date.now();
  200. log(" // STARTED task =", task.action);
  201. }
  202. const promise = this.executeTask(task);
  203. if (DEBUG) {
  204. promise.then(() => log(" // ENDED task =", task.action, "delay =", Date.now() - startTime));
  205. }
  206. return promise;
  207. }));
  208. }
  209. if (DEBUG) {
  210. log("**** ENDED STAGE", step, "****");
  211. }
  212. }
  213. executeTask(task) {
  214. if (!task.option || ((task.option.startsWith("!") && !this.options[task.option]) || this.options[task.option])) {
  215. return this.processor[task.action]();
  216. }
  217. }
  218. }
  219. // --------
  220. // BatchRequest
  221. // --------
  222. class BatchRequest {
  223. constructor() {
  224. this.requests = new Map();
  225. }
  226. async addURL(resourceURL, asDataURI = true) {
  227. return new Promise((resolve, reject) => {
  228. const requestKey = JSON.stringify([resourceURL, asDataURI]);
  229. const resourceRequests = this.requests.get(requestKey);
  230. if (resourceRequests) {
  231. resourceRequests.push({ resolve, reject });
  232. } else {
  233. this.requests.set(requestKey, [{ resolve, reject }]);
  234. }
  235. });
  236. }
  237. getMaxResources() {
  238. return Array.from(this.requests.keys()).length;
  239. }
  240. async run(onloadListener, options) {
  241. const resourceURLs = Array.from(this.requests.keys());
  242. let indexResource = 0;
  243. return Promise.all(resourceURLs.map(async requestKey => {
  244. const [resourceURL, asDataURI] = JSON.parse(requestKey);
  245. const resourceRequests = this.requests.get(requestKey);
  246. try {
  247. const resourceContent = await Download.getContent(resourceURL, { asDataURI, maxResourceSize: options.maxResourceSize, maxResourceSizeEnabled: options.maxResourceSizeEnabled });
  248. indexResource = indexResource + 1;
  249. onloadListener({ index: indexResource, url: resourceURL });
  250. resourceRequests.forEach(resourceRequest => resourceRequest.resolve({ content: resourceContent, indexResource, duplicate: Boolean(resourceRequests.length > 1) }));
  251. } catch (error) {
  252. indexResource = indexResource + 1;
  253. onloadListener({ index: indexResource, url: resourceURL });
  254. resourceRequests.forEach(resourceRequest => resourceRequest.reject(error));
  255. }
  256. this.requests.delete(requestKey);
  257. }));
  258. }
  259. }
  260. // ------------
  261. // DOMProcessor
  262. // ------------
  263. const EMPTY_DATA_URI = "data:base64,";
  264. const EMPTY_IMAGE = "data:image/gif;base64,R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw==";
  265. const SCRIPT_TAG_FOUND = /<script/gi;
  266. const NOSCRIPT_TAG_FOUND = /<noscript/gi;
  267. class DOMProcessor {
  268. constructor(options, batchRequest) {
  269. this.options = options;
  270. this.stats = new Stats(options);
  271. this.baseURI = DomUtil.normalizeURL(options.baseURI || options.url);
  272. this.batchRequest = batchRequest;
  273. }
  274. initialize() {
  275. this.maxResources = this.batchRequest.getMaxResources();
  276. if (!this.options.removeFrames) {
  277. this.options.framesData.forEach(frameData => this.maxResources += frameData.maxResources || 0);
  278. }
  279. this.stats.set("processed", "resources", this.maxResources);
  280. }
  281. async loadPage(pageContent) {
  282. if (!pageContent || this.options.saveRawPage) {
  283. pageContent = await Download.getContent(this.baseURI, { asDataURI: false, maxResourceSize: this.options.maxResourceSize, maxResourceSizeEnabled: this.options.maxResourceSizeEnabled });
  284. }
  285. this.doc = DOM.createDoc(pageContent, this.baseURI);
  286. this.onEventAttributeNames = DOM.getOnEventAttributeNames(this.doc);
  287. }
  288. async getPageData() {
  289. DOM.postProcessDoc(this.doc, this.options);
  290. const titleElement = this.doc.querySelector("title");
  291. this.options.title = titleElement ? titleElement.textContent.trim() : "";
  292. this.options.info = {};
  293. const descriptionElement = this.doc.querySelector("meta[name=description]");
  294. this.options.info.description = descriptionElement ? descriptionElement.content.trim() : "";
  295. this.options.info.lang = this.doc.documentElement.lang;
  296. const authorElement = this.doc.querySelector("meta[name=author]");
  297. this.options.info.author = authorElement ? authorElement.content.trim() : "";
  298. const creatorElement = this.doc.querySelector("meta[name=creator]");
  299. this.options.info.creator = creatorElement ? creatorElement.content.trim() : "";
  300. const publisherElement = this.doc.querySelector("meta[name=publisher]");
  301. this.options.info.publisher = publisherElement ? publisherElement.content.trim() : "";
  302. const url = new URL(this.baseURI);
  303. let size;
  304. if (this.options.displayStats) {
  305. size = DOM.getContentSize(this.doc.documentElement.outerHTML);
  306. }
  307. const content = DOM.serialize(this.doc, this.options.compressHTML);
  308. if (this.options.displayStats) {
  309. const contentSize = DOM.getContentSize(content);
  310. this.stats.set("processed", "HTML bytes", contentSize);
  311. this.stats.add("discarded", "HTML bytes", size - contentSize);
  312. }
  313. const filename = await DomProcessorHelper.getFilename(this.options, content);
  314. const matchTitle = this.baseURI.match(/([^/]*)\/?(\.html?.*)$/) || this.baseURI.match(/\/\/([^/]*)\/?$/);
  315. return {
  316. stats: this.stats.data,
  317. title: this.options.title || (this.baseURI && matchTitle ? matchTitle[1] : (url.hostname ? url.hostname : "")),
  318. filename,
  319. content
  320. };
  321. }
  322. removeUnselectedElements() {
  323. const rootElement = this.doc.querySelector("[" + SELECTED_CONTENT_ROOT_ATTRIBUTE_NAME + "]");
  324. if (rootElement) {
  325. DomProcessorHelper.isolateElements(rootElement);
  326. rootElement.removeAttribute(SELECTED_CONTENT_ROOT_ATTRIBUTE_NAME);
  327. rootElement.removeAttribute(SELECTED_CONTENT_ATTRIBUTE_NAME);
  328. }
  329. }
  330. setInputValues() {
  331. this.doc.querySelectorAll("input").forEach(input => {
  332. const value = input.getAttribute(DOM.inputValueAttributeName(this.options.sessionId));
  333. input.setAttribute("value", value || "");
  334. });
  335. this.doc.querySelectorAll("textarea").forEach(textarea => {
  336. const value = textarea.getAttribute(DOM.inputValueAttributeName(this.options.sessionId));
  337. textarea.textContent = value || "";
  338. });
  339. this.doc.querySelectorAll("select").forEach(select => {
  340. select.querySelectorAll("option").forEach(option => {
  341. const selected = option.getAttribute(DOM.inputValueAttributeName(this.options.sessionId)) != null;
  342. if (selected) {
  343. option.setAttribute("selected", "");
  344. }
  345. });
  346. });
  347. }
  348. removeDiscardedResources() {
  349. const objectElements = this.doc.querySelectorAll("applet, meta[http-equiv=refresh], object[data]:not([type=\"image/svg+xml\"]):not([type=\"image/svg-xml\"]):not([type=\"text/html\"]), embed[src]:not([src*=\".svg\"])");
  350. this.stats.set("discarded", "objects", objectElements.length);
  351. this.stats.set("processed", "objects", objectElements.length);
  352. objectElements.forEach(element => element.remove());
  353. const replacedAttributeValue = this.doc.querySelectorAll("link[rel~=preconnect], link[rel~=prerender], link[rel~=dns-prefetch], link[rel~=preload], link[rel~=prefetch]");
  354. replacedAttributeValue.forEach(element => {
  355. const relValue = element.getAttribute("rel").replace(/(preconnect|prerender|dns-prefetch|preload|prefetch)/g, "").trim();
  356. if (relValue.length) {
  357. element.setAttribute("rel", relValue);
  358. } else {
  359. element.remove();
  360. }
  361. });
  362. this.doc.querySelectorAll("meta[http-equiv=\"content-security-policy\"]").forEach(element => element.remove());
  363. if (this.options.compressHTML) {
  364. this.doc.querySelectorAll("input[type=hidden]").forEach(element => element.remove());
  365. }
  366. this.doc.querySelectorAll("a[ping]").forEach(element => element.removeAttribute("ping"));
  367. if (this.options.removeScripts) {
  368. this.onEventAttributeNames.forEach(attributeName => this.doc.querySelectorAll("[" + attributeName + "]").forEach(element => element.removeAttribute(attributeName)));
  369. this.doc.querySelectorAll("[href]").forEach(element => {
  370. if (element.href && element.href.match && element.href.match(/^\s*javascript:/)) {
  371. element.setAttribute("href", "");
  372. }
  373. });
  374. this.doc.querySelectorAll("[src]").forEach(element => {
  375. if (element.src && element.src.match(/^\s*javascript:/)) {
  376. element.removeAttribute("src");
  377. }
  378. });
  379. }
  380. const audioSourceElements = this.doc.querySelectorAll("audio[src], audio > source[src]");
  381. this.stats.set("processed", "audio sources", audioSourceElements.length);
  382. if (this.options.removeAudioSrc) {
  383. this.stats.set("discarded", "audio sources", audioSourceElements.length);
  384. audioSourceElements.forEach(element => element.removeAttribute("src"));
  385. }
  386. const videoSourceElements = this.doc.querySelectorAll("video[src], video > source[src]");
  387. this.stats.set("processed", "video sources", videoSourceElements.length);
  388. if (this.options.removeVideoSrc) {
  389. this.stats.set("discarded", "video sources", videoSourceElements.length);
  390. videoSourceElements.forEach(element => element.removeAttribute("src"));
  391. }
  392. }
  393. cleanup() {
  394. const metaCharset = this.doc.head.querySelector("meta[charset]");
  395. if (metaCharset) {
  396. this.doc.head.insertBefore(metaCharset, this.doc.head.firstChild);
  397. }
  398. this.doc.querySelectorAll("style[data-single-file-sheet]").forEach(element => element.removeAttribute("data-single-file-sheet"));
  399. this.doc.querySelectorAll("base").forEach(element => element.remove());
  400. if (this.doc.head.querySelectorAll("*").length == 1 && metaCharset && this.doc.body.childNodes.length == 0) {
  401. this.doc.head.querySelector("meta[charset]").remove();
  402. }
  403. }
  404. preProcessPage() {
  405. this.doc.querySelectorAll("singlefile-infobar, singlefile-mask").forEach(element => element.remove());
  406. if (this.options.win) {
  407. this.doc.body.querySelectorAll(":not(svg) title, meta").forEach(element => element instanceof this.options.win.HTMLElement && this.doc.head.appendChild(element));
  408. }
  409. if (this.options.imageData) {
  410. const dataAttributeName = DOM.imagesAttributeName(this.options.sessionId);
  411. this.doc.querySelectorAll("img").forEach(imgElement => {
  412. const imgData = this.options.imageData[Number(imgElement.getAttribute(dataAttributeName))];
  413. if (imgData.currentSrc) {
  414. imgElement.setAttribute("src", imgData.currentSrc);
  415. }
  416. });
  417. if (this.options.lazyLoadImages) {
  418. this.doc.querySelectorAll("img[data-src]").forEach(imgElement => {
  419. const imgData = this.options.imageData[Number(imgElement.getAttribute(dataAttributeName))];
  420. if ((!imgElement.getAttribute("src") || imgElement.getAttribute("src") == EMPTY_IMAGE) && imgElement.getAttribute("data-src")) {
  421. imgData.src = imgElement.dataset.src;
  422. imgElement.setAttribute("src", imgElement.dataset.src);
  423. imgElement.removeAttribute("data-src");
  424. }
  425. });
  426. this.doc.querySelectorAll("img[data-srcset]").forEach(imgElement => {
  427. if (!imgElement.getAttribute("srcset") && imgElement.getAttribute("data-srcset")) {
  428. imgElement.setAttribute("srcset", imgElement.dataset.srcset);
  429. imgElement.removeAttribute("data-srcset");
  430. }
  431. });
  432. }
  433. }
  434. }
  435. removeScripts() {
  436. const scriptElements = this.doc.querySelectorAll("script:not([type=\"application/ld+json\"])");
  437. this.stats.set("discarded", "scripts", scriptElements.length);
  438. this.stats.set("processed", "scripts", scriptElements.length);
  439. scriptElements.forEach(element => element.remove());
  440. }
  441. removeFrames() {
  442. const frameElements = this.doc.querySelectorAll("iframe, frame, object[type=\"text/html\"][data]");
  443. this.stats.set("discarded", "frames", frameElements.length);
  444. this.stats.set("processed", "frames", frameElements.length);
  445. this.doc.querySelectorAll("iframe, frame, object[type=\"text/html\"][data]").forEach(element => element.remove());
  446. }
  447. removeImports() {
  448. const importElements = this.doc.querySelectorAll("link[rel=import]");
  449. this.stats.set("discarded", "HTML imports", importElements.length);
  450. this.stats.set("processed", "HTML imports", importElements.length);
  451. importElements.forEach(element => element.remove());
  452. }
  453. resetCharsetMeta() {
  454. let charSet;
  455. this.doc.querySelectorAll("meta[charset], meta[http-equiv=\"content-type\"]").forEach(element => {
  456. const charSetDeclaration = element.content.split(";")[1];
  457. if (charSetDeclaration && !charSet) {
  458. charSet = charSetDeclaration.split("=")[1];
  459. if (charSet) {
  460. this.charSet = charSet.trim().toLowerCase();
  461. }
  462. }
  463. element.remove();
  464. });
  465. const metaElement = this.doc.createElement("meta");
  466. metaElement.setAttribute("charset", "utf-8");
  467. if (this.doc.head.firstChild) {
  468. this.doc.head.insertBefore(metaElement, this.doc.head.firstChild);
  469. } else {
  470. this.doc.head.appendChild(metaElement);
  471. }
  472. }
  473. insertFaviconLink() {
  474. let faviconElement = this.doc.querySelector("link[href][rel=\"icon\"]");
  475. if (!faviconElement) {
  476. faviconElement = this.doc.querySelector("link[href][rel=\"shortcut icon\"]");
  477. }
  478. if (!faviconElement) {
  479. faviconElement = this.doc.createElement("link");
  480. faviconElement.setAttribute("type", "image/x-icon");
  481. faviconElement.setAttribute("rel", "shortcut icon");
  482. faviconElement.setAttribute("href", "/favicon.ico");
  483. }
  484. this.doc.head.appendChild(faviconElement);
  485. }
  486. resolveHrefs() {
  487. this.doc.querySelectorAll("[href]").forEach(element => {
  488. if (element.href) {
  489. const href = element.href.baseVal ? element.href.baseVal : element.href;
  490. const normalizedHref = DomUtil.normalizeURL(href);
  491. if (normalizedHref == href && href) {
  492. element.setAttribute("href", href);
  493. }
  494. }
  495. });
  496. }
  497. removeUnusedStyles() {
  498. if (!this.mediaAllInfo) {
  499. this.mediaAllInfo = DOM.getMediaAllInfo(this.doc);
  500. }
  501. const stats = DOM.minifyCSS(this.doc, this.mediaAllInfo);
  502. this.stats.set("processed", "CSS rules", stats.processed);
  503. this.stats.set("discarded", "CSS rules", stats.discarded);
  504. }
  505. removeUnusedFonts() {
  506. DOM.removeUnusedFonts(this.doc, this.options);
  507. }
  508. removeAlternativeFonts() {
  509. DOM.removeAlternativeFonts(this.doc);
  510. }
  511. removeAlternativeImages() {
  512. DOM.removeAlternativeImages(this.doc);
  513. }
  514. removeHiddenElements() {
  515. const hiddenElements = this.doc.querySelectorAll("[" + DOM.removedContentAttributeName(this.options.sessionId) + "]");
  516. this.stats.set("discarded", "hidden elements", hiddenElements.length);
  517. this.stats.set("processed", "hidden elements", hiddenElements.length);
  518. hiddenElements.forEach(element => element.remove());
  519. }
  520. compressHTML() {
  521. let size;
  522. if (this.options.displayStats) {
  523. size = DOM.getContentSize(this.doc.documentElement.outerHTML);
  524. }
  525. DOM.minifyHTML(this.doc, { preservedSpaceAttributeName: DOM.preservedSpaceAttributeName(this.options.sessionId) });
  526. if (this.options.displayStats) {
  527. this.stats.add("discarded", "HTML bytes", size - DOM.getContentSize(this.doc.documentElement.outerHTML));
  528. }
  529. }
  530. removeAlternativeMedias() {
  531. const stats = DOM.minifyMedias(this.doc);
  532. this.stats.set("processed", "medias", stats.processed);
  533. this.stats.set("discarded", "medias", stats.discarded);
  534. }
  535. compressCSS() {
  536. this.doc.querySelectorAll("style").forEach(styleElement => {
  537. if (styleElement) {
  538. styleElement.textContent = DOM.compressCSS(styleElement.textContent);
  539. }
  540. });
  541. this.doc.querySelectorAll("[style]").forEach(element => {
  542. element.setAttribute("style", DOM.compressCSS(element.getAttribute("style")));
  543. });
  544. }
  545. insertSingleFileComment() {
  546. const commentNode = this.doc.createComment("\n Archive processed by SingleFile \n url: " + this.options.url + " \n saved date: " + new Date() + " \n");
  547. this.doc.documentElement.insertBefore(commentNode, this.doc.documentElement.firstChild);
  548. }
  549. replaceCanvasElements() {
  550. if (this.options.canvasData) {
  551. this.doc.querySelectorAll("canvas").forEach((canvasElement, indexCanvasElement) => {
  552. const canvasData = this.options.canvasData[indexCanvasElement];
  553. if (canvasData) {
  554. DomProcessorHelper.setBackgroundImage(canvasElement, "url(" + canvasData.dataURI + ")");
  555. this.stats.add("processed", "canvas", 1);
  556. }
  557. });
  558. }
  559. }
  560. insertFonts() {
  561. if (this.options.fontsData && this.options.fontsData.length) {
  562. let stylesheetContent = "";
  563. this.options.fontsData.forEach(fontStyles => {
  564. if (fontStyles["font-family"] && fontStyles.src) {
  565. stylesheetContent += "@font-face{";
  566. let stylesContent = "";
  567. Object.keys(fontStyles).forEach(fontStyle => {
  568. if (stylesContent) {
  569. stylesContent += ";";
  570. }
  571. stylesContent += fontStyle + ":" + fontStyles[fontStyle];
  572. });
  573. stylesheetContent += stylesContent + "}";
  574. }
  575. });
  576. if (stylesheetContent) {
  577. const styleElement = this.doc.createElement("style");
  578. styleElement.textContent = stylesheetContent;
  579. const existingStyleElement = this.doc.querySelector("style");
  580. if (existingStyleElement) {
  581. existingStyleElement.parentElement.insertBefore(styleElement, existingStyleElement);
  582. } else {
  583. this.doc.head.insertBefore(styleElement, this.doc.head.firstChild);
  584. }
  585. }
  586. }
  587. }
  588. replaceStyleContents() {
  589. if (this.options.stylesheetContents) {
  590. this.doc.querySelectorAll("style").forEach((styleElement, styleIndex) => {
  591. if (this.options.stylesheetContents[styleIndex]) {
  592. styleElement.textContent = this.options.stylesheetContents[styleIndex];
  593. }
  594. });
  595. }
  596. }
  597. insertVideoPosters() {
  598. if (this.options.postersData) {
  599. this.doc.querySelectorAll("video[src], video > source[src]").forEach((element, videoIndex) => {
  600. let videoElement;
  601. if (element.tagName == "VIDEO") {
  602. videoElement = element;
  603. } else {
  604. videoElement = element.parentElement;
  605. }
  606. if (!videoElement.poster && this.options.postersData[videoIndex]) {
  607. videoElement.setAttribute("poster", this.options.postersData[videoIndex]);
  608. }
  609. });
  610. }
  611. }
  612. async processPageResources() {
  613. const resourcePromises = [
  614. DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("link[href][rel*=\"icon\"]"), "href", "data:", this.baseURI, this.options, this.batchRequest, false, true),
  615. DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("object[type=\"image/svg+xml\"], object[type=\"image/svg-xml\"]"), "data", PREFIX_DATA_URI_IMAGE_SVG, this.baseURI, this.options, this.batchRequest),
  616. DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("img[src], input[src][type=image]"), "src", PREFIX_DATA_URI_IMAGE, this.baseURI, this.options, this.batchRequest, true),
  617. DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("embed[src*=\".svg\"]"), "src", PREFIX_DATA_URI_IMAGE_SVG, this.baseURI, this.options, this.batchRequest),
  618. DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("video[poster]"), "poster", PREFIX_DATA_URI_IMAGE, this.baseURI, this.options, this.batchRequest),
  619. DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("*[background]"), "background", PREFIX_DATA_URI_IMAGE, this.baseURI, this.options, this.batchRequest),
  620. DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("image"), "xlink:href", PREFIX_DATA_URI_IMAGE, this.baseURI, this.options, this.batchRequest),
  621. DomProcessorHelper.processXLinks(this.doc.querySelectorAll("use"), this.baseURI, this.options, this.batchRequest),
  622. DomProcessorHelper.processSrcset(this.doc.querySelectorAll("img[srcset], source[srcset]"), "srcset", PREFIX_DATA_URI_IMAGE, this.baseURI, this.options, this.batchRequest)
  623. ];
  624. if (!this.options.removeAudioSrc) {
  625. resourcePromises.push(DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("audio[src], audio > source[src]"), "src", PREFIX_DATA_URI_AUDIO, this.baseURI, this.options, this.batchRequest));
  626. }
  627. if (!this.options.removeVideoSrc) {
  628. resourcePromises.push(DomProcessorHelper.processAttribute(this.doc, this.doc.querySelectorAll("video[src], video > source[src]"), "src", PREFIX_DATA_URI_VIDEO, this.baseURI, this.options, this.batchRequest));
  629. }
  630. await resourcePromises;
  631. if (this.options.removeAlternativeImages) {
  632. let shortcutIcon = findShortcutIcon(Array.from(this.doc.querySelectorAll("link[href][rel=\"icon\"], link[href][rel=\"shortcut icon\"]")));
  633. if (!shortcutIcon) {
  634. shortcutIcon = findShortcutIcon(Array.from(this.doc.querySelectorAll("link[href][rel*=\"icon\"]")));
  635. if (shortcutIcon) {
  636. shortcutIcon.rel = "icon";
  637. }
  638. }
  639. if (shortcutIcon) {
  640. this.doc.querySelectorAll("link[href][rel*=\"icon\"]").forEach(linkElement => {
  641. if (linkElement != shortcutIcon) {
  642. linkElement.remove();
  643. }
  644. });
  645. }
  646. }
  647. function findShortcutIcon(shortcutIcons) {
  648. shortcutIcons = shortcutIcons.filter(linkElement => linkElement.href != EMPTY_IMAGE);
  649. shortcutIcons.sort((linkElement1, linkElement2) => (parseInt(linkElement2.sizes, 10) || 16) - (parseInt(linkElement1.sizes, 10) || 16));
  650. return shortcutIcons[0];
  651. }
  652. }
  653. async resolveStylesheetURLs() {
  654. await Promise.all(Array.from(this.doc.querySelectorAll("style"))
  655. .map(async styleElement => {
  656. const options = { url: this.options.url, maxResourceSize: this.options.maxResourceSize, maxResourceSizeEnabled: this.options.maxResourceSizeEnabled };
  657. const textContent = await DomProcessorHelper.resolveImportURLs(styleElement.textContent, this.baseURI, options);
  658. styleElement.textContent = textContent;
  659. }));
  660. }
  661. async processStylesheets() {
  662. await Promise.all(Array.from(this.doc.querySelectorAll("style")).map(async styleElement => {
  663. this.stats.add("processed", "stylesheets", 1);
  664. if (styleElement.sheet) {
  665. styleElement.textContent = await DomProcessorHelper.processStylesheet(this.doc, styleElement.textContent, styleElement.sheet.cssRules, this.baseURI, this.options, this.batchRequest);
  666. } else {
  667. styleElement.remove();
  668. }
  669. }));
  670. }
  671. async processScripts() {
  672. await Promise.all(Array.from(this.doc.querySelectorAll("script[src]")).map(async scriptElement => {
  673. if (scriptElement.src) {
  674. this.stats.add("processed", "scripts", 1);
  675. const scriptContent = await Download.getContent(scriptElement.src, { asDataURI: false, maxResourceSize: this.options.maxResourceSize, maxResourceSizeEnabled: this.options.maxResourceSizeEnabled });
  676. scriptElement.textContent = scriptContent.replace(/<\/script>/gi, "<\\/script>");
  677. }
  678. scriptElement.removeAttribute("src");
  679. }));
  680. }
  681. async resolveFrameURLs() {
  682. if (this.options.framesData) {
  683. const frameElements = Array.from(this.doc.querySelectorAll("iframe, frame, object[type=\"text/html\"][data]"));
  684. await Promise.all(frameElements.map(async frameElement => {
  685. DomProcessorHelper.setFrameEmptySrc(frameElement);
  686. const frameWindowId = frameElement.getAttribute(DOM.windowIdAttributeName(this.options.sessionId));
  687. if (frameWindowId) {
  688. const frameData = this.options.framesData.find(frame => frame.windowId == frameWindowId);
  689. if (frameData) {
  690. await initializeProcessor(frameData, frameElement, frameWindowId, this.batchRequest, Object.create(this.options));
  691. }
  692. }
  693. }));
  694. }
  695. async function initializeProcessor(frameData, frameElement, frameWindowId, batchRequest, options) {
  696. options.insertSingleFileComment = false;
  697. options.insertFaviconLink = false;
  698. options.doc = null;
  699. options.win = null;
  700. options.url = frameData.baseURI;
  701. options.windowId = frameWindowId;
  702. if (frameData.content) {
  703. options.content = frameData.content;
  704. options.canvasData = frameData.canvasData;
  705. options.stylesheetContents = frameData.stylesheetContents;
  706. options.fontsData = frameData.fontsData;
  707. options.imageData = frameData.imageData;
  708. options.usedFonts = frameData.usedFonts;
  709. frameData.processor = new PageProcessor(options);
  710. frameData.frameElement = frameElement;
  711. await frameData.processor.loadPage();
  712. await frameData.processor.initialize();
  713. frameData.maxResources = batchRequest.getMaxResources();
  714. }
  715. }
  716. }
  717. async processFrames() {
  718. if (this.options.framesData) {
  719. const frameElements = Array.from(this.doc.querySelectorAll("iframe, frame, object[type=\"text/html\"][data]"));
  720. await Promise.all(frameElements.map(async frameElement => {
  721. const frameWindowId = frameElement.getAttribute(DOM.windowIdAttributeName(this.options.sessionId));
  722. if (frameWindowId) {
  723. const frameData = this.options.framesData.find(frame => frame.windowId == frameWindowId);
  724. if (frameData) {
  725. if (frameData.processor) {
  726. this.stats.add("processed", "frames", 1);
  727. await frameData.processor.run();
  728. const pageData = await frameData.processor.getPageData();
  729. frameElement.removeAttribute(DOM.windowIdAttributeName(this.options.sessionId));
  730. if (pageData.content.match(NOSCRIPT_TAG_FOUND) || pageData.content.match(SCRIPT_TAG_FOUND)) {
  731. frameElement.setAttribute("sandbox", "allow-scripts allow-same-origin");
  732. } else {
  733. frameElement.setAttribute("sandbox", "");
  734. }
  735. DomProcessorHelper.setFrameContent(frameElement, pageData.content);
  736. this.stats.addAll(pageData);
  737. } else {
  738. this.stats.add("discarded", "frames", 1);
  739. }
  740. }
  741. }
  742. }));
  743. }
  744. }
  745. async resolveHtmlImportURLs() {
  746. const linkElements = Array.from(this.doc.querySelectorAll("link[rel=import][href]"));
  747. if (!this.relImportProcessors) {
  748. this.relImportProcessors = new Map();
  749. }
  750. await Promise.all(linkElements.map(async linkElement => {
  751. const resourceURL = linkElement.href;
  752. linkElement.removeAttribute("href");
  753. const options = Object.create(this.options);
  754. options.insertSingleFileComment = false;
  755. options.insertFaviconLink = false;
  756. options.doc = null;
  757. options.win = null;
  758. options.url = resourceURL;
  759. if (!DomUtil.testIgnoredPath(resourceURL) && DomUtil.testValidPath(resourceURL, this.baseURI, this.options.url)) {
  760. const processor = new PageProcessor(options);
  761. this.relImportProcessors.set(linkElement, processor);
  762. await processor.loadPage();
  763. return processor.initialize();
  764. }
  765. }));
  766. }
  767. async processHtmlImports() {
  768. const linkElements = Array.from(this.doc.querySelectorAll("link[rel=import][href]"));
  769. await Promise.all(linkElements.map(async linkElement => {
  770. const processor = this.relImportProcessors.get(linkElement);
  771. if (processor) {
  772. this.stats.add("processed", "HTML imports", 1);
  773. this.relImportProcessors.delete(linkElement);
  774. const pageData = await processor.getPageData();
  775. linkElement.setAttribute("href", "data:text/html," + pageData.content);
  776. this.stats.addAll(pageData);
  777. } else {
  778. this.stats.add("discarded", "HTML imports", 1);
  779. }
  780. }));
  781. }
  782. resolveStyleAttributeURLs() {
  783. Array.from(this.doc.querySelectorAll("[style]"))
  784. .map(element => element.setAttribute("style", DomProcessorHelper.resolveStylesheetURLs(element.getAttribute("style"), this.baseURI, this.options)));
  785. }
  786. async processStyleAttributes() {
  787. await Promise.all(Array.from(this.doc.querySelectorAll("[style]"))
  788. .map(async element => {
  789. const rules = [{ type: CSSRule.STYLE_RULE, cssText: element.getAttribute("style") }];
  790. const textContent = await DomProcessorHelper.processStylesheet(this.doc, element.getAttribute("style"), rules, this.baseURI, this.options, this.batchRequest);
  791. element.setAttribute("style", textContent);
  792. }));
  793. }
  794. async resolveLinkedStylesheetURLs() {
  795. await Promise.all(Array.from(this.doc.querySelectorAll("link[rel*=stylesheet]")).map(async linkElement => {
  796. const options = { maxResourceSize: this.options.maxResourceSize, maxResourceSizeEnabled: this.options.maxResourceSizeEnabled, charSet: this.charSet };
  797. const stylesheetContent = await DomProcessorHelper.resolveLinkStylesheetURLs(linkElement.href, this.baseURI, options);
  798. if (stylesheetContent) {
  799. const styleElement = this.doc.createElement("style");
  800. if (linkElement.media) {
  801. styleElement.media = linkElement.media;
  802. }
  803. styleElement.textContent = stylesheetContent;
  804. linkElement.parentElement.replaceChild(styleElement, linkElement);
  805. } else {
  806. linkElement.remove();
  807. }
  808. }));
  809. }
  810. }
  811. // ---------
  812. // DomHelper
  813. // ---------
  814. const REGEXP_AMP = /&/g;
  815. const REGEXP_NBSP = /\u00a0/g;
  816. const REGEXP_START_TAG = /</g;
  817. const REGEXP_END_TAG = />/g;
  818. const REGEXP_URL_HASH = /(#.+?)$/;
  819. const PREFIX_DATA_URI_IMAGE = "data:image/";
  820. const PREFIX_DATA_URI_AUDIO = "data:audio/";
  821. const PREFIX_DATA_URI_VIDEO = "data:video/";
  822. const PREFIX_DATA_URI_IMAGE_SVG = "data:image/svg+xml";
  823. const PREFIX_DATA_URI_NO_MIMETYPE = "data:;";
  824. const PREFIX_DATA_URI_OCTET_STREAM = /^data:(application|binary)\/octet-stream/;
  825. const PREFIX_DATA_URI_VND = "data:application/vnd.";
  826. const SINGLE_FILE_VARIABLE_NAME_PREFIX = "--sf-img-";
  827. class DomProcessorHelper {
  828. static async getFilename(options, content) {
  829. let filename = options.filenameTemplate;
  830. const date = new Date();
  831. const url = new URL(options.url);
  832. filename = await DomUtil.evalTemplateVariable(filename, "page-title", () => options.title || "No title");
  833. filename = await DomUtil.evalTemplateVariable(filename, "page-language", () => options.info.lang || "No language");
  834. filename = await DomUtil.evalTemplateVariable(filename, "page-description", () => options.info.description || "No description");
  835. filename = await DomUtil.evalTemplateVariable(filename, "page-author", () => options.info.author || "No author");
  836. filename = await DomUtil.evalTemplateVariable(filename, "page-creator", () => options.info.creator || "No creator");
  837. filename = await DomUtil.evalTemplateVariable(filename, "page-publisher", () => options.info.publisher || "No publisher");
  838. filename = await DomUtil.evalTemplateVariable(filename, "datetime-iso", () => date.toISOString());
  839. filename = await DomUtil.evalTemplateVariable(filename, "date-iso", () => date.toISOString().split("T")[0]);
  840. filename = await DomUtil.evalTemplateVariable(filename, "time-iso", () => date.toISOString().split("T")[1].split("Z")[0]);
  841. filename = await DomUtil.evalTemplateVariable(filename, "date-locale", () => date.toLocaleDateString());
  842. filename = await DomUtil.evalTemplateVariable(filename, "time-locale", () => date.toLocaleTimeString());
  843. filename = await DomUtil.evalTemplateVariable(filename, "day-locale", () => String(date.getDate()).padStart(2, "0"));
  844. filename = await DomUtil.evalTemplateVariable(filename, "month-locale", () => String(date.getMonth() + 1).padStart(2, "0"));
  845. filename = await DomUtil.evalTemplateVariable(filename, "year-locale", () => String(date.getFullYear()));
  846. filename = await DomUtil.evalTemplateVariable(filename, "datetime-locale", () => date.toLocaleString());
  847. filename = await DomUtil.evalTemplateVariable(filename, "datetime-utc", () => date.toUTCString());
  848. filename = await DomUtil.evalTemplateVariable(filename, "day-utc", () => String(date.getUTCDate()).padStart(2, "0"));
  849. filename = await DomUtil.evalTemplateVariable(filename, "month-utc", () => String(date.getUTCMonth() + 1).padStart(2, "0"));
  850. filename = await DomUtil.evalTemplateVariable(filename, "year-utc", () => String(date.getUTCFullYear()));
  851. filename = await DomUtil.evalTemplateVariable(filename, "hours-locale", () => String(date.getHours()).padStart(2, "0"));
  852. filename = await DomUtil.evalTemplateVariable(filename, "minutes-locale", () => String(date.getMinutes()).padStart(2, "0"));
  853. filename = await DomUtil.evalTemplateVariable(filename, "seconds-locale", () => String(date.getSeconds()).padStart(2, "0"));
  854. filename = await DomUtil.evalTemplateVariable(filename, "hours-utc", () => String(date.getUTCHours()).padStart(2, "0"));
  855. filename = await DomUtil.evalTemplateVariable(filename, "minutes-utc", () => String(date.getUTCMinutes()).padStart(2, "0"));
  856. filename = await DomUtil.evalTemplateVariable(filename, "seconds-utc", () => String(date.getUTCSeconds()).padStart(2, "0"));
  857. filename = await DomUtil.evalTemplateVariable(filename, "url-hash", () => url.hash.substring(1));
  858. filename = await DomUtil.evalTemplateVariable(filename, "url-host", () => url.host.replace(/\/$/, ""));
  859. filename = await DomUtil.evalTemplateVariable(filename, "url-hostname", () => url.hostname.replace(/\/$/, ""));
  860. filename = await DomUtil.evalTemplateVariable(filename, "url-href", () => url.href);
  861. filename = await DomUtil.evalTemplateVariable(filename, "url-password", () => url.password);
  862. filename = await DomUtil.evalTemplateVariable(filename, "url-pathname", () => url.pathname.replace(/^\//, "").replace(/\/$/, ""), true);
  863. filename = await DomUtil.evalTemplateVariable(filename, "url-port", () => url.port);
  864. filename = await DomUtil.evalTemplateVariable(filename, "url-protocol", () => url.protocol);
  865. filename = await DomUtil.evalTemplateVariable(filename, "url-search", () => url.search.substring(1));
  866. filename = await DomUtil.evalTemplateVariable(filename, "url-username", () => url.username);
  867. filename = await DomUtil.evalTemplateVariable(filename, "tab-id", () => String(options.tabId || "No tab id"));
  868. filename = await DomUtil.evalTemplateVariable(filename, "url-last-segment", () => DomUtil.getLastSegment(url));
  869. filename = await DomUtil.evalTemplateVariable(filename, "digest-sha-256", async () => DOM.digest("SHA-256", content));
  870. filename = await DomUtil.evalTemplateVariable(filename, "digest-sha-384", async () => DOM.digest("SHA-384", content));
  871. filename = await DomUtil.evalTemplateVariable(filename, "digest-sha-512", async () => DOM.digest("SHA-512", content));
  872. filename = filename.replace(/[~\\?%*:|"<>\x00-\x1f\x7F]+/g, "_"); // eslint-disable-line no-control-regex
  873. filename = filename.replace(/\.\.\//g, "").replace(/^\/+/, "").replace(/\/+/g, "/").replace(/\/$/, "");
  874. if (!options.backgroundSave) {
  875. filename = filename.replace(/\//g, "_");
  876. }
  877. if (filename.length > 192) {
  878. const extensionMatch = filename.match(/(\.[^.]{3,4})$/);
  879. const extension = extensionMatch && extensionMatch[0] && extensionMatch[0].length > 1 ? extensionMatch[0] : "";
  880. filename = filename.substring(0, 192 - extension.length) + "…" + extension;
  881. }
  882. if (!filename) {
  883. filename = "Unnamed page";
  884. }
  885. return filename;
  886. }
  887. static setBackgroundImage(element, url, style) {
  888. element.style.setProperty("background-blend-mode", "normal", "important");
  889. element.style.setProperty("background-clip", "content-box", "important");
  890. element.style.setProperty("background-position", style && style["background-position"] ? style["background-position"] : "center", "important");
  891. element.style.setProperty("background-color", style && style["background-color"] ? style["background-color"] : "transparent", "important");
  892. element.style.setProperty("background-image", url, "important");
  893. element.style.setProperty("background-size", style && style["background-size"] ? style["background-size"] : "100% 100%", "important");
  894. element.style.setProperty("background-origin", "content-box", "important");
  895. element.style.setProperty("background-repeat", "no-repeat", "important");
  896. }
  897. static setFrameEmptySrc(frameElement) {
  898. if (frameElement.tagName == "OBJECT") {
  899. frameElement.setAttribute("data", "data:text/html,");
  900. } else {
  901. frameElement.setAttribute("srcdoc", "");
  902. frameElement.removeAttribute("src");
  903. }
  904. }
  905. static setFrameContent(frameElement, content) {
  906. if (frameElement.tagName == "OBJECT") {
  907. frameElement.setAttribute("data", "data:text/html," + content);
  908. } else {
  909. frameElement.setAttribute("srcdoc", content);
  910. frameElement.removeAttribute("src");
  911. }
  912. }
  913. static isolateElements(rootElement) {
  914. rootElement.querySelectorAll("*:not(style)").forEach(element => {
  915. if (element.getAttribute(SELECTED_CONTENT_ATTRIBUTE_NAME) == "") {
  916. element.removeAttribute(SELECTED_CONTENT_ATTRIBUTE_NAME);
  917. } else if (!element.querySelector("[" + SELECTED_CONTENT_ATTRIBUTE_NAME + "]")) {
  918. element.remove();
  919. }
  920. });
  921. isolateParentElements(rootElement.parentElement, rootElement);
  922. function isolateParentElements(parentElement, element) {
  923. let elementFound = false;
  924. Array.from(parentElement.childNodes).forEach(node => {
  925. elementFound = elementFound || (node == element);
  926. const tagName = node.tagName && node.tagName.toLowerCase();
  927. if (node != element && tagName != "svg" && tagName != "style" && tagName != "link") {
  928. if (elementFound) {
  929. node.remove();
  930. } else {
  931. node.hidden = true;
  932. if (node.childNodes && node.childNodes.length) {
  933. Array.from(node.childNodes).forEach(node => node.remove());
  934. }
  935. }
  936. }
  937. });
  938. element = element.parentElement;
  939. if (element && element.parentElement && element.parentElement.tagName != "HTML") {
  940. isolateParentElements(element.parentElement, element);
  941. }
  942. }
  943. }
  944. static async resolveImportURLs(stylesheetContent, baseURI, options) {
  945. stylesheetContent = DomProcessorHelper.resolveStylesheetURLs(stylesheetContent, baseURI, options);
  946. stylesheetContent = DomUtil.removeCssComments(stylesheetContent);
  947. const imports = DomUtil.getImportFunctions(stylesheetContent);
  948. await Promise.all(imports.map(async cssImport => {
  949. const match = DomUtil.matchImport(cssImport);
  950. if (match) {
  951. let resourceURL = DomUtil.normalizeURL(match.resourceURL);
  952. if (!DomUtil.testIgnoredPath(resourceURL) && DomUtil.testValidPath(resourceURL, baseURI, options.url)) {
  953. resourceURL = new URL(match.resourceURL, baseURI).href;
  954. if (DomUtil.testValidURL(resourceURL, baseURI, options.url)) {
  955. const downloadOptions = { asDataURI: false, maxResourceSize: options.maxResourceSize, maxResourceSizeEnabled: options.maxResourceSizeEnabled, validateTextContentType: true };
  956. let importedStylesheetContent = await Download.getContent(resourceURL, downloadOptions);
  957. importedStylesheetContent = DomUtil.wrapMediaQuery(importedStylesheetContent, match.media);
  958. if (stylesheetContent.includes(cssImport)) {
  959. importedStylesheetContent = await DomProcessorHelper.resolveImportURLs(importedStylesheetContent, resourceURL, options);
  960. stylesheetContent = stylesheetContent.replace(DomUtil.getRegExp(cssImport), importedStylesheetContent);
  961. }
  962. }
  963. }
  964. }
  965. }));
  966. return stylesheetContent;
  967. }
  968. static resolveStylesheetURLs(stylesheetContent, baseURI, options) {
  969. const urlFunctions = DomUtil.getUrlFunctions(stylesheetContent);
  970. urlFunctions.map(urlFunction => {
  971. const originalResourceURL = DomUtil.matchURL(urlFunction);
  972. const resourceURL = DomUtil.normalizeURL(originalResourceURL);
  973. if (!DomUtil.testIgnoredPath(resourceURL)) {
  974. if (DomUtil.testValidPath(resourceURL, baseURI, options.url)) {
  975. const resolvedURL = new URL(resourceURL, baseURI).href;
  976. if (DomUtil.testValidURL(resolvedURL, baseURI, options.url) && resourceURL != resolvedURL && stylesheetContent.includes(urlFunction)) {
  977. stylesheetContent = stylesheetContent.replace(DomUtil.getRegExp(urlFunction), urlFunction.replace(originalResourceURL, resolvedURL));
  978. }
  979. }
  980. } else {
  981. if (resourceURL.startsWith(DATA_URI_PREFIX)) {
  982. const escapedResourceURL = resourceURL.replace(REGEXP_AMP, "&amp;").replace(REGEXP_NBSP, "&nbsp;").replace(REGEXP_START_TAG, "&lt;").replace(REGEXP_END_TAG, "&gt;");
  983. if (escapedResourceURL != resourceURL && stylesheetContent.includes(urlFunction)) {
  984. stylesheetContent = stylesheetContent.replace(DomUtil.getRegExp(urlFunction), urlFunction.replace(originalResourceURL, escapedResourceURL));
  985. }
  986. }
  987. }
  988. });
  989. return stylesheetContent;
  990. }
  991. static async resolveLinkStylesheetURLs(resourceURL, baseURI, options) {
  992. resourceURL = DomUtil.normalizeURL(resourceURL);
  993. if (resourceURL && resourceURL != baseURI && resourceURL != ABOUT_BLANK_URI) {
  994. const downloadOptions = { asDataURI: false, maxResourceSize: options.maxResourceSize, maxResourceSizeEnabled: options.maxResourceSizeEnabled, charSet: options.charSet };
  995. let stylesheetContent = await Download.getContent(resourceURL, downloadOptions);
  996. stylesheetContent = await DomProcessorHelper.resolveImportURLs(stylesheetContent, resourceURL, options);
  997. return stylesheetContent;
  998. }
  999. }
  1000. static async processStylesheet(doc, stylesheetContent, cssRules, baseURI, options, batchRequest) {
  1001. const urlFunctions = getURLFunctions(cssRules);
  1002. const styleResources = Promise.all(urlFunctions.style.map(urlFunction => processURLFunction(urlFunction)));
  1003. const otherResources = Promise.all(urlFunctions.other.map(urlFunction => processURLFunction(urlFunction, true)));
  1004. await Promise.all([styleResources, otherResources]);
  1005. return stylesheetContent;
  1006. async function processURLFunction(urlFunction, processFont) {
  1007. const originalResourceURL = DomUtil.matchURL(urlFunction);
  1008. const resourceURL = DomUtil.normalizeURL(originalResourceURL);
  1009. if (!DomUtil.testIgnoredPath(resourceURL)) {
  1010. if (DomUtil.testValidURL(resourceURL, baseURI, options.url)) {
  1011. let { content, indexResource, duplicate } = await batchRequest.addURL(resourceURL);
  1012. let regExpUrlFunction = DomUtil.getRegExp(urlFunction);
  1013. if (!stylesheetContent.match(regExpUrlFunction)) {
  1014. urlFunction = "url(" + originalResourceURL + ")";
  1015. regExpUrlFunction = DomUtil.getRegExp(urlFunction);
  1016. }
  1017. if (!stylesheetContent.match(regExpUrlFunction)) {
  1018. urlFunction = "url('" + originalResourceURL + "')";
  1019. regExpUrlFunction = DomUtil.getRegExp(urlFunction);
  1020. }
  1021. if (stylesheetContent.match(regExpUrlFunction)) {
  1022. if (duplicate && options.groupDuplicateImages && !processFont) {
  1023. stylesheetContent = stylesheetContent.replace(regExpUrlFunction, "var(" + SINGLE_FILE_VARIABLE_NAME_PREFIX + indexResource + ")");
  1024. DomUtil.insertVariable(doc, indexResource, content, options);
  1025. } else {
  1026. stylesheetContent = stylesheetContent.replace(regExpUrlFunction, urlFunction.replace(originalResourceURL, content));
  1027. }
  1028. }
  1029. }
  1030. }
  1031. }
  1032. function getURLFunctions(cssRules, urlFunctions = { style: [], other: [] }) {
  1033. Array.from(cssRules).forEach(cssRule => {
  1034. if (cssRule.type == CSSRule.MEDIA_RULE) {
  1035. getURLFunctions(cssRule.cssRules, urlFunctions);
  1036. } else if (cssRule.type == CSSRule.STYLE_RULE) {
  1037. getURLFunctionsRule(cssRule.cssText, urlFunctions.style);
  1038. } else {
  1039. getURLFunctionsRule(cssRule.cssText, urlFunctions.other);
  1040. }
  1041. });
  1042. return { style: Array.from(urlFunctions.style), other: Array.from(urlFunctions.other) };
  1043. }
  1044. function getURLFunctionsRule(cssText, urlFunctions) {
  1045. const urlFunctionsRule = DomUtil.getUrlFunctions(cssText);
  1046. urlFunctionsRule.forEach(urlFunction => {
  1047. const originalResourceURL = DomUtil.matchURL(urlFunction);
  1048. const resourceURL = DomUtil.normalizeURL(originalResourceURL);
  1049. if (!DomUtil.testIgnoredPath(resourceURL)) {
  1050. if (DomUtil.testValidURL(resourceURL, baseURI, options.url) && cssText.includes(urlFunction)) {
  1051. urlFunctions.push(urlFunction);
  1052. }
  1053. }
  1054. });
  1055. return urlFunctions;
  1056. }
  1057. }
  1058. static async processAttribute(doc, resourceElements, attributeName, prefixDataURI, baseURI, options, batchRequest, processDuplicates, removeElementIfMissing) {
  1059. await Promise.all(Array.from(resourceElements).map(async resourceElement => {
  1060. let resourceURL = resourceElement.getAttribute(attributeName);
  1061. resourceURL = DomUtil.normalizeURL(resourceURL);
  1062. if (!DomUtil.testIgnoredPath(resourceURL)) {
  1063. resourceElement.setAttribute(attributeName, EMPTY_IMAGE);
  1064. if (DomUtil.testValidPath(resourceURL, baseURI, options.url)) {
  1065. resourceURL = new URL(resourceURL, baseURI).href;
  1066. if (DomUtil.testValidURL(resourceURL, baseURI, options.url)) {
  1067. const { content, indexResource, duplicate } = await batchRequest.addURL(resourceURL);
  1068. if (removeElementIfMissing && content == EMPTY_DATA_URI) {
  1069. resourceElement.remove();
  1070. } else {
  1071. if (content.startsWith(prefixDataURI) || content.startsWith(PREFIX_DATA_URI_NO_MIMETYPE) || content.match(PREFIX_DATA_URI_OCTET_STREAM)) {
  1072. const isSVG = content.startsWith(PREFIX_DATA_URI_IMAGE_SVG);
  1073. if (processDuplicates && duplicate && options.groupDuplicateImages && !isSVG) {
  1074. if (DomUtil.replaceImageSource(resourceElement, SINGLE_FILE_VARIABLE_NAME_PREFIX + indexResource, options)) {
  1075. DomUtil.insertVariable(doc, indexResource, content, options);
  1076. } else {
  1077. resourceElement.setAttribute(attributeName, content);
  1078. }
  1079. } else {
  1080. resourceElement.setAttribute(attributeName, content);
  1081. }
  1082. }
  1083. }
  1084. }
  1085. }
  1086. }
  1087. }));
  1088. }
  1089. static async processXLinks(resourceElements, baseURI, options, batchRequest) {
  1090. const attributeName = "xlink:href";
  1091. await Promise.all(Array.from(resourceElements).map(async resourceElement => {
  1092. const originalResourceURL = resourceElement.getAttribute(attributeName);
  1093. let resourceURL = DomUtil.normalizeURL(originalResourceURL);
  1094. if (DomUtil.testValidPath(resourceURL, baseURI, options.url) && !DomUtil.testIgnoredPath(resourceURL)) {
  1095. resourceElement.setAttribute(attributeName, EMPTY_IMAGE);
  1096. resourceURL = new URL(resourceURL, baseURI).href;
  1097. if (DomUtil.testValidURL(resourceURL, baseURI, options.url)) {
  1098. try {
  1099. const { content } = await batchRequest.addURL(resourceURL, false);
  1100. const DOMParser = DOM.getParser();
  1101. if (DOMParser) {
  1102. let svgDoc = new DOMParser().parseFromString(content, "image/svg+xml");
  1103. const hashMatch = originalResourceURL.match(REGEXP_URL_HASH);
  1104. if (hashMatch && hashMatch[0]) {
  1105. const symbolElement = svgDoc.querySelector(hashMatch[0]);
  1106. if (symbolElement) {
  1107. resourceElement.setAttribute(attributeName, hashMatch[0]);
  1108. resourceElement.parentElement.insertBefore(symbolElement, resourceElement.parentElement.firstChild);
  1109. }
  1110. } else {
  1111. resourceElement.setAttribute(attributeName, "data:image/svg+xml," + content);
  1112. }
  1113. } else {
  1114. resourceElement.setAttribute(attributeName, "data:image/svg+xml," + content);
  1115. }
  1116. } catch (error) {
  1117. /* ignored */
  1118. }
  1119. }
  1120. } else if (resourceURL == options.url) {
  1121. resourceElement.setAttribute(attributeName, originalResourceURL.substring(resourceURL.length));
  1122. }
  1123. }));
  1124. }
  1125. static async processSrcset(resourceElements, attributeName, prefixDataURI, baseURI, options, batchRequest) {
  1126. await Promise.all(Array.from(resourceElements).map(async resourceElement => {
  1127. const srcset = DOM.parseSrcset(resourceElement.getAttribute(attributeName));
  1128. const srcsetValues = await Promise.all(srcset.map(async srcsetValue => {
  1129. let resourceURL = DomUtil.normalizeURL(srcsetValue.url);
  1130. if (!DomUtil.testIgnoredPath(resourceURL)) {
  1131. if (DomUtil.testValidPath(resourceURL, baseURI, options.url)) {
  1132. resourceURL = new URL(resourceURL, baseURI).href;
  1133. if (DomUtil.testValidURL(resourceURL, baseURI, options.url)) {
  1134. const { content } = await batchRequest.addURL(resourceURL);
  1135. if (!content.startsWith(prefixDataURI) && !content.startsWith(PREFIX_DATA_URI_NO_MIMETYPE) && !content.match(PREFIX_DATA_URI_OCTET_STREAM)) {
  1136. resourceElement.setAttribute(attributeName, EMPTY_IMAGE);
  1137. }
  1138. return content + (srcsetValue.w ? " " + srcsetValue.w + "w" : srcsetValue.d ? " " + srcsetValue.d + "x" : "");
  1139. } else {
  1140. return "";
  1141. }
  1142. } else {
  1143. return "";
  1144. }
  1145. }
  1146. }));
  1147. resourceElement.setAttribute(attributeName, srcsetValues.join(", "));
  1148. }));
  1149. }
  1150. }
  1151. // -------
  1152. // DomUtil
  1153. // -------
  1154. const DATA_URI_PREFIX = "data:";
  1155. const BLOB_URI_PREFIX = "blob:";
  1156. const HTTP_URI_PREFIX = /^https?:\/\//;
  1157. const FILE_URI_PREFIX = /^file:\/\//;
  1158. const EMPTY_URL = /^https?:\/\/+\s*$/;
  1159. const ABOUT_BLANK_URI = "about:blank";
  1160. const NOT_EMPTY_URL = /^(https?|file):\/\/.+/;
  1161. const REGEXP_URL_FN = /(url\s*\(\s*'(.*?)'\s*\))|(url\s*\(\s*"(.*?)"\s*\))|(url\s*\(\s*(.*?)\s*\))/gi;
  1162. const REGEXP_URL_SIMPLE_QUOTES_FN = /^url\s*\(\s*'(.*?)'\s*\)$/i;
  1163. const REGEXP_URL_DOUBLE_QUOTES_FN = /^url\s*\(\s*"(.*?)"\s*\)$/i;
  1164. const REGEXP_URL_NO_QUOTES_FN = /^url\s*\(\s*(.*?)\s*\)$/i;
  1165. const REGEXP_IMPORT_FN = /(@import\s*url\s*\(\s*'(.*?)'\s*\)\s*(.*?)(;|$|}))|(@import\s*url\s*\(\s*"(.*?)"\s*\)\s*(.*?)(;|$|}))|(@import\s*url\s*\(\s*(.*?)\s*\)\s*(.*?)(;|$|}))|(@import\s*'(.*?)'\s*(.*?)(;|$|}))|(@import\s*"(.*?)"\s*(.*?)(;|$|}))|(@import\s*(.*?)\s*(.*?)(;|$|}))/gi;
  1166. const REGEXP_IMPORT_URL_SIMPLE_QUOTES_FN = /@import\s*url\s*\(\s*'(.*?)'\s*\)\s*(.*?)(;|$|})/i;
  1167. const REGEXP_IMPORT_URL_DOUBLE_QUOTES_FN = /@import\s*url\s*\(\s*"(.*?)"\s*\)\s*(.*?)(;|$|})/i;
  1168. const REGEXP_IMPORT_URL_NO_QUOTES_FN = /@import\s*url\s*\(\s*(.*?)\s*\)\s*(.*?)(;|$|})/i;
  1169. const REGEXP_IMPORT_SIMPLE_QUOTES_FN = /@import\s*'(.*?)'\s*(.*?)(;|$|})/i;
  1170. const REGEXP_IMPORT_DOUBLE_QUOTES_FN = /@import\s*"(.*?)"\s*(.*?)(;|$|})/i;
  1171. const REGEXP_IMPORT_NO_QUOTES_FN = /@import\s*(.*?)\s*(.*?)(;|$|})/i;
  1172. const REGEXP_ESCAPE = /([{}()^$&.*?/+|[\\\\]|\]|-)/g;
  1173. class DomUtil {
  1174. static normalizeURL(url) {
  1175. if (!url || url.startsWith(DATA_URI_PREFIX)) {
  1176. return url;
  1177. } else {
  1178. return url.split("#")[0];
  1179. }
  1180. }
  1181. static async evalTemplateVariable(template, variableName, valueGetter, dontReplaceSlash) {
  1182. const replaceRegExp = new RegExp("{\\s*" + variableName + "\\s*}", "g");
  1183. if (template.match(replaceRegExp)) {
  1184. let value = await valueGetter();
  1185. if (!dontReplaceSlash) {
  1186. value = value.replace(/\/+/g, "_");
  1187. }
  1188. return template.replace(replaceRegExp, value);
  1189. }
  1190. return template;
  1191. }
  1192. static getLastSegment(url) {
  1193. let lastSegmentMatch = url.pathname.match(/\/([^/]+)$/), lastSegment = lastSegmentMatch && lastSegmentMatch[0];
  1194. if (!lastSegment) {
  1195. lastSegmentMatch = url.href.match(/([^/]+)\/?$/);
  1196. lastSegment = lastSegmentMatch && lastSegmentMatch[0];
  1197. }
  1198. if (!lastSegment) {
  1199. lastSegmentMatch = lastSegment.match(/(.*)<\.[^.]+$/);
  1200. lastSegment = lastSegmentMatch && lastSegmentMatch[0];
  1201. }
  1202. if (!lastSegment) {
  1203. lastSegment = url.hostname.replace(/\/+/g, "_").replace(/\/$/, "");
  1204. }
  1205. lastSegment.replace(/\/$/, "").replace(/^\//, "");
  1206. return lastSegment;
  1207. }
  1208. static getRegExp(string) {
  1209. return new RegExp(string.replace(REGEXP_ESCAPE, "\\$1"), "gi");
  1210. }
  1211. static getUrlFunctions(stylesheetContent) {
  1212. return Array.from(new Set(stylesheetContent.match(REGEXP_URL_FN) || []));
  1213. }
  1214. static getImportFunctions(stylesheetContent) {
  1215. return stylesheetContent.match(REGEXP_IMPORT_FN) || [];
  1216. }
  1217. static matchURL(stylesheetContent) {
  1218. const match = stylesheetContent.match(REGEXP_URL_SIMPLE_QUOTES_FN) ||
  1219. stylesheetContent.match(REGEXP_URL_DOUBLE_QUOTES_FN) ||
  1220. stylesheetContent.match(REGEXP_URL_NO_QUOTES_FN);
  1221. return match && match[1];
  1222. }
  1223. static testIgnoredPath(resourceURL) {
  1224. return resourceURL && (resourceURL.startsWith(DATA_URI_PREFIX) || resourceURL.startsWith(BLOB_URI_PREFIX) || resourceURL == ABOUT_BLANK_URI);
  1225. }
  1226. static testValidPath(resourceURL, baseURI, docURL) {
  1227. return resourceURL && resourceURL != baseURI && resourceURL != docURL && !resourceURL.match(EMPTY_URL);
  1228. }
  1229. static testValidURL(resourceURL, baseURI, docURL) {
  1230. return DomUtil.testValidPath(resourceURL, baseURI, docURL) && (resourceURL.match(HTTP_URI_PREFIX) || resourceURL.match(FILE_URI_PREFIX)) && resourceURL.match(NOT_EMPTY_URL);
  1231. }
  1232. static matchImport(stylesheetContent) {
  1233. const match = stylesheetContent.match(REGEXP_IMPORT_URL_SIMPLE_QUOTES_FN) ||
  1234. stylesheetContent.match(REGEXP_IMPORT_URL_DOUBLE_QUOTES_FN) ||
  1235. stylesheetContent.match(REGEXP_IMPORT_URL_NO_QUOTES_FN) ||
  1236. stylesheetContent.match(REGEXP_IMPORT_SIMPLE_QUOTES_FN) ||
  1237. stylesheetContent.match(REGEXP_IMPORT_DOUBLE_QUOTES_FN) ||
  1238. stylesheetContent.match(REGEXP_IMPORT_NO_QUOTES_FN);
  1239. if (match) {
  1240. const [, resourceURL, media] = match;
  1241. return { resourceURL, media };
  1242. }
  1243. }
  1244. static removeCssComments(stylesheetContent) {
  1245. let start, end;
  1246. do {
  1247. start = stylesheetContent.indexOf("/*");
  1248. end = stylesheetContent.indexOf("*/", start + 2);
  1249. if (start != -1 && end != -1) {
  1250. stylesheetContent = stylesheetContent.substring(0, start) + stylesheetContent.substr(end + 2);
  1251. }
  1252. } while (start != -1 && end != -1);
  1253. return stylesheetContent;
  1254. }
  1255. static wrapMediaQuery(stylesheetContent, mediaQuery) {
  1256. if (mediaQuery) {
  1257. return "@media " + mediaQuery + "{ " + stylesheetContent + " }";
  1258. } else {
  1259. return stylesheetContent;
  1260. }
  1261. }
  1262. static insertVariable(doc, indexResource, content, options) {
  1263. const sheetAttributeName = DOM.sheetAttributeName(options.sessionId);
  1264. let styleElement = doc.querySelector("style[" + sheetAttributeName + "]"), insertedVariables;
  1265. if (!styleElement) {
  1266. styleElement = doc.createElement("style");
  1267. if (doc.head.firstChild) {
  1268. doc.head.insertBefore(styleElement, doc.head.firstChild);
  1269. } else {
  1270. doc.head.appendChild(styleElement);
  1271. }
  1272. styleElement.setAttribute(sheetAttributeName, "[]");
  1273. insertedVariables = [];
  1274. } else {
  1275. insertedVariables = JSON.parse(styleElement.getAttribute(sheetAttributeName));
  1276. }
  1277. if (!insertedVariables.includes(indexResource)) {
  1278. insertedVariables.push(indexResource);
  1279. styleElement.textContent = styleElement.textContent + `:root{${SINGLE_FILE_VARIABLE_NAME_PREFIX + indexResource}:url("${content}")}`;
  1280. styleElement.setAttribute(sheetAttributeName, JSON.stringify(insertedVariables));
  1281. }
  1282. }
  1283. static replaceImageSource(imgElement, variableName, options) {
  1284. const dataAttributeName = DOM.imagesAttributeName(options.sessionId);
  1285. if (imgElement.getAttribute(dataAttributeName) != null) {
  1286. const imgData = options.imageData[Number(imgElement.getAttribute(dataAttributeName))];
  1287. if (imgData.replaceable) {
  1288. imgElement.setAttribute("src", `data:image/svg+xml,<svg xmlns="http://www.w3.org/2000/svg" width="${imgData.size.pxWidth}" height="${imgData.size.pxHeight}"><rect fill-opacity="0"/></svg>`);
  1289. const backgroundStyle = {};
  1290. const backgroundSize = (imgData.objectFit == "content" || imgData.objectFit == "cover") && imgData.objectFit;
  1291. if (backgroundSize) {
  1292. backgroundStyle["background-size"] = imgData.objectFit;
  1293. }
  1294. if (imgData.objectPosition) {
  1295. backgroundStyle["background-position"] = imgData.objectPosition;
  1296. }
  1297. if (imgData.backgroundColor) {
  1298. backgroundStyle["background-color"] = imgData.backgroundColor;
  1299. }
  1300. DomProcessorHelper.setBackgroundImage(imgElement, "var(" + variableName + ")", backgroundStyle);
  1301. imgElement.removeAttribute(dataAttributeName);
  1302. return true;
  1303. }
  1304. }
  1305. }
  1306. }
  1307. function log(...args) {
  1308. console.log("S-File <core> ", ...args); // eslint-disable-line no-console
  1309. }
  1310. // -----
  1311. // Stats
  1312. // -----
  1313. const STATS_DEFAULT_VALUES = {
  1314. discarded: {
  1315. "HTML bytes": 0,
  1316. "hidden elements": 0,
  1317. "HTML imports": 0,
  1318. scripts: 0,
  1319. objects: 0,
  1320. "audio sources": 0,
  1321. "video sources": 0,
  1322. frames: 0,
  1323. "CSS rules": 0,
  1324. canvas: 0,
  1325. stylesheets: 0,
  1326. resources: 0,
  1327. medias: 0
  1328. },
  1329. processed: {
  1330. "HTML bytes": 0,
  1331. "hidden elements": 0,
  1332. "HTML imports": 0,
  1333. scripts: 0,
  1334. objects: 0,
  1335. "audio sources": 0,
  1336. "video sources": 0,
  1337. frames: 0,
  1338. "CSS rules": 0,
  1339. canvas: 0,
  1340. stylesheets: 0,
  1341. resources: 0,
  1342. medias: 0
  1343. }
  1344. };
  1345. class Stats {
  1346. constructor(options) {
  1347. this.options = options;
  1348. if (options.displayStats) {
  1349. this.data = JSON.parse(JSON.stringify(STATS_DEFAULT_VALUES));
  1350. }
  1351. }
  1352. set(type, subType, value) {
  1353. if (this.options.displayStats) {
  1354. this.data[type][subType] = value;
  1355. }
  1356. }
  1357. add(type, subType, value) {
  1358. if (this.options.displayStats) {
  1359. this.data[type][subType] += value;
  1360. }
  1361. }
  1362. addAll(pageData) {
  1363. if (this.options.displayStats) {
  1364. Object.keys(this.data.discarded).forEach(key => this.add("discarded", key, pageData.stats.discarded[key] || 0));
  1365. Object.keys(this.data.processed).forEach(key => this.add("processed", key, pageData.stats.processed[key] || 0));
  1366. }
  1367. }
  1368. }
  1369. return { getClass };
  1370. })();