ref.js 2.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. import _typeof from "@babel/runtime/helpers/esm/typeof";
  2. import { isValidElement, version } from 'react';
  3. import { ForwardRef, isMemo } from 'react-is';
  4. import useMemo from "./hooks/useMemo";
  5. import isFragment from "./React/isFragment";
  6. var ReactMajorVersion = Number(version.split('.')[0]);
  7. export var fillRef = function fillRef(ref, node) {
  8. if (typeof ref === 'function') {
  9. ref(node);
  10. } else if (_typeof(ref) === 'object' && ref && 'current' in ref) {
  11. ref.current = node;
  12. }
  13. };
  14. /**
  15. * Merge refs into one ref function to support ref passing.
  16. */
  17. export var composeRef = function composeRef() {
  18. for (var _len = arguments.length, refs = new Array(_len), _key = 0; _key < _len; _key++) {
  19. refs[_key] = arguments[_key];
  20. }
  21. var refList = refs.filter(Boolean);
  22. if (refList.length <= 1) {
  23. return refList[0];
  24. }
  25. return function (node) {
  26. refs.forEach(function (ref) {
  27. fillRef(ref, node);
  28. });
  29. };
  30. };
  31. export var useComposeRef = function useComposeRef() {
  32. for (var _len2 = arguments.length, refs = new Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
  33. refs[_key2] = arguments[_key2];
  34. }
  35. return useMemo(function () {
  36. return composeRef.apply(void 0, refs);
  37. }, refs, function (prev, next) {
  38. return prev.length !== next.length || prev.every(function (ref, i) {
  39. return ref !== next[i];
  40. });
  41. });
  42. };
  43. export var supportRef = function supportRef(nodeOrComponent) {
  44. var _type$prototype, _nodeOrComponent$prot;
  45. if (!nodeOrComponent) {
  46. return false;
  47. }
  48. // React 19 no need `forwardRef` anymore. So just pass if is a React element.
  49. if (isReactElement(nodeOrComponent) && ReactMajorVersion >= 19) {
  50. return true;
  51. }
  52. var type = isMemo(nodeOrComponent) ? nodeOrComponent.type.type : nodeOrComponent.type;
  53. // Function component node
  54. if (typeof type === 'function' && !((_type$prototype = type.prototype) !== null && _type$prototype !== void 0 && _type$prototype.render) && type.$$typeof !== ForwardRef) {
  55. return false;
  56. }
  57. // Class component
  58. if (typeof nodeOrComponent === 'function' && !((_nodeOrComponent$prot = nodeOrComponent.prototype) !== null && _nodeOrComponent$prot !== void 0 && _nodeOrComponent$prot.render) && nodeOrComponent.$$typeof !== ForwardRef) {
  59. return false;
  60. }
  61. return true;
  62. };
  63. function isReactElement(node) {
  64. return /*#__PURE__*/isValidElement(node) && !isFragment(node);
  65. }
  66. export var supportNodeRef = function supportNodeRef(node) {
  67. return isReactElement(node) && supportRef(node);
  68. };
  69. /**
  70. * In React 19. `ref` is not a property from node.
  71. * But a property from `props.ref`.
  72. * To check if `props.ref` exist or fallback to `ref`.
  73. */
  74. export var getNodeRef = function getNodeRef(node) {
  75. if (node && isReactElement(node)) {
  76. var ele = node;
  77. // Source from:
  78. // https://github.com/mui/material-ui/blob/master/packages/mui-utils/src/getReactNodeRef/getReactNodeRef.ts
  79. return ele.props.propertyIsEnumerable('ref') ? ele.props.ref : ele.ref;
  80. }
  81. return null;
  82. };