Provide an API to the example window manager app that supports closing a window.

mojo/examples/embedded_app/embedded_app.cc

 // Copyright 2014 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 "base/basictypes.h"
 #include "base/bind.h"
+#include "mojo/examples/window_manager/window_manager.mojom.h"
 #include "mojo/public/cpp/application/application.h"
 #include "mojo/services/public/cpp/view_manager/view.h"
 #include "mojo/services/public/cpp/view_manager/view_manager.h"
+#include "mojo/services/public/cpp/view_manager/view_observer.h"
 #include "mojo/services/public/cpp/view_manager/view_tree_node.h"
+#include "mojo/services/public/cpp/view_manager/view_tree_node_observer.h"
+#include "ui/events/event_constants.h"
+
+using mojo::view_manager::View;
+using mojo::view_manager::ViewManager;
+using mojo::view_manager::ViewObserver;
+using mojo::view_manager::ViewTreeNode;
+using mojo::view_manager::ViewTreeNodeObserver;
 
 namespace mojo {
 namespace examples {
+namespace {
 
-class EmbeddedApp : public Application {
+const SkColor kColors[] = { SK_ColorYELLOW,
+                            SK_ColorRED,
+                            SK_ColorGREEN,
+                            SK_ColorMAGENTA };
+
+}  // namespace
+
+class EmbeddedApp : public Application,
+                    public ViewObserver,
+                    public ViewTreeNodeObserver {
  public:
   EmbeddedApp() : view_manager_(NULL) {}
   virtual ~EmbeddedApp() {}
 
  private:
   // Overridden from Application:
   virtual void Initialize() MOJO_OVERRIDE {
-    view_manager::ViewManager::Create(this,
-        base::Bind(&EmbeddedApp::OnRootAdded, base::Unretained(this)));
+    ViewManager::Create(this,
+        base::Bind(&EmbeddedApp::OnRootAdded, base::Unretained(this)),
+        base::Bind(&EmbeddedApp::OnRootRemoved, base::Unretained(this)));
+    ConnectTo<IWindowManager>("mojo:mojo_window_manager", &window_manager_);
   }
 
-  void OnRootAdded(view_manager::ViewManager* view_manager) {
+  // Overridden from ViewObserver:
+  virtual void OnViewInputEvent(View* view, EventPtr event) OVERRIDE {
+    if (event->action == ui::ET_MOUSE_RELEASED)
+      window_manager_->CloseWindow(view->node()->id());
+  }
+
+  // Overridden from ViewTreeNodeObserver:
+  virtual void OnNodeActiveViewChange(
+      ViewTreeNode* node,
+      View* old_view,
+      View* new_view,
+      ViewTreeNodeObserver::DispositionChangePhase phase) OVERRIDE {
+    if (new_view == 0)
+      views_to_reap_[node] = old_view;
+  }
+
+  void OnRootAdded(ViewManager* view_manager, ViewTreeNode* root) {
     if (!view_manager_)
       view_manager_ = view_manager;
     DCHECK_EQ(view_manager_, view_manager);
 
-    if (view_manager_->roots().size() == 1) {
-      view_manager::View* view = view_manager::View::Create(view_manager_);
-      view_manager_->roots().front()->SetActiveView(view);
-      view->SetColor(SK_ColorYELLOW);
-    } else {
-      view_manager::View* view = view_manager::View::Create(view_manager_);
-      view_manager_->roots().back()->SetActiveView(view);
-      view->SetColor(SK_ColorRED);
-    }
+    View* view = View::Create(view_manager_);
+    view->AddObserver(this);
+    root->SetActiveView(view);
+    root->AddObserver(this);
+    size_t index = view_manager_->roots().size() - 1;
+    view->SetColor(kColors[index % arraysize(kColors)]);
   }
 
-  view_manager::ViewManager* view_manager_;
+  void OnRootRemoved(ViewManager* view_manager,
+                     ViewTreeNode* root) {
+    std::map<ViewTreeNode*, View*>::const_iterator it =
+        views_to_reap_.find(root);
+    if (it != views_to_reap_.end())
+      it->second->Destroy();
+  }
+
+  IWindowManagerPtr window_manager_;
+  ViewManager* view_manager_;
+  std::map<ViewTreeNode*, View*> views_to_reap_;
 
   DISALLOW_COPY_AND_ASSIGN(EmbeddedApp);
 };
 
 }  // namespace examples
 
 // static
 Application* Application::Create() {
   return new examples::EmbeddedApp;
 }
 
 }  // namespace mojo