import { useInjectRange } from '../../RangeContext'
import type { GenerateConfig } from '../../generate'
import useCellClassName from '../../hooks/useCellClassName'
import useMergeProps from '../../hooks/useMergeProps'
import type { Locale } from '../../interface'
import { formatValue, isSameQuarter } from '../../utils/dateUtil'
import PanelBody from '../PanelBody'

export const QUARTER_COL_COUNT = 4
const QUARTER_ROW_COUNT = 1

export interface QuarterBodyProps<DateType> {
  prefixCls: string
  locale: Locale
  generateConfig: GenerateConfig<DateType>
  value?: DateType | null
  viewDate: DateType
  disabledDate?: (date: DateType) => boolean
  onSelect: (value: DateType) => void
}

function QuarterBody<DateType>(_props: QuarterBodyProps<DateType>) {
  const props = useMergeProps(_props)
  const { prefixCls, locale, value, viewDate, generateConfig } = props

  const { rangedValue, hoverRangedValue } = useInjectRange()

  const cellPrefixCls = `${prefixCls}-cell`

  const getCellClassName = useCellClassName({
    cellPrefixCls,
    value,
    generateConfig,
    rangedValue: rangedValue.value,
    hoverRangedValue: hoverRangedValue.value,
    isSameCell: (current, target) => isSameQuarter(generateConfig, current, target),
    isInView: () => true,
    offsetCell: (date, offset) => generateConfig.addMonth(date, offset * 3),
  })

  const baseQuarter = generateConfig.setDate(generateConfig.setMonth(viewDate, 0), 1)

  return (
    <PanelBody
      {...props}
      rowNum={QUARTER_ROW_COUNT}
      colNum={QUARTER_COL_COUNT}
      baseDate={baseQuarter}
      getCellText={(date) =>
        formatValue(date, {
          locale,
          format: locale.quarterFormat || '[Q]Q',
          generateConfig,
        })
      }
      getCellClassName={getCellClassName}
      getCellDate={(date, offset) => generateConfig.addMonth(date, offset * 3)}
      titleCell={(date) =>
        formatValue(date, {
          locale,
          format: 'YYYY-[Q]Q',
          generateConfig,
        })
      }
    />
  )
}

QuarterBody.displayName = 'QuarterBody'
QuarterBody.inheritAttrs = false
export default QuarterBody
