mus: Server-side implementation of modal windows

components/mus/ws/display.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/display.h"
 
 #include "base/debug/debugger.h"
 #include "base/strings/utf_string_conversions.h"
 #include "components/mus/common/types.h"
 #include "components/mus/ws/connection_manager.h"
@@ skipping 195 matching lines @@
              ? nullptr
              : window_manager_state_map_.begin()->second.get();
 }
 
 const WindowManagerState* Display::GetWindowManagerStateForUser(
     const UserId& user_id) const {
   auto iter = window_manager_state_map_.find(user_id);
   return iter == window_manager_state_map_.end() ? nullptr : iter->second.get();
 }
 
-void Display::SetCapture(ServerWindow* window, bool in_nonclient_area) {
+bool Display::SetCapture(ServerWindow* window, bool in_nonclient_area) {
   ServerWindow* capture_window = event_dispatcher_.capture_window();
   if (capture_window == window)
+    return true;
+  return event_dispatcher_.SetCaptureWindow(window, in_nonclient_area);
+}
+
+void Display::ReleaseCaptureForModalWindow(ServerWindow* window) {
+  if (!window->is_modal() || !window->IsDrawn())
     return;
-  event_dispatcher_.SetCaptureWindow(window, in_nonclient_area);
+
+  ServerWindow* capture_window = GetCaptureWindow();
+  if (window->transient_parent() &&
+      !window->transient_parent()->Contains(capture_window)) {
+    return;
+  }
+
+  SetCapture(nullptr, false);
 }
 
 mojom::Rotation Display::GetRotation() const {
   return platform_display_->GetRotation();
 }
 
 void Display::SetFocusedWindow(ServerWindow* new_focused_window) {
   ServerWindow* old_focused_window = focus_controller_->GetFocusedWindow();
   if (old_focused_window == new_focused_window)
     return;
@@ skipping 430 matching lines @@
   DCHECK_EQ(0u, window_manager_state_map_.count(id));
 }
 
 void Display::OnWindowManagerFactorySet(WindowManagerFactoryService* service) {
   if (!binding_)
     CreateWindowManagerStateFromService(service);
 }
 
 }  // namespace ws
 }  // namespace mus