ref.js 3.5 KB

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