Conflito ao usar simultaneamente eventos do teclado para rolagem e snap à rolagem CSS

Você pode horizontalmente rolar minha página de demonstração, pressionando Space Bar, Page Up / Page Downe Left Arrow / Right Arrowchaves. Você também pode fazer o snap com o mouse ou o trackpad.

Mas apenas um ou outro funciona.

Existe uma maneira de coexistir eventos de teclado e snap à rolagem CSS? o que estou perdendo? Qualquer ajuda seria muito apreciada, pois estou lutando com esse problema há mais de uma semana.

Confira minha demonstração no CodePen

(Descomente a parte relevante do código CSS para ativar o efeito de encaixe da rolagem para ver se os atalhos do teclado param de funcionar.)

import animate from "https://cdn.jsdelivr.net/npm/animateplus@2/animateplus.js"

const sections = Array.from(document.querySelectorAll("section")).sort(
  (s1, s2) => {
    return s1.getBoundingClientRect().left - s2.getBoundingClientRect().left
  }
)

const getSectionInView = () => {
  const halfWidth = window.innerWidth / 2
  const index = sections.findIndex(
    section =>
      section.getBoundingClientRect().left <= halfWidth &&
      section.getBoundingClientRect().right > halfWidth
  )
  return index
}

const getNextSection = dir => {
  const sectionInViewIndex = getSectionInView()
  const nextIndex = sectionInViewIndex + dir
  const numSections = sections.length
  const nextSectionIndex =
    nextIndex < 0 || nextIndex >= numSections ? sectionInViewIndex : nextIndex
  return sections[nextSectionIndex]
}

const container = document.scrollingElement

const animateScroll = dir => {
  const from = container.scrollLeft
  const { left } = getNextSection(dir).getBoundingClientRect()
  return progress => (container.scrollLeft = from + progress * left)
}

window.onload = () => {
  document.body.onkeydown = event => {
    switch (event.key) {
      case " ": // Space Bar
      case "PageDown":
      case "ArrowRight": {      
        animate({
          easing: "out-quintic",
          change: animateScroll(1)
        })
        break
      }
      case "PageUp":
      case "ArrowLeft":  {      
        animate({
          easing: "out-quintic",
          change: animateScroll(-1)
        })
        break
      }
    }
  }
}

Nota: Estou usando um módulo pequeno e elegante chamado Animate Plus para obter a animação de rolagem suave.

Atualização: a solução da @ Kostja funciona no Chrome, mas não no Safari para Mac ou iOS, e é crucial para mim que funcione no Safari.

Eu acho que não há, o css substitui o javascript. Mas você pode simplesmente adicionar o listener de eventos do wheel como:

window.addEventListener("wheel", function() {
    if(event.deltaY > 0){
      animate({
        easing: "out-quintic",
        change: animateScroll(1)
      })      
    }
      if(event.deltaY < 0){
      animate({
        easing: "out-quintic",
        change: animateScroll(-1)
      })      
    }      
});

https://codepen.io/kostjaaa/pen/NWWVBKd

window.onload = () => {
  window.addEventListener("wheel", () => {
    const direction = event.deltaY > 0 ? 1 : event.deltaY < 0 ? -1 : false;

    if (direction) {
      animate({
        easing: "out-quintic",
        change: animateScroll(direction)
      });
    }
  });

  document.body.onkeydown = event => {
    switch (event.key) {
      case " ": // Space Bar
      case "PageDown":
      case "ArrowRight": {
        animate({
          easing: "out-quintic",
          change: animateScroll(1)
        });
        break;
      }
      case "PageUp":
      case "ArrowLeft": {
        animate({
          easing: "out-quintic",
          change: animateScroll(-1)
        });
        break;
      }
    }
  };
};    

Isso deve funcionar.

https://codepen.io/JZ6/pen/XWWQqRK