24 lines
675 B
TypeScript
24 lines
675 B
TypeScript
import { MutableRefObject, RefCallback, useCallback } from "react";
|
|
|
|
/**
|
|
* Combines multiple refs into one, useful for attaching multiple refs to the
|
|
* same DOM node.
|
|
*/
|
|
export const useMergedRefs = <T>(
|
|
...refs: (MutableRefObject<T | null> | RefCallback<T | null>)[]
|
|
): RefCallback<T | null> =>
|
|
useCallback(
|
|
(value) =>
|
|
refs.forEach((ref) => {
|
|
if (typeof ref === "function") {
|
|
ref(value);
|
|
} else {
|
|
ref.current = value;
|
|
}
|
|
}),
|
|
// Since this isn't an array literal, we can't use the static dependency
|
|
// checker, but that's okay
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
refs
|
|
);
|