import { mergeProps } from 'zag-ripple';
import { trackSplit, track } from 'ripple';
import { AvatarApiContext } from './use-avatar-context';
import { useAvatar, type UseAvatarProps } from './use-avatar.ripple';
import { splitAvatarProps } from './split-avatar-props.ripple';
import { ark } from '../factory';
import type { HTMLProps, MaybeTracked, PolymorphicProps } from '../../types';

export interface AvatarRootBaseProps extends PolymorphicProps<'div'>, UseAvatarProps {}
export interface AvatarRootProps extends HTMLProps<'div'>, AvatarRootBaseProps {}

export component AvatarRoot(props: MaybeTracked<AvatarRootProps>) {
  const [children, rest] = trackSplit(props, ['children']);
  const [useAvatarProps, localProps] = splitAvatarProps(@rest);
  const avatarApi = useAvatar(useAvatarProps);

  AvatarApiContext.set(avatarApi);

  let mergedProps = track(() => mergeProps(@avatarApi.getRootProps(), @localProps));

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