StyleInvalidator.cpp [plain text]
#include "config.h"
#include "StyleInvalidator.h"
#include "CSSSelectorList.h"
#include "Document.h"
#include "ElementIterator.h"
#include "ElementRuleCollector.h"
#include "HTMLSlotElement.h"
#include "SelectorFilter.h"
#include "ShadowRoot.h"
#include "StyleRuleImport.h"
#include "StyleSheetContents.h"
namespace WebCore {
namespace Style {
static bool shouldDirtyAllStyle(const Vector<RefPtr<StyleRuleBase>>& rules)
{
for (auto& rule : rules) {
if (is<StyleRuleMedia>(*rule)) {
const auto* childRules = downcast<StyleRuleMedia>(*rule).childRulesWithoutDeferredParsing();
if (childRules && shouldDirtyAllStyle(*childRules))
return true;
continue;
}
if (!is<StyleRule>(*rule))
return true;
}
return false;
}
static bool shouldDirtyAllStyle(const StyleSheetContents& sheet)
{
for (auto& import : sheet.importRules()) {
if (!import->styleSheet())
continue;
if (shouldDirtyAllStyle(*import->styleSheet()))
return true;
}
if (shouldDirtyAllStyle(sheet.childRules()))
return true;
return false;
}
static bool shouldDirtyAllStyle(const Vector<StyleSheetContents*>& sheets)
{
for (auto& sheet : sheets) {
if (shouldDirtyAllStyle(*sheet))
return true;
}
return false;
}
Invalidator::Invalidator(const Vector<StyleSheetContents*>& sheets, const MediaQueryEvaluator& mediaQueryEvaluator)
: m_ownedRuleSet(std::make_unique<RuleSet>())
, m_ruleSet(*m_ownedRuleSet)
, m_dirtiesAllStyle(shouldDirtyAllStyle(sheets))
{
if (m_dirtiesAllStyle)
return;
m_ownedRuleSet->disableAutoShrinkToFit();
for (auto& sheet : sheets)
m_ownedRuleSet->addRulesFromSheet(*sheet, mediaQueryEvaluator);
m_hasShadowPseudoElementRulesInAuthorSheet = m_ruleSet.hasShadowPseudoElementRules();
}
Invalidator::Invalidator(const RuleSet& ruleSet)
: m_ruleSet(ruleSet)
, m_hasShadowPseudoElementRulesInAuthorSheet(ruleSet.hasShadowPseudoElementRules())
{
}
Invalidator::CheckDescendants Invalidator::invalidateIfNeeded(Element& element, const SelectorFilter* filter)
{
if (m_hasShadowPseudoElementRulesInAuthorSheet) {
if (element.shadowRoot())
element.invalidateStyleForSubtree();
}
bool shouldCheckForSlots = !m_ruleSet.slottedPseudoElementRules().isEmpty() && !m_didInvalidateHostChildren;
if (shouldCheckForSlots && is<HTMLSlotElement>(element)) {
auto* containingShadowRoot = element.containingShadowRoot();
if (containingShadowRoot && containingShadowRoot->host()) {
for (auto& possiblySlotted : childrenOfType<Element>(*containingShadowRoot->host()))
possiblySlotted.invalidateStyle();
}
m_didInvalidateHostChildren = true;
}
switch (element.styleValidity()) {
case Style::Validity::Valid: {
ElementRuleCollector ruleCollector(element, m_ruleSet, filter);
ruleCollector.setMode(SelectorChecker::Mode::CollectingRulesIgnoringVirtualPseudoElements);
ruleCollector.matchAuthorRules(false);
if (ruleCollector.hasMatchedRules())
element.invalidateStyle();
return CheckDescendants::Yes;
}
case Style::Validity::ElementInvalid:
return CheckDescendants::Yes;
case Style::Validity::SubtreeInvalid:
case Style::Validity::SubtreeAndRenderersInvalid:
if (shouldCheckForSlots)
return CheckDescendants::Yes;
return CheckDescendants::No;
}
ASSERT_NOT_REACHED();
return CheckDescendants::Yes;
}
void Invalidator::invalidateStyleForTree(Element& root, SelectorFilter* filter)
{
if (invalidateIfNeeded(root, filter) == CheckDescendants::No)
return;
Vector<Element*, 20> parentStack;
Element* previousElement = &root;
auto descendants = descendantsOfType<Element>(root);
for (auto it = descendants.begin(), end = descendants.end(); it != end;) {
auto& descendant = *it;
auto* parent = descendant.parentElement();
if (parentStack.isEmpty() || parentStack.last() != parent) {
if (parent == previousElement) {
parentStack.append(parent);
if (filter)
filter->pushParent(parent);
} else {
while (parentStack.last() != parent) {
parentStack.removeLast();
if (filter)
filter->popParent();
}
}
}
previousElement = &descendant;
if (invalidateIfNeeded(descendant, filter) == CheckDescendants::Yes)
it.traverseNext();
else
it.traverseNextSkippingChildren();
}
}
void Invalidator::invalidateStyle(Document& document)
{
ASSERT(!m_dirtiesAllStyle);
Element* documentElement = document.documentElement();
if (!documentElement)
return;
SelectorFilter filter;
invalidateStyleForTree(*documentElement, &filter);
}
void Invalidator::invalidateStyle(ShadowRoot& shadowRoot)
{
ASSERT(!m_dirtiesAllStyle);
if (!m_ruleSet.hostPseudoClassRules().isEmpty() && shadowRoot.host())
shadowRoot.host()->invalidateStyle();
for (auto& child : childrenOfType<Element>(shadowRoot)) {
SelectorFilter filter;
invalidateStyleForTree(child, &filter);
}
}
void Invalidator::invalidateStyle(Element& element)
{
ASSERT(!m_dirtiesAllStyle);
invalidateStyleForTree(element, nullptr);
}
}
}