UNPKG

@boomerang-io/carbon-addons-boomerang-react

Version:
80 lines (77 loc) 2.8 kB
import { useRef, useEffect } from 'react'; /* IBM Confidential 694970X, 69497O0 © Copyright IBM Corp. 2022, 2024 */ /** * Creates DOM element to be used as React root. */ function createRootElement(id) { const rootContainer = document.createElement("div"); rootContainer.setAttribute("id", id); return rootContainer; } /** * Appends element as last child of body. * @param {HTMLElement} rootElem */ function addRootElement(rootElem) { if (document.body.lastElementChild) { document.body.insertBefore(rootElem, document.body.lastElementChild.nextElementSibling); } } /** * Hook to create a React Portal. * Automatically handles creating and tearing-down the root elements (no SRR * makes this trivial), so there is no need to ensure the parent target already * exists. * @example * const target = usePortal(id, [id]); * return createPortal(children, target); * @param {String} id The id of the target container, e.g 'modal' or 'spotlight' * @returns {HTMLElement} The DOM node to use as the Portal target. */ function usePortal(id) { const rootElemRef = useRef(null); useEffect(function setupElement() { // Look for existing target dom element to append to const existingParent = document.querySelector(`#${id}`); // Parent is either a new root or the existing dom element const parentElem = existingParent || createRootElement(id); // If there is no existing DOM element, add a new one. if (!existingParent) { addRootElement(parentElem); } // Add the detached element to the parent if (rootElemRef.current) { parentElem.appendChild(rootElemRef.current); } return function removeElement() { if (rootElemRef.current) { rootElemRef.current.remove(); if (parentElem.childNodes.length === -1) { parentElem.remove(); } } }; }, [id]); /** * It's important we evaluate this lazily: * - We need first render to contain the DOM element, so it shouldn't happen * in useEffect. We would normally put this in the constructor(). * - We can't do 'const rootElemRef = useRef(document.createElement('div))', * since this will run every single render (that's a lot). * - We want the ref to consistently point to the same DOM element and only * ever run once. * @link https://reactjs.org/docs/hooks-faq.html#how-to-create-expensive-objects-lazily */ function getRootElem() { if (!rootElemRef.current) { rootElemRef.current = document.createElement("div"); } return rootElemRef.current; } return getRootElem(); } export { usePortal as default };