export type InterpolationConfig = {
  /**
   * The start value of the input range.
   * @default 0
   */
  startValue?: number;
  /**
   * The end value of the input range.
   * @default 1
   */
  endValue?: number;
  /**
   * The number of iterations to perform.
   * @default 50
   */
  totalIterations?: number;
};

/**
 * Creates an input and output range based on the logic of the callback function.
 * @param callback The function used to create the output range for the interpolation.
 * @param interpolationConfig The configuration for the input range and the iterations to perform.
 * @returns The object containing the inputRange and the outputRange for the interpolation
 * @example
 * ```ts
 * const callback = (value: number) => value * 2;
 * const interpolationConfig = {
 *  startValue: 0,
 * endValue: 10,
 * totalIterations: 5,
 * };
 *
 * const result = withFunction(callback, interpolationConfig);
 *
 * console.log(result.inputRange); // [0, 2, 4, 6, 8, 10]
 * console.log(result.outputRange); // [0, 4, 8, 12, 16, 20]
 */
export const withFunction = <T extends number | string>(
  callback: (value: number) => T,
  interpolationConfig?: InterpolationConfig
) => {
  const {
    startValue = 0,
    endValue = 1,
    totalIterations = 50,
  } = interpolationConfig ?? {};

  if (totalIterations < 0)
    throw new Error('totalIterations cannot be negative');

  const inputRange: number[] = [];
  const outputRange: T[] = [];

  if (totalIterations === 0) return { inputRange, outputRange };

  const step = (index: number) =>
    ((endValue - startValue) * index) / totalIterations;

  for (let i = 0; i <= totalIterations; i += 1) {
    const key = startValue + step(i);
    inputRange.push(key);
    outputRange.push(callback(key));
  }
  return { inputRange, outputRange };
};