import React from 'react'
import { DefaultProps } from 'icons/typings'
import Vector from 'atoms/Vector'
import { renderWrapper as defaultRenderWrapper } from '../renderProps'
import GridViewDefaultIcon from './GridViewDefaultIcon'
import GridViewLargeIcon from './GridViewLargeIcon'

// extending interface from Label component
interface Props extends DefaultProps {
  // trying to make it optional?
  type?: string
}

// returning breed based on the breedType
function fromBreedToComponent(type: string) {
  switch (type) {
    case 'large':
      return GridViewLargeIcon
    default:
      return GridViewDefaultIcon
  }
}

class GridViewIcon extends React.PureComponent<Props> {
  static defaultProps = {
    renderWrapper: defaultRenderWrapper,
  }

  render() {
    const { type, renderWrapper, ...remainingProps } = this.props
    const Component = fromBreedToComponent(type)
    const componentView = <Component {...remainingProps} />
    return renderWrapper(componentView)
  }
}

export { GridViewIcon }
export default GridViewIcon