import { Id, Optional } from '@ephox/katamari';
import { Attribute, type SugarElement } from '@ephox/sugar';

const describedBy = (describedElement: SugarElement<Element>, describeElement: SugarElement<Element>): void => {
  const describeId = Optional.from(Attribute.get(describedElement, 'id'))
    .getOrThunk(() => {
      const id = Id.generate('aria');
      Attribute.set(describeElement, 'id', id);
      return id;
    });

  Attribute.set(describedElement, 'aria-describedby', describeId);
};

const remove = (describedElement: SugarElement<Element>): void => {
  Attribute.remove(describedElement, 'aria-describedby');
};

export {
  describedBy,
  remove
};