refToCallback.js 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", { value: true });
  3. exports.useRefToCallback = exports.refToCallback = void 0;
  4. /**
  5. * Unmemoized version of {@link useRefToCallback}
  6. * @see {@link useRefToCallback}
  7. * @param ref
  8. */
  9. function refToCallback(ref) {
  10. return function (newValue) {
  11. if (typeof ref === 'function') {
  12. ref(newValue);
  13. }
  14. else if (ref) {
  15. ref.current = newValue;
  16. }
  17. };
  18. }
  19. exports.refToCallback = refToCallback;
  20. var nullCallback = function () { return null; };
  21. // lets maintain a weak ref to, well, ref :)
  22. // not using `kashe` to keep this package small
  23. var weakMem = new WeakMap();
  24. var weakMemoize = function (ref) {
  25. var usedRef = ref || nullCallback;
  26. var storedRef = weakMem.get(usedRef);
  27. if (storedRef) {
  28. return storedRef;
  29. }
  30. var cb = refToCallback(usedRef);
  31. weakMem.set(usedRef, cb);
  32. return cb;
  33. };
  34. /**
  35. * Transforms a given `ref` into `callback`.
  36. *
  37. * To transform `callback` into ref use {@link useCallbackRef|useCallbackRef(undefined, callback)}
  38. *
  39. * @param {ReactRef} ref
  40. * @returns {Function}
  41. *
  42. * @see https://github.com/theKashey/use-callback-ref#reftocallback
  43. *
  44. * @example
  45. * const ref = useRef(0);
  46. * const setRef = useRefToCallback(ref);
  47. * 👉 setRef(10);
  48. * ✅ ref.current === 10
  49. */
  50. function useRefToCallback(ref) {
  51. return weakMemoize(ref);
  52. }
  53. exports.useRefToCallback = useRefToCallback;