All files / src/DropdownMenu DropdownMenu.tsx

100% Statements 13/13
100% Branches 0/0
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 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78                                      1x               1x   85x 85x   85x   85x         85x               85x   85x                               1x   85x 85x                  
import * as React from 'react';
import { Box as ReakitBox } from 'reakit';
 
import { useClassName, createComponent, createElement, createHook } from '../utils';
import { Box, BoxProps } from '../Box';
 
import { DropdownMenuButton } from './DropdownMenuButton';
import { DropdownMenuPopover } from './DropdownMenuPopover';
import { useDropdownMenuState, DropdownMenuInitialState, DropdownMenuStateReturn } from './DropdownMenuState';
import * as styles from './styles';
 
export type LocalDropdownMenuProps = {
  baseId?: DropdownMenuInitialState['baseId'];
  children: React.ReactElement<any>;
  content: any;
  visible?: DropdownMenuInitialState['visible'];
};
export type DropdownMenuProps = BoxProps & LocalDropdownMenuProps;
 
export const DropdownMenuContext = React.createContext<{
  dropdownMenu: Partial<DropdownMenuStateReturn>;
  overrides: any;
}>({
  dropdownMenu: {},
  overrides: {}
});
 
const useProps = createHook<DropdownMenuProps>(
  (props, { themeKey, themeKeyOverride }) => {
    const { baseId, children, content, overrides, visible, ...restProps } = props;
    const boxProps = Box.useProps(restProps);
 
    const dropdownMenu = useDropdownMenuState({ baseId, visible });
 
    const dropdownMenuButtonProps = DropdownMenuButton.useProps({
      ...dropdownMenu,
      overrides
    });
 
    const className = useClassName({
      style: styles.DropdownMenu,
      styleProps: props,
      themeKey,
      themeKeyOverride,
      prevClassName: boxProps.className
    });
 
    const contextValue = React.useMemo(() => ({ dropdownMenu, overrides }), [dropdownMenu, overrides]);
 
    return {
      ...boxProps,
      className,
      children: (
        <DropdownMenuContext.Provider value={contextValue}>
          {React.cloneElement(children, { ...dropdownMenuButtonProps })}
          <DropdownMenuPopover {...dropdownMenu} overrides={overrides}>
            {content}
          </DropdownMenuPopover>
        </DropdownMenuContext.Provider>
      )
    };
  },
  { themeKey: 'DropdownMenu' }
);
 
export const DropdownMenu = createComponent<DropdownMenuProps>(
  props => {
    const textProps = useProps(props);
    return createElement({ children: props.children, component: ReakitBox, use: props.use, htmlProps: textProps });
  },
  {
    attach: {
      useProps
    },
    themeKey: 'DropdownMenu'
  }
);