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

export interface CheckboxIndicatorBaseProps extends PolymorphicProps<'div'> {
  indeterminate?: boolean;
}
export interface CheckboxIndicatorProps extends HTMLProps<'div'>, CheckboxIndicatorBaseProps {}

export component CheckboxIndicator(props: MaybeTracked<CheckboxIndicatorProps>) {
  const [indeterminate, children, localProps] = trackSplit(props, ['indeterminate', 'children']);
  const checkbox = useCheckboxContext();

  const isVisible = track(() => @indeterminate ? @checkbox.indeterminate : @checkbox.checked);

  let mergedProps = track(
    () => mergeProps(@checkbox.getIndicatorProps(), @localProps, { hidden: !@isVisible }),
  );

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