Screen Orientation API: Add support for lockOrientation() / unlockOrientation()

Source/modules/screen_orientation/ScreenOrientation.cpp

 // 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 "config.h"
 #include "modules/screen_orientation/ScreenOrientation.h"
 
 #include "core/frame/DOMWindow.h"
 #include "core/frame/Frame.h"
 #include "core/frame/Screen.h"
 #include "modules/screen_orientation/ScreenOrientationController.h"
-#include "public/platform/WebScreenOrientation.h"
+#include "public/platform/Platform.h"
 
 namespace WebCore {
 
+static const unsigned WebScreenOrientationDefault = 0;
+
 struct ScreenOrientationInfo {
     const AtomicString& name;
     blink::WebScreenOrientation orientation;
 };
 
 static ScreenOrientationInfo* orientationsMap(unsigned& length)
 {
     DEFINE_STATIC_LOCAL(const AtomicString, portraitPrimary, ("portrait-primary", AtomicString::ConstructFromLiteral));
     DEFINE_STATIC_LOCAL(const AtomicString, portraitSecondary, ("portrait-secondary", AtomicString::ConstructFromLiteral));
     DEFINE_STATIC_LOCAL(const AtomicString, landscapePrimary, ("landscape-primary", AtomicString::ConstructFromLiteral));
@@ skipping 16 matching lines @@
     for (unsigned i = 0; i < length; ++i) {
         if (orientationMap[i].orientation == orientation)
             return orientationMap[i].name;
     }
     // We do no handle OrientationInvalid and OrientationAny but this is fine because screen.orientation
     // should never return these and WebScreenOrientation does not define those values.
     ASSERT_NOT_REACHED();
     return nullAtom;
 }
 
+static blink::WebScreenOrientations stringToOrientations(const AtomicString& orientationString)
+{
+    DEFINE_STATIC_LOCAL(const AtomicString, portrait, ("portrait", AtomicString::ConstructFromLiteral));
+    DEFINE_STATIC_LOCAL(const AtomicString, landscape, ("landscape", AtomicString::ConstructFromLiteral));
+
+    if (orientationString == portrait)
+        return blink::WebScreenOrientationPortraitPrimary | blink::WebScreenOrientationPortraitSecondary;
+    if (orientationString == landscape)
+        return blink::WebScreenOrientationLandscapePrimary | blink::WebScreenOrientationLandscapeSecondary;
+
+    unsigned length = 0;
+    ScreenOrientationInfo* orientationMap = orientationsMap(length);
+    for (unsigned i = 0; i < length; ++i) {
+        if (orientationMap[i].name == orientationString)
+            return orientationMap[i].orientation;
+    }
+    return 0;
+}
+
 ScreenOrientation::ScreenOrientation(Screen& screen)
     : DOMWindowProperty(screen.frame())
+    , m_orientationLockTimer(this, &ScreenOrientation::orientationLockTimerFired)
+    , m_lockedOrientations(WebScreenOrientationDefault)
 {
 }
 
+void ScreenOrientation::lockOrientationAsync(blink::WebScreenOrientations orientations)
+{
+    if (m_lockedOrientations == orientations)
+        return;
+    m_lockedOrientations = orientations;
+    if (!m_orientationLockTimer.isActive())
+        m_orientationLockTimer.startOneShot(0);
+}
+
+void ScreenOrientation::orientationLockTimerFired(Timer<ScreenOrientation>*)
+{
+    if (m_lockedOrientations == WebScreenOrientationDefault)
+        blink::Platform::current()->unlockOrientation();
+    else
+        blink::Platform::current()->lockOrientation(m_lockedOrientations);
+}
+
 const char* ScreenOrientation::supplementName()
 {
     return "ScreenOrientation";
 }
 
 Document& ScreenOrientation::document() const
 {
     ASSERT(m_associatedDOMWindow);
     ASSERT(m_associatedDOMWindow->document());
     return *m_associatedDOMWindow->document();
@@ skipping 13 matching lines @@
 {
 }
 
 const AtomicString& ScreenOrientation::orientation(Screen& screen)
 {
     ScreenOrientation& screenOrientation = ScreenOrientation::from(screen);
     ScreenOrientationController& controller = ScreenOrientationController::from(screenOrientation.document());
     return orientationToString(controller.orientation());
 }
 
-bool ScreenOrientation::lockOrientation(Screen& screen, const Vector<String>& orientations)
+bool ScreenOrientation::lockOrientation(Screen& screen, const Vector<String>& orientationsVector)
 {
-    // FIXME: Implement.
-    return false;
+    blink::WebScreenOrientations orientations = 0;
+    for (size_t i = 0; i < orientationsVector.size(); ++i) {
+        blink::WebScreenOrientations currentOrientation = stringToOrientations(AtomicString(orientationsVector[i]));
+        if (!currentOrientation)
+            return false;
+        orientations |= currentOrientation;
+    }
+    if (!orientations)
+        return false;
+    ScreenOrientation::from(screen).lockOrientationAsync(orientations);
+    return true;
 }
 
-bool ScreenOrientation::lockOrientation(Screen& screen, const AtomicString& orientation)
+bool ScreenOrientation::lockOrientation(Screen& screen, const AtomicString& orientationString)
 {
-    // FIXME: Implement.
-    return false;
+    blink::WebScreenOrientations orientations = stringToOrientations(orientationString);
+    if (!orientations)
+        return false;
+    ScreenOrientation::from(screen).lockOrientationAsync(orientations);
+    return true;
 }
 
 void ScreenOrientation::unlockOrientation(Screen& screen)
 {
-    // FIXME: Implement.
+    ScreenOrientation::from(screen).lockOrientationAsync(WebScreenOrientationDefault);
 }
 
 } // namespace WebCore