import {
  ScaleBand,
  scaleLinear,
  ScaleLinear,
  ScaleLogarithmic,
  ScalePoint,
  ScaleSymLog,
  scaleSymlog,
  scaleTime,
  ScaleTime,
} from 'd3-scale';

import { IAxes } from '../';

export type AnyScale = ScaleLinear<number, number>
  | ScaleTime<any, any>
  | ScaleLogarithmic<any, any>
  | ScaleSymLog<any, any>
  | ScaleBand<string>
  | ScalePoint<string>;

export const buildScales = (axis: IAxes) => {
  let x: AnyScale;
  let y: AnyScale;

  switch (axis.x.scale) {
    case 'log':
      x = scaleSymlog().clamp(true); // clamp values below 1 to be equal to 0
      break;
    case 'time':
      x = scaleTime();
      break;
    default:
      x = scaleLinear();
      break;
  }

  switch (axis.y.scale) {
    case 'log':
      y = scaleSymlog().clamp(true); // clamp values below 1 to be equal to 0
      break;
    case 'time':
      y = scaleTime();
      break;
    default:
      y = scaleLinear();
      break;
  }
  return [x, y];
};