@wordpress/compose
Version:
WordPress higher-order components (HOCs).
60 lines (59 loc) • 1.74 kB
JavaScript
// packages/compose/src/hooks/use-merge-refs/index.ts
import { useRef, useCallback, useLayoutEffect } from "@wordpress/element";
function assignRef(ref, value) {
if (typeof ref === "function") {
const returned = ref(value);
return typeof returned === "function" ? returned : void 0;
} else if (ref && ref.hasOwnProperty("current")) {
ref.current = value;
}
return void 0;
}
function detachRef(ref, index, cleanups) {
const cleanup = cleanups[index];
if (cleanup) {
cleanups[index] = void 0;
cleanup();
} else {
assignRef(ref, null);
}
}
function useMergeRefs(refs) {
const elementRef = useRef(null);
const attachedRefsRef = useRef([]);
const currentRefsRef = useRef(refs);
const cleanupsRef = useRef([]);
currentRefsRef.current = refs;
useLayoutEffect(() => {
const element = elementRef.current;
if (element === null) {
return;
}
refs.forEach((ref, index) => {
const attachedRef = attachedRefsRef.current[index];
if (ref !== attachedRef) {
detachRef(attachedRef, index, cleanupsRef.current);
cleanupsRef.current[index] = assignRef(ref, element);
}
});
attachedRefsRef.current = refs;
}, refs);
return useCallback((value) => {
elementRef.current = value;
if (value === null) {
attachedRefsRef.current.forEach((ref, index) => {
detachRef(ref, index, cleanupsRef.current);
});
attachedRefsRef.current = [];
} else {
attachedRefsRef.current = currentRefsRef.current;
attachedRefsRef.current.forEach((ref, index) => {
cleanupsRef.current[index] = assignRef(ref, value);
});
}
}, []);
}
export {
useMergeRefs as default
};
//# sourceMappingURL=index.mjs.map