UNPKG

@yandex/ui

Version:

Yandex UI components

23 lines (22 loc) 518 B
import { useMemo } from 'react'; export function useForkRef(refA, refB) { return useMemo(function () { if (!refA && !refB) { return null; } return function (refValue) { if (refA) setRef(refA, refValue); if (refB) setRef(refB, refValue); }; }, [refA, refB]); } function setRef(ref, value) { if (typeof ref === 'function') { ref(value); } else if (ref) { ref.current = value; } }