Devtools Animations: Add method to fetch CSS keyframed animations

third_party/WebKit/Source/core/css/resolver/ScopedStyleResolver.cpp

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  * Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011 Apple Inc. All rights reserved.
  * Copyright (C) 2012 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1.  Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
@@ skipping 37 matching lines @@
 
 ScopedStyleResolver* ScopedStyleResolver::parent() const
 {
     for (TreeScope* scope = treeScope().parentTreeScope(); scope; scope = scope->parentTreeScope()) {
         if (ScopedStyleResolver* resolver = scope->scopedStyleResolver())
             return resolver;
     }
     return nullptr;
 }
 
-void ScopedStyleResolver::addKeyframeRules(const RuleSet& ruleSet)
+void ScopedStyleResolver::addKeyframeRules(const RuleSet& ruleSet, CSSStyleSheet* styleSheet)
 {
     const WillBeHeapVector<RawPtrWillBeMember<StyleRuleKeyframes>> keyframesRules = ruleSet.keyframesRules();
     for (unsigned i = 0; i < keyframesRules.size(); ++i)
-        addKeyframeStyle(keyframesRules[i]);
+        addKeyframeStyle(keyframesRules[i], styleSheet);
 }
 
 void ScopedStyleResolver::addFontFaceRules(const RuleSet& ruleSet)
 {
     // FIXME(BUG 72461): We don't add @font-face rules of scoped style sheets for the moment.
     if (!treeScope().rootNode().isDocumentNode())
         return;
 
     Document& document = treeScope().document();
     CSSFontSelector* cssFontSelector = document.styleEngine().fontSelector();
     const WillBeHeapVector<RawPtrWillBeMember<StyleRuleFontFace>> fontFaceRules = ruleSet.fontFaceRules();
     for (auto& fontFaceRule : fontFaceRules) {
         if (RefPtrWillBeRawPtr<FontFace> fontFace = FontFace::create(&document, fontFaceRule))
             cssFontSelector->fontFaceCache()->add(cssFontSelector, fontFaceRule, fontFace);
     }
     if (fontFaceRules.size())
         document.styleResolver()->invalidateMatchedPropertiesCache();
 }
 
 void ScopedStyleResolver::appendCSSStyleSheet(CSSStyleSheet& cssSheet, const MediaQueryEvaluator& medium)
 {
     unsigned index = m_authorStyleSheets.size();
     m_authorStyleSheets.append(&cssSheet);
     StyleSheetContents* sheet = cssSheet.contents();
     AddRuleFlags addRuleFlags = treeScope().document().securityOrigin()->canRequest(sheet->baseURL()) ? RuleHasDocumentSecurityOrigin : RuleHasNoSpecialState;
     const RuleSet& ruleSet = sheet->ensureRuleSet(medium, addRuleFlags);
 
-    addKeyframeRules(ruleSet);
+    addKeyframeRules(ruleSet, &cssSheet);
     addFontFaceRules(ruleSet);
     addTreeBoundaryCrossingRules(ruleSet, &cssSheet, index);
     treeScope().document().styleResolver()->addMediaQueryResults(ruleSet.viewportDependentMediaQueryResults());
 }
 
 void ScopedStyleResolver::collectFeaturesTo(RuleFeatureSet& features, HashSet<const StyleSheetContents*>& visitedSharedStyleSheetContents) const
 {
     for (size_t i = 0; i < m_authorStyleSheets.size(); ++i) {
         ASSERT(m_authorStyleSheets[i]->ownerNode());
         StyleSheetContents* contents = m_authorStyleSheets[i]->contents();
         if (contents->hasOneClient() || visitedSharedStyleSheetContents.add(contents).isNewEntry)
             features.add(contents->ruleSet().features());
     }
 
     if (!m_treeBoundaryCrossingRuleSet)
         return;
 
     for (const auto& rules : *m_treeBoundaryCrossingRuleSet)
         features.add(rules->m_ruleSet->features());
 }
 
 void ScopedStyleResolver::resetAuthorStyle()
 {
     m_authorStyleSheets.clear();
     m_keyframesRuleMap.clear();
+    m_keyframesStyleSheetMap.clear();
     m_treeBoundaryCrossingRuleSet = nullptr;
 }
 
 StyleRuleKeyframes* ScopedStyleResolver::keyframeStylesForAnimation(const StringImpl* animationName)
 {
     if (m_keyframesRuleMap.isEmpty())
         return nullptr;
 
     KeyframesRuleMap::iterator it = m_keyframesRuleMap.find(animationName);
     if (it == m_keyframesRuleMap.end())
         return nullptr;
 
     return it->value.get();
 }
 
-void ScopedStyleResolver::addKeyframeStyle(PassRefPtrWillBeRawPtr<StyleRuleKeyframes> rule)
+CSSStyleSheet* ScopedStyleResolver::styleSheetForAnimation(const StringImpl* animationName)
+{
+    if (m_keyframesStyleSheetMap.isEmpty())
+        return nullptr;
+
+    WillBeHeapHashMap<const StringImpl*, CSSStyleSheet*>::iterator it = m_keyframesStyleSheetMap.find(animationName);
+    if (it == m_keyframesStyleSheetMap.end())
+        return nullptr;
+
+    return it->value;
+}
+
+void ScopedStyleResolver::addKeyframeStyle(PassRefPtrWillBeRawPtr<StyleRuleKeyframes> rule, CSSStyleSheet* styleSheet)
 {
     AtomicString s(rule->name());
 
     if (rule->isVendorPrefixed()) {
         KeyframesRuleMap::iterator it = m_keyframesRuleMap.find(s.impl());
-        if (it == m_keyframesRuleMap.end())
+        if (it == m_keyframesRuleMap.end()) {
             m_keyframesRuleMap.set(s.impl(), rule);
-        else if (it->value->isVendorPrefixed())
+            m_keyframesStyleSheetMap.set(s.impl(), styleSheet);
+        } else if (it->value->isVendorPrefixed()) {
             m_keyframesRuleMap.set(s.impl(), rule);
+            m_keyframesStyleSheetMap.set(s.impl(), styleSheet);
+        }
     } else {
         m_keyframesRuleMap.set(s.impl(), rule);
+        m_keyframesStyleSheetMap.set(s.impl(), styleSheet);
     }
 }
 
 void ScopedStyleResolver::collectMatchingAuthorRules(ElementRuleCollector& collector, bool includeEmptyRules, CascadeOrder cascadeOrder)
 {
     ASSERT(!collector.scopeContainsLastMatchedElement());
     collector.setScopeContainsLastMatchedElement(true);
     for (size_t i = 0; i < m_authorStyleSheets.size(); ++i) {
         ASSERT(m_authorStyleSheets[i]->ownerNode());
         MatchRequest matchRequest(&m_authorStyleSheets[i]->contents()->ruleSet(), includeEmptyRules, &m_scope->rootNode(), m_authorStyleSheets[i], i);
@@ skipping 76 matching lines @@
     m_treeBoundaryCrossingRuleSet->append(RuleSubSet::create(parentStyleSheet, sheetIndex, ruleSetForScope.release()));
 }
 
 DEFINE_TRACE(ScopedStyleResolver::RuleSubSet)
 {
     visitor->trace(m_parentStyleSheet);
     visitor->trace(m_ruleSet);
 }
 
 } // namespace blink