import { mergeProps } from 'zag-ripple';
import { trackSplit, track } from 'ripple';
import { useMenuContext } from './use-menu-context';
import { usePresenceContext } from '../presence/use-presence-context';
import { ark } from '../factory';
import type { HTMLProps, MaybeTracked, PolymorphicProps } from '../../types';

export interface MenuTriggerBaseProps extends PolymorphicProps<'button'> {}
export interface MenuTriggerProps extends HTMLProps<'button'>, MenuTriggerBaseProps {}

export component MenuTrigger(props: MaybeTracked<MenuTriggerProps>) {
  const [children, localProps] = trackSplit(props, ['children']);
  const menu = useMenuContext();
  const presence = usePresenceContext();

  let mergedProps = track(() => {
    return mergeProps(
      {
        ...@menu.getTriggerProps(),
        'aria-controls': @presence.unmounted ? undefined : @menu.getTriggerProps()['aria-controls'],
      },
      @localProps,
    );
  });

  <ark.button {...@mergedProps}>
    <@children />
  </ark.button>
}