123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223 |
- 'use strict';
-
- const Mixin = require('../../utils/mixin');
- const Tokenizer = require('../../tokenizer');
- const LocationInfoTokenizerMixin = require('./tokenizer-mixin');
- const LocationInfoOpenElementStackMixin = require('./open-element-stack-mixin');
- const HTML = require('../../common/html');
-
- //Aliases
- const $ = HTML.TAG_NAMES;
-
- class LocationInfoParserMixin extends Mixin {
- constructor(parser) {
- super(parser);
-
- this.parser = parser;
- this.treeAdapter = this.parser.treeAdapter;
- this.posTracker = null;
- this.lastStartTagToken = null;
- this.lastFosterParentingLocation = null;
- this.currentToken = null;
- }
-
- _setStartLocation(element) {
- let loc = null;
-
- if (this.lastStartTagToken) {
- loc = Object.assign({}, this.lastStartTagToken.location);
- loc.startTag = this.lastStartTagToken.location;
- }
-
- this.treeAdapter.setNodeSourceCodeLocation(element, loc);
- }
-
- _setEndLocation(element, closingToken) {
- const loc = this.treeAdapter.getNodeSourceCodeLocation(element);
-
- if (loc) {
- if (closingToken.location) {
- const ctLoc = closingToken.location;
- const tn = this.treeAdapter.getTagName(element);
-
- // NOTE: For cases like <p> <p> </p> - First 'p' closes without a closing
- // tag and for cases like <td> <p> </td> - 'p' closes without a closing tag.
- const isClosingEndTag = closingToken.type === Tokenizer.END_TAG_TOKEN && tn === closingToken.tagName;
- const endLoc = {};
- if (isClosingEndTag) {
- endLoc.endTag = Object.assign({}, ctLoc);
- endLoc.endLine = ctLoc.endLine;
- endLoc.endCol = ctLoc.endCol;
- endLoc.endOffset = ctLoc.endOffset;
- } else {
- endLoc.endLine = ctLoc.startLine;
- endLoc.endCol = ctLoc.startCol;
- endLoc.endOffset = ctLoc.startOffset;
- }
-
- this.treeAdapter.updateNodeSourceCodeLocation(element, endLoc);
- }
- }
- }
-
- _getOverriddenMethods(mxn, orig) {
- return {
- _bootstrap(document, fragmentContext) {
- orig._bootstrap.call(this, document, fragmentContext);
-
- mxn.lastStartTagToken = null;
- mxn.lastFosterParentingLocation = null;
- mxn.currentToken = null;
-
- const tokenizerMixin = Mixin.install(this.tokenizer, LocationInfoTokenizerMixin);
-
- mxn.posTracker = tokenizerMixin.posTracker;
-
- Mixin.install(this.openElements, LocationInfoOpenElementStackMixin, {
- onItemPop: function(element) {
- mxn._setEndLocation(element, mxn.currentToken);
- }
- });
- },
-
- _runParsingLoop(scriptHandler) {
- orig._runParsingLoop.call(this, scriptHandler);
-
- // NOTE: generate location info for elements
- // that remains on open element stack
- for (let i = this.openElements.stackTop; i >= 0; i--) {
- mxn._setEndLocation(this.openElements.items[i], mxn.currentToken);
- }
- },
-
- //Token processing
- _processTokenInForeignContent(token) {
- mxn.currentToken = token;
- orig._processTokenInForeignContent.call(this, token);
- },
-
- _processToken(token) {
- mxn.currentToken = token;
- orig._processToken.call(this, token);
-
- //NOTE: <body> and <html> are never popped from the stack, so we need to updated
- //their end location explicitly.
- const requireExplicitUpdate =
- token.type === Tokenizer.END_TAG_TOKEN &&
- (token.tagName === $.HTML || (token.tagName === $.BODY && this.openElements.hasInScope($.BODY)));
-
- if (requireExplicitUpdate) {
- for (let i = this.openElements.stackTop; i >= 0; i--) {
- const element = this.openElements.items[i];
-
- if (this.treeAdapter.getTagName(element) === token.tagName) {
- mxn._setEndLocation(element, token);
- break;
- }
- }
- }
- },
-
- //Doctype
- _setDocumentType(token) {
- orig._setDocumentType.call(this, token);
-
- const documentChildren = this.treeAdapter.getChildNodes(this.document);
- const cnLength = documentChildren.length;
-
- for (let i = 0; i < cnLength; i++) {
- const node = documentChildren[i];
-
- if (this.treeAdapter.isDocumentTypeNode(node)) {
- this.treeAdapter.setNodeSourceCodeLocation(node, token.location);
- break;
- }
- }
- },
-
- //Elements
- _attachElementToTree(element) {
- //NOTE: _attachElementToTree is called from _appendElement, _insertElement and _insertTemplate methods.
- //So we will use token location stored in this methods for the element.
- mxn._setStartLocation(element);
- mxn.lastStartTagToken = null;
- orig._attachElementToTree.call(this, element);
- },
-
- _appendElement(token, namespaceURI) {
- mxn.lastStartTagToken = token;
- orig._appendElement.call(this, token, namespaceURI);
- },
-
- _insertElement(token, namespaceURI) {
- mxn.lastStartTagToken = token;
- orig._insertElement.call(this, token, namespaceURI);
- },
-
- _insertTemplate(token) {
- mxn.lastStartTagToken = token;
- orig._insertTemplate.call(this, token);
-
- const tmplContent = this.treeAdapter.getTemplateContent(this.openElements.current);
-
- this.treeAdapter.setNodeSourceCodeLocation(tmplContent, null);
- },
-
- _insertFakeRootElement() {
- orig._insertFakeRootElement.call(this);
- this.treeAdapter.setNodeSourceCodeLocation(this.openElements.current, null);
- },
-
- //Comments
- _appendCommentNode(token, parent) {
- orig._appendCommentNode.call(this, token, parent);
-
- const children = this.treeAdapter.getChildNodes(parent);
- const commentNode = children[children.length - 1];
-
- this.treeAdapter.setNodeSourceCodeLocation(commentNode, token.location);
- },
-
- //Text
- _findFosterParentingLocation() {
- //NOTE: store last foster parenting location, so we will be able to find inserted text
- //in case of foster parenting
- mxn.lastFosterParentingLocation = orig._findFosterParentingLocation.call(this);
-
- return mxn.lastFosterParentingLocation;
- },
-
- _insertCharacters(token) {
- orig._insertCharacters.call(this, token);
-
- const hasFosterParent = this._shouldFosterParentOnInsertion();
-
- const parent =
- (hasFosterParent && mxn.lastFosterParentingLocation.parent) ||
- this.openElements.currentTmplContent ||
- this.openElements.current;
-
- const siblings = this.treeAdapter.getChildNodes(parent);
-
- const textNodeIdx =
- hasFosterParent && mxn.lastFosterParentingLocation.beforeElement
- ? siblings.indexOf(mxn.lastFosterParentingLocation.beforeElement) - 1
- : siblings.length - 1;
-
- const textNode = siblings[textNodeIdx];
-
- //NOTE: if we have location assigned by another token, then just update end position
- const tnLoc = this.treeAdapter.getNodeSourceCodeLocation(textNode);
-
- if (tnLoc) {
- const { endLine, endCol, endOffset } = token.location;
- this.treeAdapter.updateNodeSourceCodeLocation(textNode, { endLine, endCol, endOffset });
- } else {
- this.treeAdapter.setNodeSourceCodeLocation(textNode, token.location);
- }
- }
- };
- }
- }
-
- module.exports = LocationInfoParserMixin;
|