Function setStyle

packages/react-aria/src/utils/domHelpers.ts:98–131

(
  target: HTMLElement | HTMLElement[] | null,
  property: string,
  value: string,
  priority?: string
)

 * Sets a CSS property on an element and returns a cleanup function.
 */
export function setStyle(
  target: HTMLElement | HTMLElement[] | null,
  property: string,
  value: string,
  priority?: string
): () => void {
  if (target == null) {
    return () => {};
  }

  let restore = new Array<Function>();
  let styleTargets = Array.isArray(target) ? target : [target];

  for (let styleTarget of styleTargets) {
    let initialValue = styleTarget.style.getPropertyValue(property);
    let initialPriority = styleTarget.style.getPropertyPriority(property);

    styleTarget.style.setProperty(property, value, priority);

    restore.unshift(() => {
      if (initialValue) {
        styleTarget.style.setProperty(property, initialValue, initialPriority);
      } else {
        styleTarget.style.removeProperty(property);
      }
    });
  }

  return () => {
    for (let cleanup of restore) {
      cleanup();
    }
  };
}