Headless demo

headless/lib/web_element.cc

+// 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 "headless/public/web_element.h"
+
+#include "third_party/WebKit/public/platform/WebRect.h"
+#include "third_party/WebKit/public/web/WebElement.h"
+
+namespace headless {
+
+WebElement::WebElement(const blink::WebElement& web_element)
+    : WebNode(web_element), attributes_(nullptr) {}
+
+WebElement::WebElement(const WebElement& web_element)
+    : WebNode(web_element.web_node_), attributes_(nullptr) {}
+
+WebElement::~WebElement() {}
+
+std::string WebElement::TagName() const {
+  return web_element().tagName().utf8();
+}
+
+gfx::Rect WebElement::BoundingBox() const {
+  blink::WebElement& element = const_cast<blink::WebElement&>(web_element());
+  return element.boundsInViewportSpace();
+}
+
+const std::map<std::string, std::string>& WebElement::Attributes() const {
+  if (!attributes_) {
+    attributes_.reset(new std::map<std::string, std::string>());
+    for (size_t i = 0; i < web_element().attributeCount(); ++i) {
+      (*attributes_)[web_element().attributeLocalName(i).utf8()] =
+          web_element().attributeValue(i).utf8();
+    }
+  }
+  return *attributes_;
+}
+
+blink::WebElement& WebElement::web_element() {
+  return static_cast<blink::WebElement&>(web_node_);
+}
+
+const blink::WebElement& WebElement::web_element() const {
+  return static_cast<const blink::WebElement&>(web_node_);
+}
+
+}  // namespace headless