import {getTextMetrics} from './textMetrics'
import {maxBy} from './extrema'

import {
  TICK_PADDING_TOP,
  TICK_PADDING_RIGHT,
  AXIS_LABEL_PADDING_BOTTOM,
} from '../constants'

export const getMargins = (
  showAxes: boolean,
  xAxisLabel: string,
  yAxisLabel: string,
  yTicks: Array<string | number>,
  yTickFormatter: (tick: number | string) => string,
  tickFont: string
) => {
  if (!showAxes) {
    return {top: 1, right: 1, bottom: 1, left: 1}
  }

  const formattedYTicks = yTicks.map(t => yTickFormatter(t))
  const longestYTick = maxBy(d => d.length, formattedYTicks)

  const {width: maxTextWidth, height: textHeight} = getTextMetrics(
    tickFont,
    longestYTick
  )

  const xAxisLabelHeight = xAxisLabel
    ? textHeight + AXIS_LABEL_PADDING_BOTTOM
    : 0

  const yAxisLabelHeight = yAxisLabel
    ? textHeight + AXIS_LABEL_PADDING_BOTTOM
    : 0

  return {
    top: textHeight / 2,
    right: 1,
    bottom: textHeight + TICK_PADDING_TOP + xAxisLabelHeight,
    left: maxTextWidth + TICK_PADDING_RIGHT + yAxisLabelHeight,
  }
}