useMergeRef.js 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. exports.useMergeRefs = void 0;
  4. var tslib_1 = require("tslib");
  5. var React = tslib_1.__importStar(require("react"));
  6. var assignRef_1 = require("./assignRef");
  7. var useRef_1 = require("./useRef");
  8. var useIsomorphicLayoutEffect = typeof window !== 'undefined' ? React.useLayoutEffect : React.useEffect;
  9. var currentValues = new WeakMap();
  10. /**
  11. * Merges two or more refs together providing a single interface to set their value
  12. * @param {RefObject|Ref} refs
  13. * @returns {MutableRefObject} - a new ref, which translates all changes to {refs}
  14. *
  15. * @see {@link mergeRefs} a version without buit-in memoization
  16. * @see https://github.com/theKashey/use-callback-ref#usemergerefs
  17. * @example
  18. * const Component = React.forwardRef((props, ref) => {
  19. * const ownRef = useRef();
  20. * const domRef = useMergeRefs([ref, ownRef]); // 👈 merge together
  21. * return <div ref={domRef}>...</div>
  22. * }
  23. */
  24. function useMergeRefs(refs, defaultValue) {
  25. var callbackRef = (0, useRef_1.useCallbackRef)(defaultValue || null, function (newValue) {
  26. return refs.forEach(function (ref) { return (0, assignRef_1.assignRef)(ref, newValue); });
  27. });
  28. // handle refs changes - added or removed
  29. useIsomorphicLayoutEffect(function () {
  30. var oldValue = currentValues.get(callbackRef);
  31. if (oldValue) {
  32. var prevRefs_1 = new Set(oldValue);
  33. var nextRefs_1 = new Set(refs);
  34. var current_1 = callbackRef.current;
  35. prevRefs_1.forEach(function (ref) {
  36. if (!nextRefs_1.has(ref)) {
  37. (0, assignRef_1.assignRef)(ref, null);
  38. }
  39. });
  40. nextRefs_1.forEach(function (ref) {
  41. if (!prevRefs_1.has(ref)) {
  42. (0, assignRef_1.assignRef)(ref, current_1);
  43. }
  44. });
  45. }
  46. currentValues.set(callbackRef, refs);
  47. }, [refs]);
  48. return callbackRef;
  49. }
  50. exports.useMergeRefs = useMergeRefs;