metamask-extension/ui/components/component-library/badge-wrapper/badge-wrapper.tsx

import React from 'react';
import classnames from 'classnames';

import { Display } from '../../../helpers/constants/design-system';
import { Box } from '..';
import type { PolymorphicRef } from '../box';

import {
  BadgeWrapperPosition,
  BadgeWrapperAnchorElementShape,
  BadgeWrapperProps,
  BadgeWrapperComponent,
} from './badge-wrapper.types';

export const BadgeWrapper: BadgeWrapperComponent = React.forwardRef(
  <C extends React.ElementType = 'div'>(
    {
      children,
      badge,
      badgeContainerProps,
      position = BadgeWrapperPosition.topRight,
      positionObj,
      anchorElementShape = BadgeWrapperAnchorElementShape.circular,
      className = '',
      ...props
    }: BadgeWrapperProps<C>,
    ref?: PolymorphicRef<C>,
  ) => (
    <Box
      className={classnames('mm-badge-wrapper', className)}
      ref={ref}
      display={Display.InlineBlock}
      {...props} // TODO: There is a typing issue with spreading props to the Box component. It still works but TypeScript complains.
    >
      {/* Generally the AvatarAccount or AvatarToken */}
      {children}
      <Box
        className={classnames('mm-badge-wrapper__badge-container', {
          [`mm-badge-wrapper__badge-container--${anchorElementShape}-${position}`]:
            !positionObj,
        })}
        style={{ ...positionObj }}
        {...badgeContainerProps}
      >
        {/* Generally the AvatarNetwork at SIZES.XS */}
        {badge}
      </Box>
    </Box>
  ),
);