TreeScopeAdopter.cpp [plain text]
#include "config.h"
#include "TreeScopeAdopter.h"
#include "Attr.h"
#include "ElementRareData.h"
#include "NodeTraversal.h"
namespace WebCore {
void TreeScopeAdopter::moveTreeToNewScope(Node& root) const
{
ASSERT(needsScopeChange());
Document& oldDocument = m_oldScope.documentScope();
Document& newDocument = m_newScope.documentScope();
bool willMoveToNewDocument = &oldDocument != &newDocument;
if (willMoveToNewDocument) {
oldDocument.incrementReferencingNodeCount();
oldDocument.incDOMTreeVersion();
}
for (Node* node = &root; node; node = NodeTraversal::next(*node, &root)) {
updateTreeScope(*node);
if (willMoveToNewDocument)
moveNodeToNewDocument(*node, oldDocument, newDocument);
else if (node->hasRareData()) {
NodeRareData* rareData = node->rareData();
if (rareData->nodeLists())
rareData->nodeLists()->adoptTreeScope();
}
if (!is<Element>(*node))
continue;
if (node->hasSyntheticAttrChildNodes()) {
for (auto& attr : downcast<Element>(*node).attrNodeList())
moveTreeToNewScope(*attr);
}
if (auto* shadow = node->shadowRoot()) {
shadow->setParentTreeScope(m_newScope);
if (willMoveToNewDocument)
moveShadowTreeToNewDocument(*shadow, oldDocument, newDocument);
}
}
if (willMoveToNewDocument)
oldDocument.decrementReferencingNodeCount();
}
void TreeScopeAdopter::moveShadowTreeToNewDocument(ShadowRoot& shadowRoot, Document& oldDocument, Document& newDocument) const
{
for (Node* node = &shadowRoot; node; node = NodeTraversal::next(*node, &shadowRoot)) {
moveNodeToNewDocument(*node, oldDocument, newDocument);
if (auto* shadow = node->shadowRoot())
moveShadowTreeToNewDocument(*shadow, oldDocument, newDocument);
}
}
#ifndef NDEBUG
static bool didMoveToNewDocumentWasCalled = false;
static Document* oldDocumentDidMoveToNewDocumentWasCalledWith = nullptr;
void TreeScopeAdopter::ensureDidMoveToNewDocumentWasCalled(Document& oldDocument)
{
ASSERT(!didMoveToNewDocumentWasCalled);
ASSERT_UNUSED(oldDocument, &oldDocument == oldDocumentDidMoveToNewDocumentWasCalledWith);
didMoveToNewDocumentWasCalled = true;
}
#endif
inline void TreeScopeAdopter::updateTreeScope(Node& node) const
{
ASSERT(!node.isTreeScope());
ASSERT(&node.treeScope() == &m_oldScope);
node.setTreeScope(m_newScope);
}
inline void TreeScopeAdopter::moveNodeToNewDocument(Node& node, Document& oldDocument, Document& newDocument) const
{
ASSERT(!node.inDocument() || &oldDocument != &newDocument);
newDocument.incrementReferencingNodeCount();
oldDocument.decrementReferencingNodeCount();
if (node.hasRareData()) {
NodeRareData* rareData = node.rareData();
if (auto* nodeLists = rareData->nodeLists())
nodeLists->adoptDocument(oldDocument, newDocument);
}
oldDocument.moveNodeIteratorsToNewDocument(node, newDocument);
if (is<ShadowRoot>(node))
downcast<ShadowRoot>(node).setDocumentScope(newDocument);
#ifndef NDEBUG
didMoveToNewDocumentWasCalled = false;
oldDocumentDidMoveToNewDocumentWasCalledWith = &oldDocument;
#endif
node.didMoveToNewDocument(oldDocument);
ASSERT(didMoveToNewDocumentWasCalled);
}
}