Implement iterable<DOMString, (StyleValue or sequence<StyleValue>)> for InlineStylePropertyMap

third_party/WebKit/Source/core/css/cssom/StylePropertyMap.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/StylePropertyMap.h"
 
 #include "bindings/core/v8/ExceptionState.h"
 #include "core/css/cssom/SimpleLength.h"
 #include "core/css/cssom/StyleValue.h"
 #include "core/css/cssom/StyleValueFactory.h"
 
 namespace blink {
 
+namespace {
+
+class StylePropertyMapIterationSource final : public PairIterable<String, StyleValueOrStyleValueSequence>::IterationSource {
+public:
+    explicit StylePropertyMapIterationSource(HeapVector<StylePropertyMap::StylePropertyMapEntry> values)
+        : m_index(0)
+        , m_values(values)
+    {
+    }
+
+    bool next(ScriptState*, String& key, StyleValueOrStyleValueSequence& value, ExceptionState&) override
+    {
+        if (m_index >= m_values.size())
+            return false;
+
+        StylePropertyMap::StylePropertyMapEntry pair = m_values.at(m_index++);

[Timothy Loh, 2016/05/13 04:57:45]

const Foo&

[meade_UTC10, 2016/05/13 05:46:34]

Done.

+        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<StylePropertyMap::StylePropertyMapEntry> m_values;
+};
+
+} // namespace
+
 StyleValue* StylePropertyMap::get(const String& propertyName, ExceptionState& exceptionState)
 {
     CSSPropertyID propertyID = cssPropertyID(propertyName);
     if (propertyID == CSSPropertyInvalid) {
         // TODO(meade): Handle custom properties here.
         exceptionState.throwTypeError("Invalid propertyName: " + propertyName);
         return nullptr;
     }
 
     StyleValueVector styleVector = getAll(propertyID);
@@ skipping 72 matching lines @@
     for (const Member<CSSValue> value : *toCSSValueList(&cssValue)) {
         StyleValue* styleValue = StyleValueFactory::create(propertyID, *value);
         if (!styleValue)
             return StyleValueVector();
         styleValueVector.append(styleValue);
     }
 
     return styleValueVector;
 }
 
+StylePropertyMap::IterationSource* StylePropertyMap::startIteration(ScriptState*, ExceptionState&)
+{
+    return new StylePropertyMapIterationSource(getIterationEntries());
+}
+
 } // namespace blink