import { connect } from '@formily/react-schema-renderer'
import { DatePicker as NextDatePicker } from '@alifd/next'
import { mapStyledProps, mapTextComponent, compose, isStr } from '../shared'
import moment from 'moment'

const transformMoment = (value, format = 'YYYY-MM-DD HH:mm:ss') => {
  return value && value.format ? value.format(format) : value
}

// showTime可以是非单纯的布尔值，里面还可以设置时间的format
const getFormatFromProps = props => {
  if (props.showTime) {
    if (typeof props.showTime === 'boolean') {
      return props.format || 'YYYY-MM-DD HH:mm:ss'
    }
  }
  return props.format || 'YYYY-MM-DD'
}

export const DatePicker = connect<
  'RangePicker' | 'WeekPicker' | 'MonthPicker' | 'YearPicker'
>({
  getValueFromEvent(value) {
    return transformMoment(value, getFormatFromProps(this.props))
  },
  getProps: mapStyledProps,
  getComponent: mapTextComponent
})(NextDatePicker)

DatePicker.RangePicker = connect({
  getValueFromEvent([startDate, endDate]) {
    const format = getFormatFromProps(this.props)
    return [
      transformMoment(startDate, format),
      transformMoment(endDate, format)
    ]
  },
  getProps: mapStyledProps,
  getComponent: mapTextComponent
})(NextDatePicker.RangePicker)

DatePicker.WeekPicker = connect({
  getValueFromEvent(value) {
    return transformMoment(value, 'gggg-wo')
  },
  getProps: compose(mapStyledProps, (props, fieldProps) => {
    if (!fieldProps.editable) return props
    if (isStr(props.value) && props.value) {
      const parsed = props.value.match(/\D*(\d+)\D*(\d+)\D*/) || ['', '', '']
      props.value = moment(parsed[1], 'YYYY').add(parsed[2] - 1, 'weeks')
    }
    return props
  }),
  getComponent: mapTextComponent
})(NextDatePicker.WeekPicker)

DatePicker.MonthPicker = connect({
  getValueFromEvent(value) {
    return transformMoment(value, 'YYYY-MM')
  },
  getProps: mapStyledProps,
  getComponent: mapTextComponent
})(NextDatePicker.MonthPicker)

DatePicker.YearPicker = connect({
  getValueFromEvent(value) {
    return transformMoment(value, 'YYYY')
  },
  getProps: mapStyledProps,
  getComponent: mapTextComponent
})(NextDatePicker.YearPicker)

export default DatePicker