Unify the code paths for handling mouse events when pointer is locked on

third_party/WebKit/Source/web/WebFrameWidgetBase.cpp

 // Copyright 2016 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 "web/WebFrameWidgetBase.h"
 
+#include "core/dom/DocumentUserGestureToken.h"
 #include "core/frame/FrameHost.h"
+#include "core/frame/FrameView.h"
 #include "core/frame/VisualViewport.h"
 #include "core/input/EventHandler.h"
 #include "core/page/DragActions.h"
 #include "core/page/DragController.h"
 #include "core/page/DragData.h"
 #include "core/page/DragSession.h"
 #include "core/page/Page.h"
+#include "core/page/PointerLockController.h"
+#include "platform/UserGestureIndicator.h"
 #include "public/web/WebAutofillClient.h"
 #include "public/web/WebDocument.h"
 #include "public/web/WebWidgetClient.h"
+#include "web/WebInputEventConversion.h"
 #include "web/WebLocalFrameImpl.h"
 #include "web/WebViewImpl.h"
 
 namespace blink {
 
 namespace {
 
 // Helper to get LocalFrame* from WebLocalFrame*.
 // TODO(dcheng): This should be moved into WebLocalFrame.
 LocalFrame* toCoreFrame(WebLocalFrame* frame) {
@@ skipping 183 matching lines @@
 }
 
 WebViewImpl* WebFrameWidgetBase::view() const {
   return toWebLocalFrameImpl(localRoot())->viewImpl();
 }
 
 Page* WebFrameWidgetBase::page() const {
   return view()->page();
 }
 
+void WebFrameWidgetBase::didAcquirePointerLock() {
+  page()->pointerLockController().didAcquirePointerLock();
+}
+
+void WebFrameWidgetBase::didNotAcquirePointerLock() {
+  page()->pointerLockController().didNotAcquirePointerLock();
+}
+
+void WebFrameWidgetBase::didLosePointerLock() {
+  m_pointerLockGestureToken.clear();
+  page()->pointerLockController().didLosePointerLock();
+}
+
+void WebFrameWidgetBase::pointerLockMouseEvent(const WebInputEvent& event) {
+  std::unique_ptr<UserGestureIndicator> gestureIndicator;
+  AtomicString eventType;
+  switch (event.type()) {
+    case WebInputEvent::MouseDown:
+      eventType = EventTypeNames::mousedown;
+      if (!page() || !page()->pointerLockController().element())
+        break;
+      gestureIndicator = WTF::wrapUnique(
+          new UserGestureIndicator(DocumentUserGestureToken::create(
+              &page()->pointerLockController().element()->document(),
+              UserGestureToken::NewGesture)));
+      m_pointerLockGestureToken = gestureIndicator->currentToken();
+      break;
+    case WebInputEvent::MouseUp:
+      eventType = EventTypeNames::mouseup;
+      gestureIndicator = WTF::wrapUnique(
+          new UserGestureIndicator(m_pointerLockGestureToken.release()));
+      break;
+    case WebInputEvent::MouseMove:
+      eventType = EventTypeNames::mousemove;
+      break;
+    default:
+      NOTREACHED();
+  }
+
+  const WebMouseEvent& mouseEvent = static_cast<const WebMouseEvent&>(event);
+
+  if (page()) {
+    WebMouseEvent transformedEvent = TransformWebMouseEvent(
+        toWebLocalFrameImpl(localRoot())->frameView(), mouseEvent);
+    page()->pointerLockController().dispatchLockedMouseEvent(transformedEvent,
+                                                             eventType);
+  }
+}
+
 }  // namespace blink