mus: Implement Window Server Capture

components/mus/view_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/view_tree_host_impl.h"
 
 #include "base/strings/utf_string_conversions.h"
 #include "components/mus/connection_manager.h"
 #include "components/mus/display_manager.h"
 #include "components/mus/focus_controller.h"
@@ skipping 13 matching lines @@
     const scoped_refptr<GpuState>& gpu_state,
     const scoped_refptr<SurfacesState>& surfaces_state)
     : delegate_(nullptr),
       connection_manager_(connection_manager),
       client_(client.Pass()),
       event_dispatcher_(this),
       display_manager_(DisplayManager::Create(is_headless,
                                               app_impl,
                                               gpu_state,
                                               surfaces_state)),
-      focus_controller_(new FocusController(this)) {
+      focus_controller_(new FocusController(this)),
+      captured_view_(nullptr) {
   display_manager_->Init(this);
   if (client_) {
     client_.set_connection_error_handler(
         base::Bind(&ViewTreeHostImpl::OnClientClosed, base::Unretained(this)));
   }
 }
 
 ViewTreeHostImpl::~ViewTreeHostImpl() {}
 
 void ViewTreeHostImpl::Init(ViewTreeHostDelegate* delegate) {
@@ skipping 16 matching lines @@
     display_manager_->SchedulePaint(view, bounds);
     return true;
   }
   return false;
 }
 
 const mojo::ViewportMetrics& ViewTreeHostImpl::GetViewportMetrics() const {
   return display_manager_->GetViewportMetrics();
 }
 
+void ViewTreeHostImpl::SetCapture(ServerView* view) {
+  if (captured_view_) {
+    ViewTreeImpl* connection =
+        connection_manager_->GetConnection(captured_view_->id().connection_id);
+    if (connection) {
+      connection->ProcessLostCapture(captured_view_);
+    }
+  }
+  captured_view_ = view;
+  if (captured_view_)
+    display_manager_->SetCapture();
+}
+
 void ViewTreeHostImpl::SetFocusedView(ServerView* new_focused_view) {
   ServerView* old_focused_view = focus_controller_->GetFocusedView();
   if (old_focused_view == new_focused_view)
     return;
   DCHECK(root_view()->Contains(new_focused_view));
   focus_controller_->SetFocusedView(new_focused_view);
   // TODO(beng): have the FocusController notify us via FocusControllerDelegate.
   OnFocusChanged(old_focused_view, new_focused_view);
 }
 
@@ skipping 66 matching lines @@
 }
 
 ServerView* ViewTreeHostImpl::GetRootView() {
   return root_.get();
 }
 
 void ViewTreeHostImpl::OnEvent(mojo::EventPtr event) {
   event_dispatcher_.OnEvent(event.Pass());
 }
 
+void ViewTreeHostImpl::OnLostCapture() {
+  SetCapture(nullptr);
+}
+
 void ViewTreeHostImpl::OnDisplayClosed() {
   if (delegate_)
     delegate_->OnDisplayClosed();
 }
 
 void ViewTreeHostImpl::OnViewportMetricsChanged(
     const mojo::ViewportMetrics& old_metrics,
     const mojo::ViewportMetrics& new_metrics) {
   if (!root_) {
     root_.reset(connection_manager_->CreateServerView(
@@ skipping 65 matching lines @@
       root_tree != embedded_connection_old &&
       root_tree != owning_connection_new &&
       root_tree != embedded_connection_new) {
     root_tree->ProcessFocusChanged(old_focused_view, new_focused_view);
   }
 
   UpdateTextInputState(new_focused_view, new_focused_view->text_input_state());
 }
 
 }  // namespace mus