SCM-Manager/scm-ui/ui-components/src/usePortalRootElement.ts

/*
 * Copyright (c) 2020 - present Cloudogu GmbH
 *
 * This program is free software: you can redistribute it and/or modify it under
 * the terms of the GNU Affero General Public License as published by the Free
 * Software Foundation, version 3.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
 * details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see https://www.gnu.org/licenses/.
 */

import { useEffect, useState } from "react";

const createElement = (id: string) => {
  const element = document.createElement("div");
  element.setAttribute("id", id);
  return element;
};

const appendRootElement = (rootElement: HTMLElement) => {
  document.body.appendChild(rootElement);
};

const usePortalRootElement = (id: string) => {
  const [rootElement, setRootElement] = useState<HTMLElement>();
  useEffect(() => {
    let element = document.getElementById(id);
    if (!element) {
      element = createElement(id);
      appendRootElement(element);
    }
    setRootElement(element);
    return () => {
      setRootElement(undefined);
    };
  }, [id]);

  return rootElement;
};

export default usePortalRootElement;