Updates shadow when active window changes

components/mus/public/cpp/lib/window_tree_client_impl.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 "components/mus/public/cpp/lib/window_tree_client_impl.h"
 
 #include "base/bind.h"
 #include "components/mus/common/util.h"
 #include "components/mus/public/cpp/lib/in_flight_change.h"
 #include "components/mus/public/cpp/lib/window_private.h"
 #include "components/mus/public/cpp/window_manager_delegate.h"
 #include "components/mus/public/cpp/window_observer.h"
 #include "components/mus/public/cpp/window_tree_connection.h"
+#include "components/mus/public/cpp/window_tree_connection_observer.h"
 #include "components/mus/public/cpp/window_tree_delegate.h"
 #include "mojo/application/public/cpp/application_impl.h"
 #include "mojo/application/public/cpp/connect.h"
 #include "mojo/application/public/cpp/service_provider_impl.h"
 #include "mojo/application/public/interfaces/service_provider.mojom.h"
 #include "mojo/converters/geometry/geometry_type_converters.h"
 #include "ui/gfx/geometry/insets.h"
 #include "ui/gfx/geometry/size.h"
 
 namespace mus {
@@ skipping 322 matching lines @@
   connection_id_ = connection_id;
   is_embed_root_ =
       (access_policy & mojom::WindowTree::ACCESS_POLICY_EMBED_ROOT) != 0;
 
   DCHECK(!root_);
   root_ = AddWindowToConnection(this, nullptr, root_data);
 
   focused_window_ = GetWindowById(focused_window_id);
 
   delegate_->OnEmbed(root_);
+
+  if (focused_window_) {
+    FOR_EACH_OBSERVER(WindowTreeConnectionObserver, observers_,
+                      OnWindowTreeFocusChanged(focused_window_, nullptr));
+  }
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 // WindowTreeClientImpl, WindowTreeConnection implementation:
 
 Window* WindowTreeClientImpl::GetRoot() {
   return root_;
 }
 
 Window* WindowTreeClientImpl::GetWindowById(Id id) {
@@ skipping 29 matching lines @@
   return is_embed_root_;
 }
 
 ConnectionSpecificId WindowTreeClientImpl::GetConnectionId() {
   return connection_id_;
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 // WindowTreeClientImpl, WindowTreeClient implementation:
 
+void WindowTreeClientImpl::AddObserver(WindowTreeConnectionObserver* observer) {
+  observers_.AddObserver(observer);
+}
+
+void WindowTreeClientImpl::RemoveObserver(
+    WindowTreeConnectionObserver* observer) {
+  observers_.RemoveObserver(observer);
+}
+
 void WindowTreeClientImpl::OnEmbed(ConnectionSpecificId connection_id,
                                    mojom::WindowDataPtr root_data,
                                    mojom::WindowTreePtr tree,
                                    Id focused_window_id,
                                    uint32 access_policy) {
   DCHECK(!tree_ptr_);
   tree_ptr_ = tree.Pass();
   tree_ptr_.set_connection_error_handler([this]() { delete this; });
   OnEmbedImpl(tree_ptr_.get(), connection_id, root_data.Pass(),
               focused_window_id, access_policy);
@@ skipping 173 matching lines @@
   // |WindowTreeConnection::GetFocusedWindow()| etc.
   focused_window_ = focused;
   if (blurred) {
     FOR_EACH_OBSERVER(WindowObserver, *WindowPrivate(blurred).observers(),
                       OnWindowFocusChanged(focused, blurred));
   }
   if (focused) {
     FOR_EACH_OBSERVER(WindowObserver, *WindowPrivate(focused).observers(),
                       OnWindowFocusChanged(focused, blurred));
   }
+  FOR_EACH_OBSERVER(WindowTreeConnectionObserver, observers_,
+                    OnWindowTreeFocusChanged(focused, blurred));
 }
 
 void WindowTreeClientImpl::OnChangeCompleted(uint32 change_id, bool success) {
   scoped_ptr<InFlightChange> change(std::move(in_flight_map_[change_id]));
   in_flight_map_.erase(change_id);
   if (!change)
     return;
 
   if (!success)
     change->ChangeFailed();
@@ skipping 50 matching lines @@
 }
 
 ////////////////////////////////////////////////////////////////////////////////
 // WindowTreeClientImpl, private:
 
 mojo::Callback<void(bool)> WindowTreeClientImpl::ActionCompletedCallback() {
   return [this](bool success) {};
 }
 
 }  // namespace mus