import React from 'react';
import type { RadioProps, RadioGroupProps } from 'antd';
import { Radio } from 'antd';
import ProField from '@ant-design/pro-field';
import type { ProSchema } from '@ant-design/pro-utils';
import { runFunction } from '@ant-design/pro-utils';
import createField from '../../BaseForm/createField';
import type { ProFormItemProps } from '../../interface';

export type ProFormRadioGroupProps = ProFormItemProps<RadioGroupProps> & {
  layout?: 'horizontal' | 'vertical';
  radioType?: 'button' | 'radio';
  options?: RadioGroupProps['options'];
  valueEnum?: ProSchema['valueEnum'];
  request?: ProSchema['request'];
};

const RadioGroup: React.FC<ProFormRadioGroupProps> = React.forwardRef(
  ({ fieldProps, options, radioType, layout, proFieldProps, valueEnum, ...rest }, ref: any) => {
    return (
      <ProField
        mode="edit"
        valueType={radioType === 'button' ? 'radioButton' : 'radio'}
        ref={ref}
        valueEnum={runFunction<[any]>(valueEnum, undefined)}
        {...rest}
        fieldProps={{
          options,
          layout,
          ...fieldProps,
        }}
        {...proFieldProps}
      />
    );
  },
);

/**
 * Radio
 *
 * @param
 */
const ProFormRadioComponents: React.FC<ProFormItemProps<RadioProps>> = React.forwardRef(
  ({ fieldProps, children }, ref: any) => {
    return (
      <Radio {...fieldProps} ref={ref}>
        {children}
      </Radio>
    );
  },
);

const ProFormRadio = createField<ProFormItemProps<RadioProps>>(ProFormRadioComponents, {
  valuePropName: 'checked',
  ignoreWidth: true,
});

const Group = createField(RadioGroup, {
  customLightMode: true,
});

const WrappedProFormRadio: typeof ProFormRadio & {
  Group: typeof Group;
  Button: typeof Radio.Button;
} = ProFormRadio as any;

WrappedProFormRadio.Group = Group;

WrappedProFormRadio.Button = Radio.Button;

export default WrappedProFormRadio;