Implement iterable<DOMString, (StyleValue or sequence<StyleValue>)> for 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/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/SimpleLength.h"
 #include "core/css/cssom/StyleValueFactory.h"
 
 namespace blink {
 
-StyleValueVector InlineStylePropertyMap::getAll(CSSPropertyID propertyID)
+namespace {
+
+class InlineStylePropertyMapIterationSource final : public PairIterable<String, StyleValueOrStyleValueSequence>::IterationSource {
+public:
+    explicit InlineStylePropertyMapIterationSource(HeapVector<std::pair<String, StyleValueOrStyleValueSequence>> values)

[Timothy Loh, 2016/04/15 05:27:53]

let's just make a using declaration for the pair somewhere, so we don't have to
keep tying out the name.

[meade_UTC10, 2016/04/27 05:39:01]

Done.

+        : m_index(0)
+        , m_values(values)
+    {
+    }
+
+    bool next(ScriptState*, String& key, StyleValueOrStyleValueSequence& value, ExceptionState&) override
+    {
+        if (m_index >= m_values.size())
+            return false;
+
+        std::pair<String, StyleValueOrStyleValueSequence> pair = m_values.at(m_index++);
+        key = pair.first;
+        value = pair.second;
+        return true;
+    }
+
+    DEFINE_INLINE_VIRTUAL_TRACE()
+    {
+        visitor->trace(m_values);
+        PairIterable<String, StyleValueOrStyleValueSequence>::IterationSource::trace(visitor);
+    }
+
+private:
+    size_t m_index;
+    const HeapVector<std::pair<String, StyleValueOrStyleValueSequence>> m_values;
+};
+
+} // namespace
+
+StylePropertyMap::StyleValueVector InlineStylePropertyMap::getAll(CSSPropertyID propertyID)
 {
     CSSValue* cssValue = m_ownerElement->ensureMutableInlineStyle().getPropertyCSSValue(propertyID);
     if (!cssValue)
         return StyleValueVector();
 
     return cssValueToStyleValueVector(propertyID, *cssValue);
 }
 
 bool InlineStylePropertyMap::has(CSSPropertyID propertyID)
 {
@@ skipping 92 matching lines @@
     }
 
     m_ownerElement->setInlineStyleProperty(propertyID, cssValueList);
 }
 
 void InlineStylePropertyMap::remove(CSSPropertyID propertyID, ExceptionState& exceptionState)
 {
     m_ownerElement->removeInlineStyleProperty(propertyID);
 }
 
+InlineStylePropertyMap::IterationSource* InlineStylePropertyMap::startIteration(ScriptState* scriptState, ExceptionState& exceptionState)

[Timothy Loh, 2016/04/15 05:27:53]

can we make something that works for all the maps?

[meade_UTC10, 2016/04/27 05:39:01]

Done.

+{
+    HeapVector<std::pair<String, StyleValueOrStyleValueSequence>> result;
+    StylePropertySet& inlineStyleSet = m_ownerElement->ensureMutableInlineStyle();
+    for (unsigned i = 0; i < inlineStyleSet.propertyCount(); i++) {
+        StylePropertySet::PropertyReference propertyReference = inlineStyleSet.propertyAt(i);
+        CSSPropertyID propertyID = propertyReference.id();
+        StyleValueVector styleValueVector = cssValueToStyleValueVector(propertyID, *propertyReference.value());
+        StyleValueOrStyleValueSequence value;
+        if (styleValueVector.size() == 1)
+            value.setStyleValue(styleValueVector[0]);
+        else
+            value.setStyleValueSequence(styleValueVector);
+        result.append(std::make_pair(getPropertyNameString(propertyID), value));
+    }
+    return new InlineStylePropertyMapIterationSource(result);
+}
+
 } // namespace blink