import React from 'react';
import { MenuItemProps } from 'antd/lib/menu/MenuItem';
import ModalTrigger from './ModalTrigger';
import { Menu } from 'antd';

export function MenuItemModelTrigger(
  props: MenuItemProps & {
    modalProps?: Omit<React.ComponentProps<typeof ModalTrigger>, 'children'>;
  },
) {
  const { onClick: propOnClick, modalProps = {}, ...restProps } = props;
  return (
    <ModalTrigger {...modalProps}>
      {({ open }) => {
        return (
          <Menu.Item
            onClick={(info) => {
              propOnClick && propOnClick(info);
              open();
            }}
            {...restProps}
          />
        );
      }}
    </ModalTrigger>
  );
}