Create a unified UIElement interface for Widget, View and Window.

ash/devtools/widget_element.cc

+// Copyright 2017 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 "ash/devtools/widget_element.h"
+
+#include "ash/devtools/ui_element_delegate.h"
+
+namespace ash {
+namespace devtools {
+
+WidgetElement::WidgetElement(views::Widget* widget,
+                             UIElementDelegate* ui_element_delegate)
+    : widget_(widget) {
+  this->SetUIElementDelegate(ui_element_delegate);
+  this->SetType(UIElementType::WIDGET);
+  if (!widget_->HasRemovalsObserver(this))
+    widget_->AddRemovalsObserver(this);
+}
+
+WidgetElement::WidgetElement(views::Widget* widget, int node_id)
+    : widget_(widget) {
+  this->SetType(UIElementType::WIDGET);
+  if (!widget_->HasRemovalsObserver(this))
+    widget_->AddRemovalsObserver(this);
+  this->SetNodeId(node_id);
+}
+
+WidgetElement::~WidgetElement() {
+  if (widget_ && widget_->HasRemovalsObserver(this))
+    widget_->RemoveRemovalsObserver(this);
+}
+
+views::Widget* WidgetElement::widget() {
+  return widget_;
+}
+
+void WidgetElement::OnWillRemoveView(views::Widget* widget, views::View* view) {
+  if (view == widget->GetRootView())
+    GetUIElementDelegate()->OnUIElementRemoved(
+        this->GetChildren()[0]->GetNodeId());
+}
+
+void WidgetElement::OnWidgetBoundsChanged(views::Widget* widget,
+                                          const gfx::Rect& new_bounds) {
+  if (widget == this->widget_)
+    GetUIElementDelegate()->OnUIElementBoundsChanged(this->GetNodeId());
+}
+
+void WidgetElement::Destroy() {
+  this->~WidgetElement();
+}
+
+bool WidgetElement::GetBounds(gfx::Rect* bounds) {
+  if (widget_) {
+    *bounds = widget_->GetRestoredBounds();
+    return true;
+  }
+  return false;
+}
+
+bool WidgetElement::SetBounds(const gfx::Rect& bounds) {
+  if (widget_) {
+    widget_->SetBounds(bounds);
+    return true;
+  }
+  return false;
+}
+
+bool WidgetElement::GetVisible(bool* visible) {
+  if (widget_) {
+    *visible = widget_->IsVisible();
+    return true;
+  }
+  return false;
+}
+
+bool WidgetElement::SetVisible(bool visible) {
+  if (widget_) {
+    if (visible != widget_->IsVisible()) {
+      if (visible)
+        widget_->Show();
+      else
+        widget_->Hide();
+    }
+    return true;
+  }
+  return false;
+}
+
+std::pair<WmWindow*, gfx::Rect> WidgetElement::GetNodeWindowAndBounds() {
+  if (widget_)
+    return std::make_pair(WmWindow::Get(widget_->GetNativeWindow()),
+                          widget_->GetWindowBoundsInScreen());
+  return std::make_pair(nullptr, gfx::Rect());
+}
+
+// static
+views::Widget* WidgetElement::From(UIElement* element) {
+  DCHECK_EQ(UIElementType::WIDGET, element->GetType());
+  return static_cast<WidgetElement*>(element)->widget_;
+}
+
+}  // namespace devtools
+}  // namespace ash