single-file-helper.js 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532
  1. /*
  2. * Copyright 2010-2020 Gildas Lormeau
  3. * contact : gildas.lormeau <at> gmail.com
  4. *
  5. * This file is part of SingleFile.
  6. *
  7. * The code in this file is free software: you can redistribute it and/or
  8. * modify it under the terms of the GNU Affero General Public License
  9. * (GNU AGPL) as published by the Free Software Foundation, either version 3
  10. * of the License, or (at your option) any later version.
  11. *
  12. * The code in this file 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 GNU Affero
  15. * General Public License for more details.
  16. *
  17. * As additional permission under GNU AGPL version 3 section 7, you may
  18. * distribute UNMODIFIED VERSIONS OF THIS file without the copy of the GNU
  19. * AGPL normally required by section 4, provided you include this license
  20. * notice and a URL through which recipients can access the Corresponding
  21. * Source.
  22. */
  23. /* global globalThis, CustomEvent */
  24. import * as cssUnescape from "./vendor/css-unescape.js";
  25. import * as hooksFrames from "./processors/hooks/content/content-hooks-frames";
  26. const ON_BEFORE_CAPTURE_EVENT_NAME = "single-file-on-before-capture";
  27. const ON_AFTER_CAPTURE_EVENT_NAME = "single-file-on-after-capture";
  28. const REMOVED_CONTENT_ATTRIBUTE_NAME = "data-single-file-removed-content";
  29. const HIDDEN_CONTENT_ATTRIBUTE_NAME = "data-single-file-hidden-content";
  30. const KEPT_CONTENT_ATTRIBUTE_NAME = "data-single-file-kept-content";
  31. const HIDDEN_FRAME_ATTRIBUTE_NAME = "data-single-file-hidden-frame";
  32. const PRESERVED_SPACE_ELEMENT_ATTRIBUTE_NAME = "data-single-file-preserved-space-element";
  33. const SHADOW_ROOT_ATTRIBUTE_NAME = "data-single-file-shadow-root-element";
  34. const WIN_ID_ATTRIBUTE_NAME = "data-single-file-win-id";
  35. const IMAGE_ATTRIBUTE_NAME = "data-single-file-image";
  36. const POSTER_ATTRIBUTE_NAME = "data-single-file-poster";
  37. const VIDEO_ATTRIBUTE_NAME = "data-single-file-video";
  38. const CANVAS_ATTRIBUTE_NAME = "data-single-file-canvas";
  39. const HTML_IMPORT_ATTRIBUTE_NAME = "data-single-file-import";
  40. const STYLE_ATTRIBUTE_NAME = "data-single-file-movable-style";
  41. const INPUT_VALUE_ATTRIBUTE_NAME = "data-single-file-input-value";
  42. const LAZY_SRC_ATTRIBUTE_NAME = "data-single-file-lazy-loaded-src";
  43. const STYLESHEET_ATTRIBUTE_NAME = "data-single-file-stylesheet";
  44. const DISABLED_NOSCRIPT_ATTRIBUTE_NAME = "data-single-file-disabled-noscript";
  45. const SELECTED_CONTENT_ATTRIBUTE_NAME = "data-single-file-selected-content";
  46. const ASYNC_SCRIPT_ATTRIBUTE_NAME = "data-single-file-async-script";
  47. const FLOW_ELEMENTS_SELECTOR = "*:not(base):not(link):not(meta):not(noscript):not(script):not(style):not(template):not(title)";
  48. const KEPT_TAG_NAMES = ["NOSCRIPT", "DISABLED-NOSCRIPT", "META", "LINK", "STYLE", "TITLE", "TEMPLATE", "SOURCE", "OBJECT", "SCRIPT", "HEAD"];
  49. const REGEXP_SIMPLE_QUOTES_STRING = /^'(.*?)'$/;
  50. const REGEXP_DOUBLE_QUOTES_STRING = /^"(.*?)"$/;
  51. const FONT_WEIGHTS = {
  52. regular: "400",
  53. normal: "400",
  54. bold: "700",
  55. bolder: "700",
  56. lighter: "100"
  57. };
  58. const COMMENT_HEADER = "Page saved with SingleFile";
  59. const COMMENT_HEADER_LEGACY = "Archive processed by SingleFile";
  60. const SINGLE_FILE_UI_ELEMENT_CLASS = "single-file-ui-element";
  61. const addEventListener = (type, listener, options) => globalThis.addEventListener(type, listener, options);
  62. const dispatchEvent = event => globalThis.dispatchEvent(event);
  63. export {
  64. initUserScriptHandler,
  65. initDoc,
  66. preProcessDoc,
  67. postProcessDoc,
  68. serialize,
  69. removeQuotes,
  70. flatten,
  71. getFontWeight,
  72. normalizeFontFamily,
  73. getShadowRoot,
  74. ON_BEFORE_CAPTURE_EVENT_NAME,
  75. ON_AFTER_CAPTURE_EVENT_NAME,
  76. WIN_ID_ATTRIBUTE_NAME,
  77. PRESERVED_SPACE_ELEMENT_ATTRIBUTE_NAME,
  78. REMOVED_CONTENT_ATTRIBUTE_NAME,
  79. HIDDEN_CONTENT_ATTRIBUTE_NAME,
  80. HIDDEN_FRAME_ATTRIBUTE_NAME,
  81. IMAGE_ATTRIBUTE_NAME,
  82. POSTER_ATTRIBUTE_NAME,
  83. VIDEO_ATTRIBUTE_NAME,
  84. CANVAS_ATTRIBUTE_NAME,
  85. INPUT_VALUE_ATTRIBUTE_NAME,
  86. SHADOW_ROOT_ATTRIBUTE_NAME,
  87. HTML_IMPORT_ATTRIBUTE_NAME,
  88. STYLE_ATTRIBUTE_NAME,
  89. LAZY_SRC_ATTRIBUTE_NAME,
  90. STYLESHEET_ATTRIBUTE_NAME,
  91. SELECTED_CONTENT_ATTRIBUTE_NAME,
  92. ASYNC_SCRIPT_ATTRIBUTE_NAME,
  93. COMMENT_HEADER,
  94. COMMENT_HEADER_LEGACY,
  95. SINGLE_FILE_UI_ELEMENT_CLASS
  96. };
  97. function initUserScriptHandler() {
  98. addEventListener("single-file-user-script-init", () => globalThis._singleFile_waitForUserScript = async eventPrefixName => {
  99. const event = new CustomEvent(eventPrefixName + "-request", { cancelable: true });
  100. const promiseResponse = new Promise(resolve => addEventListener(eventPrefixName + "-response", resolve));
  101. dispatchEvent(event);
  102. if (event.defaultPrevented) {
  103. await promiseResponse;
  104. }
  105. });
  106. }
  107. function initDoc(doc) {
  108. doc.querySelectorAll("meta[http-equiv=refresh]").forEach(element => {
  109. element.removeAttribute("http-equiv");
  110. element.setAttribute("disabled-http-equiv", "refresh");
  111. });
  112. }
  113. function preProcessDoc(doc, win, options) {
  114. doc.querySelectorAll("noscript:not([" + DISABLED_NOSCRIPT_ATTRIBUTE_NAME + "])").forEach(element => {
  115. element.setAttribute(DISABLED_NOSCRIPT_ATTRIBUTE_NAME, element.textContent);
  116. element.textContent = "";
  117. });
  118. initDoc(doc);
  119. if (doc.head) {
  120. doc.head.querySelectorAll(FLOW_ELEMENTS_SELECTOR).forEach(element => element.hidden = true);
  121. }
  122. doc.querySelectorAll("svg foreignObject").forEach(element => {
  123. const flowElements = element.querySelectorAll("html > head > " + FLOW_ELEMENTS_SELECTOR + ", html > body > " + FLOW_ELEMENTS_SELECTOR);
  124. if (flowElements.length) {
  125. Array.from(element.childNodes).forEach(node => node.remove());
  126. flowElements.forEach(flowElement => element.appendChild(flowElement));
  127. }
  128. });
  129. let elementsInfo;
  130. if (win && doc.documentElement) {
  131. elementsInfo = getElementsInfo(win, doc, doc.documentElement, options);
  132. if (options.moveStylesInHead) {
  133. doc.querySelectorAll("body style, body ~ style").forEach(element => {
  134. const computedStyle = win.getComputedStyle(element);
  135. if (computedStyle && testHiddenElement(element, computedStyle)) {
  136. element.setAttribute(STYLE_ATTRIBUTE_NAME, "");
  137. elementsInfo.markedElements.push(element);
  138. }
  139. });
  140. }
  141. } else {
  142. elementsInfo = {
  143. canvases: [],
  144. images: [],
  145. posters: [],
  146. videos: [],
  147. usedFonts: [],
  148. shadowRoots: [],
  149. imports: [],
  150. markedElements: []
  151. };
  152. }
  153. return {
  154. canvases: elementsInfo.canvases,
  155. fonts: getFontsData(doc),
  156. stylesheets: getStylesheetsData(doc),
  157. images: elementsInfo.images,
  158. posters: elementsInfo.posters,
  159. videos: elementsInfo.videos,
  160. usedFonts: Array.from(elementsInfo.usedFonts.values()),
  161. shadowRoots: elementsInfo.shadowRoots,
  162. imports: elementsInfo.imports,
  163. referrer: doc.referrer,
  164. markedElements: elementsInfo.markedElements
  165. };
  166. }
  167. function getElementsInfo(win, doc, element, options, data = { usedFonts: new Map(), canvases: [], images: [], posters: [], videos: [], shadowRoots: [], imports: [], markedElements: [] }, ascendantHidden) {
  168. const elements = Array.from(element.childNodes).filter(node => (node instanceof win.HTMLElement) || (node instanceof win.SVGElement));
  169. elements.forEach(element => {
  170. let elementHidden, elementKept, computedStyle;
  171. if (!options.autoSaveExternalSave && (options.removeHiddenElements || options.removeUnusedFonts || options.compressHTML)) {
  172. computedStyle = win.getComputedStyle(element);
  173. if (element instanceof win.HTMLElement) {
  174. if (options.removeHiddenElements) {
  175. elementKept = ((ascendantHidden || element.closest("html > head")) && KEPT_TAG_NAMES.includes(element.tagName)) || element.closest("details");
  176. if (!elementKept) {
  177. elementHidden = ascendantHidden || testHiddenElement(element, computedStyle);
  178. if (elementHidden) {
  179. element.setAttribute(HIDDEN_CONTENT_ATTRIBUTE_NAME, "");
  180. data.markedElements.push(element);
  181. }
  182. }
  183. }
  184. }
  185. if (!elementHidden) {
  186. if (options.compressHTML && computedStyle) {
  187. const whiteSpace = computedStyle.getPropertyValue("white-space");
  188. if (whiteSpace && whiteSpace.startsWith("pre")) {
  189. element.setAttribute(PRESERVED_SPACE_ELEMENT_ATTRIBUTE_NAME, "");
  190. data.markedElements.push(element);
  191. }
  192. }
  193. if (options.removeUnusedFonts) {
  194. getUsedFont(computedStyle, options, data.usedFonts);
  195. getUsedFont(win.getComputedStyle(element, ":first-letter"), options, data.usedFonts);
  196. getUsedFont(win.getComputedStyle(element, ":before"), options, data.usedFonts);
  197. getUsedFont(win.getComputedStyle(element, ":after"), options, data.usedFonts);
  198. }
  199. }
  200. }
  201. getResourcesInfo(win, doc, element, options, data, elementHidden, computedStyle);
  202. const shadowRoot = !(element instanceof win.SVGElement) && getShadowRoot(element);
  203. if (shadowRoot && !element.classList.contains(SINGLE_FILE_UI_ELEMENT_CLASS)) {
  204. const shadowRootInfo = {};
  205. element.setAttribute(SHADOW_ROOT_ATTRIBUTE_NAME, data.shadowRoots.length);
  206. data.markedElements.push(element);
  207. data.shadowRoots.push(shadowRootInfo);
  208. getElementsInfo(win, doc, shadowRoot, options, data, elementHidden);
  209. shadowRootInfo.content = shadowRoot.innerHTML;
  210. shadowRootInfo.delegatesFocus = shadowRoot.delegatesFocus;
  211. shadowRootInfo.mode = shadowRoot.mode;
  212. if (shadowRoot.adoptedStyleSheets && shadowRoot.adoptedStyleSheets.length) {
  213. shadowRootInfo.adoptedStyleSheets = Array.from(shadowRoot.adoptedStyleSheets).map(stylesheet => Array.from(stylesheet.cssRules).map(cssRule => cssRule.cssText).join("\n"));
  214. }
  215. }
  216. getElementsInfo(win, doc, element, options, data, elementHidden);
  217. if (!options.autoSaveExternalSave && options.removeHiddenElements && ascendantHidden) {
  218. if (elementKept || element.getAttribute(KEPT_CONTENT_ATTRIBUTE_NAME) == "") {
  219. if (element.parentElement) {
  220. element.parentElement.setAttribute(KEPT_CONTENT_ATTRIBUTE_NAME, "");
  221. data.markedElements.push(element.parentElement);
  222. }
  223. } else if (elementHidden) {
  224. element.setAttribute(REMOVED_CONTENT_ATTRIBUTE_NAME, "");
  225. data.markedElements.push(element);
  226. }
  227. }
  228. });
  229. return data;
  230. }
  231. function getResourcesInfo(win, doc, element, options, data, elementHidden, computedStyle) {
  232. if (element.tagName == "CANVAS") {
  233. try {
  234. data.canvases.push({ dataURI: element.toDataURL("image/png", "") });
  235. element.setAttribute(CANVAS_ATTRIBUTE_NAME, data.canvases.length - 1);
  236. data.markedElements.push(element);
  237. } catch (error) {
  238. // ignored
  239. }
  240. }
  241. if (element.tagName == "IMG") {
  242. const imageData = {
  243. currentSrc: elementHidden ?
  244. "data:image/gif;base64,R0lGODlhAQABAAAAACH5BAEKAAEALAAAAAABAAEAAAICTAEAOw==" :
  245. (options.loadDeferredImages && element.getAttribute(LAZY_SRC_ATTRIBUTE_NAME)) || element.currentSrc
  246. };
  247. data.images.push(imageData);
  248. element.setAttribute(IMAGE_ATTRIBUTE_NAME, data.images.length - 1);
  249. data.markedElements.push(element);
  250. element.removeAttribute(LAZY_SRC_ATTRIBUTE_NAME);
  251. computedStyle = computedStyle || win.getComputedStyle(element);
  252. if (computedStyle) {
  253. imageData.size = getSize(win, element, computedStyle);
  254. const boxShadow = computedStyle.getPropertyValue("box-shadow");
  255. const backgroundImage = computedStyle.getPropertyValue("background-image");
  256. if ((!boxShadow || boxShadow == "none") &&
  257. (!backgroundImage || backgroundImage == "none") &&
  258. (imageData.size.pxWidth > 1 || imageData.size.pxHeight > 1)) {
  259. imageData.replaceable = true;
  260. imageData.backgroundColor = computedStyle.getPropertyValue("background-color");
  261. imageData.objectFit = computedStyle.getPropertyValue("object-fit");
  262. imageData.boxSizing = computedStyle.getPropertyValue("box-sizing");
  263. imageData.objectPosition = computedStyle.getPropertyValue("object-position");
  264. }
  265. }
  266. }
  267. if (element.tagName == "VIDEO") {
  268. if (element.currentSrc) {
  269. data.videos.push(element.currentSrc);
  270. element.setAttribute(VIDEO_ATTRIBUTE_NAME, data.videos.length - 1);
  271. }
  272. if (!element.poster) {
  273. const canvasElement = doc.createElement("canvas");
  274. const context = canvasElement.getContext("2d");
  275. canvasElement.width = element.clientWidth;
  276. canvasElement.height = element.clientHeight;
  277. try {
  278. context.drawImage(element, 0, 0, canvasElement.width, canvasElement.height);
  279. data.posters.push(canvasElement.toDataURL("image/png", ""));
  280. element.setAttribute(POSTER_ATTRIBUTE_NAME, data.posters.length - 1);
  281. data.markedElements.push(element);
  282. } catch (error) {
  283. // ignored
  284. }
  285. }
  286. }
  287. if (element.tagName == "IFRAME") {
  288. if (elementHidden && options.removeHiddenElements) {
  289. element.setAttribute(HIDDEN_FRAME_ATTRIBUTE_NAME, "");
  290. data.markedElements.push(element);
  291. }
  292. }
  293. if (element.tagName == "LINK") {
  294. if (element.import && element.import.documentElement) {
  295. data.imports.push({ content: serialize(element.import) });
  296. element.setAttribute(HTML_IMPORT_ATTRIBUTE_NAME, data.imports.length - 1);
  297. data.markedElements.push(element);
  298. }
  299. }
  300. if (element.tagName == "INPUT") {
  301. if (element.type != "password") {
  302. element.setAttribute(INPUT_VALUE_ATTRIBUTE_NAME, element.value);
  303. data.markedElements.push(element);
  304. }
  305. if (element.type == "radio" || element.type == "checkbox") {
  306. element.setAttribute(INPUT_VALUE_ATTRIBUTE_NAME, element.checked);
  307. data.markedElements.push(element);
  308. }
  309. }
  310. if (element.tagName == "TEXTAREA") {
  311. element.setAttribute(INPUT_VALUE_ATTRIBUTE_NAME, element.value);
  312. data.markedElements.push(element);
  313. }
  314. if (element.tagName == "SELECT") {
  315. element.querySelectorAll("option").forEach(option => {
  316. if (option.selected) {
  317. option.setAttribute(INPUT_VALUE_ATTRIBUTE_NAME, "");
  318. data.markedElements.push(option);
  319. }
  320. });
  321. }
  322. if (element.tagName == "SCRIPT") {
  323. if (element.async && element.getAttribute("async") != "" && element.getAttribute("async") != "async") {
  324. element.setAttribute(ASYNC_SCRIPT_ATTRIBUTE_NAME, "");
  325. data.markedElements.push(element);
  326. }
  327. element.textContent = element.textContent.replace(/<\/script>/gi, "<\\/script>");
  328. }
  329. }
  330. function getUsedFont(computedStyle, options, usedFonts) {
  331. if (computedStyle) {
  332. const fontStyle = computedStyle.getPropertyValue("font-style") || "normal";
  333. computedStyle.getPropertyValue("font-family").split(",").forEach(fontFamilyName => {
  334. fontFamilyName = normalizeFontFamily(fontFamilyName);
  335. if (!options.loadedFonts || options.loadedFonts.find(font => normalizeFontFamily(font.family) == fontFamilyName && font.style == fontStyle)) {
  336. const fontWeight = getFontWeight(computedStyle.getPropertyValue("font-weight"));
  337. const fontVariant = computedStyle.getPropertyValue("font-variant") || "normal";
  338. const value = [fontFamilyName, fontWeight, fontStyle, fontVariant];
  339. usedFonts.set(JSON.stringify(value), [fontFamilyName, fontWeight, fontStyle, fontVariant]);
  340. }
  341. });
  342. }
  343. }
  344. function getShadowRoot(element) {
  345. const chrome = globalThis.chrome;
  346. if (element.openOrClosedShadowRoot) {
  347. return element.openOrClosedShadowRoot;
  348. } else if (chrome && chrome.dom && chrome.dom.openOrClosedShadowRoot) {
  349. try {
  350. return chrome.dom.openOrClosedShadowRoot(element);
  351. } catch (error) {
  352. return element.shadowRoot;
  353. }
  354. } else {
  355. return element.shadowRoot;
  356. }
  357. }
  358. function normalizeFontFamily(fontFamilyName = "") {
  359. return removeQuotes(cssUnescape.process(fontFamilyName.trim())).toLowerCase();
  360. }
  361. function testHiddenElement(element, computedStyle) {
  362. let hidden = false;
  363. if (computedStyle) {
  364. const display = computedStyle.getPropertyValue("display");
  365. const opacity = computedStyle.getPropertyValue("opacity");
  366. const visibility = computedStyle.getPropertyValue("visibility");
  367. hidden = display == "none";
  368. if (!hidden && (opacity == "0" || visibility == "hidden") && element.getBoundingClientRect) {
  369. const boundingRect = element.getBoundingClientRect();
  370. hidden = !boundingRect.width && !boundingRect.height;
  371. }
  372. }
  373. return Boolean(hidden);
  374. }
  375. function postProcessDoc(doc, markedElements) {
  376. doc.querySelectorAll("[" + DISABLED_NOSCRIPT_ATTRIBUTE_NAME + "]").forEach(element => {
  377. element.textContent = element.getAttribute(DISABLED_NOSCRIPT_ATTRIBUTE_NAME);
  378. element.removeAttribute(DISABLED_NOSCRIPT_ATTRIBUTE_NAME);
  379. });
  380. doc.querySelectorAll("meta[disabled-http-equiv]").forEach(element => {
  381. element.setAttribute("http-equiv", element.getAttribute("disabled-http-equiv"));
  382. element.removeAttribute("disabled-http-equiv");
  383. });
  384. if (doc.head) {
  385. doc.head.querySelectorAll("*:not(base):not(link):not(meta):not(noscript):not(script):not(style):not(template):not(title)").forEach(element => element.removeAttribute("hidden"));
  386. }
  387. if (!markedElements) {
  388. const singleFileAttributes = [REMOVED_CONTENT_ATTRIBUTE_NAME, HIDDEN_FRAME_ATTRIBUTE_NAME, HIDDEN_CONTENT_ATTRIBUTE_NAME, PRESERVED_SPACE_ELEMENT_ATTRIBUTE_NAME, IMAGE_ATTRIBUTE_NAME, POSTER_ATTRIBUTE_NAME, VIDEO_ATTRIBUTE_NAME, CANVAS_ATTRIBUTE_NAME, INPUT_VALUE_ATTRIBUTE_NAME, SHADOW_ROOT_ATTRIBUTE_NAME, HTML_IMPORT_ATTRIBUTE_NAME, STYLESHEET_ATTRIBUTE_NAME, ASYNC_SCRIPT_ATTRIBUTE_NAME];
  389. markedElements = doc.querySelectorAll(singleFileAttributes.map(name => "[" + name + "]").join(","));
  390. }
  391. markedElements.forEach(element => {
  392. element.removeAttribute(REMOVED_CONTENT_ATTRIBUTE_NAME);
  393. element.removeAttribute(HIDDEN_CONTENT_ATTRIBUTE_NAME);
  394. element.removeAttribute(KEPT_CONTENT_ATTRIBUTE_NAME);
  395. element.removeAttribute(HIDDEN_FRAME_ATTRIBUTE_NAME);
  396. element.removeAttribute(PRESERVED_SPACE_ELEMENT_ATTRIBUTE_NAME);
  397. element.removeAttribute(IMAGE_ATTRIBUTE_NAME);
  398. element.removeAttribute(POSTER_ATTRIBUTE_NAME);
  399. element.removeAttribute(VIDEO_ATTRIBUTE_NAME);
  400. element.removeAttribute(CANVAS_ATTRIBUTE_NAME);
  401. element.removeAttribute(INPUT_VALUE_ATTRIBUTE_NAME);
  402. element.removeAttribute(SHADOW_ROOT_ATTRIBUTE_NAME);
  403. element.removeAttribute(HTML_IMPORT_ATTRIBUTE_NAME);
  404. element.removeAttribute(STYLESHEET_ATTRIBUTE_NAME);
  405. element.removeAttribute(ASYNC_SCRIPT_ATTRIBUTE_NAME);
  406. element.removeAttribute(STYLE_ATTRIBUTE_NAME);
  407. });
  408. }
  409. function getStylesheetsData(doc) {
  410. if (doc) {
  411. const contents = [];
  412. doc.querySelectorAll("style").forEach((styleElement, styleIndex) => {
  413. try {
  414. const tempStyleElement = doc.createElement("style");
  415. tempStyleElement.textContent = styleElement.textContent;
  416. doc.body.appendChild(tempStyleElement);
  417. const stylesheet = tempStyleElement.sheet;
  418. tempStyleElement.remove();
  419. if (!stylesheet || stylesheet.cssRules.length != styleElement.sheet.cssRules.length) {
  420. styleElement.setAttribute(STYLESHEET_ATTRIBUTE_NAME, styleIndex);
  421. contents[styleIndex] = Array.from(styleElement.sheet.cssRules).map(cssRule => cssRule.cssText).join("\n");
  422. }
  423. } catch (error) {
  424. // ignored
  425. }
  426. });
  427. return contents;
  428. }
  429. }
  430. function getSize(win, imageElement, computedStyle) {
  431. let pxWidth = imageElement.naturalWidth;
  432. let pxHeight = imageElement.naturalHeight;
  433. if (!pxWidth && !pxHeight) {
  434. const noStyleAttribute = imageElement.getAttribute("style") == null;
  435. computedStyle = computedStyle || win.getComputedStyle(imageElement);
  436. let removeBorderWidth = false;
  437. if (computedStyle.getPropertyValue("box-sizing") == "content-box") {
  438. const boxSizingValue = imageElement.style.getPropertyValue("box-sizing");
  439. const boxSizingPriority = imageElement.style.getPropertyPriority("box-sizing");
  440. const clientWidth = imageElement.clientWidth;
  441. imageElement.style.setProperty("box-sizing", "border-box", "important");
  442. removeBorderWidth = imageElement.clientWidth != clientWidth;
  443. if (boxSizingValue) {
  444. imageElement.style.setProperty("box-sizing", boxSizingValue, boxSizingPriority);
  445. } else {
  446. imageElement.style.removeProperty("box-sizing");
  447. }
  448. }
  449. let paddingLeft, paddingRight, paddingTop, paddingBottom, borderLeft, borderRight, borderTop, borderBottom;
  450. paddingLeft = getWidth("padding-left", computedStyle);
  451. paddingRight = getWidth("padding-right", computedStyle);
  452. paddingTop = getWidth("padding-top", computedStyle);
  453. paddingBottom = getWidth("padding-bottom", computedStyle);
  454. if (removeBorderWidth) {
  455. borderLeft = getWidth("border-left-width", computedStyle);
  456. borderRight = getWidth("border-right-width", computedStyle);
  457. borderTop = getWidth("border-top-width", computedStyle);
  458. borderBottom = getWidth("border-bottom-width", computedStyle);
  459. } else {
  460. borderLeft = borderRight = borderTop = borderBottom = 0;
  461. }
  462. pxWidth = Math.max(0, imageElement.clientWidth - paddingLeft - paddingRight - borderLeft - borderRight);
  463. pxHeight = Math.max(0, imageElement.clientHeight - paddingTop - paddingBottom - borderTop - borderBottom);
  464. if (noStyleAttribute) {
  465. imageElement.removeAttribute("style");
  466. }
  467. }
  468. return { pxWidth, pxHeight };
  469. }
  470. function getWidth(styleName, computedStyle) {
  471. if (computedStyle.getPropertyValue(styleName).endsWith("px")) {
  472. return parseFloat(computedStyle.getPropertyValue(styleName));
  473. }
  474. }
  475. function getFontsData() {
  476. return hooksFrames.getFontsData();
  477. }
  478. function serialize(doc) {
  479. const docType = doc.doctype;
  480. let docTypeString = "";
  481. if (docType) {
  482. docTypeString = "<!DOCTYPE " + docType.nodeName;
  483. if (docType.publicId) {
  484. docTypeString += " PUBLIC \"" + docType.publicId + "\"";
  485. if (docType.systemId) {
  486. docTypeString += " \"" + docType.systemId + "\"";
  487. }
  488. } else if (docType.systemId) {
  489. docTypeString += " SYSTEM \"" + docType.systemId + "\"";
  490. } if (docType.internalSubset) {
  491. docTypeString += " [" + docType.internalSubset + "]";
  492. }
  493. docTypeString += "> ";
  494. }
  495. return docTypeString + doc.documentElement.outerHTML;
  496. }
  497. function removeQuotes(string) {
  498. if (string.match(REGEXP_SIMPLE_QUOTES_STRING)) {
  499. string = string.replace(REGEXP_SIMPLE_QUOTES_STRING, "$1");
  500. } else {
  501. string = string.replace(REGEXP_DOUBLE_QUOTES_STRING, "$1");
  502. }
  503. return string.trim();
  504. }
  505. function getFontWeight(weight) {
  506. return FONT_WEIGHTS[weight.toLowerCase().trim()] || weight;
  507. }
  508. function flatten(array) {
  509. return array.flat ? array.flat() : array.reduce((a, b) => a.concat(Array.isArray(b) ? flatten(b) : b), []);
  510. }