duncanleung / duncanleung.com Goto Github PK

import React from 'react';
import { Button as ChakraButton } from '@chakra-ui/core';

type GetRenderComponentProps<T> = T extends React.ComponentType
  ? React.ComponentProps<T>
  : T extends 'a'
  ? React.HTMLProps<T>
  : {};

type RenderComponent = React.ComponentType | 'a';

type Props<E extends RenderComponent> = {
  as?: E;
} & GetRenderComponentProps<E>;

const Button = <T extends RenderComponent>({
  as,
  children,  // <-- error
  ...props
}: Props<T>): ReturnType<React.FC<Props<T>>> => {
  return (
    <ChakraButton as={as} {...props}>
      {children}
    </ChakraButton>
  );
};

export default Button;