Unreviewed, rolling out r216069.
[WebKit-https.git] / Source / WebCore / css / DocumentRuleSets.cpp
1 /*
2  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
3  *           (C) 2004-2005 Allan Sandfeld Jensen (kde@carewolf.com)
4  * Copyright (C) 2006, 2007 Nicholas Shanks (webkit@nickshanks.com)
5  * Copyright (C) 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012 Apple Inc. All rights reserved.
6  * Copyright (C) 2007 Alexey Proskuryakov <ap@webkit.org>
7  * Copyright (C) 2007, 2008 Eric Seidel <eric@webkit.org>
8  * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
9  * Copyright (c) 2011, Code Aurora Forum. All rights reserved.
10  * Copyright (C) Research In Motion Limited 2011. All rights reserved.
11  * Copyright (C) 2013 Google Inc. All rights reserved.
12  *
13  * This library is free software; you can redistribute it and/or
14  * modify it under the terms of the GNU Library General Public
15  * License as published by the Free Software Foundation; either
16  * version 2 of the License, or (at your option) any later version.
17  *
18  * This library is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
21  * Library General Public License for more details.
22  *
23  * You should have received a copy of the GNU Library General Public License
24  * along with this library; see the file COPYING.LIB.  If not, write to
25  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
26  * Boston, MA 02110-1301, USA.
27  */
28
29 #include "config.h"
30 #include "DocumentRuleSets.h"
31
32 #include "CSSStyleSheet.h"
33 #include "ExtensionStyleSheets.h"
34 #include "MediaQueryEvaluator.h"
35 #include "StyleResolver.h"
36 #include "StyleSheetContents.h"
37
38 namespace WebCore {
39
40 DocumentRuleSets::DocumentRuleSets()
41 {
42     m_authorStyle = std::make_unique<RuleSet>();
43     m_authorStyle->disableAutoShrinkToFit();
44 }
45
46 DocumentRuleSets::~DocumentRuleSets()
47 {
48 }
49
50 void DocumentRuleSets::initUserStyle(ExtensionStyleSheets& extensionStyleSheets, const MediaQueryEvaluator& medium, StyleResolver& resolver)
51 {
52     auto tempUserStyle = std::make_unique<RuleSet>();
53     if (CSSStyleSheet* pageUserSheet = extensionStyleSheets.pageUserSheet())
54         tempUserStyle->addRulesFromSheet(pageUserSheet->contents(), medium, &resolver);
55     collectRulesFromUserStyleSheets(extensionStyleSheets.injectedUserStyleSheets(), *tempUserStyle, medium, resolver);
56     collectRulesFromUserStyleSheets(extensionStyleSheets.documentUserStyleSheets(), *tempUserStyle, medium, resolver);
57     if (tempUserStyle->ruleCount() > 0 || tempUserStyle->pageRules().size() > 0)
58         m_userStyle = WTFMove(tempUserStyle);
59 }
60
61 void DocumentRuleSets::collectRulesFromUserStyleSheets(const Vector<RefPtr<CSSStyleSheet>>& userSheets, RuleSet& userStyle, const MediaQueryEvaluator& medium, StyleResolver& resolver)
62 {
63     for (unsigned i = 0; i < userSheets.size(); ++i) {
64         ASSERT(userSheets[i]->contents().isUserStyleSheet());
65         userStyle.addRulesFromSheet(userSheets[i]->contents(), medium, &resolver);
66     }
67 }
68
69 static std::unique_ptr<RuleSet> makeRuleSet(const Vector<RuleFeature>& rules)
70 {
71     size_t size = rules.size();
72     if (!size)
73         return nullptr;
74     auto ruleSet = std::make_unique<RuleSet>();
75     for (size_t i = 0; i < size; ++i)
76         ruleSet->addRule(rules[i].rule, rules[i].selectorIndex, rules[i].hasDocumentSecurityOrigin ? RuleHasDocumentSecurityOrigin : RuleHasNoSpecialState);
77     ruleSet->shrinkToFit();
78     return ruleSet;
79 }
80
81 void DocumentRuleSets::resetAuthorStyle()
82 {
83     m_isAuthorStyleDefined = true;
84     m_authorStyle = std::make_unique<RuleSet>();
85     m_authorStyle->disableAutoShrinkToFit();
86 }
87
88 void DocumentRuleSets::appendAuthorStyleSheets(const Vector<RefPtr<CSSStyleSheet>>& styleSheets, MediaQueryEvaluator* medium, InspectorCSSOMWrappers& inspectorCSSOMWrappers, StyleResolver* resolver)
89 {
90     // This handles sheets added to the end of the stylesheet list only. In other cases the style resolver
91     // needs to be reconstructed. To handle insertions too the rule order numbers would need to be updated.
92     for (auto& cssSheet : styleSheets) {
93         ASSERT(!cssSheet->disabled());
94         if (cssSheet->mediaQueries() && !medium->evaluate(*cssSheet->mediaQueries(), resolver))
95             continue;
96         m_authorStyle->addRulesFromSheet(cssSheet->contents(), *medium, resolver);
97         inspectorCSSOMWrappers.collectFromStyleSheetIfNeeded(cssSheet.get());
98     }
99     m_authorStyle->shrinkToFit();
100     collectFeatures();
101 }
102
103 void DocumentRuleSets::collectFeatures() const
104 {
105     m_features.clear();
106     // Collect all ids and rules using sibling selectors (:first-child and similar)
107     // in the current set of stylesheets. Style sharing code uses this information to reject
108     // sharing candidates.
109     if (CSSDefaultStyleSheets::defaultStyle)
110         m_features.add(CSSDefaultStyleSheets::defaultStyle->features());
111     m_defaultStyleVersionOnFeatureCollection = CSSDefaultStyleSheets::defaultStyleVersion;
112
113     if (m_authorStyle)
114         m_features.add(m_authorStyle->features());
115     if (m_userStyle)
116         m_features.add(m_userStyle->features());
117
118     m_siblingRuleSet = makeRuleSet(m_features.siblingRules);
119     m_uncommonAttributeRuleSet = makeRuleSet(m_features.uncommonAttributeRules);
120
121     m_ancestorClassRuleSets.clear();
122     m_ancestorAttributeRuleSetsForHTML.clear();
123
124     m_features.shrinkToFit();
125 }
126
127 RuleSet* DocumentRuleSets::ancestorClassRules(const AtomicString& className) const
128 {
129     auto addResult = m_ancestorClassRuleSets.add(className, nullptr);
130     if (addResult.isNewEntry) {
131         if (auto* rules = m_features.ancestorClassRules.get(className))
132             addResult.iterator->value = makeRuleSet(*rules);
133     }
134     return addResult.iterator->value.get();
135 }
136
137 const DocumentRuleSets::AttributeRules* DocumentRuleSets::ancestorAttributeRulesForHTML(const AtomicString& attributeName) const
138 {
139     auto addResult = m_ancestorAttributeRuleSetsForHTML.add(attributeName, nullptr);
140     auto& value = addResult.iterator->value;
141     if (addResult.isNewEntry) {
142         if (auto* rules = m_features.ancestorAttributeRulesForHTML.get(attributeName)) {
143             value = std::make_unique<AttributeRules>();
144             value->attributeSelectors.reserveCapacity(rules->selectors.size());
145             for (auto* selector : rules->selectors.values())
146                 value->attributeSelectors.uncheckedAppend(selector);
147             value->ruleSet = makeRuleSet(rules->features);
148         }
149     }
150     return value.get();
151 }
152
153 } // namespace WebCore