123456789101112131415161718192021222324252627282930313233343536373839404142 |
- import * as React from 'react';
- import { assignRef } from './assignRef';
- import { useCallbackRef } from './useRef';
- const currentValues = new WeakMap();
- export function useMergeRefs(refs, defaultValue) {
- const callbackRef = useCallbackRef(defaultValue || null, (newValue) => refs.forEach((ref) => assignRef(ref, newValue)));
-
- React.useLayoutEffect(() => {
- const oldValue = currentValues.get(callbackRef);
- if (oldValue) {
- const prevRefs = new Set(oldValue);
- const nextRefs = new Set(refs);
- const current = callbackRef.current;
- prevRefs.forEach((ref) => {
- if (!nextRefs.has(ref)) {
- assignRef(ref, null);
- }
- });
- nextRefs.forEach((ref) => {
- if (!prevRefs.has(ref)) {
- assignRef(ref, current);
- }
- });
- }
- currentValues.set(callbackRef, refs);
- }, [refs]);
- return callbackRef;
- }
|