384 lines
10 KiB
TypeScript
384 lines
10 KiB
TypeScript
import {
|
|
serializedNode,
|
|
serializedNodeWithId,
|
|
NodeType,
|
|
attributes,
|
|
INode,
|
|
idNodeMap,
|
|
} from './types';
|
|
|
|
let _id = 1;
|
|
const symbolAndNumberRegex = RegExp('[^a-z]');
|
|
|
|
function genId(): number {
|
|
return _id++;
|
|
}
|
|
|
|
function getValidTagName(tagName: string): string {
|
|
const processedTagName = tagName.toLowerCase().trim();
|
|
|
|
if (symbolAndNumberRegex.test(processedTagName)) {
|
|
// if the tag name is odd and we cannot extract
|
|
// anything from the string, then we return a
|
|
// generic div
|
|
return 'div';
|
|
}
|
|
|
|
return processedTagName;
|
|
}
|
|
|
|
function getCssRulesString(s: CSSStyleSheet): string | null {
|
|
try {
|
|
const rules = s.rules || s.cssRules;
|
|
return rules
|
|
? Array.from(rules).reduce(
|
|
(prev, cur) => prev + getCssRuleString(cur),
|
|
'',
|
|
)
|
|
: null;
|
|
} catch (error) {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
function getCssRuleString(rule: CSSRule): string {
|
|
return isCSSImportRule(rule)
|
|
? getCssRulesString(rule.styleSheet) || ''
|
|
: rule.cssText;
|
|
}
|
|
|
|
function isCSSImportRule(rule: CSSRule): rule is CSSImportRule {
|
|
return 'styleSheet' in rule;
|
|
}
|
|
|
|
function extractOrigin(url: string): string {
|
|
let origin;
|
|
if (url.indexOf('//') > -1) {
|
|
origin = url
|
|
.split('/')
|
|
.slice(0, 3)
|
|
.join('/');
|
|
} else {
|
|
origin = url.split('/')[0];
|
|
}
|
|
origin = origin.split('?')[0];
|
|
return origin;
|
|
}
|
|
|
|
const URL_IN_CSS_REF = /url\((?:'([^']*)'|"([^"]*)"|([^)]*))\)/gm;
|
|
const RELATIVE_PATH = /^(?!www\.|(?:http|ftp)s?:\/\/|[A-Za-z]:\\|\/\/).*/;
|
|
const DATA_URI = /^(data:)([\w\/\+\-]+);(charset=[\w-]+|base64).*,(.*)/i;
|
|
export function absoluteToStylesheet(cssText: string, href: string): string {
|
|
return cssText.replace(URL_IN_CSS_REF, (origin, path1, path2, path3) => {
|
|
const filePath = path1 || path2 || path3;
|
|
if (!filePath) {
|
|
return origin;
|
|
}
|
|
if (!RELATIVE_PATH.test(filePath)) {
|
|
return `url('${filePath}')`;
|
|
}
|
|
if (DATA_URI.test(filePath)) {
|
|
return `url(${filePath})`;
|
|
}
|
|
if (filePath[0] === '/') {
|
|
return `url('${extractOrigin(href) + filePath}')`;
|
|
}
|
|
const stack = href.split('/');
|
|
const parts = filePath.split('/');
|
|
stack.pop();
|
|
for (const part of parts) {
|
|
if (part === '.') {
|
|
continue;
|
|
} else if (part === '..') {
|
|
stack.pop();
|
|
} else {
|
|
stack.push(part);
|
|
}
|
|
}
|
|
return `url('${stack.join('/')}')`;
|
|
});
|
|
}
|
|
|
|
function getAbsoluteSrcsetString(doc: Document, attributeValue: string) {
|
|
if (attributeValue.trim() === '') {
|
|
return attributeValue;
|
|
}
|
|
|
|
const srcsetValues = attributeValue.split(',');
|
|
// srcset attributes is defined as such:
|
|
// srcset = "url size,url1 size1"
|
|
const resultingSrcsetString = srcsetValues
|
|
.map(srcItem => {
|
|
// removing all but middle spaces
|
|
const trimmedSrcItem = srcItem.trimLeft().trimRight();
|
|
const urlAndSize = trimmedSrcItem.split(' ');
|
|
// this means we have both 0:url and 1:size
|
|
if (urlAndSize.length === 2) {
|
|
const absUrl = absoluteToDoc(doc, urlAndSize[0]);
|
|
return `${absUrl} ${urlAndSize[1]}`;
|
|
} else if (urlAndSize.length === 1) {
|
|
const absUrl = absoluteToDoc(doc, urlAndSize[0]);
|
|
return `${absUrl}`;
|
|
}
|
|
return '';
|
|
})
|
|
.join(',');
|
|
|
|
return resultingSrcsetString;
|
|
}
|
|
|
|
export function absoluteToDoc(doc: Document, attributeValue: string): string {
|
|
if (attributeValue.trim() === '') {
|
|
return attributeValue;
|
|
}
|
|
const a: HTMLAnchorElement = doc.createElement('a');
|
|
a.href = attributeValue;
|
|
return a.href;
|
|
}
|
|
|
|
function isSVGElement(el: Element): boolean {
|
|
return el.tagName === 'svg' || el instanceof SVGElement;
|
|
}
|
|
|
|
export function transformAttribute(
|
|
doc: Document,
|
|
name: string,
|
|
value: string,
|
|
): string {
|
|
// relative path in attribute
|
|
if (name === 'src' || name === 'href') {
|
|
return absoluteToDoc(doc, value);
|
|
} else if (name === 'srcset') {
|
|
return getAbsoluteSrcsetString(doc, value);
|
|
} else if (name === 'style') {
|
|
return absoluteToStylesheet(value, location.href);
|
|
} else {
|
|
return value;
|
|
}
|
|
}
|
|
|
|
function serializeNode(
|
|
n: Node,
|
|
doc: Document,
|
|
blockClass: string | RegExp,
|
|
inlineStylesheet: boolean,
|
|
maskAllInputs: boolean,
|
|
): serializedNode | false {
|
|
switch (n.nodeType) {
|
|
case n.DOCUMENT_NODE:
|
|
return {
|
|
type: NodeType.Document,
|
|
childNodes: [],
|
|
};
|
|
case n.DOCUMENT_TYPE_NODE:
|
|
return {
|
|
type: NodeType.DocumentType,
|
|
name: (n as DocumentType).name,
|
|
publicId: (n as DocumentType).publicId,
|
|
systemId: (n as DocumentType).systemId,
|
|
};
|
|
case n.ELEMENT_NODE:
|
|
let needBlock = false;
|
|
if (typeof blockClass === 'string') {
|
|
needBlock = (n as HTMLElement).classList.contains(blockClass);
|
|
} else {
|
|
(n as HTMLElement).classList.forEach(className => {
|
|
if (blockClass.test(className)) {
|
|
needBlock = true;
|
|
}
|
|
});
|
|
}
|
|
const tagName = getValidTagName((n as HTMLElement).tagName);
|
|
let attributes: attributes = {};
|
|
for (const { name, value } of Array.from((n as HTMLElement).attributes)) {
|
|
attributes[name] = transformAttribute(doc, name, value);
|
|
}
|
|
// remote css
|
|
if (tagName === 'link' && inlineStylesheet) {
|
|
const stylesheet = Array.from(doc.styleSheets).find(s => {
|
|
return s.href === (n as HTMLLinkElement).href;
|
|
});
|
|
const cssText = getCssRulesString(stylesheet as CSSStyleSheet);
|
|
if (cssText) {
|
|
delete attributes.rel;
|
|
delete attributes.href;
|
|
attributes._cssText = absoluteToStylesheet(
|
|
cssText,
|
|
stylesheet!.href!,
|
|
);
|
|
}
|
|
}
|
|
// dynamic stylesheet
|
|
if (
|
|
tagName === 'style' &&
|
|
(n as HTMLStyleElement).sheet &&
|
|
// TODO: Currently we only try to get dynamic stylesheet when it is an empty style element
|
|
!(
|
|
(n as HTMLElement).innerText ||
|
|
(n as HTMLElement).textContent ||
|
|
''
|
|
).trim().length
|
|
) {
|
|
const cssText = getCssRulesString(
|
|
(n as HTMLStyleElement).sheet as CSSStyleSheet,
|
|
);
|
|
if (cssText) {
|
|
attributes._cssText = absoluteToStylesheet(cssText, location.href);
|
|
}
|
|
}
|
|
// form fields
|
|
if (
|
|
tagName === 'input' ||
|
|
tagName === 'textarea' ||
|
|
tagName === 'select'
|
|
) {
|
|
const value = (n as HTMLInputElement | HTMLTextAreaElement).value;
|
|
if (
|
|
attributes.type !== 'radio' &&
|
|
attributes.type !== 'checkbox' &&
|
|
value
|
|
) {
|
|
attributes.value = maskAllInputs ? '*'.repeat(value.length) : value;
|
|
} else if ((n as HTMLInputElement).checked) {
|
|
attributes.checked = (n as HTMLInputElement).checked;
|
|
}
|
|
}
|
|
if (tagName === 'option') {
|
|
const selectValue = (n as HTMLOptionElement).parentElement;
|
|
if (attributes.value === (selectValue as HTMLSelectElement).value) {
|
|
attributes.selected = (n as HTMLOptionElement).selected;
|
|
}
|
|
}
|
|
// canvas image data
|
|
if (tagName === 'canvas') {
|
|
attributes.rr_dataURL = (n as HTMLCanvasElement).toDataURL();
|
|
}
|
|
if (needBlock) {
|
|
const { width, height } = (n as HTMLElement).getBoundingClientRect();
|
|
attributes.rr_width = `${width}px`;
|
|
attributes.rr_height = `${height}px`;
|
|
}
|
|
return {
|
|
type: NodeType.Element,
|
|
tagName,
|
|
attributes,
|
|
childNodes: [],
|
|
isSVG: isSVGElement(n as Element) || undefined,
|
|
needBlock,
|
|
};
|
|
case n.TEXT_NODE:
|
|
// The parent node may not be a html element which has a tagName attribute.
|
|
// So just let it be undefined which is ok in this use case.
|
|
const parentTagName =
|
|
n.parentNode && (n.parentNode as HTMLElement).tagName;
|
|
let textContent = (n as Text).textContent;
|
|
const isStyle = parentTagName === 'STYLE' ? true : undefined;
|
|
if (isStyle && textContent) {
|
|
textContent = absoluteToStylesheet(textContent, location.href);
|
|
}
|
|
if (parentTagName === 'SCRIPT') {
|
|
textContent = 'SCRIPT_PLACEHOLDER';
|
|
}
|
|
return {
|
|
type: NodeType.Text,
|
|
textContent: textContent || '',
|
|
isStyle,
|
|
};
|
|
case n.CDATA_SECTION_NODE:
|
|
return {
|
|
type: NodeType.CDATA,
|
|
textContent: '',
|
|
};
|
|
case n.COMMENT_NODE:
|
|
return {
|
|
type: NodeType.Comment,
|
|
textContent: (n as Comment).textContent || '',
|
|
};
|
|
default:
|
|
return false;
|
|
}
|
|
}
|
|
|
|
export function serializeNodeWithId(
|
|
n: Node | INode,
|
|
doc: Document,
|
|
map: idNodeMap,
|
|
blockClass: string | RegExp,
|
|
skipChild = false,
|
|
inlineStylesheet = true,
|
|
maskAllInputs = false,
|
|
): serializedNodeWithId | null {
|
|
const _serializedNode = serializeNode(
|
|
n,
|
|
doc,
|
|
blockClass,
|
|
inlineStylesheet,
|
|
maskAllInputs,
|
|
);
|
|
if (!_serializedNode) {
|
|
// TODO: dev only
|
|
console.warn(n, 'not serialized');
|
|
return null;
|
|
}
|
|
let id;
|
|
// Try to reuse the previous id
|
|
if ('__sn' in n) {
|
|
id = n.__sn.id;
|
|
} else {
|
|
id = genId();
|
|
}
|
|
const serializedNode = Object.assign(_serializedNode, { id });
|
|
(n as INode).__sn = serializedNode;
|
|
map[id] = n as INode;
|
|
let recordChild = !skipChild;
|
|
if (serializedNode.type === NodeType.Element) {
|
|
recordChild = recordChild && !serializedNode.needBlock;
|
|
// this property was not needed in replay side
|
|
delete serializedNode.needBlock;
|
|
}
|
|
if (
|
|
(serializedNode.type === NodeType.Document ||
|
|
serializedNode.type === NodeType.Element) &&
|
|
recordChild
|
|
) {
|
|
for (const childN of Array.from(n.childNodes)) {
|
|
const serializedChildNode = serializeNodeWithId(
|
|
childN,
|
|
doc,
|
|
map,
|
|
blockClass,
|
|
skipChild,
|
|
inlineStylesheet,
|
|
maskAllInputs,
|
|
);
|
|
if (serializedChildNode) {
|
|
serializedNode.childNodes.push(serializedChildNode);
|
|
}
|
|
}
|
|
}
|
|
return serializedNode;
|
|
}
|
|
|
|
function snapshot(
|
|
n: Document,
|
|
blockClass: string | RegExp = 'rr-block',
|
|
inlineStylesheet = true,
|
|
maskAllInputs = false,
|
|
): [serializedNodeWithId | null, idNodeMap] {
|
|
const idNodeMap: idNodeMap = {};
|
|
return [
|
|
serializeNodeWithId(
|
|
n,
|
|
n,
|
|
idNodeMap,
|
|
blockClass,
|
|
false,
|
|
inlineStylesheet,
|
|
maskAllInputs,
|
|
),
|
|
idNodeMap,
|
|
];
|
|
}
|
|
|
|
export default snapshot;
|