Devtools [LayoutEditor]: Show medias for the rules with the same selectors

Source/core/inspector/LayoutEditor.cpp

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "config.h"
 #include "core/inspector/LayoutEditor.h"
 
 #include "core/css/CSSComputedStyleDeclaration.h"
+#include "core/css/CSSImportRule.h"
+#include "core/css/CSSMediaRule.h"
 #include "core/css/CSSRuleList.h"
+#include "core/css/MediaList.h"
 #include "core/dom/NodeComputedStyle.h"
 #include "core/dom/StaticNodeList.h"
 #include "core/frame/FrameView.h"
 #include "core/inspector/InspectorCSSAgent.h"
 #include "core/inspector/InspectorDOMAgent.h"
 #include "core/inspector/InspectorHighlight.h"
 #include "core/layout/LayoutBox.h"
 #include "core/layout/LayoutInline.h"
 #include "core/layout/LayoutObject.h"
 #include "core/style/ComputedStyle.h"
@@ skipping 130 matching lines @@
 InspectorHighlightConfig affectedNodesHighlightConfig()
 {
     // TODO: find a better color
     InspectorHighlightConfig config;
     config.content = Color(95, 127, 162, 100);
     config.padding = Color(95, 127, 162, 100);
     config.margin = Color(95, 127, 162, 100);
     return config;
 }
 
+void collectMediaQueriesFromRule(CSSRule* rule, Vector<String>& mediaArray)
+{
+    MediaList* mediaList;
+    if (rule->type() == CSSRule::MEDIA_RULE) {
+        CSSMediaRule* mediaRule = toCSSMediaRule(rule);
+        mediaList = mediaRule->media();
+    } else if (rule->type() == CSSRule::IMPORT_RULE) {
+        CSSImportRule* importRule = toCSSImportRule(rule);
+        mediaList = importRule->media();
+    } else {
+        mediaList = nullptr;
+    }
+
+    if (mediaList && mediaList->length())
+        mediaArray.append(mediaList->mediaText());
+}
+
+void buildMediaListChain(CSSRule* rule, Vector<String>& mediaArray)
+{
+    while (rule) {
+        collectMediaQueriesFromRule(rule, mediaArray);
+        if (rule->parentRule()) {
+            rule = rule->parentRule();
+        } else {
+            CSSStyleSheet* styleSheet = rule->parentStyleSheet();
+            // TODO: should be able to replace cycle by one iteration of it.
+            while (styleSheet) {
+                MediaList* mediaList = styleSheet->media();
+                if (mediaList && mediaList->length())
+                    mediaArray.append(mediaList->mediaText());
+
+                rule = styleSheet->ownerRule();
+                if (rule)
+                    break;
+                styleSheet = styleSheet->parentStyleSheet();
+            }
+        }
+    }
+}
+
 } // namespace
 
 LayoutEditor::LayoutEditor(InspectorCSSAgent* cssAgent, InspectorDOMAgent* domAgent)
     : m_element(nullptr)
     , m_cssAgent(cssAgent)
     , m_domAgent(domAgent)
     , m_changingProperty(CSSPropertyInvalid)
     , m_propertyInitialValue(0)
     , m_isDirty(false)
+    , m_currentRuleIndex(-1)
 {
 }
 
 DEFINE_TRACE(LayoutEditor)
 {
     visitor->trace(m_element);
     visitor->trace(m_cssAgent);
     visitor->trace(m_domAgent);
     visitor->trace(m_matchedRules);
 }
@@ skipping 210 matching lines @@
         return m_cachedSelectorsInfo.get(m_currentRuleIndex);
 
     RefPtr<JSONObject> object = JSONObject::create();
     String currentSelectorText = m_currentRuleIndex == -1 ? "inline style" : m_matchedRules[m_currentRuleIndex]->selectorText();
     object->setString("selector", currentSelectorText);
 
     Document* ownerDocument = m_element->ownerDocument();
     if (!ownerDocument->isActive() || m_currentRuleIndex == -1)
         return object->toJSONString();
 
+    if (m_currentRuleIndex != -1) {
+        bool hasSameSelectors = false;
+        for (size_t i = 0; i < m_matchedRules.size(); i++) {
+            if (i != static_cast<unsigned>(m_currentRuleIndex) && m_matchedRules[i]->selectorText() == currentSelectorText) {
+                hasSameSelectors = true;
+                break;
+            }
+        }
+
+        if (hasSameSelectors) {
+            Vector<String> medias;
+            buildMediaListChain(m_matchedRules[m_currentRuleIndex].get(), medias);
+            RefPtr<JSONArray> mediasJSONArray = JSONArray::create();
+            for (size_t i = 0; i < medias.size(); ++i)
+                mediasJSONArray->pushString(medias[i]);
+
+            object->setArray("medias", mediasJSONArray.release());
+        }
+    }
+
     TrackExceptionState exceptionState;
     RefPtrWillBeRawPtr<StaticElementList> elements = ownerDocument->querySelectorAll(AtomicString(m_matchedRules[m_currentRuleIndex]->selectorText()), exceptionState);
 
     if (!elements || exceptionState.hadException())
         return object->toJSONString();
 
     RefPtr<JSONArray> highlights = JSONArray::create();
     InspectorHighlightConfig config = affectedNodesHighlightConfig();
     for (unsigned i = 0; i < elements->length(); ++i) {
         Element* element = elements->item(i);
         if (element == m_element)
             continue;
 
         InspectorHighlight highlight(element, config, false);
         highlights->pushObject(highlight.asJSONObject());
     }
 
     object->setArray("nodes", highlights.release());
     m_cachedSelectorsInfo.add(m_currentRuleIndex, object->toJSONString());
     return m_cachedSelectorsInfo.get(m_currentRuleIndex);
 }
 
 } // namespace blink