More user id tracking for mus:

components/mus/ws/window_tree_host_impl.h

 // Copyright 2015 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.
 
 #ifndef COMPONENTS_MUS_WS_WINDOW_TREE_HOST_IMPL_H_
 #define COMPONENTS_MUS_WS_WINDOW_TREE_HOST_IMPL_H_
 
 #include <stdint.h>
 
 #include <map>
 #include <queue>
 
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "components/mus/common/types.h"
 #include "components/mus/public/interfaces/window_manager_constants.mojom.h"
 #include "components/mus/public/interfaces/window_tree_host.mojom.h"
 #include "components/mus/ws/display_manager.h"
 #include "components/mus/ws/event_dispatcher.h"
 #include "components/mus/ws/event_dispatcher_delegate.h"
 #include "components/mus/ws/focus_controller_delegate.h"
 #include "components/mus/ws/focus_controller_observer.h"
 #include "components/mus/ws/server_window.h"
 #include "components/mus/ws/server_window_observer.h"
 #include "components/mus/ws/server_window_tracker.h"
+#include "components/mus/ws/user_id_tracker_observer.h"
+#include "components/mus/ws/window_manager_factory_registry_observer.h"
 
 namespace mus {
 namespace ws {
 
 class ConnectionManager;
 class FocusController;
 class WindowManagerState;
 class WindowTreeHostConnection;
 class WindowTreeImpl;
 
+namespace test {
+class WindowTreeHostTestApi;
+}
+
 // WindowTreeHostImpl is an implementation of the WindowTreeHost interface.
 // It serves as a top level root window for a window. Its lifetime is managed by
 // ConnectionManager. If the connection to the client breaks or if the user
 // closes the associated window, then this object and related state will be
 // deleted.
 //
 // WindowTreeHost has a single root window whose children are the roots for
 // a per-user WindowManager. WindowTreeHost is configured in two distinct
 // ways:
 // . with a WindowTreeHostConnection. In this mode there is only ever one
 //   WindowManager for the host, which comes from the client that created
 //   the WindowTreeHost.
 // . without a WindowTreeHostConnection. In this mode a WindowManager is
 //   automatically created per user.
 class WindowTreeHostImpl : public DisplayManagerDelegate,
                            public mojom::WindowTreeHost,
                            public FocusControllerObserver,
                            public FocusControllerDelegate,
                            public EventDispatcherDelegate,
-                           public ServerWindowObserver {
+                           public ServerWindowObserver,
+                           public UserIdTrackerObserver,
+                           public WindowManagerFactoryRegistryObserver {
  public:
   // TODO(fsamuel): All these parameters are just plumbing for creating
   // DisplayManagers. We should probably just store these common parameters
   // in the DisplayManagerFactory and pass them along on DisplayManager::Create.
   WindowTreeHostImpl(ConnectionManager* connection_manager,
                      mojo::Connector* connector,
                      const scoped_refptr<GpuState>& gpu_state,
                      const scoped_refptr<SurfacesState>& surfaces_state);
   ~WindowTreeHostImpl() override;
 
@@ skipping 31 matching lines @@
   // Returns the root of the WindowTreeHost. The root's children are the roots
   // of the corresponding WindowManagers.
   ServerWindow* root_window() { return root_.get(); }
   const ServerWindow* root_window() const { return root_.get(); }
 
   ServerWindow* GetRootWithId(const WindowId& id);
 
   WindowManagerState* GetWindowManagerStateWithRoot(const ServerWindow* window);
   // TODO(sky): this is wrong, plumb through user_id.
   WindowManagerState* GetFirstWindowManagerState();
+  WindowManagerState* GetWindowManagerStateForUser(UserId user_id);
+  size_t num_window_manger_states() const {
+    return window_manager_state_map_.size();
+  }
 
   void SetCapture(ServerWindow* window, bool in_nonclient_area);
 
   void SetFocusedWindow(ServerWindow* window);
   ServerWindow* GetFocusedWindow();
   void DestroyFocusController();
   FocusController* focus_controller() { return focus_controller_.get(); }
 
   void AddActivationParent(ServerWindow* window);
   void RemoveActivationParent(ServerWindow* window);
@@ skipping 24 matching lines @@
   void MaybeChangeCursorOnWindowTreeChange();
 
   // WindowTreeHost:
   void SetSize(mojo::SizePtr size) override;
   void SetTitle(const mojo::String& title) override;
 
   void OnEventAck(mojom::WindowTree* tree);
 
  private:
   class ProcessedEventTarget;
-  friend class WindowTreeTest;
+  friend class test::WindowTreeHostTestApi;
+
   using WindowManagerStateMap =
       std::map<uint32_t, scoped_ptr<WindowManagerState>>;
 
   // There are two types of events that may be queued, both occur only when
   // waiting for an ack from a client.
   // . We get an event from the DisplayManager. This results in |event| being
   //   set, but |processed_target| is null.
   // . We get an event from the EventDispatcher. In this case both |event| and
   //   |processed_target| are valid.
   // The second case happens if EventDispatcher generates more than one event
@@ skipping 22 matching lines @@
 
   // Dispatches the event to the appropriate client and starts the ack timer.
   void DispatchInputEventToWindowImpl(ServerWindow* target,
                                       bool in_nonclient_area,
                                       const ui::Event& event);
 
   // Creates the set of WindowManagerStates from the
   // WindowManagerFactoryRegistry.
   void CreateWindowManagerStatesFromRegistry();
 
+  void CreateWindowManagerStateFromService(
+      WindowManagerFactoryService* service);
+
   void UpdateNativeCursor(int32_t cursor_id);
 
   // DisplayManagerDelegate:
   ServerWindow* GetRootWindow() override;
   void OnEvent(const ui::Event& event) override;
   void OnNativeCaptureLost() override;
   void OnDisplayClosed() override;
   void OnViewportMetricsChanged(
       const mojom::ViewportMetrics& old_metrics,
       const mojom::ViewportMetrics& new_metrics) override;
@@ skipping 17 matching lines @@
   void SetNativeCapture() override;
   void ReleaseNativeCapture() override;
   void OnServerWindowCaptureLost(ServerWindow* window) override;
   void DispatchInputEventToWindow(ServerWindow* target,
                                   bool in_nonclient_area,
                                   const ui::Event& event) override;
 
   // ServerWindowObserver:
   void OnWindowDestroyed(ServerWindow* window) override;
 
+  // UserIdTrackerObserver:
+  void OnActiveUserIdChanged(UserId id) override;
+  void OnUserIdAdded(UserId id) override;
+  void OnUserIdRemoved(UserId id) override;
+
+  // WindowManagerFactoryRegistryObserver:
+  void OnWindowManagerFactorySet(WindowManagerFactoryService* service) override;
+
   const uint32_t id_;
   scoped_ptr<WindowTreeHostConnection> window_tree_host_connection_;
   // Set once Init() has been called.
   bool init_called_ = false;
   ConnectionManager* const connection_manager_;
   EventDispatcher event_dispatcher_;
   scoped_ptr<ServerWindow> root_;
   scoped_ptr<DisplayManager> display_manager_;
   scoped_ptr<FocusController> focus_controller_;
   mojom::WindowTree* tree_awaiting_input_ack_;
@@ skipping 14 matching lines @@
 
   mojom::FrameDecorationValuesPtr frame_decoration_values_;
 
   DISALLOW_COPY_AND_ASSIGN(WindowTreeHostImpl);
 };
 
 }  // namespace ws
 }  // namespace mus
 
 #endif  // COMPONENTS_MUS_WS_WINDOW_TREE_HOST_IMPL_H_