import React from 'react';
import type { AriaPopoverProps } from 'react-aria';

import type { Complete, Exact, SafePick } from '@shoptet/utils';
import { destruct } from '@shoptet/utils';

import { Menu } from '../Menu/Menu';
import type { MenuProps } from '../Menu/Menu';
import type { MenuListProps } from '../Menu/MenuList';

import { ContextMenuIconButton, type ContextMenuIconButtonProps } from './ContextMenuIconButton';

export interface ContextMenuInheritedMenuTriggerProps extends SafePick<
  MenuProps,
  'open' | 'defaultOpen' | 'onOpenChange' | 'disabled'
> {}
export interface ContextMenuInheritedContextMenuIconButtonProps extends Omit<ContextMenuIconButtonProps, 'disabled'> {}
export interface ContextMenuInheritedContextMenuListProps extends MenuListProps {}

export interface ContextMenuProps
  extends
    ContextMenuInheritedMenuTriggerProps,
    ContextMenuInheritedContextMenuIconButtonProps,
    ContextMenuInheritedContextMenuListProps {
  placement?: AriaPopoverProps['placement'];
}

export const ContextMenu = Object.assign(
  function ContextMenu({ placement, ...rest }: ContextMenuProps) {
    const [menuOpenProps, menuProps, iconButtonProps] = destruct(
      rest,
      ['open', 'defaultOpen', 'onOpenChange', 'disabled'],
      ['children']
    );

    menuOpenProps satisfies Exact<Complete<ContextMenuInheritedMenuTriggerProps>, typeof menuOpenProps>;
    iconButtonProps satisfies Exact<Complete<ContextMenuInheritedContextMenuIconButtonProps>, typeof iconButtonProps>;
    menuProps satisfies Exact<Complete<ContextMenuInheritedContextMenuListProps>, typeof menuProps>;

    return (
      <Menu
        {...menuOpenProps}
        {...menuProps}
        placement={placement}
        compact
        trigger={({ props }) => <ContextMenuIconButton {...props} {...iconButtonProps} />}
      />
    );
  },
  {
    Item: Menu.Item,
  }
);