mus: Implement Window Server Capture

components/mus/ws/window_tree_host_impl.cc

 // 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.
 
 #include "components/mus/ws/window_tree_host_impl.h"
 
 #include "base/strings/utf_string_conversions.h"
 #include "components/mus/common/types.h"
 #include "components/mus/ws/connection_manager.h"
 #include "components/mus/ws/display_manager.h"
@@ skipping 65 matching lines @@
   if (windows_needing_frame_destruction_.count(window))
     return;
   windows_needing_frame_destruction_.insert(window);
   window->AddObserver(this);
 }
 
 const mojom::ViewportMetrics& WindowTreeHostImpl::GetViewportMetrics() const {
   return display_manager_->GetViewportMetrics();
 }
 
+void WindowTreeHostImpl::SetCapture(ServerWindow* window) {
+  ServerWindow* capture_window = event_dispatcher_.capture_window();
+  if (capture_window) {
+    WindowTreeImpl* connection =
+        connection_manager_->GetConnection(capture_window->id().connection_id);
+    if (connection)
+      connection->ProcessLostCapture(capture_window);
+  }
+  event_dispatcher_.SetCaptureWindow(window);
+  if (window)
+    display_manager_->SetCapture();
+  else
+    display_manager_->ReleaseCapture();
+}
+
 void WindowTreeHostImpl::SetFocusedWindow(ServerWindow* new_focused_window) {
   ServerWindow* old_focused_window = focus_controller_->GetFocusedWindow();
   if (old_focused_window == new_focused_window)
     return;
   DCHECK(root_window()->Contains(new_focused_window));
   focus_controller_->SetFocusedWindow(new_focused_window);
   // TODO(beng): have the FocusController notify us via FocusControllerDelegate.
   OnFocusChanged(old_focused_window, new_focused_window);
 }
 
@@ skipping 47 matching lines @@
 }
 
 ServerWindow* WindowTreeHostImpl::GetRootWindow() {
   return root_.get();
 }
 
 void WindowTreeHostImpl::OnEvent(mojom::EventPtr event) {
   event_dispatcher_.OnEvent(event.Pass());
 }
 
+void WindowTreeHostImpl::OnLostCapture() {
+  SetCapture(nullptr);
+}
+
 void WindowTreeHostImpl::OnDisplayClosed() {
   if (delegate_)
     delegate_->OnDisplayClosed();
 }
 
 void WindowTreeHostImpl::OnViewportMetricsChanged(
     const mojom::ViewportMetrics& old_metrics,
     const mojom::ViewportMetrics& new_metrics) {
   if (!root_) {
     root_.reset(connection_manager_->CreateServerWindow(
@@ skipping 91 matching lines @@
 
 void WindowTreeHostImpl::SetFocusedWindowFromEventDispatcher(
     ServerWindow* new_focused_window) {
   SetFocusedWindow(new_focused_window);
 }
 
 ServerWindow* WindowTreeHostImpl::GetFocusedWindowForEventDispatcher() {
   return GetFocusedWindow();
 }
 
+void WindowTreeHostImpl::OnLostCapture(ServerWindow* window) {
+  DCHECK(window);
+  WindowTreeImpl* connection =
+      connection_manager_->GetConnection(window->id().connection_id);
+  if (connection)
+    connection->ProcessLostCapture(window);
+}
+
 void WindowTreeHostImpl::DispatchInputEventToWindow(ServerWindow* target,
                                                     bool in_nonclient_area,
                                                     mojom::EventPtr event) {
   // If the event is in the non-client area the event goes to the owner of
   // the window. Otherwise if the window is an embed root, forward to the
   // embedded window.
   WindowTreeImpl* connection =
       in_nonclient_area
           ? connection_manager_->GetConnection(target->id().connection_id)
           : connection_manager_->GetConnectionWithRoot(target->id());
   if (!connection) {
     DCHECK(!in_nonclient_area);
     connection = connection_manager_->GetConnection(target->id().connection_id);
   }
   connection->client()->OnWindowInputEvent(WindowIdToTransportId(target->id()),
                                            event.Pass(),
                                            base::Bind(&base::DoNothing));
 }
 
 void WindowTreeHostImpl::OnWindowDestroyed(ServerWindow* window) {
   windows_needing_frame_destruction_.erase(window);
   window->RemoveObserver(this);
 }
 
 }  // namespace ws
 }  // namespace mus