import {
  SharedValue,
  interpolate,
  useDerivedValue,
} from "react-native-reanimated";
import { useGetBoneDimensions } from "../worklets/useGetBoneDimensions";
import { useGetPositionRange } from "../worklets/useGetPositionRange";
import type {
  ICustomViewStyle,
  IComponentSize,
  ISkeletonProps,
} from "../constants";

type UseGetGradientTransformProps = Pick<
  ISkeletonProps,
  "animationDirection"
> & {
  componentSize: IComponentSize;
  boneLayout: ICustomViewStyle;
  animationValue: SharedValue<number>;
};

/**
 * Provides the animation values for the gradient transform
 * @componentSize is the size of the component
 * @boneLayout is the layout of the bone
 * @animationDirection is the direction of the animation
 * @animationValue is the value of the animation
 */
export const useGetGradientTransform = ({
  componentSize,
  boneLayout,
  animationDirection,
  animationValue,
}: UseGetGradientTransformProps): SharedValue<{
  translateX?: number;
  translateY?: number;
  rotate?: string;
}> => {
  const getBoneDimensions = useGetBoneDimensions(componentSize);
  const getPositionRange = useGetPositionRange(componentSize);

  return useDerivedValue(() => {
    let transform = {};
    const { width, height } = getBoneDimensions(boneLayout);

    if (
      animationDirection === "verticalTop" ||
      animationDirection === "verticalDown" ||
      animationDirection === "horizontalLeft" ||
      animationDirection === "horizontalRight"
    ) {
      const interpolatedPosition = interpolate(
        animationValue.value,
        [0, 1],
        getPositionRange({ animationDirection, boneLayout }),
      );

      if (
        animationDirection === "verticalTop" ||
        animationDirection === "verticalDown"
      ) {
        transform = { translateY: interpolatedPosition };
      } else {
        transform = { translateX: interpolatedPosition };
      }
    } else if (
      animationDirection === "diagonalDownRight" ||
      animationDirection === "diagonalTopRight" ||
      animationDirection === "diagonalDownLeft" ||
      animationDirection === "diagonalTopLeft"
    ) {
      const diagonal = Math.sqrt(height * height + width * width);
      const mainDimension = Math.max(height, width);
      const oppositeDimension = mainDimension === width ? height : width;
      const diagonalAngle = Math.acos(mainDimension / diagonal);
      let rotateAngle =
        animationDirection === "diagonalDownRight" ||
        animationDirection === "diagonalTopLeft"
          ? Math.PI / 2 - diagonalAngle
          : Math.PI / 2 + diagonalAngle;
      const additionalRotate =
        animationDirection === "diagonalDownRight" ||
        animationDirection === "diagonalTopLeft"
          ? 2 * diagonalAngle
          : -2 * diagonalAngle;
      const distanceFactor = (diagonal + oppositeDimension) / 2;
      if (mainDimension === width && width !== height)
        rotateAngle += additionalRotate;
      const sinComponent = Math.sin(diagonalAngle) * distanceFactor;
      const cosComponent = Math.cos(diagonalAngle) * distanceFactor;
      let xOutputRange: number[];
      let yOutputRange: number[];
      if (
        animationDirection === "diagonalDownRight" ||
        animationDirection === "diagonalTopLeft"
      ) {
        xOutputRange =
          animationDirection === "diagonalDownRight"
            ? [-sinComponent, sinComponent]
            : [sinComponent, -sinComponent];
        yOutputRange =
          animationDirection === "diagonalDownRight"
            ? [-cosComponent, cosComponent]
            : [cosComponent, -cosComponent];
      } else {
        xOutputRange =
          animationDirection === "diagonalDownLeft"
            ? [-sinComponent, sinComponent]
            : [sinComponent, -sinComponent];
        yOutputRange =
          animationDirection === "diagonalDownLeft"
            ? [cosComponent, -cosComponent]
            : [-cosComponent, cosComponent];
        if (mainDimension === height && width !== height) {
          xOutputRange.reverse();
          yOutputRange.reverse();
        }
      }

      let translateX = interpolate(animationValue.value, [0, 1], xOutputRange);
      let translateY = interpolate(animationValue.value, [0, 1], yOutputRange);

      // swapping the translates if width is the main dim
      if (mainDimension === width) {
        [translateX, translateY] = [translateY, translateX];
      }

      const rotate = `${rotateAngle}rad`;
      transform = { translateX, translateY, rotate };
    }

    return transform;
  });
};