single-file-core.js 60 KB

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