import React from 'react' import useTheme from '../use-theme' import useScale, { withScale } from '../use-scale' import useClasses from '../use-classes' interface Props { count?: number className?: string } const defaultProps = { className: '' } type NativeAttrs = Omit, keyof Props> export type AvatarGroupProps = Props & NativeAttrs const AvatarGroupComponent: React.FC> = ({ count, className, children }: AvatarGroupProps & typeof defaultProps) => { const theme = useTheme() const { SCALES } = useScale() return (
{children} {count && +{count}}
) } AvatarGroupComponent.defaultProps = defaultProps AvatarGroupComponent.displayName = 'HuiAvatarGroup' const AvatarGroup = withScale(AvatarGroupComponent) export default AvatarGroup