import React from 'react';
import { CheckboxContext, useSlottedContext } from 'react-aria-components';

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

import type { CheckboxProps } from '../../Inputs/Checkbox/Checkbox';
import { Checkbox } from '../../Inputs/Checkbox/Checkbox';

export interface TreeItemCheckboxProps extends SafeOmit<CheckboxProps, 'checked' | 'disabled'> {}

export function TreeItemCheckbox(props: TreeItemCheckboxProps) {
  const selection = useSlottedContext(CheckboxContext, 'selection');

  return <Checkbox {...props} checked={selection?.isSelected ?? false} disabled={selection?.isDisabled ?? false} />;
}