22 lines
651 B
JavaScript
22 lines
651 B
JavaScript
|
'use client';
|
||
|
|
|
||
|
|
import * as React from 'react';
|
||
|
|
import setRef from '../setRef';
|
||
|
|
export default function useForkRef(...refs) {
|
||
|
|
/**
|
||
|
|
* This will create a new function if the refs passed to this hook change and are all defined.
|
||
|
|
* This means react will call the old forkRef with `null` and the new forkRef
|
||
|
|
* with the ref. Cleanup naturally emerges from this behavior.
|
||
|
|
*/
|
||
|
|
return React.useMemo(() => {
|
||
|
|
if (refs.every(ref => ref == null)) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
return instance => {
|
||
|
|
refs.forEach(ref => {
|
||
|
|
setRef(ref, instance);
|
||
|
|
});
|
||
|
|
};
|
||
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
||
|
|
}, refs);
|
||
|
|
}
|