Create a RegisterViewAssociate method in ViewManager

services/ui/launcher/launch_instance.cc

 // 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 "services/ui/launcher/launch_instance.h"
 
+#include "base/bind.h"
 #include "base/command_line.h"
 #include "base/logging.h"
+#include "base/strings/string_split.h"
 #include "base/trace_event/trace_event.h"
 #include "mojo/application/application_runner_chromium.h"
 #include "mojo/public/c/system/main.h"
 #include "mojo/public/cpp/application/application_impl.h"
 #include "mojo/public/cpp/application/connect.h"
 #include "mojo/services/ui/views/interfaces/view_provider.mojom.h"
 #include "services/ui/launcher/launcher_view_tree.h"
 
 namespace launcher {
 
 LaunchInstance::LaunchInstance(mojo::ApplicationImpl* app_impl,
                                const std::string& app_url,
+                               const std::string& view_associate_urls,
                                const base::Closure& shutdown_callback)
     : app_impl_(app_impl),
       app_url_(app_url),
+      view_associate_urls_(view_associate_urls),
       shutdown_callback_(shutdown_callback),
       viewport_event_dispatcher_binding_(this) {}
 
 LaunchInstance::~LaunchInstance() {}
 
 void LaunchInstance::Launch() {
   DVLOG(1) << "Launching " << app_url_;
   TRACE_EVENT0("launcher", __func__);
 
   mojo::ConnectToService(app_impl_->shell(), "mojo:compositor_service",
                          GetProxy(&compositor_));
   compositor_.set_connection_error_handler(base::Bind(
       &LaunchInstance::OnCompositorConnectionError, base::Unretained(this)));
 
   mojo::ConnectToService(app_impl_->shell(), "mojo:view_manager_service",
                          GetProxy(&view_manager_));
   view_manager_.set_connection_error_handler(base::Bind(
       &LaunchInstance::OnViewManagerConnectionError, base::Unretained(this)));
 
+  // Connect to ViewAssociates.
+  std::vector<std::string> associate_urls = SplitString(
+      view_associate_urls_, ",", base::KEEP_WHITESPACE, base::SPLIT_WANT_ALL);
+  if (associate_urls.empty()) {
+    // TODO(jeffbrown): Replace this hardcoded list.
+    associate_urls.push_back("mojo:input_manager_service");
+  }
+
+  view_associate_owners_.reserve(associate_urls.size());
+
+  for (const auto& url : associate_urls) {
+    // Connect to the ViewAssociate.
+    DVLOG(2) << "Connecting to ViewAssociate " << url;
+    mojo::ui::ViewAssociatePtr view_associate;
+    mojo::ConnectToService(app_impl_->shell(), url, GetProxy(&view_associate));
+
+    // Wire up the associate to the ViewManager.
+    mojo::ui::ViewAssociateOwnerPtr view_associate_owner;
+    view_manager_->RegisterViewAssociate(view_associate.Pass(),
+                                         GetProxy(&view_associate_owner), url);
+
+    view_associate_owner.set_connection_error_handler(
+        base::Bind(&LaunchInstance::OnViewAssociateConnectionError,
+                   base::Unretained(this)));
+
+    view_associate_owners_.push_back(view_associate_owner.Pass());
+  }
+  view_manager_->FinishedRegisteringViewAssociates();
+
   InitViewport();
 
   mojo::ui::ViewProviderPtr client_view_provider;
   mojo::ConnectToService(app_impl_->shell(), app_url_,
                          GetProxy(&client_view_provider));
 
   client_view_provider->CreateView(GetProxy(&client_view_owner_), nullptr,
                                    nullptr);
 }
 
+void LaunchInstance::OnViewAssociateConnectionError() {
+  LOG(ERROR) << "Exiting due to view associate connection error.";
+  shutdown_callback_.Run();
+};
+
 void LaunchInstance::OnCompositorConnectionError() {
   LOG(ERROR) << "Exiting due to compositor connection error.";
   shutdown_callback_.Run();
 }
 
 void LaunchInstance::OnViewManagerConnectionError() {
   LOG(ERROR) << "Exiting due to view manager connection error.";
   shutdown_callback_.Run();
 }
 
@@ skipping 51 matching lines @@
 }
 
 void LaunchInstance::OnEvent(mojo::EventPtr event,
                              const mojo::Callback<void()>& callback) {
   if (view_tree_)
     view_tree_->DispatchEvent(event.Pass());
   callback.Run();
 }
 
 }  // namespace launcher