content: Let embedders know when MojoShellConnection is initialized.

chrome/browser/ui/views/chrome_browser_main_extra_parts_views.cc

 // Copyright (c) 2012 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 "chrome/browser/ui/views/chrome_browser_main_extra_parts_views.h"
 
 #include "chrome/browser/ui/views/chrome_constrained_window_views_client.h"
 #include "chrome/browser/ui/views/chrome_views_delegate.h"
 #include "components/constrained_window/constrained_window_views.h"
 
 #if defined(USE_AURA)
+#include "base/run_loop.h"
 #include "content/public/common/mojo_shell_connection.h"
 #include "services/shell/public/cpp/connector.h"
 #include "services/shell/runner/common/client_util.h"
 #include "services/ui/public/cpp/input_devices/input_device_client.h"
 #include "services/ui/public/interfaces/input_devices/input_device_server.mojom.h"
 #include "ui/display/screen.h"
 #include "ui/views/mus/window_manager_connection.h"
 #include "ui/views/widget/desktop_aura/desktop_screen.h"
 #include "ui/wm/core/wm_state.h"
 #endif  // defined(USE_AURA)
@@ skipping 17 matching lines @@
   wm_state_.reset(new wm::WMState);
 #endif
 }
 
 void ChromeBrowserMainExtraPartsViews::PreCreateThreads() {
 #if defined(USE_AURA) && !defined(OS_CHROMEOS)
   display::Screen::SetScreenInstance(views::CreateDesktopScreen());
 #endif
 }
 
-void ChromeBrowserMainExtraPartsViews::PreProfileInit() {
+void ChromeBrowserMainExtraPartsViews::MojoShellConnectionStarted(
+    content::MojoShellConnection* connection) {
+  DCHECK(connection);
 #if defined(USE_AURA)
-  content::MojoShellConnection* mojo_shell_connection =
-      content::MojoShellConnection::GetForProcess();
-  if (mojo_shell_connection && shell::ShellIsRemote()) {
+  if (shell::ShellIsRemote()) {
+    // TODO(rockot): Remove the blocking wait for init.
+    // http://crbug.com/594852.
+    base::RunLoop wait_loop;
+    connection->SetInitializeHandler(wait_loop.QuitClosure());
+    wait_loop.Run();
+
     input_device_client_.reset(new ui::InputDeviceClient());
     ui::mojom::InputDeviceServerPtr server;
-    mojo_shell_connection->GetConnector()->ConnectToInterface("mojo:ui",
-                                                              &server);
+    connection->GetConnector()->ConnectToInterface("mojo:ui", &server);
     input_device_client_->Connect(std::move(server));
 
     window_manager_connection_ = views::WindowManagerConnection::Create(
-        mojo_shell_connection->GetConnector(),
-        mojo_shell_connection->GetIdentity());
+        connection->GetConnector(), connection->GetIdentity());
   }
 #endif  // defined(USE_AURA)
-  ChromeBrowserMainExtraParts::PreProfileInit();
 }