Change FocusSynchronizer to maintain active focus client and window.

components/exo/wm_helper_mus.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 "components/exo/wm_helper_mus.h"
 
 #include "ui/aura/client/focus_client.h"
 #include "ui/aura/env.h"
+#include "ui/aura/mus/focus_synchronizer.h"
+#include "ui/aura/mus/window_tree_client.h"
 #include "ui/aura/window.h"
 #include "ui/display/manager/managed_display_info.h"
+#include "ui/views/mus/mus_client.h"
 #include "ui/wm/public/activation_client.h"
 namespace exo {
 
 ////////////////////////////////////////////////////////////////////////////////
 // WMHelperMus, public:
 
 WMHelperMus::WMHelperMus() {
-  aura::Env* env = aura::Env::GetInstance();
-  SetActiveFocusClient(env->active_focus_client(),
-                       env->active_focus_client_root());
-  env->AddObserver(this);
+  aura::FocusSynchronizer* focus_synchronizer =
+      views::MusClient::Get()->window_tree_client()->focus_synchronizer();
+  SetActiveFocusClient(focus_synchronizer->active_focus_client(),
+                       focus_synchronizer->active_focus_client_root());
+  focus_synchronizer->AddObserver(this);
 }
 
 WMHelperMus::~WMHelperMus() {
   if (active_focus_client_)
     active_focus_client_->RemoveObserver(this);
-  aura::Env::GetInstance()->RemoveObserver(this);
+  views::MusClient::Get()
+      ->window_tree_client()
+      ->focus_synchronizer()
+      ->RemoveObserver(this);
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 // WMHelperMus, private:
 
 const display::ManagedDisplayInfo WMHelperMus::GetDisplayInfo(
     int64_t display_id) const {
   // TODO(penghuang): Return real display info when it is supported in mus.
   return display::ManagedDisplayInfo(display_id, "", false);
 }
@@ skipping 43 matching lines @@
 
 bool WMHelperMus::IsSpokenFeedbackEnabled() const {
   NOTIMPLEMENTED();
   return false;
 }
 
 void WMHelperMus::PlayEarcon(int sound_key) const {
   NOTIMPLEMENTED();
 }
 
-void WMHelperMus::OnWindowInitialized(aura::Window* window) {}
-
 void WMHelperMus::OnActiveFocusClientChanged(
     aura::client::FocusClient* focus_client,
-    aura::Window* window) {
-  SetActiveFocusClient(focus_client, window);
+    aura::Window* focus_client_root) {
+  SetActiveFocusClient(focus_client, focus_client_root);
 }
 
 void WMHelperMus::OnWindowFocused(aura::Window* gained_focus,
                                   aura::Window* lost_focus) {
   if (focused_window_ == gained_focus)
     return;
 
   SetActiveWindow(GetActivationClient()->GetActiveWindow());
   SetFocusedWindow(gained_focus);
 }
@@ skipping 36 matching lines @@
   NotifyWindowFocused(focused_window_, lost_focus);
 }
 
 aura::client::ActivationClient* WMHelperMus::GetActivationClient() {
   return root_with_active_focus_client_
              ? aura::client::GetActivationClient(root_with_active_focus_client_)
              : nullptr;
 }
 
 }  // namespace exo