import React from 'react';
import DatePicker from '../../DatePicker';
import withFormItem from '../withFormItem';
import { useFormFieldContext } from '../hooks';
import { DateRangeProps } from '../../DatePicker/DateRange';
import { FieldValues } from 'react-hook-form';
import { FormItemProps } from '../Item';

const DateRangeBase = ({ onConfirm, ...props }: DateRangeProps) => {
  const { field } = useFormFieldContext();
  return (
    <DatePicker.DateRange
      {...props}
      onConfirm={(value) => {
        field.onChange(value);
        field.onBlur();
        onConfirm?.(value);
      }}
    />
  );
};

// 使用withFormItem包装，自动处理表单项能力
const FormDateRange = withFormItem(DateRangeBase, 'FormDateRange');

// 导出支持泛型的类型定义
export type FormDateRangeProps<TFieldValues extends FieldValues = FieldValues> =
  Omit<FormItemProps<TFieldValues>, 'children'> & DateRangeProps;

export default FormDateRange;