483 lines
12 KiB
JavaScript
483 lines
12 KiB
JavaScript
'use strict';
|
|
|
|
const HTML = require('../common/html');
|
|
|
|
//Aliases
|
|
const $ = HTML.TAG_NAMES;
|
|
const NS = HTML.NAMESPACES;
|
|
|
|
//Element utils
|
|
|
|
//OPTIMIZATION: Integer comparisons are low-cost, so we can use very fast tag name length filters here.
|
|
//It's faster than using dictionary.
|
|
function isImpliedEndTagRequired(tn) {
|
|
switch (tn.length) {
|
|
case 1:
|
|
return tn === $.P;
|
|
|
|
case 2:
|
|
return tn === $.RB || tn === $.RP || tn === $.RT || tn === $.DD || tn === $.DT || tn === $.LI;
|
|
|
|
case 3:
|
|
return tn === $.RTC;
|
|
|
|
case 6:
|
|
return tn === $.OPTION;
|
|
|
|
case 8:
|
|
return tn === $.OPTGROUP;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
function isImpliedEndTagRequiredThoroughly(tn) {
|
|
switch (tn.length) {
|
|
case 1:
|
|
return tn === $.P;
|
|
|
|
case 2:
|
|
return (
|
|
tn === $.RB ||
|
|
tn === $.RP ||
|
|
tn === $.RT ||
|
|
tn === $.DD ||
|
|
tn === $.DT ||
|
|
tn === $.LI ||
|
|
tn === $.TD ||
|
|
tn === $.TH ||
|
|
tn === $.TR
|
|
);
|
|
|
|
case 3:
|
|
return tn === $.RTC;
|
|
|
|
case 5:
|
|
return tn === $.TBODY || tn === $.TFOOT || tn === $.THEAD;
|
|
|
|
case 6:
|
|
return tn === $.OPTION;
|
|
|
|
case 7:
|
|
return tn === $.CAPTION;
|
|
|
|
case 8:
|
|
return tn === $.OPTGROUP || tn === $.COLGROUP;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
function isScopingElement(tn, ns) {
|
|
switch (tn.length) {
|
|
case 2:
|
|
if (tn === $.TD || tn === $.TH) {
|
|
return ns === NS.HTML;
|
|
} else if (tn === $.MI || tn === $.MO || tn === $.MN || tn === $.MS) {
|
|
return ns === NS.MATHML;
|
|
}
|
|
|
|
break;
|
|
|
|
case 4:
|
|
if (tn === $.HTML) {
|
|
return ns === NS.HTML;
|
|
} else if (tn === $.DESC) {
|
|
return ns === NS.SVG;
|
|
}
|
|
|
|
break;
|
|
|
|
case 5:
|
|
if (tn === $.TABLE) {
|
|
return ns === NS.HTML;
|
|
} else if (tn === $.MTEXT) {
|
|
return ns === NS.MATHML;
|
|
} else if (tn === $.TITLE) {
|
|
return ns === NS.SVG;
|
|
}
|
|
|
|
break;
|
|
|
|
case 6:
|
|
return (tn === $.APPLET || tn === $.OBJECT) && ns === NS.HTML;
|
|
|
|
case 7:
|
|
return (tn === $.CAPTION || tn === $.MARQUEE) && ns === NS.HTML;
|
|
|
|
case 8:
|
|
return tn === $.TEMPLATE && ns === NS.HTML;
|
|
|
|
case 13:
|
|
return tn === $.FOREIGN_OBJECT && ns === NS.SVG;
|
|
|
|
case 14:
|
|
return tn === $.ANNOTATION_XML && ns === NS.MATHML;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
//Stack of open elements
|
|
class OpenElementStack {
|
|
constructor(document, treeAdapter) {
|
|
this.stackTop = -1;
|
|
this.items = [];
|
|
this.current = document;
|
|
this.currentTagName = null;
|
|
this.currentTmplContent = null;
|
|
this.tmplCount = 0;
|
|
this.treeAdapter = treeAdapter;
|
|
}
|
|
|
|
//Index of element
|
|
_indexOf(element) {
|
|
let idx = -1;
|
|
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
if (this.items[i] === element) {
|
|
idx = i;
|
|
break;
|
|
}
|
|
}
|
|
return idx;
|
|
}
|
|
|
|
//Update current element
|
|
_isInTemplate() {
|
|
return this.currentTagName === $.TEMPLATE && this.treeAdapter.getNamespaceURI(this.current) === NS.HTML;
|
|
}
|
|
|
|
_updateCurrentElement() {
|
|
this.current = this.items[this.stackTop];
|
|
this.currentTagName = this.current && this.treeAdapter.getTagName(this.current);
|
|
|
|
this.currentTmplContent = this._isInTemplate() ? this.treeAdapter.getTemplateContent(this.current) : null;
|
|
}
|
|
|
|
//Mutations
|
|
push(element) {
|
|
this.items[++this.stackTop] = element;
|
|
this._updateCurrentElement();
|
|
|
|
if (this._isInTemplate()) {
|
|
this.tmplCount++;
|
|
}
|
|
}
|
|
|
|
pop() {
|
|
this.stackTop--;
|
|
|
|
if (this.tmplCount > 0 && this._isInTemplate()) {
|
|
this.tmplCount--;
|
|
}
|
|
|
|
this._updateCurrentElement();
|
|
}
|
|
|
|
replace(oldElement, newElement) {
|
|
const idx = this._indexOf(oldElement);
|
|
|
|
this.items[idx] = newElement;
|
|
|
|
if (idx === this.stackTop) {
|
|
this._updateCurrentElement();
|
|
}
|
|
}
|
|
|
|
insertAfter(referenceElement, newElement) {
|
|
const insertionIdx = this._indexOf(referenceElement) + 1;
|
|
|
|
this.items.splice(insertionIdx, 0, newElement);
|
|
|
|
if (insertionIdx === ++this.stackTop) {
|
|
this._updateCurrentElement();
|
|
}
|
|
}
|
|
|
|
popUntilTagNamePopped(tagName) {
|
|
while (this.stackTop > -1) {
|
|
const tn = this.currentTagName;
|
|
const ns = this.treeAdapter.getNamespaceURI(this.current);
|
|
|
|
this.pop();
|
|
|
|
if (tn === tagName && ns === NS.HTML) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
popUntilElementPopped(element) {
|
|
while (this.stackTop > -1) {
|
|
const poppedElement = this.current;
|
|
|
|
this.pop();
|
|
|
|
if (poppedElement === element) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
popUntilNumberedHeaderPopped() {
|
|
while (this.stackTop > -1) {
|
|
const tn = this.currentTagName;
|
|
const ns = this.treeAdapter.getNamespaceURI(this.current);
|
|
|
|
this.pop();
|
|
|
|
if (
|
|
tn === $.H1 ||
|
|
tn === $.H2 ||
|
|
tn === $.H3 ||
|
|
tn === $.H4 ||
|
|
tn === $.H5 ||
|
|
(tn === $.H6 && ns === NS.HTML)
|
|
) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
popUntilTableCellPopped() {
|
|
while (this.stackTop > -1) {
|
|
const tn = this.currentTagName;
|
|
const ns = this.treeAdapter.getNamespaceURI(this.current);
|
|
|
|
this.pop();
|
|
|
|
if (tn === $.TD || (tn === $.TH && ns === NS.HTML)) {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
popAllUpToHtmlElement() {
|
|
//NOTE: here we assume that root <html> element is always first in the open element stack, so
|
|
//we perform this fast stack clean up.
|
|
this.stackTop = 0;
|
|
this._updateCurrentElement();
|
|
}
|
|
|
|
clearBackToTableContext() {
|
|
while (
|
|
(this.currentTagName !== $.TABLE && this.currentTagName !== $.TEMPLATE && this.currentTagName !== $.HTML) ||
|
|
this.treeAdapter.getNamespaceURI(this.current) !== NS.HTML
|
|
) {
|
|
this.pop();
|
|
}
|
|
}
|
|
|
|
clearBackToTableBodyContext() {
|
|
while (
|
|
(this.currentTagName !== $.TBODY &&
|
|
this.currentTagName !== $.TFOOT &&
|
|
this.currentTagName !== $.THEAD &&
|
|
this.currentTagName !== $.TEMPLATE &&
|
|
this.currentTagName !== $.HTML) ||
|
|
this.treeAdapter.getNamespaceURI(this.current) !== NS.HTML
|
|
) {
|
|
this.pop();
|
|
}
|
|
}
|
|
|
|
clearBackToTableRowContext() {
|
|
while (
|
|
(this.currentTagName !== $.TR && this.currentTagName !== $.TEMPLATE && this.currentTagName !== $.HTML) ||
|
|
this.treeAdapter.getNamespaceURI(this.current) !== NS.HTML
|
|
) {
|
|
this.pop();
|
|
}
|
|
}
|
|
|
|
remove(element) {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
if (this.items[i] === element) {
|
|
this.items.splice(i, 1);
|
|
this.stackTop--;
|
|
this._updateCurrentElement();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
//Search
|
|
tryPeekProperlyNestedBodyElement() {
|
|
//Properly nested <body> element (should be second element in stack).
|
|
const element = this.items[1];
|
|
|
|
return element && this.treeAdapter.getTagName(element) === $.BODY ? element : null;
|
|
}
|
|
|
|
contains(element) {
|
|
return this._indexOf(element) > -1;
|
|
}
|
|
|
|
getCommonAncestor(element) {
|
|
let elementIdx = this._indexOf(element);
|
|
|
|
return --elementIdx >= 0 ? this.items[elementIdx] : null;
|
|
}
|
|
|
|
isRootHtmlElementCurrent() {
|
|
return this.stackTop === 0 && this.currentTagName === $.HTML;
|
|
}
|
|
|
|
//Element in scope
|
|
hasInScope(tagName) {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
const tn = this.treeAdapter.getTagName(this.items[i]);
|
|
const ns = this.treeAdapter.getNamespaceURI(this.items[i]);
|
|
|
|
if (tn === tagName && ns === NS.HTML) {
|
|
return true;
|
|
}
|
|
|
|
if (isScopingElement(tn, ns)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
hasNumberedHeaderInScope() {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
const tn = this.treeAdapter.getTagName(this.items[i]);
|
|
const ns = this.treeAdapter.getNamespaceURI(this.items[i]);
|
|
|
|
if (
|
|
(tn === $.H1 || tn === $.H2 || tn === $.H3 || tn === $.H4 || tn === $.H5 || tn === $.H6) &&
|
|
ns === NS.HTML
|
|
) {
|
|
return true;
|
|
}
|
|
|
|
if (isScopingElement(tn, ns)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
hasInListItemScope(tagName) {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
const tn = this.treeAdapter.getTagName(this.items[i]);
|
|
const ns = this.treeAdapter.getNamespaceURI(this.items[i]);
|
|
|
|
if (tn === tagName && ns === NS.HTML) {
|
|
return true;
|
|
}
|
|
|
|
if (((tn === $.UL || tn === $.OL) && ns === NS.HTML) || isScopingElement(tn, ns)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
hasInButtonScope(tagName) {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
const tn = this.treeAdapter.getTagName(this.items[i]);
|
|
const ns = this.treeAdapter.getNamespaceURI(this.items[i]);
|
|
|
|
if (tn === tagName && ns === NS.HTML) {
|
|
return true;
|
|
}
|
|
|
|
if ((tn === $.BUTTON && ns === NS.HTML) || isScopingElement(tn, ns)) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
hasInTableScope(tagName) {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
const tn = this.treeAdapter.getTagName(this.items[i]);
|
|
const ns = this.treeAdapter.getNamespaceURI(this.items[i]);
|
|
|
|
if (ns !== NS.HTML) {
|
|
continue;
|
|
}
|
|
|
|
if (tn === tagName) {
|
|
return true;
|
|
}
|
|
|
|
if (tn === $.TABLE || tn === $.TEMPLATE || tn === $.HTML) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
hasTableBodyContextInTableScope() {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
const tn = this.treeAdapter.getTagName(this.items[i]);
|
|
const ns = this.treeAdapter.getNamespaceURI(this.items[i]);
|
|
|
|
if (ns !== NS.HTML) {
|
|
continue;
|
|
}
|
|
|
|
if (tn === $.TBODY || tn === $.THEAD || tn === $.TFOOT) {
|
|
return true;
|
|
}
|
|
|
|
if (tn === $.TABLE || tn === $.HTML) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
hasInSelectScope(tagName) {
|
|
for (let i = this.stackTop; i >= 0; i--) {
|
|
const tn = this.treeAdapter.getTagName(this.items[i]);
|
|
const ns = this.treeAdapter.getNamespaceURI(this.items[i]);
|
|
|
|
if (ns !== NS.HTML) {
|
|
continue;
|
|
}
|
|
|
|
if (tn === tagName) {
|
|
return true;
|
|
}
|
|
|
|
if (tn !== $.OPTION && tn !== $.OPTGROUP) {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
//Implied end tags
|
|
generateImpliedEndTags() {
|
|
while (isImpliedEndTagRequired(this.currentTagName)) {
|
|
this.pop();
|
|
}
|
|
}
|
|
|
|
generateImpliedEndTagsThoroughly() {
|
|
while (isImpliedEndTagRequiredThoroughly(this.currentTagName)) {
|
|
this.pop();
|
|
}
|
|
}
|
|
|
|
generateImpliedEndTagsWithExclusion(exclusionTagName) {
|
|
while (isImpliedEndTagRequired(this.currentTagName) && this.currentTagName !== exclusionTagName) {
|
|
this.pop();
|
|
}
|
|
}
|
|
}
|
|
|
|
module.exports = OpenElementStack;
|