import { translateRotation, translateZeroRotation } from '@plugin/translators';
import { applyInverseRotation, hasRotation } from '@plugin/utils';

import { ShapeBaseAttributes, ShapeGeomAttributes } from '@ui/lib/types/shapes/shape';

export const transformRotation = (
  node: LayoutMixin,
  baseRotation: number
): Pick<ShapeBaseAttributes, 'transform' | 'transformInverse' | 'rotation'> => {
  const rotation = node.rotation + baseRotation;

  if (!hasRotation(rotation)) {
    return translateZeroRotation();
  }

  return translateRotation(node.absoluteTransform, rotation);
};

export const transformRotationAndPosition = (
  node: LayoutMixin,
  baseRotation: number
): Pick<ShapeBaseAttributes, 'transform' | 'transformInverse' | 'rotation'> &
  Pick<ShapeGeomAttributes, 'x' | 'y'> => {
  const rotation = node.rotation + baseRotation;
  const x = node.absoluteTransform[0][2];
  const y = node.absoluteTransform[1][2];

  if (!hasRotation(rotation) || !node.absoluteBoundingBox) {
    return {
      x,
      y,
      ...translateZeroRotation()
    };
  }

  const referencePoint = applyInverseRotation(
    { x, y },
    node.absoluteTransform,
    node.absoluteBoundingBox
  );

  return {
    ...referencePoint,
    ...translateRotation(node.absoluteTransform, rotation)
  };
};