1dfb5cef5b9460bb346325bfca6555e44133c560
[WebKit-https.git] / Source / WebCore / dom / TreeScopeAdopter.cpp
1 /*
2  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
3  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
4  *           (C) 2001 Dirk Mueller (mueller@kde.org)
5  * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011 Apple Inc. All rights reserved.
6  * Copyright (C) 2008 Nokia Corporation and/or its subsidiary(-ies)
7  * Copyright (C) 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
8  * Copyright (C) 2011 Google Inc. All rights reserved.
9  *
10  * This library is free software; you can redistribute it and/or
11  * modify it under the terms of the GNU Library General Public
12  * License as published by the Free Software Foundation; either
13  * version 2 of the License, or (at your option) any later version.
14  *
15  * This library is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
18  * Library General Public License for more details.
19  *
20  * You should have received a copy of the GNU Library General Public License
21  * along with this library; see the file COPYING.LIB.  If not, write to
22  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
23  * Boston, MA 02110-1301, USA.
24  */
25 #include "config.h"
26 #include "TreeScopeAdopter.h"
27
28 #include "Document.h"
29 #include "ElementRareData.h"
30 #include "ElementShadow.h"
31 #include "NodeRareData.h"
32 #include "RenderStyle.h"
33 #include "ShadowRoot.h"
34
35 namespace WebCore {
36
37 void TreeScopeAdopter::moveTreeToNewScope(Node* root) const
38 {
39     ASSERT(needsScopeChange());
40
41     // If an element is moved from a document and then eventually back again the collection cache for
42     // that element may contain stale data as changes made to it will have updated the DOMTreeVersion
43     // of the document it was moved to. By increasing the DOMTreeVersion of the donating document here
44     // we ensure that the collection cache will be invalidated as needed when the element is moved back.
45     Document* oldDocument = m_oldScope ? m_oldScope->rootNode()->document() : 0;
46     Document* newDocument = m_newScope->rootNode()->document();
47     bool willMoveToNewDocument = oldDocument != newDocument;
48     if (oldDocument && willMoveToNewDocument)
49         oldDocument->incDOMTreeVersion();
50
51     for (Node* node = root; node; node = node->traverseNextNode(root)) {
52         node->setTreeScope(m_newScope);
53         if (node->hasRareData()) {
54             NodeRareData* rareData = node->rareData();
55             if (rareData->nodeLists())
56                 rareData->nodeLists()->adoptTreeScope(oldDocument, newDocument);
57             if (node->isElementNode())
58                 static_cast<ElementRareData*>(rareData)->adoptTreeScope(oldDocument, newDocument);
59         }
60
61         if (willMoveToNewDocument)
62             moveNodeToNewDocument(node, oldDocument, newDocument);
63
64         for (ShadowRoot* shadow = node->youngestShadowRoot(); shadow; shadow = shadow->olderShadowRoot()) {
65             shadow->setParentTreeScope(m_newScope);
66             if (willMoveToNewDocument)
67                 moveTreeToNewDocument(shadow, oldDocument, newDocument);
68         }
69     }
70 }
71
72 void TreeScopeAdopter::moveTreeToNewDocument(Node* root, Document* oldDocument, Document* newDocument) const
73 {
74     for (Node* node = root; node; node = node->traverseNextNode(root)) {
75         moveNodeToNewDocument(node, oldDocument, newDocument);
76         for (ShadowRoot* shadow = node->youngestShadowRoot(); shadow; shadow = shadow->olderShadowRoot())
77             moveTreeToNewDocument(shadow, oldDocument, newDocument);
78     }
79 }
80
81 #ifndef NDEBUG
82 static bool didMoveToNewDocumentWasCalled = false;
83 static Document* oldDocumentDidMoveToNewDocumentWasCalledWith = 0;
84
85 void TreeScopeAdopter::ensureDidMoveToNewDocumentWasCalled(Document* oldDocument)
86 {
87     ASSERT(!didMoveToNewDocumentWasCalled);
88     ASSERT_UNUSED(oldDocument, oldDocument == oldDocumentDidMoveToNewDocumentWasCalledWith);
89     didMoveToNewDocumentWasCalled = true;
90 }
91 #endif
92
93 inline void TreeScopeAdopter::moveNodeToNewDocument(Node* node, Document* oldDocument, Document* newDocument) const
94 {
95     ASSERT(!node->inDocument() || oldDocument != newDocument);
96
97     newDocument->guardRef();
98     if (oldDocument)
99         oldDocument->moveNodeIteratorsToNewDocument(node, newDocument);
100
101 #ifndef NDEBUG
102     didMoveToNewDocumentWasCalled = false;
103     oldDocumentDidMoveToNewDocumentWasCalledWith = oldDocument;
104 #endif
105
106     node->didMoveToNewDocument(oldDocument);
107     ASSERT(didMoveToNewDocumentWasCalled);
108     
109     if (oldDocument)
110         oldDocument->guardDeref();
111 }
112
113 }