Set capture to the window being dragged when dragging a window on Ash

ash/wm/toplevel_window_event_handler.cc

 // Copyright (c) 2012 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 "ash/wm/toplevel_window_event_handler.h"
 
 #include "ash/shell.h"
 #include "ash/wm/resize_shadow_controller.h"
 #include "ash/wm/window_resizer.h"
 #include "ash/wm/window_state.h"
@@ skipping 92 matching lines @@
   void OnWindowDestroying(aura::Window* window) override;
 
   // WindowStateObserver overrides:
   void OnPreWindowStateTypeChange(wm::WindowState* window_state,
                                   wm::WindowStateType type) override;
 
  private:
   ToplevelWindowEventHandler* handler_;
   scoped_ptr<WindowResizer> resizer_;
 
+  // Whether ScopedWindowResizer grabbed capture.
+  bool grabbed_capture_;
+
   DISALLOW_COPY_AND_ASSIGN(ScopedWindowResizer);
 };
 
 ToplevelWindowEventHandler::ScopedWindowResizer::ScopedWindowResizer(
     ToplevelWindowEventHandler* handler,
     WindowResizer* resizer)
     : handler_(handler),
-      resizer_(resizer) {
-  resizer_->GetTarget()->AddObserver(this);
-  wm::GetWindowState(resizer_->GetTarget())->AddObserver(this);
+      resizer_(resizer),
+      grabbed_capture_(false) {
+  aura::Window* target = resizer_->GetTarget();
+  target->AddObserver(this);
+  wm::GetWindowState(target)->AddObserver(this);
+
+  if (!target->HasCapture()) {
+    grabbed_capture_ = true;
+    target->SetCapture();
+  }
 }
 
 ToplevelWindowEventHandler::ScopedWindowResizer::~ScopedWindowResizer() {
-  resizer_->GetTarget()->RemoveObserver(this);
-  wm::GetWindowState(resizer_->GetTarget())->RemoveObserver(this);
+  aura::Window* target = resizer_->GetTarget();
+  target->RemoveObserver(this);
+  wm::GetWindowState(target)->RemoveObserver(this);
+  if (grabbed_capture_)
+    target->ReleaseCapture();
 }
 
 bool ToplevelWindowEventHandler::ScopedWindowResizer::IsMove() const {
   return resizer_->details().bounds_change ==
       WindowResizer::kBoundsChange_Repositions;
 }
 
 void
 ToplevelWindowEventHandler::ScopedWindowResizer::OnPreWindowStateTypeChange(
     wm::WindowState* window_state,
@@ skipping 466 matching lines @@
     default:
       NOTREACHED();
   }
 }
 
 void ToplevelWindowEventHandler::ResizerWindowDestroyed() {
   CompleteDrag(DRAG_RESIZER_WINDOW_DESTROYED);
 }
 
 }  // namespace ash