puppeteer-autoscroll-down
Version:
Handle infinite scroll on websites with puppeteer
51 lines (43 loc) • 1.79 kB
JavaScript
function scrollPage(scrollDirection) {
return async (page, { delay = 100, size = 250, stepsLimit = null } = {}) => {
let lastScrollPosition = await page.evaluate(
async (pixelsToScroll, delayAfterStep, limit, direction) => {
let getElementScrollHeight = element => {
if (!element) return 0
let { clientHeight, offsetHeight, scrollHeight } = element
return Math.max(scrollHeight, offsetHeight, clientHeight)
}
let initialScrollPosition = window.pageYOffset
let availableScrollHeight = getElementScrollHeight(document.body)
let lastPosition = direction === 'bottom' ? 0 : initialScrollPosition
let scrollFn = resolve => {
let intervalId = setInterval(() => {
window.scrollBy(0, direction === 'bottom' ? pixelsToScroll : -pixelsToScroll)
lastPosition += direction === 'bottom' ? pixelsToScroll : -pixelsToScroll
if (
(direction === 'bottom' && lastPosition >= availableScrollHeight) ||
(direction === 'bottom' &&
limit !== null &&
lastPosition >= pixelsToScroll * limit) ||
(direction === 'top' && lastPosition <= 0) ||
(direction === 'top' &&
limit !== null &&
lastPosition <= initialScrollPosition - pixelsToScroll * limit)
) {
clearInterval(intervalId)
resolve(lastPosition)
}
}, delayAfterStep)
}
return new Promise(scrollFn)
},
size,
delay,
stepsLimit,
scrollDirection
)
return lastScrollPosition
}
}
export const scrollPageToBottom = scrollPage('bottom');
export const scrollPageToTop = scrollPage('top');