export function getData(element: Element, name?: string) {
  prepareData(element);

  if (name === undefined) {
    return element.__unicorn;
  }

  return element.__unicorn[name];
}

export function setData(element: Element, name: string, value: any) {
  prepareData(element);
  element.__unicorn[name] = value;
}

export function defData(element: Element, name: string, defCallback: Function) {
  prepareData(element);
  element.__unicorn[name] = element.__unicorn[name] || defCallback(element);

  return element.__unicorn[name];
}

export function removeData(element: Element, name: string) {
  prepareData(element);

  const v = element.__unicorn[name];
  delete element.__unicorn[name];

  return v;
}

export function prepareData<T extends Node>(element: T): T {
  if (!element) {
    return element;
  }

  element.__unicorn = element.__unicorn || {};
  return element;
}

declare global {
  interface Node {
    __unicorn?: any;
  }
}