single-file-core.js 52 KB

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