27 lines
827 B
JavaScript
27 lines
827 B
JavaScript
'use client';
|
|
|
|
import * as React from 'react';
|
|
import setRef from '../setRef';
|
|
export default function useForkRef() {
|
|
for (var _len = arguments.length, refs = new Array(_len), _key = 0; _key < _len; _key++) {
|
|
refs[_key] = arguments[_key];
|
|
}
|
|
/**
|
|
* 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(function () {
|
|
if (refs.every(function (ref) {
|
|
return ref == null;
|
|
})) {
|
|
return null;
|
|
}
|
|
return function (instance) {
|
|
refs.forEach(function (ref) {
|
|
setRef(ref, instance);
|
|
});
|
|
};
|
|
// eslint-disable-next-line react-hooks/exhaustive-deps
|
|
}, refs);
|
|
} |