25 lines
481 B
JavaScript
25 lines
481 B
JavaScript
import React from 'react';
|
|
|
|
export default function useCombinedRefs(...refs) {
|
|
const targetRef = React.useRef();
|
|
|
|
React.useEffect(() => {
|
|
refs.forEach(ref => {
|
|
if (!ref) return;
|
|
|
|
if (typeof ref === 'function') {
|
|
ref(targetRef.current);
|
|
} else {
|
|
ref.current = targetRef.current;
|
|
}
|
|
});
|
|
}, [refs]);
|
|
|
|
return targetRef;
|
|
}
|
|
|
|
/*
|
|
Problem described in
|
|
https://itnext.io/reusing-the-ref-from-forwardref-with-react-hooks-4ce9df693dd
|
|
*/
|