import { Extrapolation, interpolate } from 'react-native-reanimated';
import type { GalleryTransitionCallback } from '../../components/gallery/types';

export const stackTransition: GalleryTransitionCallback = (options) => {
  'worklet';

  const { index, activeIndex, vertical, gallerySize, scroll, isScrolling } =
    options;

  if (vertical) {
    const translateY = index * gallerySize.height - scroll;
    return { transform: [{ translateY }] };
  }

  if (index < activeIndex - 1 || index > activeIndex + 1) {
    return {
      opacity: 0,
      transform: [{ translateX: 0 }, { scale: 0 }],
    };
  }

  let translateX = index * gallerySize.width - scroll;
  let opacity = 1;
  let scale = 1;
  if (index !== activeIndex && !isScrolling) scale = 0;

  const isCurrent = index === activeIndex;
  const isNext = index === activeIndex + 1;
  if (isNext || (isCurrent && scroll < index * gallerySize.width)) {
    opacity = interpolate(
      scroll,
      [(index - 1) * gallerySize.width, index * gallerySize.width],
      [0, 1],
      Extrapolation.CLAMP
    );

    scale = 0.75 + 0.25 * opacity;
    translateX = 0;
  }

  return {
    opacity,
    transform: [{ translateX }, { scale }],
  };
};