Add support for custom properties and @apply when iterating InlineStylePropertyMap

third_party/WebKit/Source/core/css/cssom/InlineStylePropertyMap.cpp

 // Copyright 2016 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 "core/css/cssom/InlineStylePropertyMap.h"
 
 #include "bindings/core/v8/Iterable.h"
 #include "core/CSSPropertyNames.h"
+#include "core/css/CSSCustomIdentValue.h"
+#include "core/css/CSSCustomPropertyDeclaration.h"
 #include "core/css/CSSPrimitiveValue.h"
 #include "core/css/CSSPropertyMetadata.h"
 #include "core/css/CSSValueList.h"
 #include "core/css/StylePropertySet.h"
 #include "core/css/cssom/CSSOMTypes.h"
 #include "core/css/cssom/CSSSimpleLength.h"
+#include "core/css/cssom/CSSUnsupportedStyleValue.h"
 #include "core/css/cssom/StyleValueFactory.h"
 
 namespace blink {
 
 namespace {
 
 CSSValue* styleValueToCSSValue(CSSPropertyID propertyID, const CSSStyleValue& styleValue)
 {
     if (!CSSOMTypes::propertyCanTake(propertyID, styleValue))
         return nullptr;
@@ skipping 116 matching lines @@
     m_ownerElement->removeInlineStyleProperty(propertyID);
 }
 
 HeapVector<StylePropertyMap::StylePropertyMapEntry> InlineStylePropertyMap::getIterationEntries()
 {
     HeapVector<StylePropertyMap::StylePropertyMapEntry> result;
     StylePropertySet& inlineStyleSet = m_ownerElement->ensureMutableInlineStyle();
     for (unsigned i = 0; i < inlineStyleSet.propertyCount(); i++) {
         StylePropertySet::PropertyReference propertyReference = inlineStyleSet.propertyAt(i);
         CSSPropertyID propertyID = propertyReference.id();
-        CSSStyleValueVector styleValueVector = StyleValueFactory::cssValueToStyleValueVector(propertyID, *propertyReference.value());
+        String name;
         CSSStyleValueOrCSSStyleValueSequence value;
-        if (styleValueVector.size() == 1)
-            value.setCSSStyleValue(styleValueVector[0]);
-        else
-            value.setCSSStyleValueSequence(styleValueVector);
-        result.append(std::make_pair(getPropertyNameString(propertyID), value));
+        if (propertyID == CSSPropertyVariable) {
+            const CSSCustomPropertyDeclaration* customDeclaration = toCSSCustomPropertyDeclaration(propertyReference.value());
+            name = customDeclaration->name();
+            // TODO(meade): Eventually custom properties will support other types, so actually return them instead of always returning a CSSUnsupportedStyleValue.
+            value.setCSSStyleValue(CSSUnsupportedStyleValue::create(customDeclaration->customCSSText()));
+        } else if (propertyID == CSSPropertyApplyAtRule) {
+            name = "@apply";
+            value.setCSSStyleValue(CSSUnsupportedStyleValue::create(toCSSCustomIdentValue(propertyReference.value())->value()));
+        } else {
+            name = getPropertyNameString(propertyID);
+            CSSStyleValueVector styleValueVector = StyleValueFactory::cssValueToStyleValueVector(propertyID, *propertyReference.value());
+            if (styleValueVector.size() == 1)
+                value.setCSSStyleValue(styleValueVector[0]);
+            else
+                value.setCSSStyleValueSequence(styleValueVector);
+        }
+        result.append(std::make_pair(name, value));
     }
     return result;
 }
 
 } // namespace blink