You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
146 lines
4.6 KiB
146 lines
4.6 KiB
import { ElementType } from "domelementtype"; |
|
import { Element, Text, Comment, CDATA, Document, ProcessingInstruction, } from "./node.js"; |
|
export * from "./node.js"; |
|
// Default options |
|
const defaultOpts = { |
|
withStartIndices: false, |
|
withEndIndices: false, |
|
xmlMode: false, |
|
}; |
|
export class DomHandler { |
|
/** |
|
* @param callback Called once parsing has completed. |
|
* @param options Settings for the handler. |
|
* @param elementCB Callback whenever a tag is closed. |
|
*/ |
|
constructor(callback, options, elementCB) { |
|
/** The elements of the DOM */ |
|
this.dom = []; |
|
/** The root element for the DOM */ |
|
this.root = new Document(this.dom); |
|
/** Indicated whether parsing has been completed. */ |
|
this.done = false; |
|
/** Stack of open tags. */ |
|
this.tagStack = [this.root]; |
|
/** A data node that is still being written to. */ |
|
this.lastNode = null; |
|
/** Reference to the parser instance. Used for location information. */ |
|
this.parser = null; |
|
// Make it possible to skip arguments, for backwards-compatibility |
|
if (typeof options === "function") { |
|
elementCB = options; |
|
options = defaultOpts; |
|
} |
|
if (typeof callback === "object") { |
|
options = callback; |
|
callback = undefined; |
|
} |
|
this.callback = callback !== null && callback !== void 0 ? callback : null; |
|
this.options = options !== null && options !== void 0 ? options : defaultOpts; |
|
this.elementCB = elementCB !== null && elementCB !== void 0 ? elementCB : null; |
|
} |
|
onparserinit(parser) { |
|
this.parser = parser; |
|
} |
|
// Resets the handler back to starting state |
|
onreset() { |
|
this.dom = []; |
|
this.root = new Document(this.dom); |
|
this.done = false; |
|
this.tagStack = [this.root]; |
|
this.lastNode = null; |
|
this.parser = null; |
|
} |
|
// Signals the handler that parsing is done |
|
onend() { |
|
if (this.done) |
|
return; |
|
this.done = true; |
|
this.parser = null; |
|
this.handleCallback(null); |
|
} |
|
onerror(error) { |
|
this.handleCallback(error); |
|
} |
|
onclosetag() { |
|
this.lastNode = null; |
|
const elem = this.tagStack.pop(); |
|
if (this.options.withEndIndices) { |
|
elem.endIndex = this.parser.endIndex; |
|
} |
|
if (this.elementCB) |
|
this.elementCB(elem); |
|
} |
|
onopentag(name, attribs) { |
|
const type = this.options.xmlMode ? ElementType.Tag : undefined; |
|
const element = new Element(name, attribs, undefined, type); |
|
this.addNode(element); |
|
this.tagStack.push(element); |
|
} |
|
ontext(data) { |
|
const { lastNode } = this; |
|
if (lastNode && lastNode.type === ElementType.Text) { |
|
lastNode.data += data; |
|
if (this.options.withEndIndices) { |
|
lastNode.endIndex = this.parser.endIndex; |
|
} |
|
} |
|
else { |
|
const node = new Text(data); |
|
this.addNode(node); |
|
this.lastNode = node; |
|
} |
|
} |
|
oncomment(data) { |
|
if (this.lastNode && this.lastNode.type === ElementType.Comment) { |
|
this.lastNode.data += data; |
|
return; |
|
} |
|
const node = new Comment(data); |
|
this.addNode(node); |
|
this.lastNode = node; |
|
} |
|
oncommentend() { |
|
this.lastNode = null; |
|
} |
|
oncdatastart() { |
|
const text = new Text(""); |
|
const node = new CDATA([text]); |
|
this.addNode(node); |
|
text.parent = node; |
|
this.lastNode = text; |
|
} |
|
oncdataend() { |
|
this.lastNode = null; |
|
} |
|
onprocessinginstruction(name, data) { |
|
const node = new ProcessingInstruction(name, data); |
|
this.addNode(node); |
|
} |
|
handleCallback(error) { |
|
if (typeof this.callback === "function") { |
|
this.callback(error, this.dom); |
|
} |
|
else if (error) { |
|
throw error; |
|
} |
|
} |
|
addNode(node) { |
|
const parent = this.tagStack[this.tagStack.length - 1]; |
|
const previousSibling = parent.children[parent.children.length - 1]; |
|
if (this.options.withStartIndices) { |
|
node.startIndex = this.parser.startIndex; |
|
} |
|
if (this.options.withEndIndices) { |
|
node.endIndex = this.parser.endIndex; |
|
} |
|
parent.children.push(node); |
|
if (previousSibling) { |
|
node.prev = previousSibling; |
|
previousSibling.next = node; |
|
} |
|
node.parent = parent; |
|
this.lastNode = null; |
|
} |
|
} |
|
export default DomHandler;
|
|
|