import { IconButtonTypeMap as MuiIconButtonTypeMap } from '@material-ui/core/IconButton'
import { ForwardRefLink } from '../compat/ForwardRefLink'
import { ExtendButtonBaseTypeMap } from '@material-ui/core/ButtonBase'
import { OverrideProps } from '@material-ui/core/OverridableComponent'

export type IconButtonTypeMap<
  P = {},
  D extends React.ElementType = MuiIconButtonTypeMap['defaultComponent']
> = ExtendButtonBaseTypeMap<{
  props: P &
    MuiIconButtonTypeMap['props'] & {
      (custom props...)
    }
  defaultComponent: D
  classKey: MuiIconButtonTypeMap['classKey']
}>

export type IconButtonPropsWithTo<P = {}> = OverrideProps<
  IconButtonTypeMap<P & { to: LinkProps['to'] }, typeof ForwardRefLink>,
  typeof ForwardRefLink
>

export type IconButtonPropsWithoutTo<P = {}> = OverrideProps<
  IconButtonTypeMap<P>,
  IconButtonTypeMap['defaultComponent']
>

export type IconButtonProps<P = {}> = IconButtonPropsWithoutTo<P & { to?: undefined }> | IconButtonPropsWithTo<P>

function IconButton(props: IconButtonPropsWithTo): JSX.Element
function IconButton(props: IconButtonPropsWithoutTo): JSX.Element
function IconButton(props: IconButtonProps) {
  ...
}

type IconButtonProps<P> = P extends {to: string} ? A : B;
// more variants
type X<P> = 
  P extends X1 ? A :
  P extends X2 ? B :
  P extends X3 ? C :
  D