DocumentRuleSets.cpp [plain text]
#include "config.h"
#include "DocumentRuleSets.h"
#include "CSSStyleSheet.h"
#include "ExtensionStyleSheets.h"
#include "MediaQueryEvaluator.h"
#include "StyleResolver.h"
#include "StyleSheetContents.h"
namespace WebCore {
DocumentRuleSets::DocumentRuleSets(StyleResolver& styleResolver)
: m_styleResolver(styleResolver)
{
m_authorStyle = std::make_unique<RuleSet>();
m_authorStyle->disableAutoShrinkToFit();
}
DocumentRuleSets::~DocumentRuleSets() = default;
RuleSet* DocumentRuleSets::userAgentMediaQueryStyle() const
{
if (m_isForShadowScope)
return m_styleResolver.document().styleScope().resolver().ruleSets().userAgentMediaQueryStyle();
updateUserAgentMediaQueryStyleIfNeeded();
return m_userAgentMediaQueryStyle.get();
}
void DocumentRuleSets::updateUserAgentMediaQueryStyleIfNeeded() const
{
if (!CSSDefaultStyleSheets::mediaQueryStyleSheet)
return;
auto ruleCount = CSSDefaultStyleSheets::mediaQueryStyleSheet->ruleCount();
if (m_userAgentMediaQueryStyle && ruleCount == m_userAgentMediaQueryRuleCountOnUpdate)
return;
m_userAgentMediaQueryRuleCountOnUpdate = ruleCount;
#if !ASSERT_DISABLED
bool hadViewportDependentMediaQueries = m_styleResolver.hasViewportDependentMediaQueries();
#endif
auto& mediaQueryEvaluator = m_styleResolver.mediaQueryEvaluator();
m_userAgentMediaQueryStyle = std::make_unique<RuleSet>();
m_userAgentMediaQueryStyle->addRulesFromSheet(*CSSDefaultStyleSheets::mediaQueryStyleSheet, mediaQueryEvaluator, &m_styleResolver);
ASSERT(!m_styleResolver.hasViewportDependentMediaQueries() || hadViewportDependentMediaQueries);
}
RuleSet* DocumentRuleSets::userStyle() const
{
if (m_usesSharedUserStyle)
return m_styleResolver.document().styleScope().resolver().ruleSets().userStyle();
return m_userStyle.get();
}
void DocumentRuleSets::initializeUserStyle()
{
auto& extensionStyleSheets = m_styleResolver.document().extensionStyleSheets();
auto& mediaQueryEvaluator = m_styleResolver.mediaQueryEvaluator();
auto tempUserStyle = std::make_unique<RuleSet>();
if (CSSStyleSheet* pageUserSheet = extensionStyleSheets.pageUserSheet())
tempUserStyle->addRulesFromSheet(pageUserSheet->contents(), mediaQueryEvaluator, &m_styleResolver);
collectRulesFromUserStyleSheets(extensionStyleSheets.injectedUserStyleSheets(), *tempUserStyle, mediaQueryEvaluator, m_styleResolver);
collectRulesFromUserStyleSheets(extensionStyleSheets.documentUserStyleSheets(), *tempUserStyle, mediaQueryEvaluator, m_styleResolver);
if (tempUserStyle->ruleCount() > 0 || tempUserStyle->pageRules().size() > 0)
m_userStyle = WTFMove(tempUserStyle);
}
void DocumentRuleSets::collectRulesFromUserStyleSheets(const Vector<RefPtr<CSSStyleSheet>>& userSheets, RuleSet& userStyle, const MediaQueryEvaluator& medium, StyleResolver& resolver)
{
for (unsigned i = 0; i < userSheets.size(); ++i) {
ASSERT(userSheets[i]->contents().isUserStyleSheet());
userStyle.addRulesFromSheet(userSheets[i]->contents(), medium, &resolver);
}
}
static std::unique_ptr<RuleSet> makeRuleSet(const Vector<RuleFeature>& rules)
{
size_t size = rules.size();
if (!size)
return nullptr;
auto ruleSet = std::make_unique<RuleSet>();
for (size_t i = 0; i < size; ++i)
ruleSet->addRule(rules[i].rule, rules[i].selectorIndex);
ruleSet->shrinkToFit();
return ruleSet;
}
void DocumentRuleSets::resetAuthorStyle()
{
m_isAuthorStyleDefined = true;
m_authorStyle = std::make_unique<RuleSet>();
m_authorStyle->disableAutoShrinkToFit();
}
void DocumentRuleSets::resetUserAgentMediaQueryStyle()
{
m_userAgentMediaQueryStyle = nullptr;
}
void DocumentRuleSets::appendAuthorStyleSheets(const Vector<RefPtr<CSSStyleSheet>>& styleSheets, MediaQueryEvaluator* medium, InspectorCSSOMWrappers& inspectorCSSOMWrappers, StyleResolver* resolver)
{
for (auto& cssSheet : styleSheets) {
ASSERT(!cssSheet->disabled());
if (cssSheet->mediaQueries() && !medium->evaluate(*cssSheet->mediaQueries(), resolver))
continue;
m_authorStyle->addRulesFromSheet(cssSheet->contents(), *medium, resolver);
inspectorCSSOMWrappers.collectFromStyleSheetIfNeeded(cssSheet.get());
}
m_authorStyle->shrinkToFit();
collectFeatures();
}
void DocumentRuleSets::collectFeatures() const
{
m_features.clear();
if (CSSDefaultStyleSheets::defaultStyle)
m_features.add(CSSDefaultStyleSheets::defaultStyle->features());
m_defaultStyleVersionOnFeatureCollection = CSSDefaultStyleSheets::defaultStyleVersion;
if (auto* userAgentMediaQueryStyle = this->userAgentMediaQueryStyle())
m_features.add(userAgentMediaQueryStyle->features());
if (m_authorStyle)
m_features.add(m_authorStyle->features());
if (auto* userStyle = this->userStyle())
m_features.add(userStyle->features());
m_siblingRuleSet = makeRuleSet(m_features.siblingRules);
m_uncommonAttributeRuleSet = makeRuleSet(m_features.uncommonAttributeRules);
m_ancestorClassRuleSets.clear();
m_ancestorAttributeRuleSetsForHTML.clear();
m_features.shrinkToFit();
}
RuleSet* DocumentRuleSets::ancestorClassRules(const AtomicString& className) const
{
auto addResult = m_ancestorClassRuleSets.add(className, nullptr);
if (addResult.isNewEntry) {
if (auto* rules = m_features.ancestorClassRules.get(className))
addResult.iterator->value = makeRuleSet(*rules);
}
return addResult.iterator->value.get();
}
const DocumentRuleSets::AttributeRules* DocumentRuleSets::ancestorAttributeRulesForHTML(const AtomicString& attributeName) const
{
auto addResult = m_ancestorAttributeRuleSetsForHTML.add(attributeName, nullptr);
auto& value = addResult.iterator->value;
if (addResult.isNewEntry) {
if (auto* rules = m_features.ancestorAttributeRulesForHTML.get(attributeName)) {
value = std::make_unique<AttributeRules>();
value->attributeSelectors.reserveCapacity(rules->selectors.size());
for (auto* selector : rules->selectors.values())
value->attributeSelectors.uncheckedAppend(selector);
value->ruleSet = makeRuleSet(rules->features);
}
}
return value.get();
}
}