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

export interface DialogTriggerBaseProps extends PolymorphicProps<'button'> {}
export interface DialogTriggerProps extends HTMLProps<'button'>, DialogTriggerBaseProps {}

export component DialogTrigger(props: MaybeTracked<DialogTriggerProps>) {
  const [children, localProps] = trackSplit(props, ['children']);
  const dialog = useDialogContext();
  const presence = usePresenceContext();

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

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