Centralize ScreenOrientationProvider logic, add platform delegates

content/public/browser/screen_orientation_provider.cc

Index: content/public/browser/screen_orientation_provider.cc
diff --git a/content/public/browser/screen_orientation_provider.cc b/content/public/browser/screen_orientation_provider.cc
new file mode 100644
index 0000000000000000000000000000000000000000..07c4c9323e5b3e417e97bd06d7175594a88eebe6
--- /dev/null
+++ b/content/public/browser/screen_orientation_provider.cc
@@ -0,0 +1,207 @@
+// 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 "content/public/browser/screen_orientation_provider.h"
+
+#include "content/browser/renderer_host/render_view_host_delegate.h"
+#include "content/public/browser/render_view_host.h"
+#include "content/public/browser/screen_orientation_delegate.h"
+#include "content/public/browser/screen_orientation_dispatcher_host.h"
+#include "content/public/browser/web_contents.h"
+#include "third_party/WebKit/public/platform/WebLockOrientationError.h"
+#include "third_party/WebKit/public/platform/WebScreenInfo.h"
+
+namespace content {
+
+ScreenOrientationDelegate* ScreenOrientationProvider::delegate_ = NULL;
+
+ScreenOrientationProvider::LockInformation::LockInformation(int request_id,
+    blink::WebScreenOrientationLockType lock)
+    : request_id(request_id),
+      lock(lock) {
+}
+
+ScreenOrientationProvider::ScreenOrientationProvider(
+    ScreenOrientationDispatcherHost* dispatcher_host,
+    WebContents* web_contents)
+    : WebContentsObserver(web_contents),
+      dispatcher_(dispatcher_host),
+      lock_applied_(false) {
+}
+
+ScreenOrientationProvider::~ScreenOrientationProvider() {
+}
+
+void ScreenOrientationProvider::LockOrientation(int request_id,
+    blink::WebScreenOrientationLockType lock_orientation) {
+
+  if (!delegate_ || !delegate_->ScreenOrientationProviderSupported()) {
+    dispatcher_->NotifyLockError(request_id,
+                                 blink::WebLockOrientationErrorNotAvailable);
+    return;
+  }
+
+  if (delegate_->FullScreenRequired(web_contents())) {
+    RenderViewHost* rvh = web_contents()->GetRenderViewHost();
+    if (!rvh) {
+      dispatcher_->NotifyLockError(request_id,
+                                   blink::WebLockOrientationErrorCanceled);
+    }
+    RenderViewHostDelegate* rvhd = rvh->GetDelegate();
+    if (!rvhd) {
+      dispatcher_->NotifyLockError(request_id,
+                                   blink::WebLockOrientationErrorCanceled);
+    }
+    if (!rvhd->IsFullscreenForCurrentTab()) {
+      dispatcher_->NotifyLockError(request_id,
+          blink::WebLockOrientationErrorFullScreenRequired);
+    }
+    return;
+  }
+
+  if (lock_orientation == blink::WebScreenOrientationLockNatural) {
+    lock_orientation = GetNaturalLockType();
+    if (lock_orientation == blink::WebScreenOrientationLockDefault) {
+      // We are in a broken state, let's pretend we got canceled.
+      dispatcher_->NotifyLockError(request_id,
+                                   blink::WebLockOrientationErrorCanceled);
+      return;
+    }
+  }
+
+  lock_applied_ = true;
+  delegate_->Lock(lock_orientation);
+
+  // If two calls happen close to each other some platforms will ignore the
+  // first. A successful lock will be once orientation matches the latest
+  // request.
+  pending_lock_.reset();
+
+  // If the orientation we are locking to matches the current orientation, we
+  // should succeed immediately.
+  if (LockMatchesCurrentOrientation(lock_orientation)) {
+    dispatcher_->NotifyLockSuccess(request_id);
+    return;
+  }
+
+  pending_lock_.reset(new LockInformation(request_id, lock_orientation));
+}
+
+void ScreenOrientationProvider::UnlockOrientation() {
+  if (!lock_applied_ || !delegate_)
+    return;
+
+  delegate_->Unlock();
+
+  lock_applied_ = false;
+  pending_lock_.reset();
+}
+
+void ScreenOrientationProvider::OnOrientationChange() {
+  if (!pending_lock_.get())
+    return;
+
+  if (LockMatchesCurrentOrientation(pending_lock_->lock)) {
+    dispatcher_->NotifyLockSuccess(pending_lock_->request_id);
+    pending_lock_.reset();
+  }
+}
+
+void ScreenOrientationProvider::SetDelegate(
+    ScreenOrientationDelegate* delegate) {
+  delegate_ = delegate;
+}
+
+void ScreenOrientationProvider::DidToggleFullscreenModeForTab(
+    bool entered_fullscreen) {
+  if (!lock_applied_ || !delegate_)
+    return;
+
+  // If fullscreen is not required in order to lock orientation, don't unlock
+  // when fullscreen state changes.
+  if (!delegate_->FullScreenRequired(web_contents()))
+    return;
+
+  DCHECK(!entered_fullscreen);
+  UnlockOrientation();
+}
+
+blink::WebScreenOrientationLockType
+    ScreenOrientationProvider::GetNaturalLockType() const {
+  RenderWidgetHost* rwh = web_contents()->GetRenderViewHost();
+  if (!rwh)
+    return blink::WebScreenOrientationLockDefault;
+
+  blink::WebScreenInfo screen_info;
+  rwh->GetWebScreenInfo(&screen_info);
+
+  switch (screen_info.orientationType) {
+    case blink::WebScreenOrientationPortraitPrimary:
+    case blink::WebScreenOrientationPortraitSecondary:
+      if (screen_info.orientationAngle == 0 ||
+          screen_info.orientationAngle == 180) {
+        return blink::WebScreenOrientationLockPortraitPrimary;
+      }
+      return blink::WebScreenOrientationLockLandscapePrimary;
+    case blink::WebScreenOrientationLandscapePrimary:
+    case blink::WebScreenOrientationLandscapeSecondary:
+      if (screen_info.orientationAngle == 0 ||
+          screen_info.orientationAngle == 180) {
+        return blink::WebScreenOrientationLockLandscapePrimary;
+      }
+      return blink::WebScreenOrientationLockPortraitPrimary;
+    case blink::WebScreenOrientationUndefined:
+      NOTREACHED();
+      return blink::WebScreenOrientationLockDefault;
+  }
+
+  NOTREACHED();
+  return blink::WebScreenOrientationLockDefault;
+}
+
+bool ScreenOrientationProvider::LockMatchesCurrentOrientation(
+    blink::WebScreenOrientationLockType lock) {
+  RenderWidgetHost* rwh = web_contents()->GetRenderViewHost();
+  if (!rwh)
+    return false;
+
+  blink::WebScreenInfo screen_info;
+  rwh->GetWebScreenInfo(&screen_info);
+
+  switch (lock) {
+    case blink::WebScreenOrientationLockPortraitPrimary:
+      return screen_info.orientationType ==
+          blink::WebScreenOrientationPortraitPrimary;
+    case blink::WebScreenOrientationLockPortraitSecondary:
+      return screen_info.orientationType ==
+          blink::WebScreenOrientationPortraitSecondary;
+    case blink::WebScreenOrientationLockLandscapePrimary:
+      return screen_info.orientationType ==
+          blink::WebScreenOrientationLandscapePrimary;
+    case blink::WebScreenOrientationLockLandscapeSecondary:
+      return screen_info.orientationType ==
+          blink::WebScreenOrientationLandscapeSecondary;
+    case blink::WebScreenOrientationLockLandscape:
+      return screen_info.orientationType ==
+          blink::WebScreenOrientationLandscapePrimary ||
+          screen_info.orientationType ==
+          blink::WebScreenOrientationLandscapeSecondary;
+    case blink::WebScreenOrientationLockPortrait:
+      return screen_info.orientationType ==
+          blink::WebScreenOrientationPortraitPrimary ||
+          screen_info.orientationType ==
+          blink::WebScreenOrientationPortraitSecondary;
+    case blink::WebScreenOrientationLockAny:
+      return true;
+    case blink::WebScreenOrientationLockNatural:
+    case blink::WebScreenOrientationLockDefault:
+      NOTREACHED();
+      return false;
+  }
+
+  NOTREACHED();
+  return false;
+}
+
+} // namespace content