import { CheckSmIcon } from '../icons/check-sm-icon';
import { CheckMdIcon } from '../icons/check-md-icon';
import { CheckLgIcon } from '../icons/check-lg-icon';
import { IntermediateSmIcon } from '../icons/intermediate-sm-icon';
import { IntermediateMdIcon } from '../icons/intermediate-md-icon';
import { IntermediateLgIcon } from '../icons/intermediate-lg-icon';
import { HoversSmIcon } from '../icons/hover-sm-icon';
import { HoversMdIcon } from '../icons/hover-md-icon';
import { HoversLgIcon } from '../icons/hover-lg-icon';
import { RtCheckboxSize } from '../rt-checkbox.types';

const icons = {
  checked: {
    sm: CheckSmIcon,
    md: CheckMdIcon,
    lg: CheckLgIcon,
  },
  intermediate: {
    sm: IntermediateSmIcon,
    md: IntermediateMdIcon,
    lg: IntermediateLgIcon,
  },
  hover: {
    sm: HoversSmIcon,
    md: HoversMdIcon,
    lg: HoversLgIcon,
  },
};

export const getIcon = (
  type: 'checked' | 'intermediate' | 'hover',
  size: RtCheckboxSize
): React.FC => {
  return icons[type][size];
};