Merge branch 'dpogue-scrollingElement'

1 files changed, 18 insertions, 11 deletions

diff --git a/content_scripts/scroller.coffee b/content_scripts/scroller.coffee
index 56b9a6c6..3a1b3772 100644
--- a/content_scripts/scroller.coffee
+++ b/content_scripts/scroller.coffee
@@ -4,6 +4,12 @@
 #
 activatedElement = null
 
+# Previously, the main scrolling element was document.body.  If the "experimental web platform features" flag
+# is enabled, then we need to use document.scrollingElement instead.  There's an explanation in #2168:
+# https://github.com/philc/vimium/pull/2168#issuecomment-236488091
+
+getScrollingElement = -> document.scrollingElement ? document.body
+
 # Return 0, -1 or 1: the sign of the argument.
 # NOTE(smblott; 2014/12/17) We would like to use Math.sign().  However, according to this site
 # (https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Math/sign) Math.sign() was
@@ -37,7 +43,7 @@ getDimension = (el, direction, amount) ->
     name = amount
     # the clientSizes of the body are the dimensions of the entire page, but the viewport should only be the
     # part visible through the window
-    if name is 'viewSize' and el is document.body
+    if name is 'viewSize' and el is getScrollingElement()
       # TODO(smblott) Should we not be returning the width/height of element, here?
       if direction is 'x' then window.innerWidth else window.innerHeight
     else
@@ -82,13 +88,14 @@ isScrollableElement = (element, direction = "y", amount = 1, factor = 1) ->
 
 # From element and its parents, find the first which we should scroll and which does scroll.
 findScrollableElement = (element, direction, amount, factor) ->
-  while element != document.body and not isScrollableElement element, direction, amount, factor
-    element = DomUtils.getContainingElement(element) ? document.body
+  while element != getScrollingElement() and not isScrollableElement element, direction, amount, factor
+    element = DomUtils.getContainingElement(element) ? getScrollingElement()
   element
 
-# On some pages, document.body is not scrollable.  Here, we search the document for the largest visible
-# element which does scroll vertically. This is used to initialize activatedElement. See #1358.
-firstScrollableElement = (element=document.body) ->
+# On some pages, the scrolling element is not actually scrollable.  Here, we search the document for the
+# largest visible element which does scroll vertically. This is used to initialize activatedElement. See
+# #1358.
+firstScrollableElement = (element=getScrollingElement()) ->
   if doesScroll(element, "y", 1, 1) or doesScroll(element, "y", -1, 1)
     element
   else
@@ -234,14 +241,14 @@ Scroller =
   # :factor is needed because :amount can take on string values, which scrollBy converts to element dimensions.
   scrollBy: (direction, amount, factor = 1, continuous = true) ->
     # if this is called before domReady, just use the window scroll function
-    if (!document.body and amount instanceof Number)
+    if (!getScrollingElement() and amount instanceof Number)
       if (direction == "x")
         window.scrollBy(amount, 0)
       else
         window.scrollBy(0, amount)
       return
 
-    activatedElement ||= (document.body and firstScrollableElement()) or document.body
+    activatedElement ||= (getScrollingElement() and firstScrollableElement()) or getScrollingElement()
     return unless activatedElement
 
     # Avoid the expensive scroll calculation if it will not be used.  This reduces costs during smooth,
@@ -252,7 +259,7 @@ Scroller =
       CoreScroller.scroll element, direction, elementAmount, continuous
 
   scrollTo: (direction, pos) ->
-    activatedElement ||= (document.body and firstScrollableElement()) or document.body
+    activatedElement ||= (getScrollingElement() and firstScrollableElement()) or getScrollingElement()
     return unless activatedElement
 
     element = findScrollableElement activatedElement, direction, pos, 1
@@ -261,13 +268,13 @@ Scroller =
 
   # Is element scrollable and not the activated element?
   isScrollableElement: (element) ->
-    activatedElement ||= (document.body and firstScrollableElement()) or document.body
+    activatedElement ||= (getScrollingElement() and firstScrollableElement()) or getScrollingElement()
     element != activatedElement and isScrollableElement element
 
   # Scroll the top, bottom, left and right of element into view.  The is used by visual mode to ensure the
   # focus remains visible.
   scrollIntoView: (element) ->
-    activatedElement ||= document.body and firstScrollableElement()
+    activatedElement ||= getScrollingElement() and firstScrollableElement()
     rect = element. getClientRects()?[0]
     if rect?
       # Scroll y axis.