hyper-tuner/hyper-tuner-cloud

import { CheckOutlined, CloseOutlined } from '@ant-design/icons';
import { Switches } from '@hyper-tuner/types';
import { Radio, Select, Switch } from 'antd';

const SmartSelect = ({
  values,
  defaultValue,
  disabled,
}: {
  values: string[];
  defaultValue: string;
  disabled: boolean;
}) => {
  if (
    values.length === 2 &&
    ((values.includes(Switches.YES) && values.includes(Switches.NO)) ||
      (values.includes(Switches.ON) && values.includes(Switches.OFF)))
  ) {
    return (
      <Switch
        defaultChecked={[Switches.ON, Switches.YES].includes(defaultValue as Switches)}
        checkedChildren={<CheckOutlined />}
        unCheckedChildren={<CloseOutlined />}
      />
    );
  }

  if (values.length < 3) {
    return (
      <Radio.Group
        value={values.indexOf(defaultValue)}
        optionType="button"
        buttonStyle="solid"
        disabled={disabled}
      >
        {values.map((val: string, index) => (
          <Radio key={val} value={index}>
            {val}
          </Radio>
        ))}
      </Radio.Group>
    );
  }

  return (
    <Select
      value={values.indexOf(defaultValue)}
      optionFilterProp="label"
      disabled={disabled}
      style={{ maxWidth: 250 }}
    >
      {/* we need to preserve indexes here, skip INVALID option */}
      {values.map((val: string, index) => {
        return val === 'INVALID' ? null : (
          <Select.Option key={val} value={index} label={val}>
            {val}
          </Select.Option>
        );
      })}
    </Select>
  );
};

export default SmartSelect;