[CSSTypedOM] Computed StylePropertyMap use ComputedStyle for Lengths

third_party/WebKit/Source/core/css/cssom/ComputedStylePropertyMap.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/ComputedStylePropertyMap.h"
 
 #include "core/css/CSSComputedStyleDeclaration.h"
+#include "core/css/cssom/CSSCalcLength.h"
+#include "core/css/cssom/CSSKeywordValue.h"
+#include "core/css/cssom/CSSSimpleLength.h"
 #include "core/css/cssom/StyleValueFactory.h"
+#include "core/dom/PseudoElement.h"
 
 namespace blink {
 
-CSSStyleValueVector ComputedStylePropertyMap::getAllInternal(
-    CSSPropertyID propertyID) {
-  const CSSValue* cssValue =
-      m_computedStyleDeclaration->getPropertyCSSValueInternal(propertyID);
-  if (!cssValue)
-    return CSSStyleValueVector();
-  return StyleValueFactory::cssValueToStyleValueVector(propertyID, *cssValue);
+namespace {
+
+CSSStyleValue* styleValueForLength(const Length& length) {
+  if (length.isAuto()) {
+    return CSSKeywordValue::create("auto");
+  }
+  if (length.isFixed()) {
+    return CSSSimpleLength::create(length.pixels(),
+                                   CSSPrimitiveValue::UnitType::Pixels);
+  }
+  if (length.isPercent()) {
+    return CSSSimpleLength::create(length.percent(),
+                                   CSSPrimitiveValue::UnitType::Percentage);
+  }
+  if (length.isCalculated()) {
+    return CSSCalcLength::fromLength(length);
+  }
+  NOTREACHED();
+  return nullptr;
+}
+
+}  // namespace
+
+Node* ComputedStylePropertyMap::node() const {
+  if (m_node->isElementNode()) {
+    if (PseudoElement* element =
+            toElement(m_node)->pseudoElement(m_node->getPseudoId()))
+      return element;
+  }
+  return m_node;
 }
 
 CSSStyleValueVector ComputedStylePropertyMap::getAllInternal(
+    CSSPropertyID propertyID) {
+  Node* node = this->node();
+  Document& document = node->document();
+  // TODO(rjwright): This style recalc is copied from
+  // CSSComputedStyleDeclaration::getPropertyCSSValue(CSSPropertyID), which also
+  // does a "force
+  // full layout" step. Need to work out if I need to do that too.
+  document.updateStyleAndLayoutTreeForNode(node);
+  // TODO(rjwright): Do I need this?
+  node = this->node();
+  const ComputedStyle* style = node->ensureComputedStyle(node->getPseudoId());
+
+  CSSStyleValueVector styleValueVector;
+  CSSStyleValue* styleValue = nullptr;
+  switch (propertyID) {
+    case CSSPropertyLeft:
+      styleValue = styleValueForLength(style->left());
+      break;
+    case CSSPropertyRight:
+      styleValue = styleValueForLength(style->right());
+      break;
+    case CSSPropertyTop:
+      styleValue = styleValueForLength(style->top());
+      break;
+    case CSSPropertyBottom:
+      styleValue = styleValueForLength(style->bottom());
+      break;
+    case CSSPropertyHeight:
+      styleValue = styleValueForLength(style->height());
+      break;
+    case CSSPropertyWidth:
+      styleValue = styleValueForLength(style->width());
+      break;
+    default:
+      break;
+  }
+
+  if (styleValue) {
+    styleValueVector.append(styleValue);
+  }
+  return styleValueVector;
+}
+
+CSSStyleValueVector ComputedStylePropertyMap::getAllInternal(
     AtomicString customPropertyName) {
   const CSSValue* cssValue =
       m_computedStyleDeclaration->getPropertyCSSValueInternal(
           customPropertyName);
   if (!cssValue)
     return CSSStyleValueVector();
   return StyleValueFactory::cssValueToStyleValueVector(CSSPropertyInvalid,
                                                        *cssValue);
 }
 
 Vector<String> ComputedStylePropertyMap::getProperties() {
   Vector<String> result;
   for (unsigned i = 0; i < m_computedStyleDeclaration->length(); i++) {
     result.append(m_computedStyleDeclaration->item(i));
   }
   return result;
 }
 
 }  // namespace blink