All files / src/Badge Badge.tsx

100% Statements 12/12
100% Branches 4/4
100% Functions 3/3
100% Lines 12/12

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56                                1x   26x 26x   26x               26x 26x   25x 25x       26x         1x   25x 25x                    
import * as React from 'react';
import { Box as ReakitBox } from 'reakit';
 
import { Size } from '../types';
import { useClassName, createComponent, createElement, createHook } from '../utils';
import { Box, BoxProps } from '../Box';
 
import * as styles from './styles';
 
export type LocalBadgeProps = {
  isAttached?: boolean;
  palette?: string;
  size?: Size;
};
export type BadgeProps = BoxProps & LocalBadgeProps;
 
const useProps = createHook<BadgeProps>(
  (props, { themeKey, themeKeyOverride }) => {
    const ref = React.useRef();
    const boxProps = Box.useProps({ ...props, ref });
 
    const className = useClassName({
      style: styles.Badge,
      styleProps: props,
      themeKey,
      themeKeyOverride,
      prevClassName: boxProps.className
    });
 
    React.useEffect(() => {
      if (ref && ref.current) {
        // @ts-ignore
        const parentElement = ref.current.parentElement;
        parentElement.setAttribute('style', 'position:relative;');
      }
    }, []);
 
    return { ...boxProps, className };
  },
  { defaultProps: { palette: 'primary' }, themeKey: 'Badge' }
);
 
export const Badge = createComponent<BadgeProps>(
  props => {
    const badgeProps = useProps(props);
    return createElement({ children: props.children, component: ReakitBox, use: props.use, htmlProps: badgeProps });
  },
  {
    attach: {
      useProps
    },
    defaultProps: {},
    themeKey: 'Badge'
  }
);