import React from 'react';

import { ChainIconStyled } from './ChainIcon.styled';
import { Icon, IconProps } from './Icon';

export type ChainIconProps = Omit<IconProps, 'chainId'> & {
  chainId: number;
  'data-testid'?: string;
  size?: number;
};

export const ChainIcon: React.FunctionComponent<ChainIconProps> = ({ size, ...props }) => {
  if (!size) {
    return <Icon {...(props as IconProps)} />;
  }
  return <ChainIconStyled size={size} {...(props as IconProps)} />;
};