xuguopeng/rrweb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make block css class optional & fix test (#6)

Browse Source 
* Make block class optional

* Fix test by removing unexpected space space
This commit is contained in:
Sam Mason
2019-03-31 22:30:52 -04:00
committed by yz-yu
parent 110511ce73
commit 03c9fde822
2 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/snapshot.ts
View File

@@ -86,8 +86,11 @@ function isSVGElement(el: Element): boolean {
return el.tagName === 'svg' || el instanceof SVGElement;
}
const BLOCK_CLASS = 'rr-block';
function serializeNode(n: Node, doc: Document): serializedNode | false {
function serializeNode(
n: Node,
doc: Document,
blockClass: string,
): serializedNode | false {
switch (n.nodeType) {
case n.DOCUMENT_NODE:
return {
@@ -102,7 +105,7 @@ function serializeNode(n: Node, doc: Document): serializedNode | false {
systemId: (n as DocumentType).systemId,
};
case n.ELEMENT_NODE:
const needBlock = (n as HTMLElement).classList.contains(BLOCK_CLASS);
const needBlock = (n as HTMLElement).classList.contains(blockClass);
const tagName = (n as HTMLElement).tagName.toLowerCase();
let attributes: attributes = {};
for (const { name, value } of Array.from((n as HTMLElement).attributes)) {
@@ -216,9 +219,10 @@ export function serializeNodeWithId(
n: Node,
doc: Document,
map: idNodeMap,
blockClass: string,
skipChild = false,
): serializedNodeWithId | null {
const _serializedNode = serializeNode(n, doc);
const _serializedNode = serializeNode(n, doc, blockClass);
if (!_serializedNode) {
// TODO: dev only
console.warn(n, 'not serialized');
@@ -241,7 +245,12 @@ export function serializeNodeWithId(
recordChild
) {
for (const childN of Array.from(n.childNodes)) {
const serializedChildNode = serializeNodeWithId(childN, doc, map);
const serializedChildNode = serializeNodeWithId(
childN,
doc,
map,
blockClass,
);
if (serializedChildNode) {
serializedNode.childNodes.push(serializedChildNode);
}
@@ -250,10 +259,13 @@ export function serializeNodeWithId(
return serializedNode;
}
function snapshot(n: Document): [serializedNodeWithId | null, idNodeMap] {
function snapshot(
n: Document,
blockClass = 'rr-block',
): [serializedNodeWithId | null, idNodeMap] {
resetId();
const idNodeMap: idNodeMap = {};
return [serializeNodeWithId(n, n, idNodeMap), idNodeMap];
return [serializeNodeWithId(n, n, idNodeMap, blockClass), idNodeMap];
}
export default snapshot;