import React from 'react';
import { useButton } from 'react-aria';
import type { ButtonProps, SlotProps } from 'react-aria-components';
import { ButtonContext, useContextProps } from 'react-aria-components';

import type { Exact, SafeOmit } from '@shoptet/utils';

import type { IconButtonProps } from '../../IconButton/IconButton';
import { IconButton } from '../../IconButton/IconButton';

interface InheritedIconButtonProps extends SafeOmit<IconButtonProps, 'icon' | 'variant' | 'ghost'> {}

export interface TreeItemChevronProps extends InheritedIconButtonProps {
  isExpanded: boolean;
}

export function TreeItemChevron({ isExpanded, ...restProps }: TreeItemChevronProps) {
  restProps satisfies Exact<InheritedIconButtonProps, typeof restProps>;

  const chevronRef = React.useRef<HTMLButtonElement>(null);

  const [chevronAriaProps, ref] = useContextProps<ButtonProps, SlotProps, HTMLButtonElement>(
    { slot: 'chevron' },
    chevronRef,
    ButtonContext
  );

  const { buttonProps: chevronProps } = useButton(chevronAriaProps, ref);

  return (
    <IconButton
      {...restProps}
      {...chevronProps}
      ref={ref}
      icon={isExpanded ? 'shp-descending' : 'shp-arrow-right'}
      variant='secondary'
      ghost
    />
  );
}