mus: Implement Window Server Capture

components/mus/ws/window_tree_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/ws/window_tree_impl.h"
 
 #include "base/bind.h"
 #include "base/stl_util.h"
 #include "components/mus/ws/connection_manager.h"
 #include "components/mus/ws/default_access_policy.h"
@@ skipping 385 matching lines @@
   WindowTreeHostImpl* host = GetHost();
   if (!host || !host->window_manager())
     return false;
   // Requests coming from the WM should not be routed through the WM again.
   bool is_wm = host->GetWindowTree() == this;
   if (is_wm)
     return false;
   return true;
 }
 
+void WindowTreeImpl::ProcessLostCapture(
+    const ServerWindow* old_capture_window) {
+  client()->OnLostCapture(WindowIdToTransportId(old_capture_window->id()));
+}
+
 bool WindowTreeImpl::IsWindowKnown(const ServerWindow* window) const {
   return known_windows_.count(WindowIdToTransportId(window->id())) > 0;
 }
 
 bool WindowTreeImpl::CanReorderWindow(const ServerWindow* window,
                                       const ServerWindow* relative_window,
                                       mojom::OrderDirection direction) const {
   if (!window || !relative_window)
     return false;
 
@@ skipping 274 matching lines @@
 }
 
 void WindowTreeImpl::GetWindowTree(
     Id window_id,
     const Callback<void(Array<mojom::WindowDataPtr>)>& callback) {
   std::vector<const ServerWindow*> windows(
       GetWindowTree(WindowIdFromTransportId(window_id)));
   callback.Run(WindowsToWindowDatas(windows));
 }
 
+void WindowTreeImpl::SetCapture(uint32_t change_id, Id window_id) {
+  ServerWindow* window = GetWindow(WindowIdFromTransportId(window_id));
+  WindowTreeHostImpl* host = GetHost();
+  bool success = window && access_policy_->CanSetCapture(window) && host &&
+                 access_policy_->CanSetCapture(host->GetCaptureWindow());
+  if (success) {
+    Operation op(this, connection_manager_, OperationType::SET_CAPTURE);
+    host->SetCapture(window);
+  }
+  client_->OnChangeCompleted(change_id, success);
+}
+
+void WindowTreeImpl::ReleaseCapture(uint32_t change_id, Id window_id) {
+  ServerWindow* window = GetWindow(WindowIdFromTransportId(window_id));
+  WindowTreeHostImpl* host = GetHost();
+  bool success =
+      window && host && access_policy_->CanSetCapture(host->GetCaptureWindow());
+  if (success) {
+    Operation op(this, connection_manager_, OperationType::RELEASE_CAPTURE);
+    host->SetCapture(nullptr);
+  }
+  client_->OnChangeCompleted(change_id, success);
+}
+
 void WindowTreeImpl::SetWindowBounds(uint32_t change_id,
                                      Id window_id,
                                      mojo::RectPtr bounds) {
   ServerWindow* window = GetWindow(WindowIdFromTransportId(window_id));
   if (window && ShouldRouteToWindowManager(window)) {
     const uint32_t wm_change_id =
         connection_manager_->GenerateWindowManagerChangeId(this, change_id);
     GetHost()->GetWindowTree()->client_->WmSetBounds(wm_change_id, window_id,
                                                      bounds.Pass());
     return;
@@ skipping 157 matching lines @@
   return connection && connection != this;
 }
 
 bool WindowTreeImpl::IsDescendantOfEmbedRoot(const ServerWindow* window) {
   return is_embed_root_ && root_ && GetWindow(*root_)->Contains(window);
 }
 
 }  // namespace ws
 
 }  // namespace mus