reakit/src/Dialog/__utils/usePortalRef.ts

Toolkit for building accessible rich web apps with React

import * as React from "react";
import { closest } from "reakit-utils/closest";
import { Portal } from "../../Portal/Portal";
import { DialogOptions } from "../Dialog";

export function usePortalRef(
  dialogRef: React.RefObject<HTMLElement>,
  options: DialogOptions
) {
  const portalRef = React.useRef<HTMLElement | null>(null);

  React.useEffect(() => {
    const dialog = dialogRef.current;
    if (!dialog || !options.visible) return;
    portalRef.current = closest(dialog, Portal.__selector) as HTMLElement;
  }, [dialogRef, options.visible]);

  return portalRef;
}