import { useCallback } from 'react';
import { useGetBoneDimensions } from './useGetBoneDimensions';
import type {
  ISkeletonProps,
  ICustomViewStyle,
  IComponentSize,
} from '../constants';

type UseGetGradientEndDirectionProps = Pick<
  ISkeletonProps,
  'animationType' | 'animationDirection'
> & {
  boneLayout: ICustomViewStyle;
};

export const useGetGradientEndDirection = (componentSize: IComponentSize) => {
  const getBoneDimensions = useGetBoneDimensions(componentSize);

  return useCallback(
    ({
      animationType,
      animationDirection,
      boneLayout,
    }: UseGetGradientEndDirectionProps) => {
      let direction = { x: 0, y: 0 };

      if (animationType === 'shiver') {
        switch (animationDirection) {
          case 'horizontalLeft':
          case 'horizontalRight':
            direction = { x: 1, y: 0 };
            break;
          case 'verticalTop':
          case 'verticalDown':
            direction = { x: 0, y: 1 };
            break;
          case 'diagonalTopRight':
          case 'diagonalDownRight':
          case 'diagonalDownLeft':
          case 'diagonalTopLeft': {
            const { height, width } = getBoneDimensions(boneLayout);
            return width && height && width > height
              ? { x: 0, y: 1 }
              : { x: 1, y: 0 };
          }
        }
      }

      return direction;
    },
    [getBoneDimensions],
  );
};